✨ Add yaml file filter completion

clevyr · May 6, 2022 · b07e180 · b07e180
1 parent 9b6bd53
commit b07e180
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/cmd/cmd.go b/cmd/cmd.go
@@ -18,6 +18,7 @@ var Command = &cobra.Command{
 	Short:                 "Inline YAML templating via line-comments",
 	DisableFlagsInUseLine: true,
 	DisableAutoGenTag:     true,
+	ValidArgsFunction:     validArgs,
 	Version:               buildVersion(),
 	PreRunE:               preRun,
 	RunE:                  run,
@@ -33,6 +34,10 @@ func init() {
 	Command.Flags().StringVar(&conf.RightDelim, "right-delim", "}}", "Override the default right delimiter")
 }
 
+func validArgs(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
+	return []string{"yaml", "yml"}, cobra.ShellCompDirectiveFilterFileExt
+}
+
 func preRun(cmd *cobra.Command, args []string) error {
 	if completionFlag != "" {
 		return nil