Fix argument completion

thestormforge · Nov 30, 2023 · d80887d · d80887d
1 parent df39761
commit d80887d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/internal/command/root.go b/internal/command/root.go
@@ -37,6 +37,9 @@ func NewRootCommand(version, refspec, date string) *cobra.Command {
 		Version:          version,
 		SilenceUsage:     true,
 		TraverseChildren: true,
+		ValidArgsFunction: func(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
+			return []string{"yaml", "yml", "json"}, cobra.ShellCompDirectiveFilterFileExt
+		},
 		Annotations: map[string]string{
 			"BuildRefspec": refspec,
 			"BuildDate":    date,