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
81
82
83
84
85
86
87
88
89
90
|
package flag
import (
"fmt"
"os"
"strings"
"github.com/posener/complete"
)
// -- EnumVar and enumValue
type EnumVar struct {
Name string
Aliases []string
Usage string
Values []string
Default []string
Hidden bool
EnvVar string
Target *[]string
Completion complete.Predictor
}
func (f *Set) EnumVar(i *EnumVar) {
initial := i.Default
if v, exist := os.LookupEnv(i.EnvVar); exist {
parts := strings.Split(v, ",")
for i := range parts {
parts[i] = strings.TrimSpace(parts[i])
}
initial = parts
}
def := ""
if i.Default != nil {
def = strings.Join(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: newEnumValue(i, initial, i.Target, i.Hidden),
Completion: i.Completion,
})
}
type enumValue struct {
ev *EnumVar
hidden bool
target *[]string
}
func newEnumValue(ev *EnumVar, def []string, target *[]string, hidden bool) *enumValue {
*target = def
return &enumValue{
ev: ev,
hidden: hidden,
target: target,
}
}
func (s *enumValue) Set(vals string) error {
parts := strings.Split(vals, ",")
parts:
for _, val := range parts {
val = strings.TrimSpace(val)
for _, p := range s.ev.Values {
if p == val {
*s.target = append(*s.target, strings.TrimSpace(val))
continue parts
}
}
return fmt.Errorf("'%s' not valid. Must be one of: %s", val, strings.Join(s.ev.Values, ", "))
}
return nil
}
func (s *enumValue) Get() interface{} { return *s.target }
func (s *enumValue) String() string { return strings.Join(*s.target, ",") }
func (s *enumValue) Example() string { return "string" }
func (s *enumValue) Hidden() bool { return s.hidden }
|
