mirror of
https://github.com/kubescape/kubescape.git
synced 2026-03-05 11:10:32 +00:00
85da52ebbd
Wrote ne tsts for the following packages: - completion - config - download - fix - list Also addressed a potential crash in the compleition, download, and list subcommands when no arguement was provided to the Args, RunE or Run functions. Updated `DownloadSupportCommands`, 'ListSupportActions' function to return sorted slice of strings. Signed-off-by: VaibhavMalik4187 <vaibhavmalik2018@gmail.com>
54 lines
1.5 KiB
Go
54 lines
1.5 KiB
Go
package completion
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/kubescape/kubescape/v3/core/cautils"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var completionCmdExamples = fmt.Sprintf(`
|
|
# Enable BASH shell autocompletion
|
|
$ source <(%[1]s completion bash)
|
|
$ echo 'source <(%[1]s completion bash)' >> ~/.bashrc
|
|
|
|
# Enable ZSH shell autocompletion
|
|
$ source <(%[1]s completion zsh)
|
|
$ echo 'source <(%[1]s completion zsh)' >> "${fpath[1]}/_%[1]s"
|
|
`, cautils.ExecName())
|
|
|
|
func GetCompletionCmd() *cobra.Command {
|
|
completionCmd := &cobra.Command{
|
|
Use: "completion [bash|zsh|fish|powershell]",
|
|
Short: "Generate autocompletion script",
|
|
Long: "To load completions",
|
|
Example: completionCmdExamples,
|
|
DisableFlagsInUseLine: true,
|
|
ValidArgs: []string{"bash", "zsh", "fish", "powershell"},
|
|
Args: cobra.MatchAll(cobra.ExactArgs(1), cobra.OnlyValidArgs),
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
// Check if args array is not empty
|
|
if len(args) == 0 {
|
|
fmt.Println("No arguements provided.")
|
|
return
|
|
}
|
|
|
|
switch strings.ToLower(args[0]) {
|
|
case "bash":
|
|
cmd.Root().GenBashCompletion(os.Stdout)
|
|
case "zsh":
|
|
cmd.Root().GenZshCompletion(os.Stdout)
|
|
case "fish":
|
|
cmd.Root().GenFishCompletion(os.Stdout, true)
|
|
case "powershell":
|
|
cmd.Root().GenPowerShellCompletionWithDesc(os.Stdout)
|
|
default:
|
|
fmt.Printf("Invalid arguement %s", args[0])
|
|
}
|
|
},
|
|
}
|
|
return completionCmd
|
|
}
|