Compare commits
25 Commits
19381fabe4
...
v3.8.6
Author | SHA1 | Date | |
---|---|---|---|
19bf7ef965 | |||
2134bf2b10 | |||
21b3d5ddc8 | |||
1dd7e71881 | |||
7e5339802a | |||
b89c92f109 | |||
7a2461f7ce | |||
d949a86f34 | |||
|
b57d7c3f4e | ||
0611fd60d2 | |||
3534470eb7 | |||
349bdb005e | |||
3064141583 | |||
|
29efa4ffa5 | ||
df0ee2daff | |||
|
281aaac71a | ||
188517a0c2 | |||
6304fc94d6 | |||
|
75d04ec22b | ||
|
1b2f46eb42 | ||
|
5cbbb17ce7 | ||
|
87208845f8 | ||
|
eb24cae8db | ||
|
77667c7d4a | ||
73c66bb531 |
4
.github/workflows/build.yml
vendored
4
.github/workflows/build.yml
vendored
@@ -12,11 +12,11 @@ jobs:
|
||||
- name: setup
|
||||
uses: actions/setup-go@v2
|
||||
with:
|
||||
go-version: 1.16
|
||||
go-version: 1.17
|
||||
- name: checkout
|
||||
uses: actions/checkout@v3
|
||||
- name: cache
|
||||
uses: actions/cache@v2
|
||||
uses: actions/cache@v3
|
||||
with:
|
||||
path: ~/go/pkg/mod
|
||||
key: ${{ runner.os }}-go-${{ hashFiles('**/go.sum') }}
|
||||
|
2
.github/workflows/codeql-analysis.yml
vendored
2
.github/workflows/codeql-analysis.yml
vendored
@@ -47,7 +47,7 @@ jobs:
|
||||
- name: setup
|
||||
uses: actions/setup-go@v2
|
||||
with:
|
||||
go-version: 1.16
|
||||
go-version: 1.17
|
||||
# Initializes the CodeQL tools for scanning.
|
||||
- name: init
|
||||
uses: github/codeql-action/init@v1
|
||||
|
4
.github/workflows/pr.yml
vendored
4
.github/workflows/pr.yml
vendored
@@ -12,11 +12,11 @@ jobs:
|
||||
- name: setup
|
||||
uses: actions/setup-go@v2
|
||||
with:
|
||||
go-version: 1.16
|
||||
go-version: 1.17
|
||||
- name: checkout
|
||||
uses: actions/checkout@v3
|
||||
- name: cache
|
||||
uses: actions/cache@v2
|
||||
uses: actions/cache@v3
|
||||
with:
|
||||
path: ~/go/pkg/mod
|
||||
key: ${{ runner.os }}-go-${{ hashFiles('**/go.sum') }}
|
||||
|
14
flag.go
14
flag.go
@@ -49,11 +49,21 @@ func (c *flagConfig) Init(opts ...config.Option) error {
|
||||
if !ok {
|
||||
continue
|
||||
}
|
||||
fn, fv, fd := getFlagOpts(tf)
|
||||
fn, fd, fv := getFlagOpts(tf)
|
||||
|
||||
rcheck := true
|
||||
|
||||
switch sf.Value.Interface().(type) {
|
||||
if !sf.Value.IsValid() {
|
||||
continue
|
||||
}
|
||||
vi := sf.Value.Interface()
|
||||
if vi == nil {
|
||||
continue
|
||||
}
|
||||
if f := flag.Lookup(fn); f != nil {
|
||||
return nil
|
||||
}
|
||||
switch vi.(type) {
|
||||
case time.Duration:
|
||||
err = c.flagDuration(sf.Value, fn, fv, fd)
|
||||
rcheck = false
|
||||
|
32
flag_test.go
32
flag_test.go
@@ -17,23 +17,37 @@ func TestLoad(t *testing.T) {
|
||||
os.Args = append(os.Args, "-addr", "33,44")
|
||||
os.Args = append(os.Args, "-time", time.RFC822)
|
||||
os.Args = append(os.Args, "-metadata", "key=20")
|
||||
os.Args = append(os.Args, "-components", "all=info,api=debug")
|
||||
type NestedConfig struct {
|
||||
Value string `flag:"name=nested_value"`
|
||||
}
|
||||
type Config struct {
|
||||
Broker string `flag:"name=broker,desc='description with, comma',default='127.0.0.1:9092'"`
|
||||
Verbose bool `flag:"name=verbose,desc='verbose output',default='false'"`
|
||||
Addr []string `flag:"name=addr,desc='addrs',default='127.0.0.1:9092'"`
|
||||
Wait time.Duration `flag:"name=wait,desc='wait time',default='2s'"`
|
||||
Time time.Time `flag:"name=time,desc='some time',default='02 Jan 06 15:04 MST'"`
|
||||
Metadata map[string]int `flag:"name=metadata,desc='some meta',default=''"`
|
||||
Broker string `flag:"name=broker,desc='description with, comma',default='127.0.0.1:9092'"`
|
||||
Verbose bool `flag:"name=verbose,desc='verbose output',default='false'"`
|
||||
Addr []string `flag:"name=addr,desc='addrs',default='127.0.0.1:9092'"`
|
||||
Wait time.Duration `flag:"name=wait,desc='wait time',default='2s'"`
|
||||
Time time.Time `flag:"name=time,desc='some time',default='02 Jan 06 15:04 MST'"`
|
||||
Metadata map[string]int `flag:"name=metadata,desc='some meta',default=''"`
|
||||
WithoutDefault string `flag:"name=without_default,desc='with'"`
|
||||
WithoutDesc string `flag:"name=without_desc,default='without_default'"`
|
||||
WithoutAll string `flag:"name=without_all"`
|
||||
Components map[string]string `flag:"name=components,desc='components logging'"`
|
||||
Nested *NestedConfig
|
||||
}
|
||||
|
||||
ctx := context.Background()
|
||||
cfg := &Config{}
|
||||
cfg := &Config{Nested: &NestedConfig{}}
|
||||
|
||||
c := NewConfig(config.Struct(cfg), TimeFormat(time.RFC822))
|
||||
if err := c.Init(); err != nil {
|
||||
t.Fatalf("init failed: %v", err)
|
||||
}
|
||||
|
||||
// double init test
|
||||
if err := c.Init(); err != nil {
|
||||
t.Fatalf("init failed: %v", err)
|
||||
}
|
||||
|
||||
if err := c.Load(ctx); err != nil {
|
||||
t.Fatalf("load failed: %v", err)
|
||||
}
|
||||
@@ -44,4 +58,8 @@ func TestLoad(t *testing.T) {
|
||||
if tf := cfg.Time.Format(time.RFC822); tf != "02 Jan 06 15:04 MST" {
|
||||
t.Fatalf("parse time error: %s != %s", tf, "02 Jan 06 15:04 MST")
|
||||
}
|
||||
|
||||
if len(cfg.Components) != 2 {
|
||||
t.Fatalf("cant parse map components %#+v", cfg)
|
||||
}
|
||||
}
|
||||
|
2
go.mod
2
go.mod
@@ -2,4 +2,4 @@ module go.unistack.org/micro-config-flag/v3
|
||||
|
||||
go 1.16
|
||||
|
||||
require go.unistack.org/micro/v3 v3.8.21
|
||||
require go.unistack.org/micro/v3 v3.9.1
|
||||
|
16
go.sum
16
go.sum
@@ -22,7 +22,7 @@ github.com/envoyproxy/go-control-plane v0.9.9-0.20210512163311-63b5d3c536b0/go.m
|
||||
github.com/envoyproxy/protoc-gen-validate v0.1.0/go.mod h1:iSmxcyjqTsJpI2R4NaDN7+kN2VEUnK/pcBlmesArF7c=
|
||||
github.com/flowstack/go-jsonschema v0.1.1/go.mod h1:yL7fNggx1o8rm9RlgXv7hTBWxdBM0rVwpMwimd3F3N0=
|
||||
github.com/ghodss/yaml v1.0.0/go.mod h1:4dBDuWmgqj2HViK6kFavaiC9ZROes6MMH2rRYeMEF04=
|
||||
github.com/golang-jwt/jwt/v4 v4.2.0/go.mod h1:/xlHOz8bRuivTWchD4jCa+NbatV+wEUSzwAxVc6locg=
|
||||
github.com/golang-jwt/jwt/v4 v4.4.0/go.mod h1:/xlHOz8bRuivTWchD4jCa+NbatV+wEUSzwAxVc6locg=
|
||||
github.com/golang/glog v0.0.0-20160126235308-23def4e6c14b/go.mod h1:SBH7ygxi8pfUlaOkMMuAQtPIUF8ecWP5IEl/CR7VP2Q=
|
||||
github.com/golang/mock v1.1.1/go.mod h1:oTYuIxOrZwtPieC+H1uAHpcLFnEyAGVDL/k47Jfbm0A=
|
||||
github.com/golang/protobuf v1.2.0/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=
|
||||
@@ -37,7 +37,9 @@ github.com/golang/protobuf v1.4.1/go.mod h1:U8fpvMrcmy5pZrNK1lt4xCsGvpyWQ/VVv6QD
|
||||
github.com/golang/protobuf v1.4.2/go.mod h1:oDoupMAO8OvCJWAcko0GGGIgR6R6ocIYbsSw735rRwI=
|
||||
github.com/golang/protobuf v1.4.3/go.mod h1:oDoupMAO8OvCJWAcko0GGGIgR6R6ocIYbsSw735rRwI=
|
||||
github.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk=
|
||||
github.com/golang/protobuf v1.5.2 h1:ROPKBNFfQgOUMifHyP+KYbvpjbdoFNs+aK7DXlji0Tw=
|
||||
github.com/golang/protobuf v1.5.2/go.mod h1:XVQd3VNwM+JqD3oG2Ue2ip4fOMUkwXdXDdiuN0vRsmY=
|
||||
github.com/google/gnostic v0.6.6 h1:MVSM2r2j9aRUvYNym66JGW96Ddd5MN4sTi59yktb6yk=
|
||||
github.com/google/gnostic v0.6.6/go.mod h1:Nm8234We1lq6iB9OmlgNv3nH91XLLVZHCDayfA3xq+E=
|
||||
github.com/google/go-cmp v0.2.0/go.mod h1:oXzfMopK8JAjlY9xF4vHSVASa0yLyX7SntLO5aqRK0M=
|
||||
github.com/google/go-cmp v0.3.0/go.mod h1:8QqcDgzrUqlUb/G2PQTWiueGozuR1884gddMywk6iLU=
|
||||
@@ -50,8 +52,10 @@ github.com/google/uuid v1.1.2/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+
|
||||
github.com/grpc-ecosystem/grpc-gateway v1.16.0/go.mod h1:BDjrQk3hbvj6Nolgz8mAMFbcEtjT1g+wF4CSlocrBnw=
|
||||
github.com/imdario/mergo v0.3.12 h1:b6R2BslTbIEToALKP7LxUvijTsNI9TAe80pLWN2g/HU=
|
||||
github.com/imdario/mergo v0.3.12/go.mod h1:jmQim1M+e3UYxmgPu/WyfjB3N3VflVyUjjjwH0dnCYA=
|
||||
github.com/kr/pretty v0.2.0 h1:s5hAObm+yFO5uHYt5dYjxi2rXrsnmRpJx4OYvIWUaQs=
|
||||
github.com/kr/pretty v0.2.0/go.mod h1:ipq/a2n7PKx3OHsz4KJII5eveXtPO4qwEXGdVfWzfnI=
|
||||
github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ=
|
||||
github.com/kr/text v0.1.0 h1:45sCR5RtlFHMR4UwH9sdQ5TC8v0qDQCHnXt+kaKSTVE=
|
||||
github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI=
|
||||
github.com/patrickmn/go-cache v2.1.0+incompatible/go.mod h1:3Qf8kWWT7OJRJbdiICTKqZju1ZixQ/KpMGzzAfe6+WQ=
|
||||
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||
@@ -67,10 +71,10 @@ github.com/xeipuuv/gojsonpointer v0.0.0-20180127040702-4e3ac2762d5f/go.mod h1:N2
|
||||
github.com/xeipuuv/gojsonreference v0.0.0-20180127040603-bd5ef7bd5415/go.mod h1:GwrjFmJcFw6At/Gs6z4yjiIwzuJ1/+UwLxMQDVQXShQ=
|
||||
github.com/xeipuuv/gojsonschema v1.2.0/go.mod h1:anYRn/JVcOK2ZgGU+IjEV4nwlhoK5sQluxsYJ78Id3Y=
|
||||
go.opentelemetry.io/proto/otlp v0.7.0/go.mod h1:PqfVotwruBrMGOCsRd/89rSnXhoiJIqeYNgFYFoEGnI=
|
||||
go.unistack.org/micro-proto/v3 v3.2.1 h1:z7+V97LcAwMbBiYStmf8b6fFk2UPJTzni+rxNqk4NrI=
|
||||
go.unistack.org/micro-proto/v3 v3.2.1/go.mod h1:ZltVWNECD5yK+40+OCONzGw4OtmSdTpVi8/KFgo9dqM=
|
||||
go.unistack.org/micro/v3 v3.8.21 h1:8vYxtIlPnCBZ888BrQSP6+b0CFQcn5FiaqDpI5XqorQ=
|
||||
go.unistack.org/micro/v3 v3.8.21/go.mod h1:aBOCQ6SkAwKMRI7Q1CIsQA2S88ibyJfweJZlT1M2qpE=
|
||||
go.unistack.org/micro-proto/v3 v3.2.7 h1:zG6d69kHc+oij2lwQ3AfrCgdjiEVRG2A7TlsxjusWs4=
|
||||
go.unistack.org/micro-proto/v3 v3.2.7/go.mod h1:ZltVWNECD5yK+40+OCONzGw4OtmSdTpVi8/KFgo9dqM=
|
||||
go.unistack.org/micro/v3 v3.9.1 h1:C1TEP8Eooqr5vlNLoZ3LCpNqk+XuCiC1yBHol+qJZ4E=
|
||||
go.unistack.org/micro/v3 v3.9.1/go.mod h1:7ssIWk+PJXvb2nSl8NUnQRs32JJEId2IDi9PobrQlKo=
|
||||
golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
|
||||
golang.org/x/crypto v0.0.0-20200622213623-75b288015ac9/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto=
|
||||
golang.org/x/exp v0.0.0-20190121172915-509febef88a4/go.mod h1:CJ0aWSM057203Lf6IL+f9T1iT9GByDxfZKAQTCR3kQA=
|
||||
@@ -142,12 +146,14 @@ google.golang.org/protobuf v1.26.0/go.mod h1:9q0QmTI4eRPtz6boOQmLYwt+qCgq0jsYwAQ
|
||||
google.golang.org/protobuf v1.27.1 h1:SnqbnDw1V7RiZcXPx5MEeqPv2s79L9i7BJUlG/+RurQ=
|
||||
google.golang.org/protobuf v1.27.1/go.mod h1:9q0QmTI4eRPtz6boOQmLYwt+qCgq0jsYwAQnmE0givc=
|
||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||
gopkg.in/check.v1 v1.0.0-20190902080502-41f04d3bba15 h1:YR8cESwS4TdDjEe65xsg0ogRM/Nc3DYOhEAlW+xobZo=
|
||||
gopkg.in/check.v1 v1.0.0-20190902080502-41f04d3bba15/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||
gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
|
||||
gopkg.in/yaml.v2 v2.2.3/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
|
||||
gopkg.in/yaml.v2 v2.3.0 h1:clyUAQHOM3G0M3f5vQj7LuJrETvjVot3Z5el9nffUtU=
|
||||
gopkg.in/yaml.v2 v2.3.0/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
|
||||
gopkg.in/yaml.v3 v3.0.0-20200615113413-eeeca48fe776/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||
gopkg.in/yaml.v3 v3.0.0-20210107192922-496545a6307b h1:h8qDotaEPuJATrMmW04NCwg7v22aHH28wwpauUhK9Oo=
|
||||
gopkg.in/yaml.v3 v3.0.0-20210107192922-496545a6307b/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||
honnef.co/go/tools v0.0.0-20190102054323-c2f93a96b099/go.mod h1:rf3lG4BRIbNafJWhAfAdb/ePZxsR/4RtNHQocxwk9r4=
|
||||
honnef.co/go/tools v0.0.0-20190523083050-ea95bdfd59fc/go.mod h1:rf3lG4BRIbNafJWhAfAdb/ePZxsR/4RtNHQocxwk9r4=
|
||||
|
323
util.go
323
util.go
@@ -2,12 +2,131 @@ package flag
|
||||
|
||||
import (
|
||||
"flag"
|
||||
"fmt"
|
||||
"reflect"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
)
|
||||
|
||||
type mapValue struct {
|
||||
v reflect.Value
|
||||
delim string
|
||||
def string
|
||||
}
|
||||
|
||||
func (v mapValue) String() string {
|
||||
if v.v.Kind() != reflect.Invalid {
|
||||
var kv []string
|
||||
it := v.v.MapRange()
|
||||
for it.Next() {
|
||||
k := it.Key().Interface()
|
||||
v := it.Value().Interface()
|
||||
kv = append(kv, fmt.Sprintf("%v=%v", k, v))
|
||||
}
|
||||
return strings.Join(kv, ",")
|
||||
}
|
||||
return v.def
|
||||
}
|
||||
|
||||
func (v mapValue) Set(s string) error {
|
||||
ps := strings.Split(s, v.delim)
|
||||
if len(ps) == 0 {
|
||||
return nil
|
||||
}
|
||||
v.v.Set(reflect.MakeMapWithSize(v.v.Type(), len(ps)))
|
||||
kt := v.v.Type().Key().Kind()
|
||||
vt := v.v.Type().Elem().Kind()
|
||||
|
||||
for i := 0; i < len(ps); i++ {
|
||||
fs := strings.Split(ps[i], "=")
|
||||
switch len(fs) {
|
||||
case 0:
|
||||
return nil
|
||||
case 1:
|
||||
if len(fs[0]) == 0 {
|
||||
return nil
|
||||
}
|
||||
return ErrInvalidValue
|
||||
case 2:
|
||||
break
|
||||
default:
|
||||
return ErrInvalidValue
|
||||
}
|
||||
key, err := convertType(reflect.ValueOf(fs[0]), kt)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
val, err := convertType(reflect.ValueOf(fs[1]), vt)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
v.v.SetMapIndex(key.Convert(v.v.Type().Key()), val.Convert(v.v.Type().Elem()))
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
type sliceValue struct {
|
||||
v reflect.Value
|
||||
delim string
|
||||
def string
|
||||
}
|
||||
|
||||
func (v sliceValue) String() string {
|
||||
if v.v.Kind() != reflect.Invalid {
|
||||
var kv []string
|
||||
for idx := 0; idx < v.v.Len(); idx++ {
|
||||
kv = append(kv, fmt.Sprintf("%v", v.v.Index(idx).Interface()))
|
||||
}
|
||||
return strings.Join(kv, ",")
|
||||
}
|
||||
return v.def
|
||||
}
|
||||
|
||||
func (v sliceValue) Set(s string) error {
|
||||
p := strings.Split(s, v.delim)
|
||||
v.v.Set(reflect.MakeSlice(v.v.Type(), len(p), len(p)))
|
||||
switch v.v.Type().Elem().Kind() {
|
||||
case reflect.Int, reflect.Int64:
|
||||
for idx := range p {
|
||||
i, err := strconv.ParseInt(p[idx], 10, 64)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
v.v.Index(idx).SetInt(i)
|
||||
}
|
||||
case reflect.Uint, reflect.Uint64:
|
||||
for idx := range p {
|
||||
i, err := strconv.ParseUint(p[idx], 10, 64)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
v.v.Index(idx).SetUint(i)
|
||||
}
|
||||
case reflect.Float64:
|
||||
for idx := range p {
|
||||
i, err := strconv.ParseFloat(p[idx], 64)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
v.v.Index(idx).SetFloat(i)
|
||||
}
|
||||
case reflect.Bool:
|
||||
for idx := range p {
|
||||
i, err := strconv.ParseBool(p[idx])
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
v.v.Index(idx).SetBool(i)
|
||||
}
|
||||
case reflect.String:
|
||||
for idx := range p {
|
||||
v.v.Index(idx).SetString(p[idx])
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func convertType(v reflect.Value, t reflect.Kind) (reflect.Value, error) {
|
||||
switch v.Kind() {
|
||||
case reflect.String:
|
||||
@@ -51,50 +170,12 @@ func (c *flagConfig) flagSlice(v reflect.Value, fn, fv, fd string) error {
|
||||
}
|
||||
}
|
||||
|
||||
flag.Func(fn, fd, func(s string) error {
|
||||
p := strings.Split(s, delim)
|
||||
v.Set(reflect.MakeSlice(v.Type(), len(p), len(p)))
|
||||
switch v.Type().Elem().Kind() {
|
||||
case reflect.Int, reflect.Int64:
|
||||
for idx := range p {
|
||||
i, err := strconv.ParseInt(p[idx], 10, 64)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
v.Index(idx).SetInt(i)
|
||||
}
|
||||
case reflect.Uint, reflect.Uint64:
|
||||
for idx := range p {
|
||||
i, err := strconv.ParseUint(p[idx], 10, 64)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
v.Index(idx).SetUint(i)
|
||||
}
|
||||
case reflect.Float64:
|
||||
for idx := range p {
|
||||
i, err := strconv.ParseFloat(p[idx], 64)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
v.Index(idx).SetFloat(i)
|
||||
}
|
||||
case reflect.Bool:
|
||||
for idx := range p {
|
||||
i, err := strconv.ParseBool(p[idx])
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
v.Index(idx).SetBool(i)
|
||||
}
|
||||
case reflect.String:
|
||||
for idx := range p {
|
||||
v.Index(idx).SetString(p[idx])
|
||||
}
|
||||
}
|
||||
return nil
|
||||
})
|
||||
|
||||
v.Set(reflect.MakeSlice(v.Type(), 0, 0))
|
||||
mp := &sliceValue{v: v, def: fv, delim: delim}
|
||||
if err := mp.Set(fv); err != nil {
|
||||
return err
|
||||
}
|
||||
flag.Var(mp, fn, fd)
|
||||
return nil
|
||||
}
|
||||
|
||||
@@ -105,43 +186,12 @@ func (c *flagConfig) flagMap(v reflect.Value, fn, fv, fd string) error {
|
||||
delim = d
|
||||
}
|
||||
}
|
||||
flag.Func(fn, fd, func(s string) error {
|
||||
ps := strings.Split(s, delim)
|
||||
if len(ps) == 0 {
|
||||
return nil
|
||||
}
|
||||
v.Set(reflect.MakeMapWithSize(v.Type(), len(ps)))
|
||||
kt := v.Type().Key().Kind()
|
||||
vt := v.Type().Elem().Kind()
|
||||
|
||||
for i := 0; i < len(ps); i++ {
|
||||
fs := strings.Split(ps[i], "=")
|
||||
switch len(fs) {
|
||||
case 0:
|
||||
return nil
|
||||
case 1:
|
||||
if len(fs[0]) == 0 {
|
||||
return nil
|
||||
}
|
||||
return ErrInvalidValue
|
||||
case 2:
|
||||
break
|
||||
default:
|
||||
return ErrInvalidValue
|
||||
}
|
||||
key, err := convertType(reflect.ValueOf(fs[0]), kt)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
val, err := convertType(reflect.ValueOf(fs[1]), vt)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
v.SetMapIndex(key.Convert(v.Type().Key()), val.Convert(v.Type().Elem()))
|
||||
}
|
||||
return nil
|
||||
})
|
||||
|
||||
v.Set(reflect.MakeMapWithSize(v.Type(), 0))
|
||||
mp := &mapValue{v: v, def: fv, delim: delim}
|
||||
if err := mp.Set(fv); err != nil {
|
||||
return err
|
||||
}
|
||||
flag.Var(mp, fn, fd)
|
||||
return nil
|
||||
}
|
||||
|
||||
@@ -172,7 +222,7 @@ func (c *flagConfig) flagDuration(v reflect.Value, fn, fv, fd string) error {
|
||||
if !ok {
|
||||
return ErrInvalidValue
|
||||
}
|
||||
i, err := time.ParseDuration(fd)
|
||||
i, err := time.ParseDuration(fv)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
@@ -185,7 +235,7 @@ func (c *flagConfig) flagBool(v reflect.Value, fn, fv, fd string) error {
|
||||
if !ok {
|
||||
return ErrInvalidValue
|
||||
}
|
||||
i, err := strconv.ParseBool(fd)
|
||||
i, err := strconv.ParseBool(fv)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
@@ -207,11 +257,11 @@ func (c *flagConfig) flagInt(v reflect.Value, fn, fv, fd string) error {
|
||||
if !ok {
|
||||
return ErrInvalidValue
|
||||
}
|
||||
i, err := strconv.ParseInt(fd, 10, 64)
|
||||
i, err := strconv.ParseInt(fv, 10, 64)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
flag.IntVar(nv, fn, int(i), fd)
|
||||
flag.IntVar(nv, fn, int(i), fv)
|
||||
return nil
|
||||
}
|
||||
|
||||
@@ -220,11 +270,11 @@ func (c *flagConfig) flagInt64(v reflect.Value, fn, fv, fd string) error {
|
||||
if !ok {
|
||||
return ErrInvalidValue
|
||||
}
|
||||
i, err := strconv.ParseInt(fd, 10, 64)
|
||||
i, err := strconv.ParseInt(fv, 10, 64)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
flag.Int64Var(nv, fn, int64(i), fd)
|
||||
flag.Int64Var(nv, fn, int64(i), fv)
|
||||
return nil
|
||||
}
|
||||
|
||||
@@ -233,11 +283,11 @@ func (c *flagConfig) flagUint(v reflect.Value, fn, fv, fd string) error {
|
||||
if !ok {
|
||||
return ErrInvalidValue
|
||||
}
|
||||
i, err := strconv.ParseUint(fd, 10, 64)
|
||||
i, err := strconv.ParseUint(fv, 10, 64)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
flag.UintVar(nv, fn, uint(i), fd)
|
||||
flag.UintVar(nv, fn, uint(i), fv)
|
||||
return nil
|
||||
}
|
||||
|
||||
@@ -246,11 +296,11 @@ func (c *flagConfig) flagUint64(v reflect.Value, fn, fv, fd string) error {
|
||||
if !ok {
|
||||
return ErrInvalidValue
|
||||
}
|
||||
i, err := strconv.ParseUint(fd, 10, 64)
|
||||
i, err := strconv.ParseUint(fv, 10, 64)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
flag.Uint64Var(nv, fn, uint64(i), fd)
|
||||
flag.Uint64Var(nv, fn, uint64(i), fv)
|
||||
return nil
|
||||
}
|
||||
|
||||
@@ -259,11 +309,11 @@ func (c *flagConfig) flagFloat64(v reflect.Value, fn, fv, fd string) error {
|
||||
if !ok {
|
||||
return ErrInvalidValue
|
||||
}
|
||||
i, err := strconv.ParseFloat(fd, 64)
|
||||
i, err := strconv.ParseFloat(fv, 64)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
flag.Float64Var(nv, fn, float64(i), fd)
|
||||
flag.Float64Var(nv, fn, float64(i), fv)
|
||||
return nil
|
||||
}
|
||||
|
||||
@@ -277,31 +327,74 @@ func (c *flagConfig) flagStringSlice(v reflect.Value, fn, fv, fd string) error {
|
||||
}
|
||||
|
||||
func getFlagOpts(tf string) (string, string, string) {
|
||||
ret := make([]string, 3)
|
||||
vals := strings.Split(tf, ",")
|
||||
f := 0
|
||||
for _, val := range vals {
|
||||
p := strings.Split(val, "=")
|
||||
switch p[0] {
|
||||
var name, desc, def string
|
||||
delim := ","
|
||||
|
||||
var buf string
|
||||
for idx := 0; idx < len(tf); idx++ {
|
||||
buf += string(tf[idx])
|
||||
switch buf {
|
||||
case "name":
|
||||
f = 0
|
||||
ndx := idx + 2
|
||||
stop := ","
|
||||
var quote bool
|
||||
for ; ndx < len(tf); ndx++ {
|
||||
if string(tf[ndx]) == stop {
|
||||
if quote {
|
||||
ndx++
|
||||
}
|
||||
break
|
||||
}
|
||||
if string(tf[ndx]) == "'" {
|
||||
stop = "'"
|
||||
quote = true
|
||||
continue
|
||||
}
|
||||
name += string(tf[ndx])
|
||||
}
|
||||
idx = ndx
|
||||
buf = ""
|
||||
case "desc":
|
||||
f = 1
|
||||
ndx := idx + 2
|
||||
stop := ","
|
||||
var quote bool
|
||||
for ; ndx < len(tf); ndx++ {
|
||||
if string(tf[ndx]) == stop {
|
||||
if quote {
|
||||
ndx++
|
||||
}
|
||||
break
|
||||
}
|
||||
if string(tf[ndx]) == "'" {
|
||||
stop = "'"
|
||||
quote = true
|
||||
continue
|
||||
}
|
||||
desc += string(tf[ndx])
|
||||
}
|
||||
idx = ndx
|
||||
buf = ""
|
||||
case "default":
|
||||
f = 2
|
||||
default:
|
||||
ret[f] += "," + val
|
||||
continue
|
||||
}
|
||||
ret[f] = p[1]
|
||||
}
|
||||
for idx := range ret {
|
||||
if ret[idx][0] == '\'' {
|
||||
ret[idx] = ret[idx][1:]
|
||||
}
|
||||
if ret[idx][len(ret[idx])-1] == '\'' {
|
||||
ret[idx] = ret[idx][:len(ret[idx])-1]
|
||||
ndx := idx + 2
|
||||
stop := ","
|
||||
var quote bool
|
||||
for ; ndx < len(tf); ndx++ {
|
||||
if string(tf[ndx]) == stop && (stop != delim) {
|
||||
if quote {
|
||||
ndx++
|
||||
}
|
||||
break
|
||||
}
|
||||
if string(tf[ndx]) == "'" {
|
||||
stop = "'"
|
||||
quote = true
|
||||
continue
|
||||
}
|
||||
def += string(tf[ndx])
|
||||
}
|
||||
idx = ndx
|
||||
buf = ""
|
||||
}
|
||||
}
|
||||
return ret[0], ret[1], ret[2]
|
||||
return name, desc, def
|
||||
}
|
||||
|
Reference in New Issue
Block a user