chore(cmd): Add "get stream" subcommand completion

gabe565 · Sep 3, 2024 · c25b31b · c25b31b
1 parent bee9af3
commit c25b31b
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/cmd/get/stream/cmd.go b/cmd/get/stream/cmd.go
@@ -24,6 +24,15 @@ func NewCommand(opts ...util.Option) *cobra.Command {
 
 		PreRunE: preRun,
 		RunE:    run,
+
+		Args: cobra.MaximumNArgs(1),
+		ValidArgsFunction: func(_ *cobra.Command, args []string, _ string) ([]string, cobra.ShellCompDirective) {
+			var completion []string
+			if len(args) == 0 {
+				completion = []string{"count"}
+			}
+			return completion, cobra.ShellCompDirectiveNoFileComp
+		},
 	}
 
 	cmd.Flags().StringP("count", "c", "", "Gets stream count (active, total)")