78 lines
1.9 KiB
Go
78 lines
1.9 KiB
Go
package command
|
|
|
|
import (
|
|
"errors"
|
|
"flag"
|
|
"strings"
|
|
)
|
|
|
|
// FlagError wraps errors returned by [flag].
|
|
type FlagError struct{ error }
|
|
|
|
func (e FlagError) Success() bool { return errors.Is(e.error, flag.ErrHelp) }
|
|
func (e FlagError) Is(target error) bool {
|
|
return (e.error == nil && target == nil) ||
|
|
((e.error != nil && target != nil) && e.error.Error() == target.Error())
|
|
}
|
|
|
|
func (n *node) Flag(p any, name string, value FlagDefiner, usage string) Node {
|
|
value.Define(&n.suffix, n.set, p, name, usage)
|
|
return n
|
|
}
|
|
|
|
// StringFlag is the default value of a string flag.
|
|
type StringFlag string
|
|
|
|
func (v StringFlag) Define(b *strings.Builder, set *flag.FlagSet, p any, name, usage string) {
|
|
set.StringVar(p.(*string), name, string(v), usage)
|
|
b.WriteString(" [" + prettyFlag(name) + " <value>]")
|
|
}
|
|
|
|
// IntFlag is the default value of an int flag.
|
|
type IntFlag int
|
|
|
|
func (v IntFlag) Define(b *strings.Builder, set *flag.FlagSet, p any, name, usage string) {
|
|
set.IntVar(p.(*int), name, int(v), usage)
|
|
b.WriteString(" [" + prettyFlag(name) + " <int>]")
|
|
}
|
|
|
|
// BoolFlag is the default value of a bool flag.
|
|
type BoolFlag bool
|
|
|
|
func (v BoolFlag) Define(b *strings.Builder, set *flag.FlagSet, p any, name, usage string) {
|
|
set.BoolVar(p.(*bool), name, bool(v), usage)
|
|
b.WriteString(" [" + prettyFlag(name) + "]")
|
|
}
|
|
|
|
// RepeatableFlag implements an ordered, repeatable string flag.
|
|
type RepeatableFlag []string
|
|
|
|
func (r *RepeatableFlag) String() string {
|
|
if r == nil {
|
|
return "<nil>"
|
|
}
|
|
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) + " <value>]")
|
|
}
|
|
|
|
// this has no effect on parse outcome
|
|
func prettyFlag(name string) string {
|
|
switch len(name) {
|
|
case 0:
|
|
panic("zero length flag name")
|
|
case 1:
|
|
return "-" + name
|
|
default:
|
|
return "--" + name
|
|
}
|
|
}
|