add a helper to annotate a flag as 'required' by a command

This commit is contained in:
Eric Paris 2015-03-17 15:14:06 -04:00
parent 9b2e6822e5
commit a068307565

View file

@ -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
}