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
|
diff --git a/cmd/spectre/main.go b/cmd/spectre/main.go
index da941f83a95480819dd59ed6c1539450f724b899..a39998126babfe883cf79b0f60c202af7922f1e0 100644
--- a/cmd/spectre/main.go
+++ b/cmd/spectre/main.go
@@ -10,6 +10,15 @@ "go.jolheiser.com/go-spectre"
)
func main() {
+ pw, err := doMain(os.Args[1:])
+ if err != nil {
+ fmt.Println(err)
+ return
+ }
+ fmt.Println(pw)
+}
+
+func doMain(args []string) (string, error) {
fs := flag.NewFlagSet("spectre", flag.ExitOnError)
fs.Usage = func() {
fmt.Fprintln(fs.Output(), "spectre [FLAGS] [site]")
@@ -30,11 +39,11 @@ templateFlag, err = spectre.ParseTemplate(s)
return
})
- if err := fs.Parse(os.Args[1:]); err != nil {
- panic(err)
+ if err := fs.Parse(args); err != nil {
+ return "", err
}
if err := checkEnv(fs); err != nil {
- panic(err)
+ return "", err
}
if templateFlag == "" {
@@ -42,23 +51,45 @@ templateFlag = scopeFlag.DefaultTemplate()
}
if *usernameFlag == "" || *secretFlag == "" || fs.NArg() < 1 {
- panic("username, secret, and site are required")
+ return "", requiredArgs{
+ missingUsername: *usernameFlag == "",
+ missingSecret: *secretFlag == "",
+ missingSite: fs.NArg() < 1,
+ }
}
s, err := spectre.New(*usernameFlag, *secretFlag, spectre.WithScoper(spectre.SimpleScoper{
Key: *scoperFlag,
}))
if err != nil {
- panic(err)
+ return "", err
}
- pw := s.Site(fs.Arg(0),
+ return s.Site(fs.Arg(0),
spectre.WithScope(scopeFlag),
spectre.WithTemplate(templateFlag),
spectre.WithCounter(*counterFlag),
- )
+ ), nil
+}
- fmt.Println(pw)
+type requiredArgs struct {
+ missingUsername bool
+ missingSecret bool
+ missingSite bool
+}
+
+func (r requiredArgs) Error() string {
+ s := "--username, --secret, and <site-name> are required, missing: "
+ if r.missingUsername {
+ s += "\n- username"
+ }
+ if r.missingSecret {
+ s += "\n- secret"
+ }
+ if r.missingSite {
+ s += "\n- site name"
+ }
+ return s
}
func checkEnv(fs *flag.FlagSet) error {
|