From 9241c701a8dc59be990346d261ef9e2731907630 Mon Sep 17 00:00:00 2001
From: Mathieu Frenette <silphid@users.noreply.github.com>
Date: Sat, 17 Jul 2021 14:13:29 -0400
Subject: [PATCH] yey get containers now only list project-specific containers,
 unless --all specified

---
 src/cmd/get/containers/containers.go | 45 ++++++++++++++++++++++++----
 src/cmd/remove/remove.go             |  5 ++++
 2 files changed, 44 insertions(+), 6 deletions(-)

diff --git a/src/cmd/get/containers/containers.go b/src/cmd/get/containers/containers.go
index dbb2dc3..e400d34 100644
--- a/src/cmd/get/containers/containers.go
+++ b/src/cmd/get/containers/containers.go
@@ -9,30 +9,63 @@ import (
 	"github.com/TwinProduction/go-color"
 	"github.com/spf13/cobra"
 
+	yey "github.com/silphid/yey/src/internal"
 	"github.com/silphid/yey/src/internal/docker"
 )
 
+type Options struct {
+	All bool
+}
+
 // New creates a cobra command
 func New() *cobra.Command {
-	return &cobra.Command{
+	var options Options
+
+	cmd := &cobra.Command{
 		Use:   "containers",
 		Short: "Lists running containers",
 		Args:  cobra.NoArgs,
 		RunE: func(cmd *cobra.Command, args []string) error {
-			return run(cmd.Context())
+			return run(cmd.Context(), options)
 		},
 	}
+
+	cmd.Flags().BoolVarP(&options.All, "all", "a", false, "include all yey containers")
+
+	return cmd
 }
 
-func run(ctx context.Context) error {
-	names, err := docker.ListContainers(ctx, true)
+func run(ctx context.Context, options Options) error {
+	containers, err := docker.ListContainers(ctx, true)
 	if err != nil {
 		return err
 	}
-	if len(names) == 0 {
+	totalCount := len(containers)
+
+	if !options.All {
+		contexts, err := yey.LoadContexts()
+		if err != nil {
+			return err
+		}
+		prefix := yey.ContainerPathPrefix(contexts.Path)
+
+		var filteredContainers []string
+		for _, container := range containers {
+			if strings.HasPrefix(container, prefix) {
+				filteredContainers = append(filteredContainers, container)
+			}
+		}
+		containers = filteredContainers
+	}
+
+	if len(containers) == 0 {
+		if totalCount > 0 {
+			fmt.Fprintln(os.Stderr, color.Ize(color.Green, fmt.Sprintf("no project-specific yey containers found, but %d other(s) were found that you could include with --all flag", totalCount)))
+			return nil
+		}
 		fmt.Fprintln(os.Stderr, color.Ize(color.Green, "no yey containers found"))
 		return nil
 	}
-	fmt.Println(strings.Join(names, "\n"))
+	fmt.Println(strings.Join(containers, "\n"))
 	return nil
 }
diff --git a/src/cmd/remove/remove.go b/src/cmd/remove/remove.go
index c3f957e..d25704c 100644
--- a/src/cmd/remove/remove.go
+++ b/src/cmd/remove/remove.go
@@ -48,6 +48,7 @@ func run(ctx context.Context, names []string, options RemoveOptions) error {
 	if err != nil {
 		return fmt.Errorf("failed to list containers to prompt for removal: %w", err)
 	}
+	totalCount := len(containers)
 
 	// Compute all valid contexts
 	combos := contexts.GetCombos()
@@ -84,6 +85,10 @@ func run(ctx context.Context, names []string, options RemoveOptions) error {
 
 	// Abort if no containers to remove
 	if len(validContainers) == 0 && len(otherContainers) == 0 {
+		if totalCount > 0 {
+			fmt.Fprintln(os.Stderr, color.Ize(color.Green, fmt.Sprintf("no project-specific yey containers found, but %d other(s) were found that you could include with --all flag", totalCount)))
+			return nil
+		}
 		fmt.Fprintln(os.Stderr, color.Ize(color.Green, "no yey containers found to remove"))
 		return nil
 	}