mirror of
https://github.com/spf13/cobra
synced 2024-11-24 22:57:12 +00:00
parse lflags
This commit is contained in:
parent
55ff048a8d
commit
200e518356
1 changed files with 88 additions and 1 deletions
89
command.go
89
command.go
|
@ -1767,6 +1767,91 @@ func (c *Command) persistentFlag(name string) (flag *flag.Flag) {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (c *Command) parseLongArgs(s string, args []string) (passedArgs, restArgs []string) {
|
||||||
|
restArgs = args
|
||||||
|
|
||||||
|
split := strings.SplitN(s[2:], "=", 2)
|
||||||
|
name := split[0]
|
||||||
|
searchedFlag := c.lflags.Lookup(name)
|
||||||
|
|
||||||
|
if searchedFlag == nil {
|
||||||
|
// a flag not registered in c.lflags should be registered in c.flags
|
||||||
|
c.flags.VisitAll(func(f *flag.Flag) {
|
||||||
|
if name == f.Name {
|
||||||
|
if len(split) == 1 && f.NoOptDefVal == "" && len(args) > 0 {
|
||||||
|
// '--flag arg'
|
||||||
|
restArgs = args[1:]
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
passedArgs = append(passedArgs, fmt.Sprintf("--%s", name))
|
||||||
|
if len(split) == 1 && searchedFlag.NoOptDefVal == "" && len(args) > 0 {
|
||||||
|
passedArgs = append(passedArgs, args[0])
|
||||||
|
restArgs = args[1:]
|
||||||
|
}
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Command) parseShortArgs(s string, args []string) (passedArgs []string, restArgs []string) {
|
||||||
|
restArgs = args
|
||||||
|
|
||||||
|
shorthands := s[1:]
|
||||||
|
shorthand := string(s[1])
|
||||||
|
|
||||||
|
searchedFlag := c.lflags.ShorthandLookup(shorthand)
|
||||||
|
if searchedFlag == nil {
|
||||||
|
// a flag not registered in c.lflags should be registered in c.flags
|
||||||
|
c.flags.VisitAll(func(f *flag.Flag) {
|
||||||
|
if shorthand == f.Shorthand {
|
||||||
|
if len(shorthands) == 1 && f.NoOptDefVal == "" && len(args) > 0 {
|
||||||
|
// '-f arg'
|
||||||
|
restArgs = args[1:]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
passedArgs = append(passedArgs, s)
|
||||||
|
if len(shorthands) == 1 && searchedFlag.NoOptDefVal == "" && len(args) > 0 {
|
||||||
|
// '-f arg'
|
||||||
|
passedArgs = append(passedArgs, args[0])
|
||||||
|
restArgs = args[1:]
|
||||||
|
}
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Command) removeParentPersistentArgs(args []string) []string {
|
||||||
|
var newArgs []string
|
||||||
|
|
||||||
|
for len(args) > 0 {
|
||||||
|
s := args[0]
|
||||||
|
args = args[1:]
|
||||||
|
if s[0] != '-' {
|
||||||
|
newArgs = append(newArgs, s)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
var passedArgs, restArgs []string
|
||||||
|
if s[1] == '-' {
|
||||||
|
passedArgs, restArgs = c.parseLongArgs(s, args)
|
||||||
|
} else {
|
||||||
|
passedArgs, restArgs = c.parseShortArgs(s, args)
|
||||||
|
}
|
||||||
|
if len(passedArgs) > 0 {
|
||||||
|
newArgs = append(newArgs, passedArgs...)
|
||||||
|
}
|
||||||
|
args = restArgs
|
||||||
|
}
|
||||||
|
|
||||||
|
return newArgs
|
||||||
|
}
|
||||||
|
|
||||||
// ParseFlags parses persistent flag tree and local flags.
|
// ParseFlags parses persistent flag tree and local flags.
|
||||||
func (c *Command) ParseFlags(args []string) error {
|
func (c *Command) ParseFlags(args []string) error {
|
||||||
if c.DisableFlagParsing {
|
if c.DisableFlagParsing {
|
||||||
|
@ -1790,7 +1875,9 @@ func (c *Command) ParseFlags(args []string) error {
|
||||||
}
|
}
|
||||||
|
|
||||||
// parse Local Flags
|
// parse Local Flags
|
||||||
err = c.LocalFlags().Parse(args)
|
c.LocalFlags() // need to execute LocalFlags() to set the value in c.lflags before executing removeParentPersistentArgs
|
||||||
|
localArgs := c.removeParentPersistentArgs(args) // get only arguments related to c.lflags
|
||||||
|
err = c.lflags.Parse(localArgs)
|
||||||
// Print warnings if they occurred (e.g. deprecated flag messages).
|
// Print warnings if they occurred (e.g. deprecated flag messages).
|
||||||
if c.flagErrorBuf.Len()-beforeErrorBufLen > 0 && err == nil {
|
if c.flagErrorBuf.Len()-beforeErrorBufLen > 0 && err == nil {
|
||||||
c.Print(c.flagErrorBuf.String())
|
c.Print(c.flagErrorBuf.String())
|
||||||
|
|
Loading…
Reference in a new issue