1package 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}
161