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>
75 lines
1.7 KiB
Go
75 lines
1.7 KiB
Go
package flag
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/iancoleman/strcase"
|
|
"github.com/spf13/pflag"
|
|
"google.golang.org/protobuf/reflect/protoreflect"
|
|
)
|
|
|
|
type enumType struct {
|
|
enum protoreflect.EnumDescriptor
|
|
}
|
|
|
|
func (b enumType) NewValue(context.Context, *Builder) pflag.Value {
|
|
val := &enumValue{
|
|
enum: b.enum,
|
|
valMap: map[string]protoreflect.EnumValueDescriptor{},
|
|
}
|
|
n := b.enum.Values().Len()
|
|
for i := 0; i < n; i++ {
|
|
valDesc := b.enum.Values().Get(i)
|
|
val.valMap[enumValueName(b.enum, valDesc)] = valDesc
|
|
}
|
|
return val
|
|
}
|
|
|
|
func (b enumType) DefaultValue() string {
|
|
defValue := ""
|
|
if def := b.enum.Values().ByNumber(0); def != nil {
|
|
defValue = enumValueName(b.enum, def)
|
|
}
|
|
return defValue
|
|
}
|
|
|
|
type enumValue struct {
|
|
enum protoreflect.EnumDescriptor
|
|
value protoreflect.EnumNumber
|
|
valMap map[string]protoreflect.EnumValueDescriptor
|
|
}
|
|
|
|
func (e enumValue) Get() protoreflect.Value {
|
|
return protoreflect.ValueOfEnum(e.value)
|
|
}
|
|
|
|
func enumValueName(enum protoreflect.EnumDescriptor, enumValue protoreflect.EnumValueDescriptor) string {
|
|
name := string(enumValue.Name())
|
|
name = strings.TrimPrefix(name, strcase.ToScreamingSnake(string(enum.Name()))+"_")
|
|
return strcase.ToKebab(name)
|
|
}
|
|
|
|
func (e enumValue) String() string {
|
|
return enumValueName(e.enum, e.enum.Values().ByNumber(e.value))
|
|
}
|
|
|
|
func (e *enumValue) Set(s string) error {
|
|
valDesc, ok := e.valMap[s]
|
|
if !ok {
|
|
return fmt.Errorf("%s is not a valid value for enum %s", s, e.enum.FullName())
|
|
}
|
|
e.value = valDesc.Number()
|
|
return nil
|
|
}
|
|
|
|
func (e enumValue) Type() string {
|
|
var vals []string
|
|
n := e.enum.Values().Len()
|
|
for i := 0; i < n; i++ {
|
|
vals = append(vals, enumValueName(e.enum, e.enum.Values().Get(i)))
|
|
}
|
|
return fmt.Sprintf("%s (%s)", e.enum.Name(), strings.Join(vals, " | "))
|
|
}
|