From 89970f51970b320430852ed57dcfc9d6c841322e Mon Sep 17 00:00:00 2001 From: Ophestra Date: Sun, 23 Feb 2025 02:25:31 +0900 Subject: command/flag: implement repeatable flag Signed-off-by: Ophestra --- command/flag.go | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) (limited to 'command/flag.go') diff --git a/command/flag.go b/command/flag.go index 8e890a1a..cb85a648 100644 --- a/command/flag.go +++ b/command/flag.go @@ -44,6 +44,26 @@ func (v BoolFlag) Define(b *strings.Builder, set *flag.FlagSet, p any, name, usa b.WriteString(" [" + prettyFlag(name) + "]") } +// RepeatableFlag implements an ordered, repeatable string flag. +type RepeatableFlag []string + +func (r *RepeatableFlag) String() string { + if r == nil { + return "" + } + return strings.Join(*r, " ") +} + +func (r *RepeatableFlag) Set(v string) error { + *r = append(*r, v) + return nil +} + +func (r *RepeatableFlag) Define(b *strings.Builder, set *flag.FlagSet, _ any, name, usage string) { + set.Var(r, name, usage) + b.WriteString(" [" + prettyFlag(name) + " ]") +} + // this has no effect on parse outcome func prettyFlag(name string) string { switch len(name) { -- cgit v1.3.1