Complete --from and --to fields

Signed-off-by: Daichi Sakaue <[email protected]>

cmd/npv/app/helper_completion.go

@@ -2,6 +2,7 @@ package app
 
 import (
 	"context"
+	"strings"
 
 	"github.com/spf13/cobra"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
@@ -46,3 +47,38 @@ func completePods(cmd *cobra.Command, args []string, toComplete string) (ret []s
 	}
 	return
 }
+
+func completeNamespacePods(cmd *cobra.Command, args []string, toComplete string) (ret []string, directive cobra.ShellCompDirective) {
+	ret = make([]string, 0)
+	directive = cobra.ShellCompDirectiveNoFileComp
+
+	clientset, _, err := createK8sClients()
+	if err != nil {
+		return
+	}
+
+	li := strings.Split(toComplete, "/")
+	switch len(li) {
+	case 2: // namespace already filled
+		pods, err := listRelevantPods(context.Background(), clientset, li[0])
+		if err != nil {
+			return
+		}
+		for _, p := range pods {
+			ret = append(ret, li[0]+"/"+p.Name)
+		}
+		return
+
+	default:
+		nss, err := clientset.CoreV1().Namespaces().List(context.Background(), metav1.ListOptions{})
+		if err != nil {
+			return
+		}
+
+		for _, ns := range nss.Items {
+			ret = append(ret, ns.Name+"/")
+		}
+		directive = cobra.ShellCompDirectiveNoSpace
+		return
+	}
+}

cmd/npv/app/manifest_generate.go

@@ -31,6 +31,8 @@ func init() {
 	manifestGenerateCmd.Flags().BoolVar(&manifestGenerateOptions.deny, "deny", false, "generate deny rule")
 	manifestGenerateCmd.Flags().StringVar(&manifestGenerateOptions.from, "from", "", "egress pod")
 	manifestGenerateCmd.Flags().StringVar(&manifestGenerateOptions.to, "to", "", "ingress pod")
+	manifestGenerateCmd.RegisterFlagCompletionFunc("from", completeNamespacePods)
+	manifestGenerateCmd.RegisterFlagCompletionFunc("to", completeNamespacePods)
 	manifestCmd.AddCommand(manifestGenerateCmd)
 }
 

cmd/npv/app/manifest_range.go

@@ -18,6 +18,8 @@ var manifestRangeOptions struct {
 func init() {
 	manifestRangeCmd.Flags().StringVar(&manifestRangeOptions.from, "from", "", "egress pod")
 	manifestRangeCmd.Flags().StringVar(&manifestRangeOptions.to, "to", "", "ingress pod")
+	manifestRangeCmd.RegisterFlagCompletionFunc("from", completeNamespacePods)
+	manifestRangeCmd.RegisterFlagCompletionFunc("to", completeNamespacePods)
 	manifestCmd.AddCommand(manifestRangeCmd)
 }