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
|
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
// vespa logfmt command
// Author: arnej
package logfmt
import (
"fmt"
"strings"
)
// common code for showFlags and levelFlags
type plusMinusFlag interface {
flags() map[string]bool
name() string
unchanged() bool
}
func trimPrefix(value, prefix string) (newValue string, hadPrefix bool) {
hadPrefix = strings.HasPrefix(value, prefix)
if hadPrefix {
newValue = strings.TrimPrefix(value, prefix)
} else {
newValue = value
}
return
}
func applyPlusMinus(val string, target plusMinusFlag) error {
minus := strings.HasPrefix(val, "-")
plus := strings.HasPrefix(val, "+")
val = strings.ReplaceAll(val, "-", ",-")
val = strings.ReplaceAll(val, "+", ",+")
if target.unchanged() {
// user wants to reset flags?
if minus == false && plus == false {
for k, _ := range target.flags() {
target.flags()[k] = false
}
}
}
changeTo := !minus
for _, k := range strings.Split(val, ",") {
if suppress, minus := trimPrefix(k, "-"); minus {
k = suppress
changeTo = false
}
if surface, plus := trimPrefix(k, "+"); plus {
k = surface
changeTo = true
}
if k == "" {
continue
}
if k == "all" {
for k, _ := range target.flags() {
target.flags()[k] = changeTo
}
} else if _, ok := target.flags()[k]; !ok {
return fmt.Errorf("not a valid %s flag: '%s'", target.name(), k)
} else {
target.flags()[k] = changeTo
}
}
return nil
}
|