podman
160 строк · 5.2 Кб
1package pflag
2
3import (
4"bytes"
5"encoding/csv"
6"fmt"
7"strings"
8)
9
10// -- stringToString Value
11type stringToStringValue struct {
12value *map[string]string
13changed bool
14}
15
16func newStringToStringValue(val map[string]string, p *map[string]string) *stringToStringValue {
17ssv := new(stringToStringValue)
18ssv.value = p
19*ssv.value = val
20return ssv
21}
22
23// Format: a=1,b=2
24func (s *stringToStringValue) Set(val string) error {
25var ss []string
26n := strings.Count(val, "=")
27switch n {
28case 0:
29return fmt.Errorf("%s must be formatted as key=value", val)
30case 1:
31ss = append(ss, strings.Trim(val, `"`))
32default:
33r := csv.NewReader(strings.NewReader(val))
34var err error
35ss, err = r.Read()
36if err != nil {
37return err
38}
39}
40
41out := make(map[string]string, len(ss))
42for _, pair := range ss {
43kv := strings.SplitN(pair, "=", 2)
44if len(kv) != 2 {
45return fmt.Errorf("%s must be formatted as key=value", pair)
46}
47out[kv[0]] = kv[1]
48}
49if !s.changed {
50*s.value = out
51} else {
52for k, v := range out {
53(*s.value)[k] = v
54}
55}
56s.changed = true
57return nil
58}
59
60func (s *stringToStringValue) Type() string {
61return "stringToString"
62}
63
64func (s *stringToStringValue) String() string {
65records := make([]string, 0, len(*s.value)>>1)
66for k, v := range *s.value {
67records = append(records, k+"="+v)
68}
69
70var buf bytes.Buffer
71w := csv.NewWriter(&buf)
72if err := w.Write(records); err != nil {
73panic(err)
74}
75w.Flush()
76return "[" + strings.TrimSpace(buf.String()) + "]"
77}
78
79func stringToStringConv(val string) (interface{}, error) {
80val = strings.Trim(val, "[]")
81// An empty string would cause an empty map
82if len(val) == 0 {
83return map[string]string{}, nil
84}
85r := csv.NewReader(strings.NewReader(val))
86ss, err := r.Read()
87if err != nil {
88return nil, err
89}
90out := make(map[string]string, len(ss))
91for _, pair := range ss {
92kv := strings.SplitN(pair, "=", 2)
93if len(kv) != 2 {
94return nil, fmt.Errorf("%s must be formatted as key=value", pair)
95}
96out[kv[0]] = kv[1]
97}
98return 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) {
103val, err := f.getFlagType(name, "stringToString", stringToStringConv)
104if err != nil {
105return map[string]string{}, err
106}
107return 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) {
114f.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) {
119f.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) {
126CommandLine.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) {
131CommandLine.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 {
138p := map[string]string{}
139f.StringToStringVarP(&p, name, "", value, usage)
140return &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 {
145p := map[string]string{}
146f.StringToStringVarP(&p, name, shorthand, value, usage)
147return &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 {
154return 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 {
159return CommandLine.StringToStringP(name, shorthand, value, usage)
160}
161