68 lines
1.8 KiB
Go
68 lines
1.8 KiB
Go
package commandmeta
|
|
|
|
import "strings"
|
|
|
|
// FlagTokenKind identifies the portion of the shared option grammar a token uses.
|
|
type FlagTokenKind uint8
|
|
|
|
const (
|
|
OperandToken FlagTokenKind = iota
|
|
TerminatorToken
|
|
LongFlagToken
|
|
ShortFlagToken
|
|
)
|
|
|
|
// ClassifyFlagToken distinguishes option tokens without consuming their values.
|
|
func ClassifyFlagToken(token string) FlagTokenKind {
|
|
if token == "--" {
|
|
return TerminatorToken
|
|
}
|
|
if strings.HasPrefix(token, "--") && len(token) > 2 {
|
|
return LongFlagToken
|
|
}
|
|
if len(token) > 1 && token[0] == '-' {
|
|
return ShortFlagToken
|
|
}
|
|
return OperandToken
|
|
}
|
|
|
|
// SplitLongFlag separates the entered spelling from an optional attached value.
|
|
func SplitLongFlag(token string) (spelling, value string, attached bool) {
|
|
return strings.Cut(token, "=")
|
|
}
|
|
|
|
// FlagLookup maps canonical and applicable alias spellings to neutral flags.
|
|
type FlagLookup struct {
|
|
canonical map[string]Flag
|
|
aliases map[string]Flag
|
|
}
|
|
|
|
// NewFlagLookup includes the built-in help flag alongside the supplied flags.
|
|
func NewFlagLookup(flags []Flag, aliases []FlagAlias) FlagLookup {
|
|
canonical := make(map[string]Flag, len(flags)*2+2)
|
|
help := HelpFlag()
|
|
canonical["--help"], canonical["-h"] = help, help
|
|
for _, flag := range flags {
|
|
canonical["--"+flag.Name] = flag
|
|
if flag.Short != "" {
|
|
canonical["-"+flag.Short] = flag
|
|
}
|
|
}
|
|
lookup := FlagLookup{canonical: canonical, aliases: make(map[string]Flag, len(aliases))}
|
|
for _, alias := range aliases {
|
|
if target, ok := canonical[alias.Target]; ok {
|
|
lookup.aliases[alias.Source] = target
|
|
}
|
|
}
|
|
return lookup
|
|
}
|
|
|
|
// Find returns the canonical flag for an entered spelling.
|
|
func (lookup FlagLookup) Find(spelling string) (Flag, bool) {
|
|
flag, ok := lookup.canonical[spelling]
|
|
if !ok {
|
|
flag, ok = lookup.aliases[spelling]
|
|
}
|
|
return flag, ok
|
|
}
|