-
Notifications
You must be signed in to change notification settings - Fork 0
/
flags.go
38 lines (34 loc) · 781 Bytes
/
flags.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
package utils
import (
"strings"
"github.com/spf13/pflag"
)
// ExtractUnknownArgs extracts unknown args from a flagset
// https://github.com/spf13/cobra/issues/739#issuecomment-677999676
func ExtractUnknownArgs(flags *pflag.FlagSet, args []string) []string {
unknownArgs := []string{}
for i := 0; i < len(args); i++ {
a := args[i]
var f *pflag.Flag
if a[0] == '-' {
if a[1] == '-' {
f = flags.Lookup(strings.SplitN(a[2:], "=", 2)[0])
} else {
for _, s := range a[1:] {
f = flags.ShorthandLookup(string(s))
if f == nil {
break
}
}
}
}
if f != nil {
if f.NoOptDefVal == "" && i+1 < len(args) && f.Value.String() == args[i+1] {
i++
}
continue
}
unknownArgs = append(unknownArgs, a)
}
return unknownArgs
}