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>
53 lines
929 B
Go
53 lines
929 B
Go
package flag
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/spf13/pflag"
|
|
"google.golang.org/protobuf/reflect/protoreflect"
|
|
"google.golang.org/protobuf/types/known/durationpb"
|
|
)
|
|
|
|
type durationType struct{}
|
|
|
|
func (t durationType) NewValue(context.Context, *Builder) pflag.Value {
|
|
return &durationValue{}
|
|
}
|
|
|
|
func (t durationType) DefaultValue() string {
|
|
return ""
|
|
}
|
|
|
|
type durationValue struct {
|
|
value *durationpb.Duration
|
|
}
|
|
|
|
func (t durationValue) Get() protoreflect.Value {
|
|
if t.value == nil {
|
|
return protoreflect.Value{}
|
|
}
|
|
return protoreflect.ValueOfMessage(t.value.ProtoReflect())
|
|
}
|
|
|
|
func (v durationValue) String() string {
|
|
if v.value == nil {
|
|
return ""
|
|
}
|
|
return v.value.AsDuration().String()
|
|
}
|
|
|
|
func (v *durationValue) Set(s string) error {
|
|
dur, err := time.ParseDuration(s)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
v.value = durationpb.New(dur)
|
|
return nil
|
|
}
|
|
|
|
func (v durationValue) Type() string {
|
|
return "duration"
|
|
}
|