123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799 |
- package cli
- import (
- "flag"
- "fmt"
- "reflect"
- "runtime"
- "strconv"
- "strings"
- "syscall"
- "time"
- )
- const defaultPlaceholder = "value"
- // BashCompletionFlag enables bash-completion for all commands and subcommands
- var BashCompletionFlag = BoolFlag{
- Name: "generate-bash-completion",
- Hidden: true,
- }
- // VersionFlag prints the version for the application
- var VersionFlag = BoolFlag{
- Name: "version, v",
- Usage: "print the version",
- }
- // HelpFlag 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)
- var HelpFlag = BoolFlag{
- Name: "help, h",
- Usage: "show help",
- }
- // FlagStringer converts a flag definition to a string. This is used by help
- // to display a flag.
- var FlagStringer FlagStringFunc = stringifyFlag
- // FlagsByName is a slice of Flag.
- type FlagsByName []Flag
- func (f FlagsByName) Len() int {
- return len(f)
- }
- func (f FlagsByName) Less(i, j int) bool {
- return f[i].GetName() < f[j].GetName()
- }
- func (f FlagsByName) Swap(i, j int) {
- f[i], f[j] = f[j], f[i]
- }
- // Flag is a common interface related to parsing flags in cli.
- // For more advanced flag parsing techniques, it is recommended that
- // this interface be implemented.
- type Flag interface {
- fmt.Stringer
- // Apply Flag settings to the given flag set
- Apply(*flag.FlagSet)
- GetName() string
- }
- // errorableFlag is an interface that allows us to return errors during apply
- // it allows flags defined in this library to return errors in a fashion backwards compatible
- // TODO remove in v2 and modify the existing Flag interface to return errors
- type errorableFlag interface {
- Flag
- ApplyWithError(*flag.FlagSet) error
- }
- func flagSet(name string, flags []Flag) (*flag.FlagSet, error) {
- set := flag.NewFlagSet(name, flag.ContinueOnError)
- for _, f := range flags {
- //TODO remove in v2 when errorableFlag is removed
- if ef, ok := f.(errorableFlag); ok {
- if err := ef.ApplyWithError(set); err != nil {
- return nil, err
- }
- } else {
- f.Apply(set)
- }
- }
- return set, nil
- }
- func eachName(longName string, fn func(string)) {
- parts := strings.Split(longName, ",")
- for _, name := range parts {
- name = strings.Trim(name, " ")
- fn(name)
- }
- }
- // Generic is a generic parseable type identified by a specific flag
- type Generic interface {
- Set(value string) error
- String() string
- }
- // Apply takes the flagset and calls Set on the generic flag with the value
- // provided by the user for parsing by the flag
- // Ignores parsing errors
- func (f GenericFlag) Apply(set *flag.FlagSet) {
- f.ApplyWithError(set)
- }
- // ApplyWithError takes the flagset and calls Set on the generic flag with the value
- // provided by the user for parsing by the flag
- func (f GenericFlag) ApplyWithError(set *flag.FlagSet) error {
- val := f.Value
- if f.EnvVar != "" {
- for _, envVar := range strings.Split(f.EnvVar, ",") {
- envVar = strings.TrimSpace(envVar)
- if envVal, ok := syscall.Getenv(envVar); ok {
- if err := val.Set(envVal); err != nil {
- return fmt.Errorf("could not parse %s as value for flag %s: %s", envVal, f.Name, err)
- }
- break
- }
- }
- }
- eachName(f.Name, func(name string) {
- set.Var(f.Value, name, f.Usage)
- })
- return nil
- }
- // StringSlice is an opaque type for []string to satisfy flag.Value and flag.Getter
- type StringSlice []string
- // Set appends the string value to the list of values
- func (f *StringSlice) Set(value string) error {
- *f = append(*f, value)
- return nil
- }
- // String returns a readable representation of this value (for usage defaults)
- func (f *StringSlice) String() string {
- return fmt.Sprintf("%s", *f)
- }
- // Value returns the slice of strings set by this flag
- func (f *StringSlice) Value() []string {
- return *f
- }
- // Get returns the slice of strings set by this flag
- func (f *StringSlice) Get() interface{} {
- return *f
- }
- // Apply populates the flag given the flag set and environment
- // Ignores errors
- func (f StringSliceFlag) Apply(set *flag.FlagSet) {
- f.ApplyWithError(set)
- }
- // ApplyWithError populates the flag given the flag set and environment
- func (f StringSliceFlag) ApplyWithError(set *flag.FlagSet) error {
- if f.EnvVar != "" {
- for _, envVar := range strings.Split(f.EnvVar, ",") {
- envVar = strings.TrimSpace(envVar)
- if envVal, ok := syscall.Getenv(envVar); ok {
- newVal := &StringSlice{}
- for _, s := range strings.Split(envVal, ",") {
- s = strings.TrimSpace(s)
- if err := newVal.Set(s); err != nil {
- return fmt.Errorf("could not parse %s as string value for flag %s: %s", envVal, f.Name, err)
- }
- }
- f.Value = newVal
- break
- }
- }
- }
- eachName(f.Name, func(name string) {
- if f.Value == nil {
- f.Value = &StringSlice{}
- }
- set.Var(f.Value, name, f.Usage)
- })
- return nil
- }
- // IntSlice is an opaque type for []int to satisfy flag.Value and flag.Getter
- type IntSlice []int
- // Set parses the value into an integer and appends it to the list of values
- func (f *IntSlice) Set(value string) error {
- tmp, err := strconv.Atoi(value)
- if err != nil {
- return err
- }
- *f = append(*f, tmp)
- return nil
- }
- // String returns a readable representation of this value (for usage defaults)
- func (f *IntSlice) String() string {
- return fmt.Sprintf("%#v", *f)
- }
- // Value returns the slice of ints set by this flag
- func (f *IntSlice) Value() []int {
- return *f
- }
- // Get returns the slice of ints set by this flag
- func (f *IntSlice) Get() interface{} {
- return *f
- }
- // Apply populates the flag given the flag set and environment
- // Ignores errors
- func (f IntSliceFlag) Apply(set *flag.FlagSet) {
- f.ApplyWithError(set)
- }
- // ApplyWithError populates the flag given the flag set and environment
- func (f IntSliceFlag) ApplyWithError(set *flag.FlagSet) error {
- if f.EnvVar != "" {
- for _, envVar := range strings.Split(f.EnvVar, ",") {
- envVar = strings.TrimSpace(envVar)
- if envVal, ok := syscall.Getenv(envVar); ok {
- newVal := &IntSlice{}
- for _, s := range strings.Split(envVal, ",") {
- s = strings.TrimSpace(s)
- if err := newVal.Set(s); err != nil {
- return fmt.Errorf("could not parse %s as int slice value for flag %s: %s", envVal, f.Name, err)
- }
- }
- f.Value = newVal
- break
- }
- }
- }
- eachName(f.Name, func(name string) {
- if f.Value == nil {
- f.Value = &IntSlice{}
- }
- set.Var(f.Value, name, f.Usage)
- })
- return nil
- }
- // Int64Slice is an opaque type for []int to satisfy flag.Value and flag.Getter
- type Int64Slice []int64
- // Set parses the value into an integer and appends it to the list of values
- func (f *Int64Slice) Set(value string) error {
- tmp, err := strconv.ParseInt(value, 10, 64)
- if err != nil {
- return err
- }
- *f = append(*f, tmp)
- return nil
- }
- // String returns a readable representation of this value (for usage defaults)
- func (f *Int64Slice) String() string {
- return fmt.Sprintf("%#v", *f)
- }
- // Value returns the slice of ints set by this flag
- func (f *Int64Slice) Value() []int64 {
- return *f
- }
- // Get returns the slice of ints set by this flag
- func (f *Int64Slice) Get() interface{} {
- return *f
- }
- // Apply populates the flag given the flag set and environment
- // Ignores errors
- func (f Int64SliceFlag) Apply(set *flag.FlagSet) {
- f.ApplyWithError(set)
- }
- // ApplyWithError populates the flag given the flag set and environment
- func (f Int64SliceFlag) ApplyWithError(set *flag.FlagSet) error {
- if f.EnvVar != "" {
- for _, envVar := range strings.Split(f.EnvVar, ",") {
- envVar = strings.TrimSpace(envVar)
- if envVal, ok := syscall.Getenv(envVar); ok {
- newVal := &Int64Slice{}
- for _, s := range strings.Split(envVal, ",") {
- s = strings.TrimSpace(s)
- if err := newVal.Set(s); err != nil {
- return fmt.Errorf("could not parse %s as int64 slice value for flag %s: %s", envVal, f.Name, err)
- }
- }
- f.Value = newVal
- break
- }
- }
- }
- eachName(f.Name, func(name string) {
- if f.Value == nil {
- f.Value = &Int64Slice{}
- }
- set.Var(f.Value, name, f.Usage)
- })
- return nil
- }
- // Apply populates the flag given the flag set and environment
- // Ignores errors
- func (f BoolFlag) Apply(set *flag.FlagSet) {
- f.ApplyWithError(set)
- }
- // ApplyWithError populates the flag given the flag set and environment
- func (f BoolFlag) ApplyWithError(set *flag.FlagSet) error {
- val := false
- if f.EnvVar != "" {
- for _, envVar := range strings.Split(f.EnvVar, ",") {
- envVar = strings.TrimSpace(envVar)
- if envVal, ok := syscall.Getenv(envVar); ok {
- if envVal == "" {
- val = false
- break
- }
- envValBool, err := strconv.ParseBool(envVal)
- if err != nil {
- return fmt.Errorf("could not parse %s as bool value for flag %s: %s", envVal, f.Name, err)
- }
- val = envValBool
- break
- }
- }
- }
- eachName(f.Name, func(name string) {
- if f.Destination != nil {
- set.BoolVar(f.Destination, name, val, f.Usage)
- return
- }
- set.Bool(name, val, f.Usage)
- })
- return nil
- }
- // Apply populates the flag given the flag set and environment
- // Ignores errors
- func (f BoolTFlag) Apply(set *flag.FlagSet) {
- f.ApplyWithError(set)
- }
- // ApplyWithError populates the flag given the flag set and environment
- func (f BoolTFlag) ApplyWithError(set *flag.FlagSet) error {
- val := true
- if f.EnvVar != "" {
- for _, envVar := range strings.Split(f.EnvVar, ",") {
- envVar = strings.TrimSpace(envVar)
- if envVal, ok := syscall.Getenv(envVar); ok {
- if envVal == "" {
- val = false
- break
- }
- envValBool, err := strconv.ParseBool(envVal)
- if err != nil {
- return fmt.Errorf("could not parse %s as bool value for flag %s: %s", envVal, f.Name, err)
- }
- val = envValBool
- break
- }
- }
- }
- eachName(f.Name, func(name string) {
- if f.Destination != nil {
- set.BoolVar(f.Destination, name, val, f.Usage)
- return
- }
- set.Bool(name, val, f.Usage)
- })
- return nil
- }
- // Apply populates the flag given the flag set and environment
- // Ignores errors
- func (f StringFlag) Apply(set *flag.FlagSet) {
- f.ApplyWithError(set)
- }
- // ApplyWithError populates the flag given the flag set and environment
- func (f StringFlag) ApplyWithError(set *flag.FlagSet) error {
- if f.EnvVar != "" {
- for _, envVar := range strings.Split(f.EnvVar, ",") {
- envVar = strings.TrimSpace(envVar)
- if envVal, ok := syscall.Getenv(envVar); ok {
- f.Value = envVal
- break
- }
- }
- }
- eachName(f.Name, func(name string) {
- if f.Destination != nil {
- set.StringVar(f.Destination, name, f.Value, f.Usage)
- return
- }
- set.String(name, f.Value, f.Usage)
- })
- return nil
- }
- // Apply populates the flag given the flag set and environment
- // Ignores errors
- func (f IntFlag) Apply(set *flag.FlagSet) {
- f.ApplyWithError(set)
- }
- // ApplyWithError populates the flag given the flag set and environment
- func (f IntFlag) ApplyWithError(set *flag.FlagSet) error {
- if f.EnvVar != "" {
- for _, envVar := range strings.Split(f.EnvVar, ",") {
- envVar = strings.TrimSpace(envVar)
- if envVal, ok := syscall.Getenv(envVar); ok {
- envValInt, err := strconv.ParseInt(envVal, 0, 64)
- if err != nil {
- return fmt.Errorf("could not parse %s as int value for flag %s: %s", envVal, f.Name, err)
- }
- f.Value = int(envValInt)
- break
- }
- }
- }
- eachName(f.Name, func(name string) {
- if f.Destination != nil {
- set.IntVar(f.Destination, name, f.Value, f.Usage)
- return
- }
- set.Int(name, f.Value, f.Usage)
- })
- return nil
- }
- // Apply populates the flag given the flag set and environment
- // Ignores errors
- func (f Int64Flag) Apply(set *flag.FlagSet) {
- f.ApplyWithError(set)
- }
- // ApplyWithError populates the flag given the flag set and environment
- func (f Int64Flag) ApplyWithError(set *flag.FlagSet) error {
- if f.EnvVar != "" {
- for _, envVar := range strings.Split(f.EnvVar, ",") {
- envVar = strings.TrimSpace(envVar)
- if envVal, ok := syscall.Getenv(envVar); ok {
- envValInt, err := strconv.ParseInt(envVal, 0, 64)
- if err != nil {
- return fmt.Errorf("could not parse %s as int value for flag %s: %s", envVal, f.Name, err)
- }
- f.Value = envValInt
- break
- }
- }
- }
- eachName(f.Name, func(name string) {
- if f.Destination != nil {
- set.Int64Var(f.Destination, name, f.Value, f.Usage)
- return
- }
- set.Int64(name, f.Value, f.Usage)
- })
- return nil
- }
- // Apply populates the flag given the flag set and environment
- // Ignores errors
- func (f UintFlag) Apply(set *flag.FlagSet) {
- f.ApplyWithError(set)
- }
- // ApplyWithError populates the flag given the flag set and environment
- func (f UintFlag) ApplyWithError(set *flag.FlagSet) error {
- if f.EnvVar != "" {
- for _, envVar := range strings.Split(f.EnvVar, ",") {
- envVar = strings.TrimSpace(envVar)
- if envVal, ok := syscall.Getenv(envVar); ok {
- envValInt, err := strconv.ParseUint(envVal, 0, 64)
- if err != nil {
- return fmt.Errorf("could not parse %s as uint value for flag %s: %s", envVal, f.Name, err)
- }
- f.Value = uint(envValInt)
- break
- }
- }
- }
- eachName(f.Name, func(name string) {
- if f.Destination != nil {
- set.UintVar(f.Destination, name, f.Value, f.Usage)
- return
- }
- set.Uint(name, f.Value, f.Usage)
- })
- return nil
- }
- // Apply populates the flag given the flag set and environment
- // Ignores errors
- func (f Uint64Flag) Apply(set *flag.FlagSet) {
- f.ApplyWithError(set)
- }
- // ApplyWithError populates the flag given the flag set and environment
- func (f Uint64Flag) ApplyWithError(set *flag.FlagSet) error {
- if f.EnvVar != "" {
- for _, envVar := range strings.Split(f.EnvVar, ",") {
- envVar = strings.TrimSpace(envVar)
- if envVal, ok := syscall.Getenv(envVar); ok {
- envValInt, err := strconv.ParseUint(envVal, 0, 64)
- if err != nil {
- return fmt.Errorf("could not parse %s as uint64 value for flag %s: %s", envVal, f.Name, err)
- }
- f.Value = uint64(envValInt)
- break
- }
- }
- }
- eachName(f.Name, func(name string) {
- if f.Destination != nil {
- set.Uint64Var(f.Destination, name, f.Value, f.Usage)
- return
- }
- set.Uint64(name, f.Value, f.Usage)
- })
- return nil
- }
- // Apply populates the flag given the flag set and environment
- // Ignores errors
- func (f DurationFlag) Apply(set *flag.FlagSet) {
- f.ApplyWithError(set)
- }
- // ApplyWithError populates the flag given the flag set and environment
- func (f DurationFlag) ApplyWithError(set *flag.FlagSet) error {
- if f.EnvVar != "" {
- for _, envVar := range strings.Split(f.EnvVar, ",") {
- envVar = strings.TrimSpace(envVar)
- if envVal, ok := syscall.Getenv(envVar); ok {
- envValDuration, err := time.ParseDuration(envVal)
- if err != nil {
- return fmt.Errorf("could not parse %s as duration for flag %s: %s", envVal, f.Name, err)
- }
- f.Value = envValDuration
- break
- }
- }
- }
- eachName(f.Name, func(name string) {
- if f.Destination != nil {
- set.DurationVar(f.Destination, name, f.Value, f.Usage)
- return
- }
- set.Duration(name, f.Value, f.Usage)
- })
- return nil
- }
- // Apply populates the flag given the flag set and environment
- // Ignores errors
- func (f Float64Flag) Apply(set *flag.FlagSet) {
- f.ApplyWithError(set)
- }
- // ApplyWithError populates the flag given the flag set and environment
- func (f Float64Flag) ApplyWithError(set *flag.FlagSet) error {
- if f.EnvVar != "" {
- for _, envVar := range strings.Split(f.EnvVar, ",") {
- envVar = strings.TrimSpace(envVar)
- if envVal, ok := syscall.Getenv(envVar); ok {
- envValFloat, err := strconv.ParseFloat(envVal, 10)
- if err != nil {
- return fmt.Errorf("could not parse %s as float64 value for flag %s: %s", envVal, f.Name, err)
- }
- f.Value = float64(envValFloat)
- break
- }
- }
- }
- eachName(f.Name, func(name string) {
- if f.Destination != nil {
- set.Float64Var(f.Destination, name, f.Value, f.Usage)
- return
- }
- set.Float64(name, f.Value, f.Usage)
- })
- return nil
- }
- func visibleFlags(fl []Flag) []Flag {
- visible := []Flag{}
- for _, flag := range fl {
- if !flagValue(flag).FieldByName("Hidden").Bool() {
- visible = append(visible, flag)
- }
- }
- return visible
- }
- func prefixFor(name string) (prefix string) {
- if len(name) == 1 {
- prefix = "-"
- } else {
- prefix = "--"
- }
- return
- }
- // Returns the placeholder, if any, and the unquoted usage string.
- func unquoteUsage(usage string) (string, string) {
- for i := 0; i < len(usage); i++ {
- if usage[i] == '`' {
- for j := i + 1; j < len(usage); j++ {
- if usage[j] == '`' {
- name := usage[i+1 : j]
- usage = usage[:i] + name + usage[j+1:]
- return name, usage
- }
- }
- break
- }
- }
- return "", usage
- }
- func prefixedNames(fullName, placeholder string) string {
- var prefixed string
- parts := strings.Split(fullName, ",")
- for i, name := range parts {
- name = strings.Trim(name, " ")
- prefixed += prefixFor(name) + name
- if placeholder != "" {
- prefixed += " " + placeholder
- }
- if i < len(parts)-1 {
- prefixed += ", "
- }
- }
- return prefixed
- }
- func withEnvHint(envVar, str string) string {
- envText := ""
- if envVar != "" {
- prefix := "$"
- suffix := ""
- sep := ", $"
- if runtime.GOOS == "windows" {
- prefix = "%"
- suffix = "%"
- sep = "%, %"
- }
- envText = fmt.Sprintf(" [%s%s%s]", prefix, strings.Join(strings.Split(envVar, ","), sep), suffix)
- }
- return str + envText
- }
- func flagValue(f Flag) reflect.Value {
- fv := reflect.ValueOf(f)
- for fv.Kind() == reflect.Ptr {
- fv = reflect.Indirect(fv)
- }
- return fv
- }
- func stringifyFlag(f Flag) string {
- fv := flagValue(f)
- switch f.(type) {
- case IntSliceFlag:
- return withEnvHint(fv.FieldByName("EnvVar").String(),
- stringifyIntSliceFlag(f.(IntSliceFlag)))
- case Int64SliceFlag:
- return withEnvHint(fv.FieldByName("EnvVar").String(),
- stringifyInt64SliceFlag(f.(Int64SliceFlag)))
- case StringSliceFlag:
- return withEnvHint(fv.FieldByName("EnvVar").String(),
- stringifyStringSliceFlag(f.(StringSliceFlag)))
- }
- placeholder, usage := unquoteUsage(fv.FieldByName("Usage").String())
- needsPlaceholder := false
- defaultValueString := ""
- val := fv.FieldByName("Value")
- if val.IsValid() {
- needsPlaceholder = true
- defaultValueString = fmt.Sprintf(" (default: %v)", val.Interface())
- if val.Kind() == reflect.String && val.String() != "" {
- defaultValueString = fmt.Sprintf(" (default: %q)", val.String())
- }
- }
- if defaultValueString == " (default: )" {
- defaultValueString = ""
- }
- if needsPlaceholder && placeholder == "" {
- placeholder = defaultPlaceholder
- }
- usageWithDefault := strings.TrimSpace(fmt.Sprintf("%s%s", usage, defaultValueString))
- return withEnvHint(fv.FieldByName("EnvVar").String(),
- fmt.Sprintf("%s\t%s", prefixedNames(fv.FieldByName("Name").String(), placeholder), usageWithDefault))
- }
- func stringifyIntSliceFlag(f IntSliceFlag) string {
- defaultVals := []string{}
- if f.Value != nil && len(f.Value.Value()) > 0 {
- for _, i := range f.Value.Value() {
- defaultVals = append(defaultVals, fmt.Sprintf("%d", i))
- }
- }
- return stringifySliceFlag(f.Usage, f.Name, defaultVals)
- }
- func stringifyInt64SliceFlag(f Int64SliceFlag) string {
- defaultVals := []string{}
- if f.Value != nil && len(f.Value.Value()) > 0 {
- for _, i := range f.Value.Value() {
- defaultVals = append(defaultVals, fmt.Sprintf("%d", i))
- }
- }
- return stringifySliceFlag(f.Usage, f.Name, defaultVals)
- }
- func stringifyStringSliceFlag(f StringSliceFlag) string {
- defaultVals := []string{}
- if f.Value != nil && len(f.Value.Value()) > 0 {
- for _, s := range f.Value.Value() {
- if len(s) > 0 {
- defaultVals = append(defaultVals, fmt.Sprintf("%q", s))
- }
- }
- }
- return stringifySliceFlag(f.Usage, f.Name, defaultVals)
- }
- func stringifySliceFlag(usage, name string, defaultVals []string) string {
- placeholder, usage := unquoteUsage(usage)
- if placeholder == "" {
- placeholder = defaultPlaceholder
- }
- defaultVal := ""
- if len(defaultVals) > 0 {
- defaultVal = fmt.Sprintf(" (default: %s)", strings.Join(defaultVals, ", "))
- }
- usageWithDefault := strings.TrimSpace(fmt.Sprintf("%s%s", usage, defaultVal))
- return fmt.Sprintf("%s\t%s", prefixedNames(name, placeholder), usageWithDefault)
- }
|