summaryrefslogtreecommitdiff
path: root/vendor/github.com/spf13/pflag/bool_func.go
blob: 83d77afa89fa9dbb42e406168e6d56e6237452da (plain)
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
package pflag

// -- func Value
type boolfuncValue func(string) error

func (f boolfuncValue) Set(s string) error { return f(s) }

func (f boolfuncValue) Type() string { return "boolfunc" }

func (f boolfuncValue) String() string { return "" } // same behavior as stdlib 'flag' package

func (f boolfuncValue) IsBoolFlag() bool { return true }

// BoolFunc defines a func flag with specified name, callback function and usage string.
//
// The callback function will be called every time "--{name}" (or any form that matches the flag) is parsed
// on the command line.
func (f *FlagSet) BoolFunc(name string, usage string, fn func(string) error) {
	f.BoolFuncP(name, "", usage, fn)
}

// BoolFuncP is like BoolFunc, but accepts a shorthand letter that can be used after a single dash.
func (f *FlagSet) BoolFuncP(name, shorthand string, usage string, fn func(string) error) {
	var val Value = boolfuncValue(fn)
	flag := f.VarPF(val, name, shorthand, usage)
	flag.NoOptDefVal = "true"
}

// BoolFunc defines a func flag with specified name, callback function and usage string.
//
// The callback function will be called every time "--{name}" (or any form that matches the flag) is parsed
// on the command line.
func BoolFunc(name string, usage string, fn func(string) error) {
	CommandLine.BoolFuncP(name, "", usage, fn)
}

// BoolFuncP is like BoolFunc, but accepts a shorthand letter that can be used after a single dash.
func BoolFuncP(name, shorthand string, usage string, fn func(string) error) {
	CommandLine.BoolFuncP(name, shorthand, usage, fn)
}