|
|
|
@ -4,26 +4,25 @@ import ( |
|
|
|
|
"flag" |
|
|
|
|
"fmt" |
|
|
|
|
"os" |
|
|
|
|
"strconv" |
|
|
|
|
"strings" |
|
|
|
|
"time" |
|
|
|
|
) |
|
|
|
|
|
|
|
|
|
// This flag enables bash-completion for all commands and subcommands
|
|
|
|
|
// BashCompletionFlag - This flag enables bash-completion for all commands and subcommands
|
|
|
|
|
// it is hidden by default
|
|
|
|
|
var BashCompletionFlag = BoolFlag{ |
|
|
|
|
Name: "generate-bash-completion", |
|
|
|
|
Hide: true, |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// This flag prints the version for the application
|
|
|
|
|
// VersionFlag - This flag prints the version for the application
|
|
|
|
|
var VersionFlag = BoolFlag{ |
|
|
|
|
Name: "version, v", |
|
|
|
|
Usage: "print the version", |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// This flag prints the help for all commands and subcommands
|
|
|
|
|
// HelpFlay - This flag prints the help for all commands and subcommands
|
|
|
|
|
// Set to the zero value (BoolFlag{}) to disable flag -- keeps subcommand
|
|
|
|
|
// unless HideHelp is set to true)
|
|
|
|
|
// unless HideHelp is set to true) it is hidden by default
|
|
|
|
|
var HelpFlag = BoolFlag{ |
|
|
|
|
Name: "help, h", |
|
|
|
|
Usage: "show help", |
|
|
|
@ -107,369 +106,6 @@ func (f GenericFlag) isNotHidden() bool { |
|
|
|
|
return !f.Hide |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
type StringSlice []string |
|
|
|
|
|
|
|
|
|
func (f *StringSlice) Set(value string) error { |
|
|
|
|
*f = append(*f, value) |
|
|
|
|
return nil |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (f *StringSlice) String() string { |
|
|
|
|
return fmt.Sprintf("%s", *f) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (f *StringSlice) Value() []string { |
|
|
|
|
return *f |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
type StringSliceFlag struct { |
|
|
|
|
Name string |
|
|
|
|
Value *StringSlice |
|
|
|
|
Usage string |
|
|
|
|
EnvVar string |
|
|
|
|
Hide bool |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (f StringSliceFlag) String() string { |
|
|
|
|
firstName := strings.Trim(strings.Split(f.Name, ",")[0], " ") |
|
|
|
|
pref := prefixFor(firstName) |
|
|
|
|
return withEnvHint(f.EnvVar, fmt.Sprintf("%s [%v]\t%v", prefixedNames(f.Name), pref+firstName+" option "+pref+firstName+" option", f.Usage)) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (f StringSliceFlag) Apply(set *flag.FlagSet) { |
|
|
|
|
if f.EnvVar != "" { |
|
|
|
|
for _, envVar := range strings.Split(f.EnvVar, ",") { |
|
|
|
|
envVar = strings.TrimSpace(envVar) |
|
|
|
|
if envVal := os.Getenv(envVar); envVal != "" { |
|
|
|
|
newVal := &StringSlice{} |
|
|
|
|
for _, s := range strings.Split(envVal, ",") { |
|
|
|
|
s = strings.TrimSpace(s) |
|
|
|
|
newVal.Set(s) |
|
|
|
|
} |
|
|
|
|
f.Value = newVal |
|
|
|
|
break |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
eachName(f.Name, func(name string) { |
|
|
|
|
set.Var(f.Value, name, f.Usage) |
|
|
|
|
}) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (f StringSliceFlag) getName() string { |
|
|
|
|
return f.Name |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (f StringSliceFlag) isNotHidden() bool { |
|
|
|
|
return !f.Hide |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
type IntSlice []int |
|
|
|
|
|
|
|
|
|
func (f *IntSlice) Set(value string) error { |
|
|
|
|
|
|
|
|
|
tmp, err := strconv.Atoi(value) |
|
|
|
|
if err != nil { |
|
|
|
|
return err |
|
|
|
|
} else { |
|
|
|
|
*f = append(*f, tmp) |
|
|
|
|
} |
|
|
|
|
return nil |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (f *IntSlice) String() string { |
|
|
|
|
return fmt.Sprintf("%d", *f) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (f *IntSlice) Value() []int { |
|
|
|
|
return *f |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
type IntSliceFlag struct { |
|
|
|
|
Name string |
|
|
|
|
Value *IntSlice |
|
|
|
|
Usage string |
|
|
|
|
EnvVar string |
|
|
|
|
Hide bool |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (f IntSliceFlag) String() string { |
|
|
|
|
firstName := strings.Trim(strings.Split(f.Name, ",")[0], " ") |
|
|
|
|
pref := prefixFor(firstName) |
|
|
|
|
return withEnvHint(f.EnvVar, fmt.Sprintf("%s [%v]\t%v", prefixedNames(f.Name), pref+firstName+" option "+pref+firstName+" option", f.Usage)) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (f IntSliceFlag) Apply(set *flag.FlagSet) { |
|
|
|
|
if f.EnvVar != "" { |
|
|
|
|
for _, envVar := range strings.Split(f.EnvVar, ",") { |
|
|
|
|
envVar = strings.TrimSpace(envVar) |
|
|
|
|
if envVal := os.Getenv(envVar); envVal != "" { |
|
|
|
|
newVal := &IntSlice{} |
|
|
|
|
for _, s := range strings.Split(envVal, ",") { |
|
|
|
|
s = strings.TrimSpace(s) |
|
|
|
|
err := newVal.Set(s) |
|
|
|
|
if err != nil { |
|
|
|
|
fmt.Fprintf(os.Stderr, err.Error()) |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
f.Value = newVal |
|
|
|
|
break |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
eachName(f.Name, func(name string) { |
|
|
|
|
set.Var(f.Value, name, f.Usage) |
|
|
|
|
}) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (f IntSliceFlag) getName() string { |
|
|
|
|
return f.Name |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (f IntSliceFlag) isNotHidden() bool { |
|
|
|
|
return !f.Hide |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
type BoolFlag struct { |
|
|
|
|
Name string |
|
|
|
|
Usage string |
|
|
|
|
EnvVar string |
|
|
|
|
Hide bool |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (f BoolFlag) String() string { |
|
|
|
|
return withEnvHint(f.EnvVar, fmt.Sprintf("%s\t%v", prefixedNames(f.Name), f.Usage)) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (f BoolFlag) Apply(set *flag.FlagSet) { |
|
|
|
|
val := false |
|
|
|
|
if f.EnvVar != "" { |
|
|
|
|
for _, envVar := range strings.Split(f.EnvVar, ",") { |
|
|
|
|
envVar = strings.TrimSpace(envVar) |
|
|
|
|
if envVal := os.Getenv(envVar); envVal != "" { |
|
|
|
|
envValBool, err := strconv.ParseBool(envVal) |
|
|
|
|
if err == nil { |
|
|
|
|
val = envValBool |
|
|
|
|
} |
|
|
|
|
break |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
eachName(f.Name, func(name string) { |
|
|
|
|
set.Bool(name, val, f.Usage) |
|
|
|
|
}) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (f BoolFlag) getName() string { |
|
|
|
|
return f.Name |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (f BoolFlag) isNotHidden() bool { |
|
|
|
|
return !f.Hide |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
type BoolTFlag struct { |
|
|
|
|
Name string |
|
|
|
|
Usage string |
|
|
|
|
EnvVar string |
|
|
|
|
Hide bool |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (f BoolTFlag) String() string { |
|
|
|
|
return withEnvHint(f.EnvVar, fmt.Sprintf("%s\t%v", prefixedNames(f.Name), f.Usage)) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (f BoolTFlag) Apply(set *flag.FlagSet) { |
|
|
|
|
val := true |
|
|
|
|
if f.EnvVar != "" { |
|
|
|
|
for _, envVar := range strings.Split(f.EnvVar, ",") { |
|
|
|
|
envVar = strings.TrimSpace(envVar) |
|
|
|
|
if envVal := os.Getenv(envVar); envVal != "" { |
|
|
|
|
envValBool, err := strconv.ParseBool(envVal) |
|
|
|
|
if err == nil { |
|
|
|
|
val = envValBool |
|
|
|
|
break |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
eachName(f.Name, func(name string) { |
|
|
|
|
set.Bool(name, val, f.Usage) |
|
|
|
|
}) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (f BoolTFlag) getName() string { |
|
|
|
|
return f.Name |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (f BoolTFlag) isNotHidden() bool { |
|
|
|
|
return !f.Hide |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
type StringFlag struct { |
|
|
|
|
Name string |
|
|
|
|
Value string |
|
|
|
|
Usage string |
|
|
|
|
EnvVar string |
|
|
|
|
Hide bool |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (f StringFlag) String() string { |
|
|
|
|
var fmtString string |
|
|
|
|
fmtString = "%s %v\t%v" |
|
|
|
|
|
|
|
|
|
if len(f.Value) > 0 { |
|
|
|
|
fmtString = "%s \"%v\"\t%v" |
|
|
|
|
} else { |
|
|
|
|
fmtString = "%s %v\t%v" |
|
|
|
|
} |
|
|
|
|
return withEnvHint(f.EnvVar, fmt.Sprintf(fmtString, prefixedNames(f.Name), f.Value, f.Usage)) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (f StringFlag) Apply(set *flag.FlagSet) { |
|
|
|
|
if f.EnvVar != "" { |
|
|
|
|
for _, envVar := range strings.Split(f.EnvVar, ",") { |
|
|
|
|
envVar = strings.TrimSpace(envVar) |
|
|
|
|
if envVal := os.Getenv(envVar); envVal != "" { |
|
|
|
|
f.Value = envVal |
|
|
|
|
break |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
eachName(f.Name, func(name string) { |
|
|
|
|
set.String(name, f.Value, f.Usage) |
|
|
|
|
}) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (f StringFlag) getName() string { |
|
|
|
|
return f.Name |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (f StringFlag) isNotHidden() bool { |
|
|
|
|
return !f.Hide |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
type IntFlag struct { |
|
|
|
|
Name string |
|
|
|
|
Value int |
|
|
|
|
Usage string |
|
|
|
|
EnvVar string |
|
|
|
|
Hide bool |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (f IntFlag) String() string { |
|
|
|
|
return withEnvHint(f.EnvVar, fmt.Sprintf("%s \"%v\"\t%v", prefixedNames(f.Name), f.Value, f.Usage)) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (f IntFlag) Apply(set *flag.FlagSet) { |
|
|
|
|
if f.EnvVar != "" { |
|
|
|
|
for _, envVar := range strings.Split(f.EnvVar, ",") { |
|
|
|
|
envVar = strings.TrimSpace(envVar) |
|
|
|
|
if envVal := os.Getenv(envVar); envVal != "" { |
|
|
|
|
envValInt, err := strconv.ParseInt(envVal, 0, 64) |
|
|
|
|
if err == nil { |
|
|
|
|
f.Value = int(envValInt) |
|
|
|
|
break |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
eachName(f.Name, func(name string) { |
|
|
|
|
set.Int(name, f.Value, f.Usage) |
|
|
|
|
}) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (f IntFlag) getName() string { |
|
|
|
|
return f.Name |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (f IntFlag) isNotHidden() bool { |
|
|
|
|
return !f.Hide |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
type DurationFlag struct { |
|
|
|
|
Name string |
|
|
|
|
Value time.Duration |
|
|
|
|
Usage string |
|
|
|
|
EnvVar string |
|
|
|
|
Hide bool |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (f DurationFlag) String() string { |
|
|
|
|
return withEnvHint(f.EnvVar, fmt.Sprintf("%s \"%v\"\t%v", prefixedNames(f.Name), f.Value, f.Usage)) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (f DurationFlag) Apply(set *flag.FlagSet) { |
|
|
|
|
if f.EnvVar != "" { |
|
|
|
|
for _, envVar := range strings.Split(f.EnvVar, ",") { |
|
|
|
|
envVar = strings.TrimSpace(envVar) |
|
|
|
|
if envVal := os.Getenv(envVar); envVal != "" { |
|
|
|
|
envValDuration, err := time.ParseDuration(envVal) |
|
|
|
|
if err == nil { |
|
|
|
|
f.Value = envValDuration |
|
|
|
|
break |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
eachName(f.Name, func(name string) { |
|
|
|
|
set.Duration(name, f.Value, f.Usage) |
|
|
|
|
}) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (f DurationFlag) getName() string { |
|
|
|
|
return f.Name |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (f DurationFlag) isNotHidden() bool { |
|
|
|
|
return !f.Hide |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
type Float64Flag struct { |
|
|
|
|
Name string |
|
|
|
|
Value float64 |
|
|
|
|
Usage string |
|
|
|
|
EnvVar string |
|
|
|
|
Hide bool |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (f Float64Flag) String() string { |
|
|
|
|
return withEnvHint(f.EnvVar, fmt.Sprintf("%s \"%v\"\t%v", prefixedNames(f.Name), f.Value, f.Usage)) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (f Float64Flag) Apply(set *flag.FlagSet) { |
|
|
|
|
if f.EnvVar != "" { |
|
|
|
|
for _, envVar := range strings.Split(f.EnvVar, ",") { |
|
|
|
|
envVar = strings.TrimSpace(envVar) |
|
|
|
|
if envVal := os.Getenv(envVar); envVal != "" { |
|
|
|
|
envValFloat, err := strconv.ParseFloat(envVal, 10) |
|
|
|
|
if err == nil { |
|
|
|
|
f.Value = float64(envValFloat) |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
eachName(f.Name, func(name string) { |
|
|
|
|
set.Float64(name, f.Value, f.Usage) |
|
|
|
|
}) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (f Float64Flag) getName() string { |
|
|
|
|
return f.Name |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (f Float64Flag) isNotHidden() bool { |
|
|
|
|
return !f.Hide |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func prefixFor(name string) (prefix string) { |
|
|
|
|
if len(name) == 1 { |
|
|
|
|
prefix = "-" |
|
|
|
|