From 0845b6b0086a968684d1e5314aa402506ea45f62 Mon Sep 17 00:00:00 2001
From: Ivan Ilves <ivan.ilves@schibsted.com>
Date: Tue, 26 Sep 2017 22:59:42 +0200
Subject: [PATCH] List/pull multiple repos

---
 main.go | 99 ++++++++++++++++++++++++++++++---------------------------
 1 file changed, 52 insertions(+), 47 deletions(-)

diff --git a/main.go b/main.go
index 6cfa54b..c1531e4 100644
--- a/main.go
+++ b/main.go
@@ -27,7 +27,7 @@ type options struct {
 	TraceRequests    bool   `short:"T" long:"trace-requests" description:"Trace registry HTTP requests" env:"TRACE_REQUESTS"`
 	Version          bool   `short:"V" long:"version" description:"Show version and exit"`
 	Positional       struct {
-		Repository string `positional-arg-name:"REPOSITORY" description:"Docker repository to list tags from"`
+		Repositories []string `positional-arg-name:"REPO1 REPO2" description:"Docker repositories to operate on"`
 	} `positional-args:"yes"`
 }
 
@@ -135,8 +135,8 @@ func main() {
 		println(getVersion())
 		os.Exit(0)
 	}
-	if o.Positional.Repository == "" {
-		suicide(errors.New("You should provide a repository name, e.g. 'nginx~/^1\\\\.13/' or 'mesosphere/chronos'"))
+	if len(o.Positional.Repositories) == 0 {
+		suicide(errors.New("Need at least one repository name, e.g. 'nginx~/^1\\\\.13/' or 'mesosphere/chronos'"))
 	}
 
 	if o.InsecureRegistry {
@@ -146,74 +146,78 @@ func main() {
 
 	registry.TraceRequests = o.TraceRequests
 
-	repository, filter, err := trimFilter(o.Positional.Repository)
-	if err != nil {
-		suicide(err)
-	}
+	const format = "%-12s %-45s %-15s %-25s %s\n"
+	fmt.Printf(format, "<STATE>", "<DIGEST>", "<(local) ID>", "<Created At>", "<TAG>")
 
-	registryName := getRegistryName(repository, o.DefaultRegistry)
+	allTags := make([]*tag.Tag, 0)
+	lsRepos := make([]string, 0)
 
-	repoRegistryName := registry.FormatRepoName(repository, registryName)
-	repoLocalName := local.FormatRepoName(repository, registryName)
+	for _, r := range o.Positional.Repositories {
+		repository, filter, err := trimFilter(r)
+		if err != nil {
+			suicide(err)
+		}
 
-	username, password, err := assignCredentials(registryName, o.Username, o.Password, o.DockerJSON)
-	if err != nil {
-		suicide(err)
-	}
+		registryName := getRegistryName(repository, o.DefaultRegistry)
 
-	tresp, err := auth.NewToken(registryName, repoRegistryName, username, password)
-	if err != nil {
-		suicide(err)
-	}
+		repoRegistryName := registry.FormatRepoName(repository, registryName)
+		repoLocalName := local.FormatRepoName(repository, registryName)
 
-	authorization := getAuthorization(tresp)
+		username, password, err := assignCredentials(registryName, o.Username, o.Password, o.DockerJSON)
+		if err != nil {
+			suicide(err)
+		}
 
-	registryTags, err := registry.FetchTags(registryName, repoRegistryName, authorization, o.Concurrency)
-	if err != nil {
-		suicide(err)
-	}
-	localTags, err := local.FetchTags(repoLocalName)
-	if err != nil {
-		suicide(err)
-	}
+		tresp, err := auth.NewToken(registryName, repoRegistryName, username, password)
+		if err != nil {
+			suicide(err)
+		}
 
-	sortedKeys, names, joinedTags := tag.Join(registryTags, localTags)
+		authorization := getAuthorization(tresp)
 
-	const format = "%-12s %-45s %-15s %-25s %s\n"
-	fmt.Printf(format, "<STATE>", "<DIGEST>", "<(local) ID>", "<Created At>", "<TAG>")
-	for _, key := range sortedKeys {
-		name := names[key]
+		registryTags, err := registry.FetchTags(registryName, repoRegistryName, authorization, o.Concurrency)
+		if err != nil {
+			suicide(err)
+		}
+		localTags, err := local.FetchTags(repoLocalName)
+		if err != nil {
+			suicide(err)
+		}
+
+		sortedKeys, names, joinedTags := tag.Join(registryTags, localTags)
+
+		for _, key := range sortedKeys {
+			name := names[key]
 
-		tg := joinedTags[name]
+			tg := joinedTags[name]
 
-		if !matchesFilter(tg.GetName(), filter) {
-			continue
+			if !matchesFilter(tg.GetName(), filter) {
+				continue
+			}
+
+			allTags = append(allTags, tg)
+			lsRepos = append(lsRepos, repoLocalName)
 		}
+	}
 
+	for i, tg := range allTags {
 		fmt.Printf(
 			format,
 			tg.GetState(),
 			tg.GetShortDigest(),
 			tg.GetImageID(),
 			tg.GetCreatedString(),
-			repoLocalName+":"+tg.GetName(),
+			lsRepos[i]+":"+tg.GetName(),
 		)
+
 	}
 
 	if o.PullImages {
-		for _, key := range sortedKeys {
-			name := names[key]
-
-			tg := joinedTags[name]
-
-			if !matchesFilter(tg.GetName(), filter) {
-				continue
-			}
-
+		for i, tg := range allTags {
 			if tg.NeedsPull() {
-				ref := repoLocalName + ":" + tg.GetName()
+				ref := lsRepos[i] + ":" + tg.GetName()
 
-				fmt.Printf("Pulling: %s\n", ref)
+				fmt.Printf("PULLING: %s\n", ref)
 				err := local.PullImage(ref)
 				if err != nil {
 					suicide(err)
@@ -221,4 +225,5 @@ func main() {
 			}
 		}
 	}
+
 }