github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2026-01-05 23:47:50 +00:00
Code Issues Projects Releases Wiki Activity

Flag support in kubectl plugins

Browse Source
This commit is contained in:
Fabiano Franz
2017-06-09 14:19:55 -03:00
parent ebf24c14a9
commit 71cbad7cbb
6 changed files with 208 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
pkg/kubectl/cmd/plugin.go
View File

@@ -114,6 +114,10 @@ func NewCmdForPlugin(f cmdutil.Factory, plugin *plugins.Plugin, runner plugins.P
},
}
for _, flag := range plugin.Flags {
cmd.Flags().StringP(flag.Name, flag.Shorthand, flag.DefValue, flag.Desc)
}
for _, childPlugin := range plugin.Tree {
cmd.AddCommand(NewCmdForPlugin(f, childPlugin, runner, in, out, errout))
}
@@ -126,10 +130,14 @@ type flagsPluginEnvProvider struct {
}
func (p *flagsPluginEnvProvider) Env() (plugins.EnvList, error) {
prefix := "KUBECTL_PLUGINS_GLOBAL_FLAG_"
globalPrefix := "KUBECTL_PLUGINS_GLOBAL_FLAG_"
env := plugins.EnvList{}
p.cmd.Flags().VisitAll(func(flag *pflag.Flag) {
env = append(env, plugins.FlagToEnv(flag, prefix))
p.cmd.InheritedFlags().VisitAll(func(flag *pflag.Flag) {
env = append(env, plugins.FlagToEnv(flag, globalPrefix))
})
localPrefix := "KUBECTL_PLUGINS_LOCAL_FLAG_"
p.cmd.LocalFlags().VisitAll(func(flag *pflag.Flag) {
env = append(env, plugins.FlagToEnv(flag, localPrefix))
})
return env, nil
}