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
73
74
75
76
77
78
79
80
|
package flag
import (
"fmt"
"os"
"strings"
"github.com/posener/complete"
)
// -- EnumVar and enumValue
type EnumSingleVar struct {
Name string
Aliases []string
Usage string
Values []string
Default string
Hidden bool
EnvVar string
Target *string
SetHook func(val string)
Completion complete.Predictor
}
func (f *Set) EnumSingleVar(i *EnumSingleVar) {
initial := i.Default
if v, exist := os.LookupEnv(i.EnvVar); exist {
initial = v
}
def := i.Default
possible := strings.Join(i.Values, ", ")
f.VarFlag(&VarFlag{
Name: i.Name,
Aliases: i.Aliases,
Usage: strings.TrimRight(i.Usage, ". \t") + ". One possible value from: " + possible + ".",
Default: def,
EnvVar: i.EnvVar,
Value: newEnumSingleValue(i, initial, i.Target, i.Hidden),
Completion: i.Completion,
})
}
type enumSingleValue struct {
ev *EnumSingleVar
hidden bool
target *string
}
func newEnumSingleValue(ev *EnumSingleVar, def string, target *string, hidden bool) *enumSingleValue {
*target = def
return &enumSingleValue{
ev: ev,
hidden: hidden,
target: target,
}
}
func (s *enumSingleValue) Set(val string) error {
for _, p := range s.ev.Values {
if p == val {
*s.target = val
if s.ev.SetHook != nil {
s.ev.SetHook(val)
}
return nil
}
}
return fmt.Errorf("'%s' not valid. Must be one of: %s", val, strings.Join(s.ev.Values, ", "))
}
func (s *enumSingleValue) Get() interface{} { return *s.target }
func (s *enumSingleValue) String() string { return *s.target }
func (s *enumSingleValue) Example() string { return "string" }
func (s *enumSingleValue) Hidden() bool { return s.hidden }
|
