75 lines
1.8 KiB
Go
75 lines
1.8 KiB
Go
package flag
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"google.golang.org/protobuf/reflect/protoreflect"
|
|
|
|
"cosmossdk.io/client/v2/internal/strcase"
|
|
)
|
|
|
|
type enumType struct {
|
|
enum protoreflect.EnumDescriptor
|
|
}
|
|
|
|
func (b enumType) NewValue(*context.Context, *Builder) 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) (protoreflect.Value, error) {
|
|
return protoreflect.ValueOfEnum(e.value), nil
|
|
}
|
|
|
|
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, " | "))
|
|
}
|