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

spf13 · Mar 17, 2015 · 9d6aada · 9d6aada
1 parent 10c0e6d
commit 9d6aada
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/bash_completions.go b/bash_completions.go
@@ -309,3 +309,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
+}