/
cli.go
95 lines (83 loc) · 2.18 KB
/
cli.go
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
package upctl
import (
"errors"
"fmt"
"os"
"time"
"github.com/spf13/cobra"
"github.com/spf13/viper"
"github.com/uptime-com/uptime-client-go/v2/pkg/upapi"
)
var errNoToken = errors.New("token is required")
// cmd represents the base command when called without any subcommands
var (
api upapi.API
cmdArgs = struct {
Color bool `flag:"color" usage:"Enable color for json output"`
Output string `flag:"output" short:"o" usage:"Output format (json|spew)"`
Token string `flag:"token" usage:"Uptime.com API token"`
Trace bool `flag:"trace" usage:"Trace HTTP requests"`
}{
Color: true,
Output: "json",
}
cmd = &cobra.Command{
Use: "upctl",
Short: "Uptime.com command line API client",
Long: "", // TODO: add long description
SilenceErrors: true,
PersistentPreRunE: func(cmd *cobra.Command, _ []string) (err error) {
token := viper.GetString("token")
if token == "" {
return errNoToken
}
opts := []upapi.Option{
upapi.WithToken(viper.GetString("token")),
}
if cmdArgs.Trace {
opts = append(opts, upapi.WithTrace(os.Stderr))
}
opts = append(opts, upapi.WithRetry(10, time.Second*30, os.Stderr))
api, err = upapi.New(opts...)
return err
},
}
)
func init() {
err := Bind(cmd.PersistentFlags(), &cmdArgs)
if err != nil {
panic(err)
}
err = cmd.PersistentFlags().MarkHidden("token")
if err != nil {
return
}
err = viper.BindPFlags(cmd.PersistentFlags())
if err != nil {
panic(err)
}
err = viper.BindEnv("token", "UPCTL_TOKEN", "UPTIME_TOKEN")
if err != nil {
panic(err)
}
}
const obtainTokenMessage = `PLease obtain token from https://uptime.com/api/tokens and set it with:
export UPCTL_TOKEN=<token>
`
func Execute(version string) {
cmd.Version = version
err := cmd.Execute()
if err != nil {
var uperr = new(upapi.Error)
switch {
case errors.Is(err, errNoToken):
_, _ = fmt.Fprintf(cmd.OutOrStderr(), "\nError: %v\n\n", err)
_, _ = fmt.Fprintf(cmd.OutOrStderr(), obtainTokenMessage)
case errors.As(err, &uperr):
_, _ = fmt.Fprintf(cmd.OutOrStderr(), "\nError: %v\n\n", uperr)
default:
_, _ = fmt.Fprintf(cmd.OutOrStderr(), "\nError: %v\n\n", err)
}
os.Exit(1)
}
}