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

spf13 · Apr 6, 2015 · fb50344 · fb50344
1 parent 0d5c3d6
commit fb50344
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/bash_completions.go b/bash_completions.go
@@ -305,3 +305,16 @@ func (cmd *Command) GenCompletion(out *bytes.Buffer) {
 	gen(cmd, out)
 	postscript(out, cmd.Name())
 }
+
+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
+}