Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Implementing autocompletion for karmadactl
Signed-off-by: zhzhuang-zju <m17799853869@163.com>
- Loading branch information
1 parent
4ac95b9
commit cd341dd
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,143 @@ | ||
| /* | ||
| Copyright 2024 The Karmada Authors. | ||
| Licensed under the Apache License, Version 2.0 (the "License"); | ||
| you may not use this file except in compliance with the License. | ||
| You may obtain a copy of the License at | ||
| http://www.apache.org/licenses/LICENSE-2.0 | ||
| Unless required by applicable law or agreed to in writing, software | ||
| distributed under the License is distributed on an "AS IS" BASIS, | ||
| WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||
| See the License for the specific language governing permissions and | ||
| limitations under the License. | ||
| */ | ||
|
|
||
| package completion | ||
|
|
||
| import ( | ||
| "fmt" | ||
| "io" | ||
|
|
||
| "github.com/spf13/cobra" | ||
|
|
||
| cmdutil "k8s.io/kubectl/pkg/cmd/util" | ||
| "k8s.io/kubectl/pkg/util/i18n" | ||
| "k8s.io/kubectl/pkg/util/templates" | ||
| ) | ||
|
|
||
| const defaultBoilerPlate = ` | ||
| # Copyright 2024 The Karmada Authors. | ||
| # | ||
| # Licensed under the Apache License, Version 2.0 (the "License"); | ||
| # you may not use this file except in compliance with the License. | ||
| # You may obtain a copy of the License at | ||
| # | ||
| # http://www.apache.org/licenses/LICENSE-2.0 | ||
| # Unless required by applicable law or agreed to in writing, software | ||
| # distributed under the License is distributed on an "AS IS" BASIS, | ||
| # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||
| # See the License for the specific language governing permissions and | ||
| # limitations under the License. | ||
| ` | ||
|
|
||
| var ( | ||
| completionLong = templates.LongDesc(i18n.T(` | ||
| Output shell completion code for the specified shell (bash, zsh, fish, or powershell). | ||
| The shell code must be evaluated to provide interactive | ||
| completion of kubectl commands. This can be done by sourcing it from | ||
| the .bash_profile. | ||
| Note for zsh users: [1] zsh completions are only supported in versions of zsh >= 5.2.`)) | ||
|
|
||
| completionExample = templates.Examples(i18n.T(` | ||
| # Installing bash completion on Linux | ||
| ## If bash-completion is not installed on Linux, install the 'bash-completion' package | ||
| ## via your distribution's package manager. | ||
| ## Load the %[1]s completion code for bash into the current shell | ||
| source <(%[1]s completion bash) | ||
| ## Write bash completion code to a file and source it from .bash_profile | ||
| %[1]s completion bash > ~/.kube/completion.bash.inc | ||
| printf " | ||
| # %[1]s shell completion | ||
| source '$HOME/.kube/completion.bash.inc' | ||
| " >> $HOME/.bash_profile | ||
| source $HOME/.bash_profile | ||
| # Load the %[1]s completion code for zsh[1] into the current shell | ||
| source <(%[1]s completion zsh) | ||
| # Set the %[1]s completion code for zsh[1] to autoload on startup | ||
| %[1]s completion zsh > "${fpath[1]}/%[1]s"`)) | ||
| ) | ||
|
|
||
| var ( | ||
| completionShells = map[string]func(out io.Writer, boilerPlate string, cmd *cobra.Command) error{ | ||
| "bash": runCompletionBash, | ||
| "zsh": runCompletionZsh, | ||
| } | ||
| ) | ||
|
|
||
| // NewCmdCompletion creates the `completion` command | ||
| func NewCmdCompletion(parentCommand string, out io.Writer, boilerPlate string) *cobra.Command { | ||
| shells := []string{} | ||
| for s := range completionShells { | ||
| shells = append(shells, s) | ||
| } | ||
|
|
||
| cmd := &cobra.Command{ | ||
| Use: "completion SHELL", | ||
| DisableFlagsInUseLine: true, | ||
| Short: "Output shell completion code for the specified shell (bash, zsh)", | ||
| Long: completionLong, | ||
| Example: fmt.Sprintf(completionExample, parentCommand), | ||
| Run: func(cmd *cobra.Command, args []string) { | ||
| cmdutil.CheckErr(RunCompletion(out, boilerPlate, cmd, args)) | ||
| }, | ||
| ValidArgs: shells, | ||
| } | ||
|
|
||
| return cmd | ||
| } | ||
|
|
||
| // RunCompletion checks given arguments and executes command | ||
| func RunCompletion(out io.Writer, boilerPlate string, cmd *cobra.Command, args []string) error { | ||
| if len(args) == 0 { | ||
| return cmdutil.UsageErrorf(cmd, "Shell not specified.") | ||
| } | ||
| if len(args) > 1 { | ||
| return cmdutil.UsageErrorf(cmd, "Too many arguments. Expected only the shell type.") | ||
| } | ||
| run, found := completionShells[args[0]] | ||
| if !found { | ||
| return cmdutil.UsageErrorf(cmd, "Unsupported shell type %q.", args[0]) | ||
| } | ||
|
|
||
| return run(out, boilerPlate, cmd.Parent()) | ||
| } | ||
|
|
||
| func runCompletionBash(out io.Writer, boilerPlate string, cmd *cobra.Command) error { | ||
| if len(boilerPlate) == 0 { | ||
| boilerPlate = defaultBoilerPlate | ||
| } | ||
| if _, err := out.Write([]byte(boilerPlate)); err != nil { | ||
| return err | ||
| } | ||
|
|
||
| return cmd.GenBashCompletionV2(out, true) | ||
| } | ||
|
|
||
| func runCompletionZsh(out io.Writer, boilerPlate string, cmd *cobra.Command) error { | ||
| zshHead := fmt.Sprintf("#compdef %[1]s\ncompdef _%[1]s %[1]s\n", cmd.Name()) | ||
| out.Write([]byte(zshHead)) | ||
|
|
||
| if len(boilerPlate) == 0 { | ||
| boilerPlate = defaultBoilerPlate | ||
| } | ||
| if _, err := out.Write([]byte(boilerPlate)); err != nil { | ||
| return err | ||
| } | ||
|
|
||
| return cmd.GenZshCompletion(out) | ||
| } | ||
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.