57b4d3003b
Co-authored-by: Marko <marko@baricevic.me> Co-authored-by: Julien Robert <julien@rbrt.fr>
59 lines
1.1 KiB
Go
59 lines
1.1 KiB
Go
package flag
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"strings"
|
|
|
|
"google.golang.org/protobuf/reflect/protoreflect"
|
|
|
|
basev1beta1 "cosmossdk.io/api/cosmos/base/v1beta1"
|
|
"cosmossdk.io/client/v2/internal/coins"
|
|
)
|
|
|
|
type coinType struct{}
|
|
|
|
type coinValue struct {
|
|
value *basev1beta1.Coin
|
|
}
|
|
|
|
func (c coinType) NewValue(*context.Context, *Builder) Value {
|
|
return &coinValue{}
|
|
}
|
|
|
|
func (c coinType) DefaultValue() string {
|
|
return "zero"
|
|
}
|
|
|
|
func (c *coinValue) Get(protoreflect.Value) (protoreflect.Value, error) {
|
|
if c.value == nil {
|
|
return protoreflect.Value{}, nil
|
|
}
|
|
return protoreflect.ValueOfMessage(c.value.ProtoReflect()), nil
|
|
}
|
|
|
|
func (c *coinValue) String() string {
|
|
if c.value == nil {
|
|
return ""
|
|
}
|
|
|
|
return c.value.String()
|
|
}
|
|
|
|
func (c *coinValue) Set(stringValue string) error {
|
|
if strings.Contains(stringValue, ",") {
|
|
return errors.New("coin flag must be a single coin, specific multiple coins with multiple flags or spaces")
|
|
}
|
|
|
|
coin, err := coins.ParseCoin(stringValue)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
c.value = coin
|
|
return nil
|
|
}
|
|
|
|
func (c *coinValue) Type() string {
|
|
return "cosmos.base.v1beta1.Coin"
|
|
}
|