fix completion

sunny0826 · Dec 2, 2019 · 1b716ff · 1b716ff
1 parent 7ba1c0f
commit 1b716ff
Showing 1 changed file with 12 additions and 5 deletions.
diff --git a/cmd/completion.go b/cmd/completion.go
@@ -41,12 +41,19 @@ source <(kubecm completion zsh)
 kubecm completion zsh > "${fpath[1]}/_kubecm"
 `,
 	Run: func(cmd *cobra.Command, args []string) {
-		complet := args[0]
-		if complet == "bash" {
-			rootCmd.GenBashCompletion(os.Stdout)
-		} else if complet == "zsh" {
-			zsh.Wrap(rootCmd).GenZshCompletion(os.Stdout)
+		if len(args) == 1{
+			complet := args[0]
+			if complet == "bash" {
+				rootCmd.GenBashCompletion(os.Stdout)
+			} else if complet == "zsh" {
+				zsh.Wrap(rootCmd).GenZshCompletion(os.Stdout)
+			}else {
+				Warning.Println("Parameter error! Please input bash or zsh")
+			}
+		}else {
+			Warning.Println("Please input bash or zsh.")
 		}
+
 	},
 }