pull.go

package main

import (
	"os"
	"sync"

	survey "gopkg.in/AlecAivazis/survey.v1"
)

// Pull displays and downloads any repos that are available
func Pull(args []string, _ []Command) error {
	username, password, subreddit := GetCredentials()
	session, err := GetSession(username, password)
	if err != nil {
		return err
	}

	manifest, err := RetrieveManifestFromReddit(subreddit, session)

	if err != nil {
		return err
	}

	var options []string
	for _, repo := range manifest.Repositories {
		options = append(options, repo.Name)
	}

	prompt := &survey.MultiSelect{
		Message: "Select repositories to download",
		Options: options,
	}
	response := []string{}
	survey.AskOne(prompt, &response, nil)

	path, err := os.Getwd()
	if err != nil {
		return err
	}

	repositories := manifest.FilterReposByNames(response)

	var wg sync.WaitGroup

	downloadResponses := make(chan error, len(repositories))

	for _, repository := range repositories {
		wg.Add(1)
		go func(repository Repository) {
			defer wg.Done()
			downloadResponses <- repository.Download(path, session)
		}(repository)
	}
	wg.Wait()
	close(downloadResponses)

	for downloadResponse := range downloadResponses {
		if downloadResponse != nil {
			return downloadResponse
		}
	}
	return err
}