| // Copyright 2013-2023 The Cobra 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 cobra is a commander providing a simple interface to create powerful modern CLI interfaces. | |
| // In addition to providing an interface, Cobra simultaneously provides a controller to organize your application code. | |
| package cobra | |
| import ( | |
| "bytes" | |
| "context" | |
| "errors" | |
| "fmt" | |
| "io" | |
| "os" | |
| "path/filepath" | |
| "sort" | |
| "strings" | |
| flag "github.com/spf13/pflag" | |
| ) | |
| const ( | |
| FlagSetByCobraAnnotation = "cobra_annotation_flag_set_by_cobra" | |
| CommandDisplayNameAnnotation = "cobra_annotation_command_display_name" | |
| helpFlagName = "help" | |
| helpCommandName = "help" | |
| ) | |
| // FParseErrWhitelist configures Flag parse errors to be ignored | |
| type FParseErrWhitelist flag.ParseErrorsAllowlist | |
| // Group Structure to manage groups for commands | |
| type Group struct { | |
| ID string | |
| Title string | |
| } | |
| // Command is just that, a command for your application. | |
| // E.g. 'go run ...' - 'run' is the command. Cobra requires | |
| // you to define the usage and description as part of your command | |
| // definition to ensure usability. | |
| type Command struct { | |
| // Use is the one-line usage message. | |
| // Recommended syntax is as follows: | |
| // [ ] identifies an optional argument. Arguments that are not enclosed in brackets are required. | |
| // ... indicates that you can specify multiple values for the previous argument. | |
| // | indicates mutually exclusive information. You can use the argument to the left of the separator or the | |
| // argument to the right of the separator. You cannot use both arguments in a single use of the command. | |
| // { } delimits a set of mutually exclusive arguments when one of the arguments is required. If the arguments are | |
| // optional, they are enclosed in brackets ([ ]). | |
| // Example: add [-F file | -D dir]... [-f format] profile | |
| Use string | |
| // Aliases is an array of aliases that can be used instead of the first word in Use. | |
| Aliases []string | |
| // SuggestFor is an array of command names for which this command will be suggested - | |
| // similar to aliases but only suggests. | |
| SuggestFor []string | |
| // Short is the short description shown in the 'help' output. | |
| Short string | |
| // The group id under which this subcommand is grouped in the 'help' output of its parent. | |
| GroupID string | |
| // Long is the long message shown in the 'help <this-command>' output. | |
| Long string | |
| // Example is examples of how to use the command. | |
| Example string | |
| // ValidArgs is list of all valid non-flag arguments that are accepted in shell completions | |
| ValidArgs []Completion | |
| // ValidArgsFunction is an optional function that provides valid non-flag arguments for shell completion. | |
| // It is a dynamic version of using ValidArgs. | |
| // Only one of ValidArgs and ValidArgsFunction can be used for a command. | |
| ValidArgsFunction CompletionFunc | |
| // Expected arguments | |
| Args PositionalArgs | |
| // ArgAliases is List of aliases for ValidArgs. | |
| // These are not suggested to the user in the shell completion, | |
| // but accepted if entered manually. | |
| ArgAliases []string | |
| // BashCompletionFunction is custom bash functions used by the legacy bash autocompletion generator. | |
| // For portability with other shells, it is recommended to instead use ValidArgsFunction | |
| BashCompletionFunction string | |
| // Deprecated defines, if this command is deprecated and should print this string when used. | |
| Deprecated string | |
| // Annotations are key/value pairs that can be used by applications to identify or | |
| // group commands or set special options. | |
| Annotations map[string]string | |
| // Version defines the version for this command. If this value is non-empty and the command does not | |
| // define a "version" flag, a "version" boolean flag will be added to the command and, if specified, | |
| // will print content of the "Version" variable. A shorthand "v" flag will also be added if the | |
| // command does not define one. | |
| Version string | |
| // The *Run functions are executed in the following order: | |
| // * PersistentPreRun() | |
| // * PreRun() | |
| // * Run() | |
| // * PostRun() | |
| // * PersistentPostRun() | |
| // All functions get the same args, the arguments after the command name. | |
| // The *PreRun and *PostRun functions will only be executed if the Run function of the current | |
| // command has been declared. | |
| // | |
| // PersistentPreRun: children of this command will inherit and execute. | |
| PersistentPreRun func(cmd *Command, args []string) | |
| // PersistentPreRunE: PersistentPreRun but returns an error. | |
| PersistentPreRunE func(cmd *Command, args []string) error | |
| // PreRun: children of this command will not inherit. | |
| PreRun func(cmd *Command, args []string) | |
| // PreRunE: PreRun but returns an error. | |
| PreRunE func(cmd *Command, args []string) error | |
| // Run: Typically the actual work function. Most commands will only implement this. | |
| Run func(cmd *Command, args []string) | |
| // RunE: Run but returns an error. | |
| RunE func(cmd *Command, args []string) error | |
| // PostRun: run after the Run command. | |
| PostRun func(cmd *Command, args []string) | |
| // PostRunE: PostRun but returns an error. | |
| PostRunE func(cmd *Command, args []string) error | |
| // PersistentPostRun: children of this command will inherit and execute after PostRun. | |
| PersistentPostRun func(cmd *Command, args []string) | |
| // PersistentPostRunE: PersistentPostRun but returns an error. | |
| PersistentPostRunE func(cmd *Command, args []string) error | |
| // groups for subcommands | |
| commandgroups []*Group | |
| // args is actual args parsed from flags. | |
| args []string | |
| // flagErrorBuf contains all error messages from pflag. | |
| flagErrorBuf *bytes.Buffer | |
| // flags is full set of flags. | |
| flags *flag.FlagSet | |
| // pflags contains persistent flags. | |
| pflags *flag.FlagSet | |
| // lflags contains local flags. | |
| // This field does not represent internal state, it's used as a cache to optimise LocalFlags function call | |
| lflags *flag.FlagSet | |
| // iflags contains inherited flags. | |
| // This field does not represent internal state, it's used as a cache to optimise InheritedFlags function call | |
| iflags *flag.FlagSet | |
| // parentsPflags is all persistent flags of cmd's parents. | |
| parentsPflags *flag.FlagSet | |
| // globNormFunc is the global normalization function | |
| // that we can use on every pflag set and children commands | |
| globNormFunc func(f *flag.FlagSet, name string) flag.NormalizedName | |
| // usageFunc is usage func defined by user. | |
| usageFunc func(*Command) error | |
| // usageTemplate is usage template defined by user. | |
| usageTemplate *tmplFunc | |
| // flagErrorFunc is func defined by user and it's called when the parsing of | |
| // flags returns an error. | |
| flagErrorFunc func(*Command, error) error | |
| // helpTemplate is help template defined by user. | |
| helpTemplate *tmplFunc | |
| // helpFunc is help func defined by user. | |
| helpFunc func(*Command, []string) | |
| // helpCommand is command with usage 'help'. If it's not defined by user, | |
| // cobra uses default help command. | |
| helpCommand *Command | |
| // helpCommandGroupID is the group id for the helpCommand | |
| helpCommandGroupID string | |
| // completionCommandGroupID is the group id for the completion command | |
| completionCommandGroupID string | |
| // versionTemplate is the version template defined by user. | |
| versionTemplate *tmplFunc | |
| // versionFunc is the version func defined by user. | |
| versionFunc func(*Command) error | |
| // errPrefix is the error message prefix defined by user. | |
| errPrefix string | |
| // inReader is a reader defined by the user that replaces stdin | |
| inReader io.Reader | |
| // outWriter is a writer defined by the user that replaces stdout | |
| outWriter io.Writer | |
| // errWriter is a writer defined by the user that replaces stderr | |
| errWriter io.Writer | |
| // FParseErrWhitelist flag parse errors to be ignored | |
| FParseErrWhitelist FParseErrWhitelist | |
| // CompletionOptions is a set of options to control the handling of shell completion | |
| CompletionOptions CompletionOptions | |
| // commandsAreSorted defines, if command slice are sorted or not. | |
| commandsAreSorted bool | |
| // commandCalledAs is the name or alias value used to call this command. | |
| commandCalledAs struct { | |
| name string | |
| called bool | |
| } | |
| ctx context.Context | |
| // commands is the list of commands supported by this program. | |
| commands []*Command | |
| // parent is a parent command for this command. | |
| parent *Command | |
| // Max lengths of commands' string lengths for use in padding. | |
| commandsMaxUseLen int | |
| commandsMaxCommandPathLen int | |
| commandsMaxNameLen int | |
| // TraverseChildren parses flags on all parents before executing child command. | |
| TraverseChildren bool | |
| // Hidden defines, if this command is hidden and should NOT show up in the list of available commands. | |
| Hidden bool | |
| // SilenceErrors is an option to quiet errors down stream. | |
| SilenceErrors bool | |
| // SilenceUsage is an option to silence usage when an error occurs. | |
| SilenceUsage bool | |
| // DisableFlagParsing disables the flag parsing. | |
| // If this is true all flags will be passed to the command as arguments. | |
| DisableFlagParsing bool | |
| // DisableAutoGenTag defines, if gen tag ("Auto generated by spf13/cobra...") | |
| // will be printed by generating docs for this command. | |
| DisableAutoGenTag bool | |
| // DisableFlagsInUseLine will disable the addition of [flags] to the usage | |
| // line of a command when printing help or generating docs | |
| DisableFlagsInUseLine bool | |
| // DisableSuggestions disables the suggestions based on Levenshtein distance | |
| // that go along with 'unknown command' messages. | |
| DisableSuggestions bool | |
| // SuggestionsMinimumDistance defines minimum levenshtein distance to display suggestions. | |
| // Must be > 0. | |
| SuggestionsMinimumDistance int | |
| } | |
| // Context returns underlying command context. If command was executed | |
| // with ExecuteContext or the context was set with SetContext, the | |
| // previously set context will be returned. Otherwise, nil is returned. | |
| // | |
| // Notice that a call to Execute and ExecuteC will replace a nil context of | |
| // a command with a context.Background, so a background context will be | |
| // returned by Context after one of these functions has been called. | |
| func (c *Command) Context() context.Context { | |
| return c.ctx | |
| } | |
| // SetContext sets context for the command. This context will be overwritten by | |
| // Command.ExecuteContext or Command.ExecuteContextC. | |
| func (c *Command) SetContext(ctx context.Context) { | |
| c.ctx = ctx | |
| } | |
| // SetArgs sets arguments for the command. It is set to os.Args[1:] by default, if desired, can be overridden | |
| // particularly useful when testing. | |
| func (c *Command) SetArgs(a []string) { | |
| c.args = a | |
| } | |
| // SetOutput sets the destination for usage and error messages. | |
| // If output is nil, os.Stderr is used. | |
| // | |
| // Deprecated: Use SetOut and/or SetErr instead | |
| func (c *Command) SetOutput(output io.Writer) { | |
| c.outWriter = output | |
| c.errWriter = output | |
| } | |
| // SetOut sets the destination for usage messages. | |
| // If newOut is nil, os.Stdout is used. | |
| func (c *Command) SetOut(newOut io.Writer) { | |
| c.outWriter = newOut | |
| } | |
| // SetErr sets the destination for error messages. | |
| // If newErr is nil, os.Stderr is used. | |
| func (c *Command) SetErr(newErr io.Writer) { | |
| c.errWriter = newErr | |
| } | |
| // SetIn sets the source for input data | |
| // If newIn is nil, os.Stdin is used. | |
| func (c *Command) SetIn(newIn io.Reader) { | |
| c.inReader = newIn | |
| } | |
| // SetUsageFunc sets usage function. Usage can be defined by application. | |
| func (c *Command) SetUsageFunc(f func(*Command) error) { | |
| c.usageFunc = f | |
| } | |
| // SetUsageTemplate sets usage template. Can be defined by Application. | |
| func (c *Command) SetUsageTemplate(s string) { | |
| if s == "" { | |
| c.usageTemplate = nil | |
| return | |
| } | |
| c.usageTemplate = tmpl(s) | |
| } | |
| // SetFlagErrorFunc sets a function to generate an error when flag parsing | |
| // fails. | |
| func (c *Command) SetFlagErrorFunc(f func(*Command, error) error) { | |
| c.flagErrorFunc = f | |
| } | |
| // SetHelpFunc sets help function. Can be defined by Application. | |
| func (c *Command) SetHelpFunc(f func(*Command, []string)) { | |
| c.helpFunc = f | |
| } | |
| // SetHelpCommand sets help command. | |
| func (c *Command) SetHelpCommand(cmd *Command) { | |
| c.helpCommand = cmd | |
| } | |
| // SetHelpCommandGroupID sets the group id of the help command. | |
| func (c *Command) SetHelpCommandGroupID(groupID string) { | |
| if c.helpCommand != nil { | |
| c.helpCommand.GroupID = groupID | |
| } | |
| // helpCommandGroupID is used if no helpCommand is defined by the user | |
| c.helpCommandGroupID = groupID | |
| } | |
| // SetCompletionCommandGroupID sets the group id of the completion command. | |
| func (c *Command) SetCompletionCommandGroupID(groupID string) { | |
| // completionCommandGroupID is used if no completion command is defined by the user | |
| c.Root().completionCommandGroupID = groupID | |
| } | |
| // SetHelpTemplate sets help template to be used. Application can use it to set custom template. | |
| func (c *Command) SetHelpTemplate(s string) { | |
| if s == "" { | |
| c.helpTemplate = nil | |
| return | |
| } | |
| c.helpTemplate = tmpl(s) | |
| } | |
| // SetVersionFunc sets version function. Can be defined by Application. | |
| // | |
| // Setting this means that [Command.SetVersionTemplate] will not have any | |
| // effect any more. | |
| func (c *Command) SetVersionFunc(f func(*Command) error) { | |
| c.versionFunc = f | |
| } | |
| // SetVersionTemplate sets version template to be used. Application can use it to set custom template. | |
| func (c *Command) SetVersionTemplate(s string) { | |
| if s == "" { | |
| c.versionTemplate = nil | |
| return | |
| } | |
| c.versionTemplate = tmpl(s) | |
| } | |
| // SetErrPrefix sets error message prefix to be used. Application can use it to set custom prefix. | |
| func (c *Command) SetErrPrefix(s string) { | |
| c.errPrefix = s | |
| } | |
| // SetGlobalNormalizationFunc sets a normalization function to all flag sets and also to child commands. | |
| // The user should not have a cyclic dependency on commands. | |
| func (c *Command) SetGlobalNormalizationFunc(n func(f *flag.FlagSet, name string) flag.NormalizedName) { | |
| c.Flags().SetNormalizeFunc(n) | |
| c.PersistentFlags().SetNormalizeFunc(n) | |
| c.globNormFunc = n | |
| for _, command := range c.commands { | |
| command.SetGlobalNormalizationFunc(n) | |
| } | |
| } | |
| // OutOrStdout returns output to stdout. | |
| func (c *Command) OutOrStdout() io.Writer { | |
| return c.getOut(os.Stdout) | |
| } | |
| // OutOrStderr returns output to stderr | |
| func (c *Command) OutOrStderr() io.Writer { | |
| return c.getOut(os.Stderr) | |
| } | |
| // ErrOrStderr returns output to stderr | |
| func (c *Command) ErrOrStderr() io.Writer { | |
| return c.getErr(os.Stderr) | |
| } | |
| // InOrStdin returns input to stdin | |
| func (c *Command) InOrStdin() io.Reader { | |
| return c.getIn(os.Stdin) | |
| } | |
| func (c *Command) getOut(def io.Writer) io.Writer { | |
| if c.outWriter != nil { | |
| return c.outWriter | |
| } | |
| if c.HasParent() { | |
| return c.parent.getOut(def) | |
| } | |
| return def | |
| } | |
| func (c *Command) getErr(def io.Writer) io.Writer { | |
| if c.errWriter != nil { | |
| return c.errWriter | |
| } | |
| if c.HasParent() { | |
| return c.parent.getErr(def) | |
| } | |
| return def | |
| } | |
| func (c *Command) getIn(def io.Reader) io.Reader { | |
| if c.inReader != nil { | |
| return c.inReader | |
| } | |
| if c.HasParent() { | |
| return c.parent.getIn(def) | |
| } | |
| return def | |
| } | |
| // UsageFunc returns either the function set by SetUsageFunc for this command | |
| // or a parent, or it returns a default usage function. | |
| func (c *Command) UsageFunc() (f func(*Command) error) { | |
| if c.usageFunc != nil { | |
| return c.usageFunc | |
| } | |
| if c.HasParent() { | |
| return c.Parent().UsageFunc() | |
| } | |
| return func(c *Command) error { | |
| c.mergePersistentFlags() | |
| fn := c.getUsageTemplateFunc() | |
| err := fn(c.OutOrStderr(), c) | |
| if err != nil { | |
| c.PrintErrln(err) | |
| } | |
| return err | |
| } | |
| } | |
| // getUsageTemplateFunc returns the usage template function for the command | |
| // going up the command tree if necessary. | |
| func (c *Command) getUsageTemplateFunc() func(w io.Writer, data interface{}) error { | |
| if c.usageTemplate != nil { | |
| return c.usageTemplate.fn | |
| } | |
| if c.HasParent() { | |
| return c.parent.getUsageTemplateFunc() | |
| } | |
| return defaultUsageFunc | |
| } | |
| // Usage puts out the usage for the command. | |
| // Used when a user provides invalid input. | |
| // Can be defined by user by overriding UsageFunc. | |
| func (c *Command) Usage() error { | |
| return c.UsageFunc()(c) | |
| } | |
| // HelpFunc returns either the function set by SetHelpFunc for this command | |
| // or a parent, or it returns a function with default help behavior. | |
| func (c *Command) HelpFunc() func(*Command, []string) { | |
| if c.helpFunc != nil { | |
| return c.helpFunc | |
| } | |
| if c.HasParent() { | |
| return c.Parent().HelpFunc() | |
| } | |
| return func(c *Command, a []string) { | |
| c.mergePersistentFlags() | |
| fn := c.getHelpTemplateFunc() | |
| // The help should be sent to stdout | |
| // See https://github.com/spf13/cobra/issues/1002 | |
| err := fn(c.OutOrStdout(), c) | |
| if err != nil { | |
| c.PrintErrln(err) | |
| } | |
| } | |
| } | |
| // getHelpTemplateFunc returns the help template function for the command | |
| // going up the command tree if necessary. | |
| func (c *Command) getHelpTemplateFunc() func(w io.Writer, data interface{}) error { | |
| if c.helpTemplate != nil { | |
| return c.helpTemplate.fn | |
| } | |
| if c.HasParent() { | |
| return c.parent.getHelpTemplateFunc() | |
| } | |
| return defaultHelpFunc | |
| } | |
| // Help puts out the help for the command. | |
| // Used when a user calls help [command]. | |
| // Can be defined by user by overriding HelpFunc. | |
| func (c *Command) Help() error { | |
| c.HelpFunc()(c, []string{}) | |
| return nil | |
| } | |
| // UsageString returns usage string. | |
| func (c *Command) UsageString() string { | |
| // Storing normal writers | |
| tmpOutput := c.outWriter | |
| tmpErr := c.errWriter | |
| bb := new(bytes.Buffer) | |
| c.outWriter = bb | |
| c.errWriter = bb | |
| CheckErr(c.Usage()) | |
| // Setting things back to normal | |
| c.outWriter = tmpOutput | |
| c.errWriter = tmpErr | |
| return bb.String() | |
| } | |
| // FlagErrorFunc returns either the function set by SetFlagErrorFunc for this | |
| // command or a parent, or it returns a function which returns the original | |
| // error. | |
| func (c *Command) FlagErrorFunc() (f func(*Command, error) error) { | |
| if c.flagErrorFunc != nil { | |
| return c.flagErrorFunc | |
| } | |
| if c.HasParent() { | |
| return c.parent.FlagErrorFunc() | |
| } | |
| return func(c *Command, err error) error { | |
| return err | |
| } | |
| } | |
| var minUsagePadding = 25 | |
| // UsagePadding return padding for the usage. | |
| func (c *Command) UsagePadding() int { | |
| if c.parent == nil || minUsagePadding > c.parent.commandsMaxUseLen { | |
| return minUsagePadding | |
| } | |
| return c.parent.commandsMaxUseLen | |
| } | |
| var minCommandPathPadding = 11 | |
| // CommandPathPadding return padding for the command path. | |
| func (c *Command) CommandPathPadding() int { | |
| if c.parent == nil || minCommandPathPadding > c.parent.commandsMaxCommandPathLen { | |
| return minCommandPathPadding | |
| } | |
| return c.parent.commandsMaxCommandPathLen | |
| } | |
| var minNamePadding = 11 | |
| // NamePadding returns padding for the name. | |
| func (c *Command) NamePadding() int { | |
| if c.parent == nil || minNamePadding > c.parent.commandsMaxNameLen { | |
| return minNamePadding | |
| } | |
| return c.parent.commandsMaxNameLen | |
| } | |
| // UsageTemplate returns usage template for the command. | |
| // This function is kept for backwards-compatibility reasons. | |
| func (c *Command) UsageTemplate() string { | |
| if c.usageTemplate != nil { | |
| return c.usageTemplate.tmpl | |
| } | |
| if c.HasParent() { | |
| return c.parent.UsageTemplate() | |
| } | |
| return defaultUsageTemplate | |
| } | |
| // HelpTemplate return help template for the command. | |
| // This function is kept for backwards-compatibility reasons. | |
| func (c *Command) HelpTemplate() string { | |
| if c.helpTemplate != nil { | |
| return c.helpTemplate.tmpl | |
| } | |
| if c.HasParent() { | |
| return c.parent.HelpTemplate() | |
| } | |
| return defaultHelpTemplate | |
| } | |
| // VersionTemplate return version template for the command. | |
| // This function is kept for backwards-compatibility reasons. | |
| func (c *Command) VersionTemplate() string { | |
| if c.versionTemplate != nil { | |
| return c.versionTemplate.tmpl | |
| } | |
| if c.HasParent() { | |
| return c.parent.VersionTemplate() | |
| } | |
| return defaultVersionTemplate | |
| } | |
| // getVersionTemplateFunc returns the version template function for the command | |
| // going up the command tree if necessary. | |
| func (c *Command) getVersionTemplateFunc() func(w io.Writer, data interface{}) error { | |
| if c.versionTemplate != nil { | |
| return c.versionTemplate.fn | |
| } | |
| if c.HasParent() { | |
| return c.parent.getVersionTemplateFunc() | |
| } | |
| return defaultVersionFunc | |
| } | |
| // getVersionFunc returns the version function for the command going up the | |
| // command tree if necessary. | |
| func (c *Command) getVersionFunc() func(*Command) error { | |
| if c.versionFunc != nil { | |
| return c.versionFunc | |
| } | |
| if c.HasParent() { | |
| return c.parent.getVersionFunc() | |
| } | |
| return nil | |
| } | |
| // ErrPrefix return error message prefix for the command | |
| func (c *Command) ErrPrefix() string { | |
| if c.errPrefix != "" { | |
| return c.errPrefix | |
| } | |
| if c.HasParent() { | |
| return c.parent.ErrPrefix() | |
| } | |
| return "Error:" | |
| } | |
| func hasNoOptDefVal(name string, fs *flag.FlagSet) bool { | |
| flag := fs.Lookup(name) | |
| if flag == nil { | |
| return false | |
| } | |
| return flag.NoOptDefVal != "" | |
| } | |
| func shortHasNoOptDefVal(name string, fs *flag.FlagSet) bool { | |
| if len(name) == 0 { | |
| return false | |
| } | |
| flag := fs.ShorthandLookup(name[:1]) | |
| if flag == nil { | |
| return false | |
| } | |
| return flag.NoOptDefVal != "" | |
| } | |
| func stripFlags(args []string, c *Command) []string { | |
| if len(args) == 0 { | |
| return args | |
| } | |
| c.mergePersistentFlags() | |
| commands := []string{} | |
| flags := c.Flags() | |
| Loop: | |
| for len(args) > 0 { | |
| s := args[0] | |
| args = args[1:] | |
| switch { | |
| case s == "--": | |
| // "--" terminates the flags | |
| break Loop | |
| case strings.HasPrefix(s, "--") && !strings.Contains(s, "=") && !hasNoOptDefVal(s[2:], flags): | |
| // If '--flag arg' then | |
| // delete arg from args. | |
| fallthrough // (do the same as below) | |
| case strings.HasPrefix(s, "-") && !strings.Contains(s, "=") && len(s) == 2 && !shortHasNoOptDefVal(s[1:], flags): | |
| // If '-f arg' then | |
| // delete 'arg' from args or break the loop if len(args) <= 1. | |
| if len(args) <= 1 { | |
| break Loop | |
| } else { | |
| args = args[1:] | |
| continue | |
| } | |
| case s != "" && !strings.HasPrefix(s, "-"): | |
| commands = append(commands, s) | |
| } | |
| } | |
| return commands | |
| } | |
| // argsMinusFirstX removes only the first x from args. Otherwise, commands that look like | |
| // openshift admin policy add-role-to-user admin my-user, lose the admin argument (arg[4]). | |
| // Special care needs to be taken not to remove a flag value. | |
| func (c *Command) argsMinusFirstX(args []string, x string) []string { | |
| if len(args) == 0 { | |
| return args | |
| } | |
| c.mergePersistentFlags() | |
| flags := c.Flags() | |
| Loop: | |
| for pos := 0; pos < len(args); pos++ { | |
| s := args[pos] | |
| switch { | |
| case s == "--": | |
| // -- means we have reached the end of the parseable args. Break out of the loop now. | |
| break Loop | |
| case strings.HasPrefix(s, "--") && !strings.Contains(s, "=") && !hasNoOptDefVal(s[2:], flags): | |
| fallthrough | |
| case strings.HasPrefix(s, "-") && !strings.Contains(s, "=") && len(s) == 2 && !shortHasNoOptDefVal(s[1:], flags): | |
| // This is a flag without a default value, and an equal sign is not used. Increment pos in order to skip | |
| // over the next arg, because that is the value of this flag. | |
| pos++ | |
| continue | |
| case !strings.HasPrefix(s, "-"): | |
| // This is not a flag or a flag value. Check to see if it matches what we're looking for, and if so, | |
| // return the args, excluding the one at this position. | |
| if s == x { | |
| ret := make([]string, 0, len(args)-1) | |
| ret = append(ret, args[:pos]...) | |
| ret = append(ret, args[pos+1:]...) | |
| return ret | |
| } | |
| } | |
| } | |
| return args | |
| } | |
| func isFlagArg(arg string) bool { | |
| return ((len(arg) >= 3 && arg[0:2] == "--") || | |
| (len(arg) >= 2 && arg[0] == '-' && arg[1] != '-')) | |
| } | |
| // Find the target command given the args and command tree | |
| // Meant to be run on the highest node. Only searches down. | |
| func (c *Command) Find(args []string) (*Command, []string, error) { | |
| var innerfind func(*Command, []string) (*Command, []string) | |
| innerfind = func(c *Command, innerArgs []string) (*Command, []string) { | |
| argsWOflags := stripFlags(innerArgs, c) | |
| if len(argsWOflags) == 0 { | |
| return c, innerArgs | |
| } | |
| nextSubCmd := argsWOflags[0] | |
| cmd := c.findNext(nextSubCmd) | |
| if cmd != nil { | |
| return innerfind(cmd, c.argsMinusFirstX(innerArgs, nextSubCmd)) | |
| } | |
| return c, innerArgs | |
| } | |
| commandFound, a := innerfind(c, args) | |
| if commandFound.Args == nil { | |
| return commandFound, a, legacyArgs(commandFound, stripFlags(a, commandFound)) | |
| } | |
| return commandFound, a, nil | |
| } | |
| func (c *Command) findSuggestions(arg string) string { | |
| if c.DisableSuggestions { | |
| return "" | |
| } | |
| if c.SuggestionsMinimumDistance <= 0 { | |
| c.SuggestionsMinimumDistance = 2 | |
| } | |
| var sb strings.Builder | |
| if suggestions := c.SuggestionsFor(arg); len(suggestions) > 0 { | |
| sb.WriteString("\n\nDid you mean this?\n") | |
| for _, s := range suggestions { | |
| _, _ = fmt.Fprintf(&sb, "\t%v\n", s) | |
| } | |
| } | |
| return sb.String() | |
| } | |
| func (c *Command) findNext(next string) *Command { | |
| matches := make([]*Command, 0) | |
| for _, cmd := range c.commands { | |
| if commandNameMatches(cmd.Name(), next) || cmd.HasAlias(next) { | |
| cmd.commandCalledAs.name = next | |
| return cmd | |
| } | |
| if EnablePrefixMatching && cmd.hasNameOrAliasPrefix(next) { | |
| matches = append(matches, cmd) | |
| } | |
| } | |
| if len(matches) == 1 { | |
| // Temporarily disable gosec G602, which produces a false positive. | |
| // See https://github.com/securego/gosec/issues/1005. | |
| return matches[0] // #nosec G602 | |
| } | |
| return nil | |
| } | |
| // Traverse the command tree to find the command, and parse args for | |
| // each parent. | |
| func (c *Command) Traverse(args []string) (*Command, []string, error) { | |
| flags := []string{} | |
| inFlag := false | |
| for i, arg := range args { | |
| switch { | |
| // A long flag with a space separated value | |
| case strings.HasPrefix(arg, "--") && !strings.Contains(arg, "="): | |
| // TODO: this isn't quite right, we should really check ahead for 'true' or 'false' | |
| inFlag = !hasNoOptDefVal(arg[2:], c.Flags()) | |
| flags = append(flags, arg) | |
| continue | |
| // A short flag with a space separated value | |
| case strings.HasPrefix(arg, "-") && !strings.Contains(arg, "=") && len(arg) == 2 && !shortHasNoOptDefVal(arg[1:], c.Flags()): | |
| inFlag = true | |
| flags = append(flags, arg) | |
| continue | |
| // The value for a flag | |
| case inFlag: | |
| inFlag = false | |
| flags = append(flags, arg) | |
| continue | |
| // A flag without a value, or with an `=` separated value | |
| case isFlagArg(arg): | |
| flags = append(flags, arg) | |
| continue | |
| } | |
| cmd := c.findNext(arg) | |
| if cmd == nil { | |
| return c, args, nil | |
| } | |
| if err := c.ParseFlags(flags); err != nil { | |
| return nil, args, err | |
| } | |
| return cmd.Traverse(args[i+1:]) | |
| } | |
| return c, args, nil | |
| } | |
| // SuggestionsFor provides suggestions for the typedName. | |
| func (c *Command) SuggestionsFor(typedName string) []string { | |
| suggestions := []string{} | |
| for _, cmd := range c.commands { | |
| if cmd.IsAvailableCommand() { | |
| levenshteinDistance := ld(typedName, cmd.Name(), true) | |
| suggestByLevenshtein := levenshteinDistance <= c.SuggestionsMinimumDistance | |
| suggestByPrefix := strings.HasPrefix(strings.ToLower(cmd.Name()), strings.ToLower(typedName)) | |
| if suggestByLevenshtein || suggestByPrefix { | |
| suggestions = append(suggestions, cmd.Name()) | |
| } | |
| for _, explicitSuggestion := range cmd.SuggestFor { | |
| if strings.EqualFold(typedName, explicitSuggestion) { | |
| suggestions = append(suggestions, cmd.Name()) | |
| } | |
| } | |
| } | |
| } | |
| return suggestions | |
| } | |
| // VisitParents visits all parents of the command and invokes fn on each parent. | |
| func (c *Command) VisitParents(fn func(*Command)) { | |
| if c.HasParent() { | |
| fn(c.Parent()) | |
| c.Parent().VisitParents(fn) | |
| } | |
| } | |
| // Root finds root command. | |
| func (c *Command) Root() *Command { | |
| if c.HasParent() { | |
| return c.Parent().Root() | |
| } | |
| return c | |
| } | |
| // ArgsLenAtDash will return the length of c.Flags().Args at the moment | |
| // when a -- was found during args parsing. | |
| func (c *Command) ArgsLenAtDash() int { | |
| return c.Flags().ArgsLenAtDash() | |
| } | |
| func (c *Command) execute(a []string) (err error) { | |
| if c == nil { | |
| return fmt.Errorf("called Execute() on a nil Command") | |
| } | |
| if len(c.Deprecated) > 0 { | |
| c.Printf("Command %q is deprecated, %s\n", c.Name(), c.Deprecated) | |
| } | |
| // initialize help and version flag at the last point possible to allow for user | |
| // overriding | |
| c.InitDefaultHelpFlag() | |
| c.InitDefaultVersionFlag() | |
| err = c.ParseFlags(a) | |
| if err != nil { | |
| return c.FlagErrorFunc()(c, err) | |
| } | |
| // If help is called, regardless of other flags, return we want help. | |
| // Also say we need help if the command isn't runnable. | |
| helpVal, err := c.Flags().GetBool(helpFlagName) | |
| if err != nil { | |
| // should be impossible to get here as we always declare a help | |
| // flag in InitDefaultHelpFlag() | |
| c.Println("\"help\" flag declared as non-bool. Please correct your code") | |
| return err | |
| } | |
| if helpVal { | |
| return flag.ErrHelp | |
| } | |
| // for back-compat, only add version flag behavior if version is defined | |
| if c.Version != "" { | |
| versionVal, err := c.Flags().GetBool("version") | |
| if err != nil { | |
| c.Println("\"version\" flag declared as non-bool. Please correct your code") | |
| return err | |
| } | |
| if versionVal { | |
| if fn := c.getVersionFunc(); fn != nil { | |
| return fn(c) | |
| } | |
| fn := c.getVersionTemplateFunc() | |
| err := fn(c.OutOrStdout(), c) | |
| if err != nil { | |
| c.Println(err) | |
| } | |
| return err | |
| } | |
| } | |
| if !c.Runnable() { | |
| return flag.ErrHelp | |
| } | |
| c.preRun() | |
| defer c.postRun() | |
| argWoFlags := c.Flags().Args() | |
| if c.DisableFlagParsing { | |
| argWoFlags = a | |
| } | |
| if err := c.ValidateArgs(argWoFlags); err != nil { | |
| return err | |
| } | |
| parents := make([]*Command, 0, 5) | |
| for p := c; p != nil; p = p.Parent() { | |
| if EnableTraverseRunHooks { | |
| // When EnableTraverseRunHooks is set: | |
| // - Execute all persistent pre-runs from the root parent till this command. | |
| // - Execute all persistent post-runs from this command till the root parent. | |
| parents = append([]*Command{p}, parents...) | |
| } else { | |
| // Otherwise, execute only the first found persistent hook. | |
| parents = append(parents, p) | |
| } | |
| } | |
| for _, p := range parents { | |
| if p.PersistentPreRunE != nil { | |
| if err := p.PersistentPreRunE(c, argWoFlags); err != nil { | |
| return err | |
| } | |
| if !EnableTraverseRunHooks { | |
| break | |
| } | |
| } else if p.PersistentPreRun != nil { | |
| p.PersistentPreRun(c, argWoFlags) | |
| if !EnableTraverseRunHooks { | |
| break | |
| } | |
| } | |
| } | |
| if c.PreRunE != nil { | |
| if err := c.PreRunE(c, argWoFlags); err != nil { | |
| return err | |
| } | |
| } else if c.PreRun != nil { | |
| c.PreRun(c, argWoFlags) | |
| } | |
| if err := c.ValidateRequiredFlags(); err != nil { | |
| return err | |
| } | |
| if err := c.ValidateFlagGroups(); err != nil { | |
| return err | |
| } | |
| if c.RunE != nil { | |
| if err := c.RunE(c, argWoFlags); err != nil { | |
| return err | |
| } | |
| } else { | |
| c.Run(c, argWoFlags) | |
| } | |
| if c.PostRunE != nil { | |
| if err := c.PostRunE(c, argWoFlags); err != nil { | |
| return err | |
| } | |
| } else if c.PostRun != nil { | |
| c.PostRun(c, argWoFlags) | |
| } | |
| for p := c; p != nil; p = p.Parent() { | |
| if p.PersistentPostRunE != nil { | |
| if err := p.PersistentPostRunE(c, argWoFlags); err != nil { | |
| return err | |
| } | |
| if !EnableTraverseRunHooks { | |
| break | |
| } | |
| } else if p.PersistentPostRun != nil { | |
| p.PersistentPostRun(c, argWoFlags) | |
| if !EnableTraverseRunHooks { | |
| break | |
| } | |
| } | |
| } | |
| return nil | |
| } | |
| func (c *Command) preRun() { | |
| for _, x := range initializers { | |
| x() | |
| } | |
| } | |
| func (c *Command) postRun() { | |
| for _, x := range finalizers { | |
| x() | |
| } | |
| } | |
| // ExecuteContext is the same as Execute(), but sets the ctx on the command. | |
| // Retrieve ctx by calling cmd.Context() inside your *Run lifecycle or ValidArgs | |
| // functions. | |
| func (c *Command) ExecuteContext(ctx context.Context) error { | |
| c.ctx = ctx | |
| return c.Execute() | |
| } | |
| // Execute uses the args (os.Args[1:] by default) | |
| // and run through the command tree finding appropriate matches | |
| // for commands and then corresponding flags. | |
| func (c *Command) Execute() error { | |
| _, err := c.ExecuteC() | |
| return err | |
| } | |
| // ExecuteContextC is the same as ExecuteC(), but sets the ctx on the command. | |
| // Retrieve ctx by calling cmd.Context() inside your *Run lifecycle or ValidArgs | |
| // functions. | |
| func (c *Command) ExecuteContextC(ctx context.Context) (*Command, error) { | |
| c.ctx = ctx | |
| return c.ExecuteC() | |
| } | |
| // ExecuteC executes the command. | |
| func (c *Command) ExecuteC() (cmd *Command, err error) { | |
| if c.ctx == nil { | |
| c.ctx = context.Background() | |
| } | |
| // Regardless of what command execute is called on, run on Root only | |
| if c.HasParent() { | |
| return c.Root().ExecuteC() | |
| } | |
| // windows hook | |
| if preExecHookFn != nil { | |
| preExecHookFn(c) | |
| } | |
| // initialize help at the last point to allow for user overriding | |
| c.InitDefaultHelpCmd() | |
| args := c.args | |
| // Workaround FAIL with "go test -v" or "cobra.test -test.v", see #155 | |
| if c.args == nil && filepath.Base(os.Args[0]) != "cobra.test" { | |
| args = os.Args[1:] | |
| } | |
| // initialize the __complete command to be used for shell completion | |
| c.initCompleteCmd(args) | |
| // initialize the default completion command | |
| c.InitDefaultCompletionCmd(args...) | |
| // Now that all commands have been created, let's make sure all groups | |
| // are properly created also | |
| c.checkCommandGroups() | |
| var flags []string | |
| if c.TraverseChildren { | |
| cmd, flags, err = c.Traverse(args) | |
| } else { | |
| cmd, flags, err = c.Find(args) | |
| } | |
| if err != nil { | |
| // If found parse to a subcommand and then failed, talk about the subcommand | |
| if cmd != nil { | |
| c = cmd | |
| } | |
| if !c.SilenceErrors { | |
| c.PrintErrln(c.ErrPrefix(), err.Error()) | |
| c.PrintErrf("Run '%v --help' for usage.\n", c.CommandPath()) | |
| } | |
| return c, err | |
| } | |
| cmd.commandCalledAs.called = true | |
| if cmd.commandCalledAs.name == "" { | |
| cmd.commandCalledAs.name = cmd.Name() | |
| } | |
| // We have to pass global context to children command | |
| // if context is present on the parent command. | |
| if cmd.ctx == nil { | |
| cmd.ctx = c.ctx | |
| } | |
| err = cmd.execute(flags) | |
| if err != nil { | |
| // Always show help if requested, even if SilenceErrors is in | |
| // effect | |
| if errors.Is(err, flag.ErrHelp) { | |
| cmd.HelpFunc()(cmd, args) | |
| return cmd, nil | |
| } | |
| // If root command has SilenceErrors flagged, | |
| // all subcommands should respect it | |
| if !cmd.SilenceErrors && !c.SilenceErrors { | |
| c.PrintErrln(cmd.ErrPrefix(), err.Error()) | |
| } | |
| // If root command has SilenceUsage flagged, | |
| // all subcommands should respect it | |
| if !cmd.SilenceUsage && !c.SilenceUsage { | |
| c.Println(cmd.UsageString()) | |
| } | |
| } | |
| return cmd, err | |
| } | |
| func (c *Command) ValidateArgs(args []string) error { | |
| if c.Args == nil { | |
| return ArbitraryArgs(c, args) | |
| } | |
| return c.Args(c, args) | |
| } | |
| // ValidateRequiredFlags validates all required flags are present and returns an error otherwise | |
| func (c *Command) ValidateRequiredFlags() error { | |
| if c.DisableFlagParsing { | |
| return nil | |
| } | |
| flags := c.Flags() | |
| missingFlagNames := []string{} | |
| flags.VisitAll(func(pflag *flag.Flag) { | |
| requiredAnnotation, found := pflag.Annotations[BashCompOneRequiredFlag] | |
| if !found { | |
| return | |
| } | |
| if (requiredAnnotation[0] == "true") && !pflag.Changed { | |
| missingFlagNames = append(missingFlagNames, pflag.Name) | |
| } | |
| }) | |
| if len(missingFlagNames) > 0 { | |
| return fmt.Errorf(`required flag(s) "%s" not set`, strings.Join(missingFlagNames, `", "`)) | |
| } | |
| return nil | |
| } | |
| // checkCommandGroups checks if a command has been added to a group that does not exists. | |
| // If so, we panic because it indicates a coding error that should be corrected. | |
| func (c *Command) checkCommandGroups() { | |
| for _, sub := range c.commands { | |
| // if Group is not defined let the developer know right away | |
| if sub.GroupID != "" && !c.ContainsGroup(sub.GroupID) { | |
| panic(fmt.Sprintf("group id '%s' is not defined for subcommand '%s'", sub.GroupID, sub.CommandPath())) | |
| } | |
| sub.checkCommandGroups() | |
| } | |
| } | |
| // InitDefaultHelpFlag adds default help flag to c. | |
| // It is called automatically by executing the c or by calling help and usage. | |
| // If c already has help flag, it will do nothing. | |
| func (c *Command) InitDefaultHelpFlag() { | |
| c.mergePersistentFlags() | |
| if c.Flags().Lookup(helpFlagName) == nil { | |
| usage := "help for " | |
| name := c.DisplayName() | |
| if name == "" { | |
| usage += "this command" | |
| } else { | |
| usage += name | |
| } | |
| c.Flags().BoolP(helpFlagName, "h", false, usage) | |
| _ = c.Flags().SetAnnotation(helpFlagName, FlagSetByCobraAnnotation, []string{"true"}) | |
| } | |
| } | |
| // InitDefaultVersionFlag adds default version flag to c. | |
| // It is called automatically by executing the c. | |
| // If c already has a version flag, it will do nothing. | |
| // If c.Version is empty, it will do nothing. | |
| func (c *Command) InitDefaultVersionFlag() { | |
| if c.Version == "" { | |
| return | |
| } | |
| c.mergePersistentFlags() | |
| if c.Flags().Lookup("version") == nil { | |
| usage := "version for " | |
| if c.Name() == "" { | |
| usage += "this command" | |
| } else { | |
| usage += c.DisplayName() | |
| } | |
| if c.Flags().ShorthandLookup("v") == nil { | |
| c.Flags().BoolP("version", "v", false, usage) | |
| } else { | |
| c.Flags().Bool("version", false, usage) | |
| } | |
| _ = c.Flags().SetAnnotation("version", FlagSetByCobraAnnotation, []string{"true"}) | |
| } | |
| } | |
| // InitDefaultHelpCmd adds default help command to c. | |
| // It is called automatically by executing the c or by calling help and usage. | |
| // If c already has help command or c has no subcommands, it will do nothing. | |
| func (c *Command) InitDefaultHelpCmd() { | |
| if !c.HasSubCommands() { | |
| return | |
| } | |
| if c.helpCommand == nil { | |
| c.helpCommand = &Command{ | |
| Use: "help [command]", | |
| Short: "Help about any command", | |
| Long: `Help provides help for any command in the application. | |
| Simply type ` + c.DisplayName() + ` help [path to command] for full details.`, | |
| ValidArgsFunction: func(c *Command, args []string, toComplete string) ([]Completion, ShellCompDirective) { | |
| var completions []Completion | |
| cmd, _, e := c.Root().Find(args) | |
| if e != nil { | |
| return nil, ShellCompDirectiveNoFileComp | |
| } | |
| if cmd == nil { | |
| // Root help command. | |
| cmd = c.Root() | |
| } | |
| for _, subCmd := range cmd.Commands() { | |
| if subCmd.IsAvailableCommand() || subCmd == cmd.helpCommand { | |
| if strings.HasPrefix(subCmd.Name(), toComplete) { | |
| completions = append(completions, CompletionWithDesc(subCmd.Name(), subCmd.Short)) | |
| } | |
| } | |
| } | |
| return completions, ShellCompDirectiveNoFileComp | |
| }, | |
| Run: func(c *Command, args []string) { | |
| cmd, _, e := c.Root().Find(args) | |
| if cmd == nil || e != nil { | |
| c.Printf("Unknown help topic %#q\n", args) | |
| CheckErr(c.Root().Usage()) | |
| } else { | |
| // FLow the context down to be used in help text | |
| if cmd.ctx == nil { | |
| cmd.ctx = c.ctx | |
| } | |
| cmd.InitDefaultHelpFlag() // make possible 'help' flag to be shown | |
| cmd.InitDefaultVersionFlag() // make possible 'version' flag to be shown | |
| CheckErr(cmd.Help()) | |
| } | |
| }, | |
| GroupID: c.helpCommandGroupID, | |
| } | |
| } | |
| c.RemoveCommand(c.helpCommand) | |
| c.AddCommand(c.helpCommand) | |
| } | |
| // ResetCommands delete parent, subcommand and help command from c. | |
| func (c *Command) ResetCommands() { | |
| c.parent = nil | |
| c.commands = nil | |
| c.helpCommand = nil | |
| c.parentsPflags = nil | |
| } | |
| // Sorts commands by their names. | |
| type commandSorterByName []*Command | |
| func (c commandSorterByName) Len() int { return len(c) } | |
| func (c commandSorterByName) Swap(i, j int) { c[i], c[j] = c[j], c[i] } | |
| func (c commandSorterByName) Less(i, j int) bool { return c[i].Name() < c[j].Name() } | |
| // Commands returns a sorted slice of child commands. | |
| func (c *Command) Commands() []*Command { | |
| // do not sort commands if it already sorted or sorting was disabled | |
| if EnableCommandSorting && !c.commandsAreSorted { | |
| sort.Sort(commandSorterByName(c.commands)) | |
| c.commandsAreSorted = true | |
| } | |
| return c.commands | |
| } | |
| // AddCommand adds one or more commands to this parent command. | |
| func (c *Command) AddCommand(cmds ...*Command) { | |
| for i, x := range cmds { | |
| if cmds[i] == c { | |
| panic("Command can't be a child of itself") | |
| } | |
| cmds[i].parent = c | |
| // update max lengths | |
| usageLen := len(x.Use) | |
| if usageLen > c.commandsMaxUseLen { | |
| c.commandsMaxUseLen = usageLen | |
| } | |
| commandPathLen := len(x.CommandPath()) | |
| if commandPathLen > c.commandsMaxCommandPathLen { | |
| c.commandsMaxCommandPathLen = commandPathLen | |
| } | |
| nameLen := len(x.Name()) | |
| if nameLen > c.commandsMaxNameLen { | |
| c.commandsMaxNameLen = nameLen | |
| } | |
| // If global normalization function exists, update all children | |
| if c.globNormFunc != nil { | |
| x.SetGlobalNormalizationFunc(c.globNormFunc) | |
| } | |
| c.commands = append(c.commands, x) | |
| c.commandsAreSorted = false | |
| } | |
| } | |
| // Groups returns a slice of child command groups. | |
| func (c *Command) Groups() []*Group { | |
| return c.commandgroups | |
| } | |
| // AllChildCommandsHaveGroup returns if all subcommands are assigned to a group | |
| func (c *Command) AllChildCommandsHaveGroup() bool { | |
| for _, sub := range c.commands { | |
| if (sub.IsAvailableCommand() || sub == c.helpCommand) && sub.GroupID == "" { | |
| return false | |
| } | |
| } | |
| return true | |
| } | |
| // ContainsGroup return if groupID exists in the list of command groups. | |
| func (c *Command) ContainsGroup(groupID string) bool { | |
| for _, x := range c.commandgroups { | |
| if x.ID == groupID { | |
| return true | |
| } | |
| } | |
| return false | |
| } | |
| // AddGroup adds one or more command groups to this parent command. | |
| func (c *Command) AddGroup(groups ...*Group) { | |
| c.commandgroups = append(c.commandgroups, groups...) | |
| } | |
| // RemoveCommand removes one or more commands from a parent command. | |
| func (c *Command) RemoveCommand(cmds ...*Command) { | |
| commands := []*Command{} | |
| main: | |
| for _, command := range c.commands { | |
| for _, cmd := range cmds { | |
| if command == cmd { | |
| command.parent = nil | |
| continue main | |
| } | |
| } | |
| commands = append(commands, command) | |
| } | |
| c.commands = commands | |
| // recompute all lengths | |
| c.commandsMaxUseLen = 0 | |
| c.commandsMaxCommandPathLen = 0 | |
| c.commandsMaxNameLen = 0 | |
| for _, command := range c.commands { | |
| usageLen := len(command.Use) | |
| if usageLen > c.commandsMaxUseLen { | |
| c.commandsMaxUseLen = usageLen | |
| } | |
| commandPathLen := len(command.CommandPath()) | |
| if commandPathLen > c.commandsMaxCommandPathLen { | |
| c.commandsMaxCommandPathLen = commandPathLen | |
| } | |
| nameLen := len(command.Name()) | |
| if nameLen > c.commandsMaxNameLen { | |
| c.commandsMaxNameLen = nameLen | |
| } | |
| } | |
| } | |
| // Print is a convenience method to Print to the defined output, fallback to Stderr if not set. | |
| func (c *Command) Print(i ...interface{}) { | |
| fmt.Fprint(c.OutOrStderr(), i...) | |
| } | |
| // Println is a convenience method to Println to the defined output, fallback to Stderr if not set. | |
| func (c *Command) Println(i ...interface{}) { | |
| c.Print(fmt.Sprintln(i...)) | |
| } | |
| // Printf is a convenience method to Printf to the defined output, fallback to Stderr if not set. | |
| func (c *Command) Printf(format string, i ...interface{}) { | |
| c.Print(fmt.Sprintf(format, i...)) | |
| } | |
| // PrintErr is a convenience method to Print to the defined Err output, fallback to Stderr if not set. | |
| func (c *Command) PrintErr(i ...interface{}) { | |
| fmt.Fprint(c.ErrOrStderr(), i...) | |
| } | |
| // PrintErrln is a convenience method to Println to the defined Err output, fallback to Stderr if not set. | |
| func (c *Command) PrintErrln(i ...interface{}) { | |
| c.PrintErr(fmt.Sprintln(i...)) | |
| } | |
| // PrintErrf is a convenience method to Printf to the defined Err output, fallback to Stderr if not set. | |
| func (c *Command) PrintErrf(format string, i ...interface{}) { | |
| c.PrintErr(fmt.Sprintf(format, i...)) | |
| } | |
| // CommandPath returns the full path to this command. | |
| func (c *Command) CommandPath() string { | |
| if c.HasParent() { | |
| return c.Parent().CommandPath() + " " + c.Name() | |
| } | |
| return c.DisplayName() | |
| } | |
| // DisplayName returns the name to display in help text. Returns command Name() | |
| // If CommandDisplayNameAnnoation is not set | |
| func (c *Command) DisplayName() string { | |
| if displayName, ok := c.Annotations[CommandDisplayNameAnnotation]; ok { | |
| return displayName | |
| } | |
| return c.Name() | |
| } | |
| // UseLine puts out the full usage for a given command (including parents). | |
| func (c *Command) UseLine() string { | |
| var useline string | |
| use := strings.Replace(c.Use, c.Name(), c.DisplayName(), 1) | |
| if c.HasParent() { | |
| useline = c.parent.CommandPath() + " " + use | |
| } else { | |
| useline = use | |
| } | |
| if c.DisableFlagsInUseLine { | |
| return useline | |
| } | |
| if c.HasAvailableFlags() && !strings.Contains(useline, "[flags]") { | |
| useline += " [flags]" | |
| } | |
| return useline | |
| } | |
| // DebugFlags used to determine which flags have been assigned to which commands | |
| // and which persist. | |
| func (c *Command) DebugFlags() { | |
| c.Println("DebugFlags called on", c.Name()) | |
| var debugflags func(*Command) | |
| debugflags = func(x *Command) { | |
| if x.HasFlags() || x.HasPersistentFlags() { | |
| c.Println(x.Name()) | |
| } | |
| if x.HasFlags() { | |
| x.flags.VisitAll(func(f *flag.Flag) { | |
| if x.HasPersistentFlags() && x.persistentFlag(f.Name) != nil { | |
| c.Println(" -"+f.Shorthand+",", "--"+f.Name, "["+f.DefValue+"]", "", f.Value, " [LP]") | |
| } else { | |
| c.Println(" -"+f.Shorthand+",", "--"+f.Name, "["+f.DefValue+"]", "", f.Value, " [L]") | |
| } | |
| }) | |
| } | |
| if x.HasPersistentFlags() { | |
| x.pflags.VisitAll(func(f *flag.Flag) { | |
| if x.HasFlags() { | |
| if x.flags.Lookup(f.Name) == nil { | |
| c.Println(" -"+f.Shorthand+",", "--"+f.Name, "["+f.DefValue+"]", "", f.Value, " [P]") | |
| } | |
| } else { | |
| c.Println(" -"+f.Shorthand+",", "--"+f.Name, "["+f.DefValue+"]", "", f.Value, " [P]") | |
| } | |
| }) | |
| } | |
| c.Println(x.flagErrorBuf) | |
| if x.HasSubCommands() { | |
| for _, y := range x.commands { | |
| debugflags(y) | |
| } | |
| } | |
| } | |
| debugflags(c) | |
| } | |
| // Name returns the command's name: the first word in the use line. | |
| func (c *Command) Name() string { | |
| name := c.Use | |
| i := strings.Index(name, " ") | |
| if i >= 0 { | |
| name = name[:i] | |
| } | |
| return name | |
| } | |
| // HasAlias determines if a given string is an alias of the command. | |
| func (c *Command) HasAlias(s string) bool { | |
| for _, a := range c.Aliases { | |
| if commandNameMatches(a, s) { | |
| return true | |
| } | |
| } | |
| return false | |
| } | |
| // CalledAs returns the command name or alias that was used to invoke | |
| // this command or an empty string if the command has not been called. | |
| func (c *Command) CalledAs() string { | |
| if c.commandCalledAs.called { | |
| return c.commandCalledAs.name | |
| } | |
| return "" | |
| } | |
| // hasNameOrAliasPrefix returns true if the Name or any of aliases start | |
| // with prefix | |
| func (c *Command) hasNameOrAliasPrefix(prefix string) bool { | |
| if strings.HasPrefix(c.Name(), prefix) { | |
| c.commandCalledAs.name = c.Name() | |
| return true | |
| } | |
| for _, alias := range c.Aliases { | |
| if strings.HasPrefix(alias, prefix) { | |
| c.commandCalledAs.name = alias | |
| return true | |
| } | |
| } | |
| return false | |
| } | |
| // NameAndAliases returns a list of the command name and all aliases | |
| func (c *Command) NameAndAliases() string { | |
| return strings.Join(append([]string{c.Name()}, c.Aliases...), ", ") | |
| } | |
| // HasExample determines if the command has example. | |
| func (c *Command) HasExample() bool { | |
| return len(c.Example) > 0 | |
| } | |
| // Runnable determines if the command is itself runnable. | |
| func (c *Command) Runnable() bool { | |
| return c.Run != nil || c.RunE != nil | |
| } | |
| // HasSubCommands determines if the command has children commands. | |
| func (c *Command) HasSubCommands() bool { | |
| return len(c.commands) > 0 | |
| } | |
| // IsAvailableCommand determines if a command is available as a non-help command | |
| // (this includes all non deprecated/hidden commands). | |
| func (c *Command) IsAvailableCommand() bool { | |
| if len(c.Deprecated) != 0 || c.Hidden { | |
| return false | |
| } | |
| if c.HasParent() && c.Parent().helpCommand == c { | |
| return false | |
| } | |
| if c.Runnable() || c.HasAvailableSubCommands() { | |
| return true | |
| } | |
| return false | |
| } | |
| // IsAdditionalHelpTopicCommand determines if a command is an additional | |
| // help topic command; additional help topic command is determined by the | |
| // fact that it is NOT runnable/hidden/deprecated, and has no sub commands that | |
| // are runnable/hidden/deprecated. | |
| // Concrete example: https://github.com/spf13/cobra/issues/393#issuecomment-282741924. | |
| func (c *Command) IsAdditionalHelpTopicCommand() bool { | |
| // if a command is runnable, deprecated, or hidden it is not a 'help' command | |
| if c.Runnable() || len(c.Deprecated) != 0 || c.Hidden { | |
| return false | |
| } | |
| // if any non-help sub commands are found, the command is not a 'help' command | |
| for _, sub := range c.commands { | |
| if !sub.IsAdditionalHelpTopicCommand() { | |
| return false | |
| } | |
| } | |
| // the command either has no sub commands, or no non-help sub commands | |
| return true | |
| } | |
| // HasHelpSubCommands determines if a command has any available 'help' sub commands | |
| // that need to be shown in the usage/help default template under 'additional help | |
| // topics'. | |
| func (c *Command) HasHelpSubCommands() bool { | |
| // return true on the first found available 'help' sub command | |
| for _, sub := range c.commands { | |
| if sub.IsAdditionalHelpTopicCommand() { | |
| return true | |
| } | |
| } | |
| // the command either has no sub commands, or no available 'help' sub commands | |
| return false | |
| } | |
| // HasAvailableSubCommands determines if a command has available sub commands that | |
| // need to be shown in the usage/help default template under 'available commands'. | |
| func (c *Command) HasAvailableSubCommands() bool { | |
| // return true on the first found available (non deprecated/help/hidden) | |
| // sub command | |
| for _, sub := range c.commands { | |
| if sub.IsAvailableCommand() { | |
| return true | |
| } | |
| } | |
| // the command either has no sub commands, or no available (non deprecated/help/hidden) | |
| // sub commands | |
| return false | |
| } | |
| // HasParent determines if the command is a child command. | |
| func (c *Command) HasParent() bool { | |
| return c.parent != nil | |
| } | |
| // GlobalNormalizationFunc returns the global normalization function or nil if it doesn't exist. | |
| func (c *Command) GlobalNormalizationFunc() func(f *flag.FlagSet, name string) flag.NormalizedName { | |
| return c.globNormFunc | |
| } | |
| // Flags returns the complete FlagSet that applies | |
| // to this command (local and persistent declared here and by all parents). | |
| func (c *Command) Flags() *flag.FlagSet { | |
| if c.flags == nil { | |
| c.flags = flag.NewFlagSet(c.DisplayName(), flag.ContinueOnError) | |
| if c.flagErrorBuf == nil { | |
| c.flagErrorBuf = new(bytes.Buffer) | |
| } | |
| c.flags.SetOutput(c.flagErrorBuf) | |
| } | |
| return c.flags | |
| } | |
| // LocalNonPersistentFlags are flags specific to this command which will NOT persist to subcommands. | |
| // This function does not modify the flags of the current command, it's purpose is to return the current state. | |
| func (c *Command) LocalNonPersistentFlags() *flag.FlagSet { | |
| persistentFlags := c.PersistentFlags() | |
| out := flag.NewFlagSet(c.DisplayName(), flag.ContinueOnError) | |
| c.LocalFlags().VisitAll(func(f *flag.Flag) { | |
| if persistentFlags.Lookup(f.Name) == nil { | |
| out.AddFlag(f) | |
| } | |
| }) | |
| return out | |
| } | |
| // LocalFlags returns the local FlagSet specifically set in the current command. | |
| // This function does not modify the flags of the current command, it's purpose is to return the current state. | |
| func (c *Command) LocalFlags() *flag.FlagSet { | |
| c.mergePersistentFlags() | |
| if c.lflags == nil { | |
| c.lflags = flag.NewFlagSet(c.DisplayName(), flag.ContinueOnError) | |
| if c.flagErrorBuf == nil { | |
| c.flagErrorBuf = new(bytes.Buffer) | |
| } | |
| c.lflags.SetOutput(c.flagErrorBuf) | |
| } | |
| c.lflags.SortFlags = c.Flags().SortFlags | |
| if c.globNormFunc != nil { | |
| c.lflags.SetNormalizeFunc(c.globNormFunc) | |
| } | |
| addToLocal := func(f *flag.Flag) { | |
| // Add the flag if it is not a parent PFlag, or it shadows a parent PFlag | |
| if c.lflags.Lookup(f.Name) == nil && f != c.parentsPflags.Lookup(f.Name) { | |
| c.lflags.AddFlag(f) | |
| } | |
| } | |
| c.Flags().VisitAll(addToLocal) | |
| c.PersistentFlags().VisitAll(addToLocal) | |
| return c.lflags | |
| } | |
| // InheritedFlags returns all flags which were inherited from parent commands. | |
| // This function does not modify the flags of the current command, it's purpose is to return the current state. | |
| func (c *Command) InheritedFlags() *flag.FlagSet { | |
| c.mergePersistentFlags() | |
| if c.iflags == nil { | |
| c.iflags = flag.NewFlagSet(c.DisplayName(), flag.ContinueOnError) | |
| if c.flagErrorBuf == nil { | |
| c.flagErrorBuf = new(bytes.Buffer) | |
| } | |
| c.iflags.SetOutput(c.flagErrorBuf) | |
| } | |
| local := c.LocalFlags() | |
| if c.globNormFunc != nil { | |
| c.iflags.SetNormalizeFunc(c.globNormFunc) | |
| } | |
| c.parentsPflags.VisitAll(func(f *flag.Flag) { | |
| if c.iflags.Lookup(f.Name) == nil && local.Lookup(f.Name) == nil { | |
| c.iflags.AddFlag(f) | |
| } | |
| }) | |
| return c.iflags | |
| } | |
| // NonInheritedFlags returns all flags which were not inherited from parent commands. | |
| // This function does not modify the flags of the current command, it's purpose is to return the current state. | |
| func (c *Command) NonInheritedFlags() *flag.FlagSet { | |
| return c.LocalFlags() | |
| } | |
| // PersistentFlags returns the persistent FlagSet specifically set in the current command. | |
| func (c *Command) PersistentFlags() *flag.FlagSet { | |
| if c.pflags == nil { | |
| c.pflags = flag.NewFlagSet(c.DisplayName(), flag.ContinueOnError) | |
| if c.flagErrorBuf == nil { | |
| c.flagErrorBuf = new(bytes.Buffer) | |
| } | |
| c.pflags.SetOutput(c.flagErrorBuf) | |
| } | |
| return c.pflags | |
| } | |
| // ResetFlags deletes all flags from command. | |
| func (c *Command) ResetFlags() { | |
| c.flagErrorBuf = new(bytes.Buffer) | |
| c.flagErrorBuf.Reset() | |
| c.flags = flag.NewFlagSet(c.DisplayName(), flag.ContinueOnError) | |
| c.flags.SetOutput(c.flagErrorBuf) | |
| c.pflags = flag.NewFlagSet(c.DisplayName(), flag.ContinueOnError) | |
| c.pflags.SetOutput(c.flagErrorBuf) | |
| c.lflags = nil | |
| c.iflags = nil | |
| c.parentsPflags = nil | |
| } | |
| // HasFlags checks if the command contains any flags (local plus persistent from the entire structure). | |
| func (c *Command) HasFlags() bool { | |
| return c.Flags().HasFlags() | |
| } | |
| // HasPersistentFlags checks if the command contains persistent flags. | |
| func (c *Command) HasPersistentFlags() bool { | |
| return c.PersistentFlags().HasFlags() | |
| } | |
| // HasLocalFlags checks if the command has flags specifically declared locally. | |
| func (c *Command) HasLocalFlags() bool { | |
| return c.LocalFlags().HasFlags() | |
| } | |
| // HasInheritedFlags checks if the command has flags inherited from its parent command. | |
| func (c *Command) HasInheritedFlags() bool { | |
| return c.InheritedFlags().HasFlags() | |
| } | |
| // HasAvailableFlags checks if the command contains any flags (local plus persistent from the entire | |
| // structure) which are not hidden or deprecated. | |
| func (c *Command) HasAvailableFlags() bool { | |
| return c.Flags().HasAvailableFlags() | |
| } | |
| // HasAvailablePersistentFlags checks if the command contains persistent flags which are not hidden or deprecated. | |
| func (c *Command) HasAvailablePersistentFlags() bool { | |
| return c.PersistentFlags().HasAvailableFlags() | |
| } | |
| // HasAvailableLocalFlags checks if the command has flags specifically declared locally which are not hidden | |
| // or deprecated. | |
| func (c *Command) HasAvailableLocalFlags() bool { | |
| return c.LocalFlags().HasAvailableFlags() | |
| } | |
| // HasAvailableInheritedFlags checks if the command has flags inherited from its parent command which are | |
| // not hidden or deprecated. | |
| func (c *Command) HasAvailableInheritedFlags() bool { | |
| return c.InheritedFlags().HasAvailableFlags() | |
| } | |
| // Flag climbs up the command tree looking for matching flag. | |
| func (c *Command) Flag(name string) (flag *flag.Flag) { | |
| flag = c.Flags().Lookup(name) | |
| if flag == nil { | |
| flag = c.persistentFlag(name) | |
| } | |
| return | |
| } | |
| // Recursively find matching persistent flag. | |
| func (c *Command) persistentFlag(name string) (flag *flag.Flag) { | |
| if c.HasPersistentFlags() { | |
| flag = c.PersistentFlags().Lookup(name) | |
| } | |
| if flag == nil { | |
| c.updateParentsPflags() | |
| flag = c.parentsPflags.Lookup(name) | |
| } | |
| return | |
| } | |
| // ParseFlags parses persistent flag tree and local flags. | |
| func (c *Command) ParseFlags(args []string) error { | |
| if c.DisableFlagParsing { | |
| return nil | |
| } | |
| if c.flagErrorBuf == nil { | |
| c.flagErrorBuf = new(bytes.Buffer) | |
| } | |
| beforeErrorBufLen := c.flagErrorBuf.Len() | |
| c.mergePersistentFlags() | |
| // do it here after merging all flags and just before parse | |
| c.Flags().ParseErrorsAllowlist = flag.ParseErrorsAllowlist(c.FParseErrWhitelist) | |
| err := c.Flags().Parse(args) | |
| // Print warnings if they occurred (e.g. deprecated flag messages). | |
| if c.flagErrorBuf.Len()-beforeErrorBufLen > 0 && err == nil { | |
| c.Print(c.flagErrorBuf.String()) | |
| } | |
| return err | |
| } | |
| // Parent returns a commands parent command. | |
| func (c *Command) Parent() *Command { | |
| return c.parent | |
| } | |
| // mergePersistentFlags merges c.PersistentFlags() to c.Flags() | |
| // and adds missing persistent flags of all parents. | |
| func (c *Command) mergePersistentFlags() { | |
| c.updateParentsPflags() | |
| c.Flags().AddFlagSet(c.PersistentFlags()) | |
| c.Flags().AddFlagSet(c.parentsPflags) | |
| } | |
| // updateParentsPflags updates c.parentsPflags by adding | |
| // new persistent flags of all parents. | |
| // If c.parentsPflags == nil, it makes new. | |
| func (c *Command) updateParentsPflags() { | |
| if c.parentsPflags == nil { | |
| c.parentsPflags = flag.NewFlagSet(c.DisplayName(), flag.ContinueOnError) | |
| c.parentsPflags.SetOutput(c.flagErrorBuf) | |
| c.parentsPflags.SortFlags = false | |
| } | |
| if c.globNormFunc != nil { | |
| c.parentsPflags.SetNormalizeFunc(c.globNormFunc) | |
| } | |
| c.Root().PersistentFlags().AddFlagSet(flag.CommandLine) | |
| c.VisitParents(func(parent *Command) { | |
| c.parentsPflags.AddFlagSet(parent.PersistentFlags()) | |
| }) | |
| } | |
| // commandNameMatches checks if two command names are equal | |
| // taking into account case sensitivity according to | |
| // EnableCaseInsensitive global configuration. | |
| func commandNameMatches(s string, t string) bool { | |
| if EnableCaseInsensitive { | |
| return strings.EqualFold(s, t) | |
| } | |
| return s == t | |
| } | |
| // tmplFunc holds a template and a function that will execute said template. | |
| type tmplFunc struct { | |
| tmpl string | |
| fn func(io.Writer, interface{}) error | |
| } | |
| var defaultUsageTemplate = `Usage:{{if .Runnable}} | |
| {{.UseLine}}{{end}}{{if .HasAvailableSubCommands}} | |
| {{.CommandPath}} [command]{{end}}{{if gt (len .Aliases) 0}} | |
| Aliases: | |
| {{.NameAndAliases}}{{end}}{{if .HasExample}} | |
| Examples: | |
| {{.Example}}{{end}}{{if .HasAvailableSubCommands}}{{$cmds := .Commands}}{{if eq (len .Groups) 0}} | |
| Available Commands:{{range $cmds}}{{if (or .IsAvailableCommand (eq .Name "help"))}} | |
| {{rpad .Name .NamePadding }} {{.Short}}{{end}}{{end}}{{else}}{{range $group := .Groups}} | |
| {{.Title}}{{range $cmds}}{{if (and (eq .GroupID $group.ID) (or .IsAvailableCommand (eq .Name "help")))}} | |
| {{rpad .Name .NamePadding }} {{.Short}}{{end}}{{end}}{{end}}{{if not .AllChildCommandsHaveGroup}} | |
| Additional Commands:{{range $cmds}}{{if (and (eq .GroupID "") (or .IsAvailableCommand (eq .Name "help")))}} | |
| {{rpad .Name .NamePadding }} {{.Short}}{{end}}{{end}}{{end}}{{end}}{{end}}{{if .HasAvailableLocalFlags}} | |
| Flags: | |
| {{.LocalFlags.FlagUsages | trimTrailingWhitespaces}}{{end}}{{if .HasAvailableInheritedFlags}} | |
| Global Flags: | |
| {{.InheritedFlags.FlagUsages | trimTrailingWhitespaces}}{{end}}{{if .HasHelpSubCommands}} | |
| Additional help topics:{{range .Commands}}{{if .IsAdditionalHelpTopicCommand}} | |
| {{rpad .CommandPath .CommandPathPadding}} {{.Short}}{{end}}{{end}}{{end}}{{if .HasAvailableSubCommands}} | |
| Use "{{.CommandPath}} [command] --help" for more information about a command.{{end}} | |
| ` | |
| // defaultUsageFunc is equivalent to executing defaultUsageTemplate. The two should be changed in sync. | |
| func defaultUsageFunc(w io.Writer, in interface{}) error { | |
| c := in.(*Command) | |
| fmt.Fprint(w, "Usage:") | |
| if c.Runnable() { | |
| fmt.Fprintf(w, "\n %s", c.UseLine()) | |
| } | |
| if c.HasAvailableSubCommands() { | |
| fmt.Fprintf(w, "\n %s [command]", c.CommandPath()) | |
| } | |
| if len(c.Aliases) > 0 { | |
| fmt.Fprintf(w, "\n\nAliases:\n") | |
| fmt.Fprintf(w, " %s", c.NameAndAliases()) | |
| } | |
| if c.HasExample() { | |
| fmt.Fprintf(w, "\n\nExamples:\n") | |
| fmt.Fprintf(w, "%s", c.Example) | |
| } | |
| if c.HasAvailableSubCommands() { | |
| cmds := c.Commands() | |
| if len(c.Groups()) == 0 { | |
| fmt.Fprintf(w, "\n\nAvailable Commands:") | |
| for _, subcmd := range cmds { | |
| if subcmd.IsAvailableCommand() || subcmd.Name() == helpCommandName { | |
| fmt.Fprintf(w, "\n %s %s", rpad(subcmd.Name(), subcmd.NamePadding()), subcmd.Short) | |
| } | |
| } | |
| } else { | |
| for _, group := range c.Groups() { | |
| fmt.Fprintf(w, "\n\n%s", group.Title) | |
| for _, subcmd := range cmds { | |
| if subcmd.GroupID == group.ID && (subcmd.IsAvailableCommand() || subcmd.Name() == helpCommandName) { | |
| fmt.Fprintf(w, "\n %s %s", rpad(subcmd.Name(), subcmd.NamePadding()), subcmd.Short) | |
| } | |
| } | |
| } | |
| if !c.AllChildCommandsHaveGroup() { | |
| fmt.Fprintf(w, "\n\nAdditional Commands:") | |
| for _, subcmd := range cmds { | |
| if subcmd.GroupID == "" && (subcmd.IsAvailableCommand() || subcmd.Name() == helpCommandName) { | |
| fmt.Fprintf(w, "\n %s %s", rpad(subcmd.Name(), subcmd.NamePadding()), subcmd.Short) | |
| } | |
| } | |
| } | |
| } | |
| } | |
| if c.HasAvailableLocalFlags() { | |
| fmt.Fprintf(w, "\n\nFlags:\n") | |
| fmt.Fprint(w, trimRightSpace(c.LocalFlags().FlagUsages())) | |
| } | |
| if c.HasAvailableInheritedFlags() { | |
| fmt.Fprintf(w, "\n\nGlobal Flags:\n") | |
| fmt.Fprint(w, trimRightSpace(c.InheritedFlags().FlagUsages())) | |
| } | |
| if c.HasHelpSubCommands() { | |
| fmt.Fprintf(w, "\n\nAdditional help topics:") | |
| for _, subcmd := range c.Commands() { | |
| if subcmd.IsAdditionalHelpTopicCommand() { | |
| fmt.Fprintf(w, "\n %s %s", rpad(subcmd.CommandPath(), subcmd.CommandPathPadding()), subcmd.Short) | |
| } | |
| } | |
| } | |
| if c.HasAvailableSubCommands() { | |
| fmt.Fprintf(w, "\n\nUse \"%s [command] --help\" for more information about a command.", c.CommandPath()) | |
| } | |
| fmt.Fprintln(w) | |
| return nil | |
| } | |
| var defaultHelpTemplate = `{{with (or .Long .Short)}}{{. | trimTrailingWhitespaces}} | |
| {{end}}{{if or .Runnable .HasSubCommands}}{{.UsageString}}{{end}}` | |
| // defaultHelpFunc is equivalent to executing defaultHelpTemplate. The two should be changed in sync. | |
| func defaultHelpFunc(w io.Writer, in interface{}) error { | |
| c := in.(*Command) | |
| usage := c.Long | |
| if usage == "" { | |
| usage = c.Short | |
| } | |
| usage = trimRightSpace(usage) | |
| if usage != "" { | |
| fmt.Fprintln(w, usage) | |
| fmt.Fprintln(w) | |
| } | |
| if c.Runnable() || c.HasSubCommands() { | |
| fmt.Fprint(w, c.UsageString()) | |
| } | |
| return nil | |
| } | |
| var defaultVersionTemplate = `{{with .DisplayName}}{{printf "%s " .}}{{end}}{{printf "version %s" .Version}} | |
| ` | |
| // defaultVersionFunc is equivalent to executing defaultVersionTemplate. The two should be changed in sync. | |
| func defaultVersionFunc(w io.Writer, in interface{}) error { | |
| c := in.(*Command) | |
| _, err := fmt.Fprintf(w, "%s version %s\n", c.DisplayName(), c.Version) | |
| return err | |
| } |