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
|
package options
import (
"flag"
"fmt"
"time"
)
const (
DefaultString = "default string"
DefaultUint64 = 18446744073709551615
)
// New initializes a flag set using the provided arguments.
//
// - args should start with the (sub)command's name
// - usage is a function that prints a usage message
// - set is a function that sets the command's flag arguments
//
func New(args []string, usage func(), set func(*flag.FlagSet)) *flag.FlagSet {
if len(args) == 0 {
args = append(args, "")
}
fs := flag.NewFlagSet(args[0], flag.ExitOnError)
fs.Usage = func() {
usage()
}
set(fs)
fs.Parse(args[1:])
return fs
}
// AddString adds a string option to a flag set
func AddString(fs *flag.FlagSet, opt *string, short, long, value string) {
fs.StringVar(opt, short, value, "")
fs.StringVar(opt, long, value, "")
}
// AddUint64 adds an uint64 option to a flag set
func AddUint64(fs *flag.FlagSet, opt *uint64, short, long string, value uint64) {
fs.Uint64Var(opt, short, value, "")
fs.Uint64Var(opt, long, value, "")
}
// AddDuration adds a duration option to a flag set
func AddDuration(fs *flag.FlagSet, opt *time.Duration, short, long string, value time.Duration) {
fs.DurationVar(opt, short, value, "")
fs.DurationVar(opt, long, value, "")
}
// CheckString checks that a string option has a non-default value
func CheckString(optionName, value string, err error) error {
if err != nil {
return err
}
if value == DefaultString {
return fmt.Errorf("%s is a required option", optionName)
}
return nil
}
// CheckUint64 checks that an uint64 option has a non-default value
func CheckUint64(optionName string, value uint64, err error) error {
if err != nil {
return err
}
if value == DefaultUint64 {
return fmt.Errorf("%s is a required option", optionName)
}
return nil
}
|
