blob: 890a01afc030044c028766cc5d988255db86eacb [file] [log] [blame]
Zack Williamse940c7a2019-08-21 14:25:39 -07001package pflag
2
3import (
4 "bytes"
5 "encoding/csv"
6 "fmt"
7 "strings"
8)
9
10// -- stringToString Value
11type stringToStringValue struct {
12 value *map[string]string
13 changed bool
14}
15
16func newStringToStringValue(val map[string]string, p *map[string]string) *stringToStringValue {
17 ssv := new(stringToStringValue)
18 ssv.value = p
19 *ssv.value = val
20 return ssv
21}
22
23// Format: a=1,b=2
24func (s *stringToStringValue) Set(val string) error {
25 var ss []string
26 n := strings.Count(val, "=")
27 switch n {
28 case 0:
29 return fmt.Errorf("%s must be formatted as key=value", val)
30 case 1:
31 ss = append(ss, strings.Trim(val, `"`))
32 default:
33 r := csv.NewReader(strings.NewReader(val))
34 var err error
35 ss, err = r.Read()
36 if err != nil {
37 return err
38 }
39 }
40
41 out := make(map[string]string, len(ss))
42 for _, pair := range ss {
43 kv := strings.SplitN(pair, "=", 2)
44 if len(kv) != 2 {
45 return fmt.Errorf("%s must be formatted as key=value", pair)
46 }
47 out[kv[0]] = kv[1]
48 }
49 if !s.changed {
50 *s.value = out
51 } else {
52 for k, v := range out {
53 (*s.value)[k] = v
54 }
55 }
56 s.changed = true
57 return nil
58}
59
60func (s *stringToStringValue) Type() string {
61 return "stringToString"
62}
63
64func (s *stringToStringValue) String() string {
65 records := make([]string, 0, len(*s.value)>>1)
66 for k, v := range *s.value {
67 records = append(records, k+"="+v)
68 }
69
70 var buf bytes.Buffer
71 w := csv.NewWriter(&buf)
72 if err := w.Write(records); err != nil {
73 panic(err)
74 }
75 w.Flush()
76 return "[" + strings.TrimSpace(buf.String()) + "]"
77}
78
79func stringToStringConv(val string) (interface{}, error) {
80 val = strings.Trim(val, "[]")
81 // An empty string would cause an empty map
82 if len(val) == 0 {
83 return map[string]string{}, nil
84 }
85 r := csv.NewReader(strings.NewReader(val))
86 ss, err := r.Read()
87 if err != nil {
88 return nil, err
89 }
90 out := make(map[string]string, len(ss))
91 for _, pair := range ss {
92 kv := strings.SplitN(pair, "=", 2)
93 if len(kv) != 2 {
94 return nil, fmt.Errorf("%s must be formatted as key=value", pair)
95 }
96 out[kv[0]] = kv[1]
97 }
98 return out, nil
99}
100
101// GetStringToString return the map[string]string value of a flag with the given name
102func (f *FlagSet) GetStringToString(name string) (map[string]string, error) {
103 val, err := f.getFlagType(name, "stringToString", stringToStringConv)
104 if err != nil {
105 return map[string]string{}, err
106 }
107 return val.(map[string]string), nil
108}
109
110// StringToStringVar defines a string flag with specified name, default value, and usage string.
111// The argument p points to a map[string]string variable in which to store the values of the multiple flags.
112// The value of each argument will not try to be separated by comma
113func (f *FlagSet) StringToStringVar(p *map[string]string, name string, value map[string]string, usage string) {
114 f.VarP(newStringToStringValue(value, p), name, "", usage)
115}
116
117// StringToStringVarP is like StringToStringVar, but accepts a shorthand letter that can be used after a single dash.
118func (f *FlagSet) StringToStringVarP(p *map[string]string, name, shorthand string, value map[string]string, usage string) {
119 f.VarP(newStringToStringValue(value, p), name, shorthand, usage)
120}
121
122// StringToStringVar defines a string flag with specified name, default value, and usage string.
123// The argument p points to a map[string]string variable in which to store the value of the flag.
124// The value of each argument will not try to be separated by comma
125func StringToStringVar(p *map[string]string, name string, value map[string]string, usage string) {
126 CommandLine.VarP(newStringToStringValue(value, p), name, "", usage)
127}
128
129// StringToStringVarP is like StringToStringVar, but accepts a shorthand letter that can be used after a single dash.
130func StringToStringVarP(p *map[string]string, name, shorthand string, value map[string]string, usage string) {
131 CommandLine.VarP(newStringToStringValue(value, p), name, shorthand, usage)
132}
133
134// StringToString defines a string flag with specified name, default value, and usage string.
135// The return value is the address of a map[string]string variable that stores the value of the flag.
136// The value of each argument will not try to be separated by comma
137func (f *FlagSet) StringToString(name string, value map[string]string, usage string) *map[string]string {
138 p := map[string]string{}
139 f.StringToStringVarP(&p, name, "", value, usage)
140 return &p
141}
142
143// StringToStringP is like StringToString, but accepts a shorthand letter that can be used after a single dash.
144func (f *FlagSet) StringToStringP(name, shorthand string, value map[string]string, usage string) *map[string]string {
145 p := map[string]string{}
146 f.StringToStringVarP(&p, name, shorthand, value, usage)
147 return &p
148}
149
150// StringToString defines a string flag with specified name, default value, and usage string.
151// The return value is the address of a map[string]string variable that stores the value of the flag.
152// The value of each argument will not try to be separated by comma
153func StringToString(name string, value map[string]string, usage string) *map[string]string {
154 return CommandLine.StringToStringP(name, "", value, usage)
155}
156
157// StringToStringP is like StringToString, but accepts a shorthand letter that can be used after a single dash.
158func StringToStringP(name, shorthand string, value map[string]string, usage string) *map[string]string {
159 return CommandLine.StringToStringP(name, shorthand, value, usage)
160}