From 41182b029c6a536f917139a7e27be08cc3a50d23 Mon Sep 17 00:00:00 2001 From: Eyal Posener Date: Sat, 9 May 2020 13:36:38 +0300 Subject: Autogenerate compflag code --- compflag/gen/flags.go.gotmpl | 83 ++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 83 insertions(+) create mode 100644 compflag/gen/flags.go.gotmpl (limited to 'compflag/gen/flags.go.gotmpl') diff --git a/compflag/gen/flags.go.gotmpl b/compflag/gen/flags.go.gotmpl new file mode 100644 index 0000000..9b65dd5 --- /dev/null +++ b/compflag/gen/flags.go.gotmpl @@ -0,0 +1,83 @@ +package compflag + +// Code auto generated with `go run ./gen`. DO NOT EDIT + +//go:generate go run ./gen + + +{{ range . }} + +// {{ .Name }} if a flag function for a flag of type {{ .Type }}. +func {{ .Name }}(name string, value {{ .Type }}, usage string, options ...predict.Option) *{{ .Type }} { + return CommandLine.{{ .Name }}(name, value, usage, options...) +} + +// {{ .Name }}Var if a flag function for a flag of already exiting variable of type {{ .Type }}. +func {{ .Name }}Var(v *{{ .Type }}, name string, value {{ .Type }}, usage string, options ...predict.Option) { + CommandLine.{{ .Name }}Var(v, name, value, usage, options...) +} + +{{ end }} + +{{ range . }} + +// {{ .Name }} if a flag function for a flag of type {{ .Type }}. +func (fs *FlagSet) {{ .Name }}(name string, value {{ .Type }}, usage string, options ...predict.Option) *{{ .Type }} { + p := new({{ .Type }}) + fs.{{.Name}}Var(p, name, value, usage, options...) + return p +} + +// {{ .Name }}Var if a flag function for a flag of already exiting variable of type {{ .Type }}. +func (fs *FlagSet) {{ .Name }}Var(p *{{ .Type }}, name string, value {{ .Type }}, usage string, options ...predict.Option) { + (*flag.FlagSet)(fs).Var({{ .NewInternalTypeFuncName }}(value, p, predict.Options(options...)), name, usage) +} + +{{ end }} + +{{ range . }} + +// ============================================================================================== // + +type {{ .InternalTypeName }} struct { + v *{{ .Type }} + predict.Config +} + +func {{ .NewInternalTypeFuncName }}(val {{ .Type }}, p *{{ .Type }}, c predict.Config) *{{ .InternalTypeName }} { + *p = val + return &{{ .InternalTypeName}} {v: p, Config: c} +} + +func (v *{{ .InternalTypeName }}) Set(val string) error { + var err error + *v.v, err = parse{{ .Name }}(val) + if err != nil { + return fmt.Errorf("bad value for {{ .Name }} flag") + } + return v.Check(val) +} + +func (v *{{ .InternalTypeName }}) Get() interface{} { + return *v.v +} + +func (v *{{ .InternalTypeName }}) String() string { + if v == nil || v.v == nil { + return "" + } + return format{{ .Name }}(*v.v) +} + +{{ if .IsBool }} +func (v *{{ .InternalTypeName }}) IsBoolFlag() bool { return true } +{{ end}} + +func (v *{{ .InternalTypeName }}) Predict(prefix string) []string { + if v.Predictor != nil { + return v.Predictor.Predict(prefix) + } + return {{ if .CustomPredict }}predict{{ .Name }}(*v.v, prefix){{ else }}[]string{""}{{ end }} +} + +{{ end }} \ No newline at end of file -- cgit v1.2.3