From 1b716ff2beba23980ce4c86ac3a30f8c46869d99 Mon Sep 17 00:00:00 2001 From: guoxudong Date: Mon, 2 Dec 2019 17:32:38 +0800 Subject: [PATCH] fix completion --- cmd/completion.go | 17 ++++++++++++----- 1 file changed, 12 insertions(+), 5 deletions(-) diff --git a/cmd/completion.go b/cmd/completion.go index 75748235..78b736e5 100644 --- 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.") } + }, }