github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2025-08-17 15:50:10 +00:00
Code Issues Projects Releases Wiki Activity

Simplify plugin resolution check function

Browse Source
This commit is contained in:
Arda Güçlü 2023-05-15 08:51:46 +03:00
parent 4634073d0b
commit daebf1c735
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
staging/src/k8s.io/kubectl/pkg/cmd/cmd.go
View File

@ -142,7 +142,7 @@ func NewDefaultKubectlCommandWithArgs(o KubectlOptions) *cobra.Command {
if cmdutil.CmdPluginAsSubcommand.IsEnabled() {
// Command exists(e.g. kubectl create), but it is not certain that
// subcommand also exists (e.g. kubectl create networkpolicy)
if _, ok := GetAllowedCmdsAsSubcommandPlugins()[foundCmd.Name()]; ok {
if IsSubcommandPluginAllowed(foundCmd.Name()) {
var subcommand string
for _, arg := range foundArgs { // first "non-flag" argument as subcommand
if !strings.HasPrefix(arg, "-") {
@ -172,11 +172,12 @@ func NewDefaultKubectlCommandWithArgs(o KubectlOptions) *cobra.Command {
return cmd
}
// GetAllowedCmdsAsSubcommandPlugins returns the list of builtin commands
// that plugins are allowed to be used as subcommands only if the passed subcommand
// does not exist as builtin.
func GetAllowedCmdsAsSubcommandPlugins() map[string]struct{} {
return map[string]struct{}{"create": {}}
// IsSubcommandPluginAllowed returns the given command is allowed
// to use plugin as subcommand if the subcommand does not exist as builtin.
func IsSubcommandPluginAllowed(foundCmd string) bool {
allowedCmds := map[string]struct{}{"create": {}}
_, ok := allowedCmds[foundCmd]
return ok
}
// PluginHandler is capable of parsing command line arguments