commands: implement ls

Signed-off-by: Tonis Tiigi <tonistiigi@gmail.com>

commands/ls.go

@@ -1,30 +1,27 @@
 package commands
 
 import (
+	"context"
 	"fmt"
+	"io"
+	"os"
+	"strings"
+	"text/tabwriter"
+	"time"
 
 	"github.com/docker/cli/cli"
 	"github.com/docker/cli/cli/command"
+	"github.com/moby/buildkit/util/appcontext"
 	"github.com/spf13/cobra"
+	"github.com/tonistiigi/buildx/store"
+	"golang.org/x/sync/errgroup"
 )
 
 type lsOptions struct {
 }
 
 func runLs(dockerCli command.Cli, in lsOptions) error {
-	ep, err := getCurrentEndpoint(dockerCli)
-	fmt.Printf("current endpoint: %+v %v\n", ep, err)
-
-	fmt.Printf("current config file: %+v\n", dockerCli.ConfigFile())
-	fmt.Printf("current config file: %+v\n", dockerCli)
-
-	list, err := dockerCli.ContextStore().ListContexts()
-	if err != nil {
-		return err
-	}
-	for i, l := range list {
-		fmt.Printf("context%d: %+v\n", i, l)
-	}
+	ctx := appcontext.Context()
 
 	txn, release, err := getStore(dockerCli)
 	if err != nil {
@@ -32,17 +29,112 @@ func runLs(dockerCli command.Cli, in lsOptions) error {
 	}
 	defer release()
 
+	ctx, cancel := context.WithTimeout(ctx, 7*time.Second)
+	defer cancel()
+
 	ll, err := txn.List()
 	if err != nil {
 		return err
 	}
-	for i, l := range ll {
-		fmt.Printf("store %d: %+v\n", i, l)
+
+	builders := make([]*nginfo, len(ll))
+	for i, ng := range ll {
+		builders[i] = &nginfo{ng: ng}
 	}
 
+	list, err := dockerCli.ContextStore().ListContexts()
+	if err != nil {
+		return err
+	}
+	ctxbuilders := make([]*nginfo, len(list))
+	for i, l := range list {
+		ctxbuilders[i] = &nginfo{ng: &store.NodeGroup{
+			Name: l.Name,
+			Nodes: []store.Node{{
+				Name:     l.Name,
+				Endpoint: l.Name,
+			}},
+		}}
+	}
+
+	builders = append(builders, ctxbuilders...)
+
+	eg, _ := errgroup.WithContext(ctx)
+
+	for _, b := range builders {
+		func(b *nginfo) {
+			eg.Go(func() error {
+				err = loadNodeGroupData(ctx, dockerCli, b)
+				if b.err == nil && err != nil {
+					b.err = err
+				}
+				return nil
+			})
+		}(b)
+	}
+
+	if err := eg.Wait(); err != nil {
+		return err
+	}
+
+	currentName := "default"
+	current, err := getCurrentInstance(txn, dockerCli)
+	if err != nil {
+		return err
+	}
+	if current != nil {
+		currentName = current.Name
+		if current.Name == "default" {
+			currentName = current.Nodes[0].Endpoint
+		}
+	}
+
+	w := tabwriter.NewWriter(os.Stdout, 0, 0, 1, ' ', 0)
+	fmt.Fprintf(w, "NAME\tDRIVER/ENDPOINT\tSTATUS\tPLATFORMS\n")
+
+	currentSet := false
+	for _, b := range builders {
+		if !currentSet && b.ng.Name == currentName {
+			b.ng.Name += " *"
+			currentSet = true
+		}
+		printngi(w, b)
+	}
+
+	w.Flush()
+
 	return nil
 }
 
+func printngi(w io.Writer, ngi *nginfo) {
+	var err string
+	if ngi.err != nil {
+		err = ngi.err.Error()
+	}
+	fmt.Fprintf(w, "%s\t%s\t%s\t\n", ngi.ng.Name, ngi.ng.Driver, err)
+	if ngi.err == nil {
+		for idx, n := range ngi.ng.Nodes {
+			d := ngi.drivers[idx]
+			var err string
+			if d.err != nil {
+				err = d.err.Error()
+			} else if d.di.Err != nil {
+				err = d.di.Err.Error()
+			}
+			var status string
+			if d.info != nil {
+				status = d.info.Status.String()
+			}
+			p := append(n.Platforms, d.platforms...)
+			if err != "" {
+				fmt.Fprintf(w, "  %s\t%s\t%s\n", n.Name, n.Endpoint, err)
+			} else {
+				fmt.Fprintf(w, "  %s\t%s\t%s\t%s\n", n.Name, n.Endpoint, status, strings.Join(p, ", "))
+			}
+		}
+	}
+}
+
 func lsCmd(dockerCli command.Cli) *cobra.Command {
 	var options lsOptions
 
@@ -55,11 +147,5 @@ func lsCmd(dockerCli command.Cli) *cobra.Command {
 		},
 	}
 
-	flags := cmd.Flags()
-
-	// flags.StringArrayVarP(&options.outputs, "output", "o", []string{}, "Output destination (format: type=local,dest=path)")
-
-	_ = flags
-
 	return cmd
 }