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
91
92
93
94
95
96
97
|
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
package cmd
import (
"fmt"
"slices"
"strconv"
"strings"
)
var arrayFields = []string{
"urls",
"follow",
"allowedDomains",
"blockedDomains",
"allowedURLs",
"blockedURLs",
"proxies",
}
func parseConfigArgs(args []string) (map[string]any, error) {
updates := map[string]any{}
flag := ""
for _, arg := range normalizeArgs(args) {
if flag == "" && !isFlag(arg) {
return nil, fmt.Errorf("expected flag, got %q instead", arg)
}
if flag != "" && isFlag(arg) {
updates[flag[2:]] = true
flag = ""
continue
}
if flag != "" {
if v, ok := updates[flag[2:]]; ok {
if vv, ok := v.([]any); ok {
updates[flag[2:]] = append(vv, parseArg(arg))
} else {
updates[flag[2:]] = []any{v, parseArg(arg)}
}
} else {
if slices.Contains(arrayFields, flag[2:]) {
updates[flag[2:]] = []any{parseArg(arg)}
} else {
updates[flag[2:]] = parseArg(arg)
}
}
flag = ""
continue
}
flag = arg
}
if flag != "" {
updates[flag[2:]] = true
flag = ""
}
return updates, nil
}
func normalizeArgs(args []string) []string {
var norm []string
for _, arg := range args {
if !strings.HasPrefix(arg, "--") {
norm = append(norm, arg)
} else {
norm = append(norm, strings.SplitN(arg, "=", 2)...)
}
}
return norm
}
func parseArg(arg string) any {
if arg == "true" {
return true
}
if arg == "false" {
return false
}
if num, err := strconv.Atoi(arg); err == nil {
return num
}
return arg
}
func isFlag(arg string) bool {
return strings.HasPrefix(arg, "--")
}
|