docs(cmd): Add util.CompleteROM

gabe565 · Oct 14, 2024 · 5269100 · 5269100
1 parent bda8c43
commit 5269100
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 6 deletions.
diff --git a/cmd/gones/cmd.go b/cmd/gones/cmd.go
@@ -9,6 +9,7 @@ import (
 
 	"gabe565.com/gones/cmd/options"
 	"gabe565.com/gones/internal/config"
+	"gabe565.com/gones/internal/util"
 	"github.com/spf13/cobra"
 )
 
@@ -17,10 +18,8 @@ func New(opts ...options.Option) *cobra.Command {
 		Use:   "gones ROM",
 		Short: "NES emulator written in Go",
 		RunE:  runCobra,
-		ValidArgsFunction: func(_ *cobra.Command, _ []string, _ string) ([]string, cobra.ShellCompDirective) {
-			return []string{"nes"}, cobra.ShellCompDirectiveFilterFileExt
-		},
 
+		ValidArgsFunction: util.CompleteROM,
 		SilenceErrors:     true,
 		DisableAutoGenTag: true,
 	}

diff --git a/cmd/nesutil/ls/cmd.go b/cmd/nesutil/ls/cmd.go
@@ -39,9 +39,7 @@ func New() *cobra.Command {
 		Aliases: []string{"list"},
 		RunE:    run,
 
-		ValidArgsFunction: func(_ *cobra.Command, _ []string, _ string) ([]string, cobra.ShellCompDirective) {
-			return []string{"nes"}, cobra.ShellCompDirectiveFilterFileExt
-		},
+		ValidArgsFunction: util.CompleteROM,
 	}
 
 	flag := cmd.Flags()

diff --git a/internal/util/cobra.go b/internal/util/cobra.go
@@ -0,0 +1,7 @@
+package util
+
+import "github.com/spf13/cobra"
+
+func CompleteROM(_ *cobra.Command, _ []string, _ string) ([]string, cobra.ShellCompDirective) {
+	return []string{"nes"}, cobra.ShellCompDirectiveFilterFileExt
+}