reg/list.go

package main

import (
	"fmt"
	"os"
	"sort"
	"strings"
	"sync"
	"text/tabwriter"

	"github.com/urfave/cli"
)

var listCommand = cli.Command{
	Name:    "list",
	Aliases: []string{"ls"},
	Usage:   "list all repositories",
	Action: func(c *cli.Context) error {
		if len(c.Args()) < 1 {
			return fmt.Errorf("pass the domain of the registry")
		}

		// Create the registry client.
		r, err := createRegistryClient(c, c.Args().First())
		if err != nil {
			return err
		}

		// Get the repositories via catalog.
		repos, err := r.Catalog("")
		if err != nil {
			return err
		}
		sort.Strings(repos)

		fmt.Printf("Repositories for %s\n", r.Domain)

		var (
			wg       sync.WaitGroup
			repoTags = map[string][]string{}
		)

		wg.Add(len(repos))
		for _, repo := range repos {
			go func(repo string) {
				// Get the tags.
				tags, err := r.Tags(repo)
				if err != nil {
					fmt.Printf("Get tags of [%s] error: %s", repo, err)
				}
				// Sort the tags
				sort.Strings(tags)
				repoTags[repo] = tags

				wg.Done()
			}(repo)
		}
		wg.Wait()

		// Setup the tab writer.
		w := tabwriter.NewWriter(os.Stdout, 20, 1, 3, ' ', 0)

		// Print header.
		fmt.Fprintln(w, "REPO\tTAGS")

		// Sort the repos.
		for _, repo := range repos {
			w.Write([]byte(fmt.Sprintf("%s\t%s\n", repo, strings.Join(repoTags[repo], ", "))))
		}

		w.Flush()

		return nil
	},
}