32 lines
913 B
Go
32 lines
913 B
Go
package cmd
|
|
|
|
import (
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
func NewCompletionCmd() *cobra.Command {
|
|
var cmd = &cobra.Command{
|
|
Use: "completion [bash|zsh|fish|powershell]",
|
|
Short: "Generate completion script",
|
|
Long: "Generate completion script for various shells",
|
|
DisableFlagsInUseLine: true,
|
|
ValidArgs: []string{"bash", "zsh", "fish", "powershell"},
|
|
Args: cobra.ExactValidArgs(1),
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
switch args[0] {
|
|
case "bash":
|
|
return cmd.Root().GenBashCompletion(cmd.OutOrStdout())
|
|
case "zsh":
|
|
return cmd.Root().GenZshCompletion(cmd.OutOrStdout())
|
|
case "fish":
|
|
return cmd.Root().GenFishCompletion(cmd.OutOrStdout(), true)
|
|
case "powershell":
|
|
return cmd.Root().GenPowerShellCompletion(cmd.OutOrStdout())
|
|
default:
|
|
return nil
|
|
}
|
|
},
|
|
}
|
|
return cmd
|
|
}
|