1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
package flag
import (
"os"
"github.com/posener/complete"
)
// -- StringVar and stringValue
type StringVar struct {
Name string
Aliases []string
Usage string
Default string
Hidden bool
EnvVar string
Target *string
Completion complete.Predictor
SetHook func(val string)
}
func (f *Set) StringVar(i *StringVar) {
initial := i.Default
if v, exist := os.LookupEnv(i.EnvVar); exist {
initial = v
}
def := ""
if i.Default != "" {
def = i.Default
}
f.VarFlag(&VarFlag{
Name: i.Name,
Aliases: i.Aliases,
Usage: i.Usage,
Default: def,
EnvVar: i.EnvVar,
Value: newStringValue(i, initial, i.Target, i.Hidden),
Completion: i.Completion,
})
}
type stringValue struct {
v *StringVar
hidden bool
target *string
}
func newStringValue(v *StringVar, def string, target *string, hidden bool) *stringValue {
*target = def
return &stringValue{
v: v,
hidden: hidden,
target: target,
}
}
func (s *stringValue) Set(val string) error {
*s.target = val
if s.v.SetHook != nil {
s.v.SetHook(val)
}
return nil
}
func (s *stringValue) Get() interface{} { return *s.target }
func (s *stringValue) String() string { return *s.target }
func (s *stringValue) Example() string { return "string" }
func (s *stringValue) Hidden() bool { return s.hidden }
|
