1c8a2d9069
* WIP on auto-generating CLi * WIP * WIP * WIP * add pagination.go * handle more flag types * WIP on refactoring * WIP * working tests * add docs * echo all flags * add repeated tests * remove comment * fix compositeListValue issue Co-authored-by: Anil Kumar Kammari <anil@vitwit.com> Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com>
55 lines
1.7 KiB
Go
55 lines
1.7 KiB
Go
package flag
|
|
|
|
import (
|
|
"github.com/spf13/pflag"
|
|
"google.golang.org/protobuf/reflect/protoreflect"
|
|
)
|
|
|
|
func bindSimpleFlag(flagSet *pflag.FlagSet, kind protoreflect.Kind, name, shorthand, usage string) SimpleValue {
|
|
switch kind {
|
|
case protoreflect.BytesKind:
|
|
val := flagSet.BytesBase64P(name, shorthand, nil, usage)
|
|
return simpleValue(func() protoreflect.Value {
|
|
return protoreflect.ValueOfBytes(*val)
|
|
})
|
|
case protoreflect.StringKind:
|
|
val := flagSet.StringP(name, shorthand, "", usage)
|
|
return simpleValue(func() protoreflect.Value {
|
|
return protoreflect.ValueOfString(*val)
|
|
})
|
|
case protoreflect.Uint32Kind, protoreflect.Fixed32Kind:
|
|
val := flagSet.Uint32P(name, shorthand, 0, usage)
|
|
return simpleValue(func() protoreflect.Value {
|
|
return protoreflect.ValueOfUint32(*val)
|
|
})
|
|
case protoreflect.Uint64Kind, protoreflect.Fixed64Kind:
|
|
val := flagSet.Uint64P(name, shorthand, 0, usage)
|
|
return simpleValue(func() protoreflect.Value {
|
|
return protoreflect.ValueOfUint64(*val)
|
|
})
|
|
case protoreflect.Int32Kind, protoreflect.Sint32Kind, protoreflect.Sfixed32Kind:
|
|
val := flagSet.Int32P(name, shorthand, 0, usage)
|
|
return simpleValue(func() protoreflect.Value {
|
|
return protoreflect.ValueOfInt32(*val)
|
|
})
|
|
case protoreflect.Int64Kind, protoreflect.Sint64Kind, protoreflect.Sfixed64Kind:
|
|
val := flagSet.Int64P(name, shorthand, 0, usage)
|
|
return simpleValue(func() protoreflect.Value {
|
|
return protoreflect.ValueOfInt64(*val)
|
|
})
|
|
case protoreflect.BoolKind:
|
|
val := flagSet.BoolP(name, shorthand, false, usage)
|
|
return simpleValue(func() protoreflect.Value {
|
|
return protoreflect.ValueOfBool(*val)
|
|
})
|
|
default:
|
|
return nil
|
|
}
|
|
}
|
|
|
|
type simpleValue func() protoreflect.Value
|
|
|
|
func (f simpleValue) Get() protoreflect.Value {
|
|
return f()
|
|
}
|