mirror of https://github.com/spf13/cobra.git
add a helper to annotate a flag as 'required' by a command
This commit is contained in:
parent
9b2e6822e5
commit
a068307565
|
@ -328,3 +328,16 @@ func (cmd *Command) GenBashCompletionFile(filename string) error {
|
|||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (cmd *Command) MarkFlagRequired(name string) {
|
||||
flag := cmd.Flags().Lookup(name)
|
||||
if flag == nil {
|
||||
return
|
||||
}
|
||||
if flag.Annotations == nil {
|
||||
flag.Annotations = make(map[string][]string)
|
||||
}
|
||||
annotation := make([]string, 1)
|
||||
annotation[0] = "true"
|
||||
flag.Annotations[BashCompOneRequiredFlag] = annotation
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue