forked from ctrysbita/smartctl_exporter
-
Notifications
You must be signed in to change notification settings - Fork 0
/
flags.go
50 lines (38 loc) · 1.02 KB
/
flags.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
package main
import (
"flag"
"log"
"os"
"strings"
)
type Flags struct {
Address *string
Port *int
Path *string
user *string
pass *string
disableAuth *bool
tls *bool
Version *bool
}
func (f *Flags) init() {
f.Address = flag.String("server-addr", "", "Address.")
f.Port = flag.Int("server-port", 9111, "Port.")
f.Path = flag.String("server-path", "/metrics", "Path.")
f.user = flag.String("server-user", "admin", "Username.")
f.pass = flag.String("server-pass", "admin", "Password.")
f.disableAuth = flag.Bool("disable-auth", false, "Disable basic authentication.")
f.tls = flag.Bool("server-tls", true, "Enable TLS.")
f.Version = flag.Bool("version", false, "Show version.")
flag.Parse()
// System Environment Variables
flag.VisitAll(func(f *flag.Flag) {
name := strings.ToUpper(strings.Replace(f.Name, "-", "_", -1))
if value, ok := os.LookupEnv(name); ok {
err := flag.Set(f.Name, value)
if err != nil {
log.Panicln("failed setting flag from environment:", err)
}
}
})
}