🐛 Fix CLI completion generation (QD-9907)

JetBrains · Sep 16, 2024 · cac836c · cac836c
1 parent 0183d62
commit cac836c
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/cmd/root.go b/cmd/root.go
@@ -31,6 +31,10 @@ func isHelpOrVersion(args []string) bool {
 	return len(args) == 2 && (args[1] == "--help" || args[1] == "-h" || args[1] == "help" || args[1] == "--version" || args[1] == "-v")
 }
 
+func isCompletionRequested(args []string) bool {
+	return args[1] == "completion"
+}
+
 // isCommandRequested checks if any command is requested.
 func isCommandRequested(commands []*cobra.Command, args []string) string {
 	for _, c := range commands {
@@ -45,7 +49,7 @@ func isCommandRequested(commands []*cobra.Command, args []string) string {
 
 // setDefaultCommandIfNeeded sets default scan command if no other command is requested.
 func setDefaultCommandIfNeeded(rootCmd *cobra.Command, args []string) {
-	if !(isHelpOrVersion(args) || isCommandRequested(rootCmd.Commands(), args[1:]) != "") {
+	if !(isHelpOrVersion(args) || isCommandRequested(rootCmd.Commands(), args[1:]) != "" || isCompletionRequested(args)) {
 		newArgs := append([]string{"scan"}, args[1:]...)
 		rootCmd.SetArgs(newArgs)
 	}