Add some more --for aliases

thestormforge · Aug 18, 2022 · a1aedb8 · a1aedb8
1 parent 505b292
commit a1aedb8
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/pkg/command/applications.go b/pkg/command/applications.go
@@ -294,6 +294,10 @@ func NewGetApplicationsCommand(cfg Config, p Printer) *cobra.Command {
 	cmd.Flags().StringVar(&product, "for", product, "show only clusters for a specific `product`; one of: optimize-pro|optimize-live")
 	cmd.Flags().IntVar(&batchSize, "batch-size", batchSize, "fetch large lists in chu`n`ks rather then all at once")
 
+	_ = cmd.RegisterFlagCompletionFunc("for", func(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
+		return []string{"optimize-pro", "optimize-live"}, cobra.ShellCompDirectiveDefault
+	})
+
 	cmd.RunE = func(cmd *cobra.Command, args []string) error {
 		ctx, out := cmd.Context(), cmd.OutOrStdout()
 		client, err := api.NewClient(cfg.Address(), nil)
@@ -347,7 +351,7 @@ func NewGetApplicationsCommand(cfg Config, p Printer) *cobra.Command {
 				// Only skip applications if we know it is one or the other
 				if isPro || isLive {
 					switch product {
-					case "optimize-pro", "pro":
+					case "optimize-pro", "pro", "optimize-controller", "controller":
 						if !isPro {
 							continue
 						}

diff --git a/pkg/command/clusters.go b/pkg/command/clusters.go
@@ -106,7 +106,7 @@ func NewGetClustersCommand(cfg Config, p Printer) *cobra.Command {
 		} else {
 			q := applications.ClusterListQuery{}
 			switch product {
-			case "optimize-pro", "pro":
+			case "optimize-pro", "pro", "optimize-controller", "controller":
 				q.SetModules(applications.ClusterScenarios)
 			case "optimize-live", "live":
 				q.SetModules(applications.ClusterRecommendations)