add command flag completion

rancher · Nov 10, 2023 · 60922f0 · 60922f0
1 parent 329814a
commit 60922f0
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/pkg/opni/commands/debug.go b/pkg/opni/commands/debug.go
@@ -543,6 +543,12 @@ func BuildDebugAgentLogStreamGetCmd() *cobra.Command {
 	cmd.Flags().StringVar(&level, "level", "info", "Minimum log level severity (debug, info, warn, error)")
 	cmd.Flags().StringVar(&output, "output", "text", "Output format")
 	cmd.Flags().BoolVar(&follow, "follow", false, "Follow logs")
+	cmd.RegisterFlagCompletionFunc("level", func(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
+		return []string{"debug", "info", "warn", "error"}, cobra.ShellCompDirectiveDefault
+	})
+	cmd.RegisterFlagCompletionFunc("output", func(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
+		return []string{"text", "json"}, cobra.ShellCompDirectiveDefault
+	})
 	return cmd
 }