blob: dcbc2b758c3e54aa0c87467680ceed3896b7f429 [file] [log] [blame]
Zack Williamse940c7a2019-08-21 14:25:39 -07001package pflag
2
3import "strconv"
4
5// -- uint Value
6type uintValue uint
7
8func newUintValue(val uint, p *uint) *uintValue {
9 *p = val
10 return (*uintValue)(p)
11}
12
13func (i *uintValue) Set(s string) error {
14 v, err := strconv.ParseUint(s, 0, 64)
15 *i = uintValue(v)
16 return err
17}
18
19func (i *uintValue) Type() string {
20 return "uint"
21}
22
23func (i *uintValue) String() string { return strconv.FormatUint(uint64(*i), 10) }
24
25func uintConv(sval string) (interface{}, error) {
26 v, err := strconv.ParseUint(sval, 0, 0)
27 if err != nil {
28 return 0, err
29 }
30 return uint(v), nil
31}
32
33// GetUint return the uint value of a flag with the given name
34func (f *FlagSet) GetUint(name string) (uint, error) {
35 val, err := f.getFlagType(name, "uint", uintConv)
36 if err != nil {
37 return 0, err
38 }
39 return val.(uint), nil
40}
41
42// UintVar defines a uint flag with specified name, default value, and usage string.
43// The argument p points to a uint variable in which to store the value of the flag.
44func (f *FlagSet) UintVar(p *uint, name string, value uint, usage string) {
45 f.VarP(newUintValue(value, p), name, "", usage)
46}
47
48// UintVarP is like UintVar, but accepts a shorthand letter that can be used after a single dash.
49func (f *FlagSet) UintVarP(p *uint, name, shorthand string, value uint, usage string) {
50 f.VarP(newUintValue(value, p), name, shorthand, usage)
51}
52
53// UintVar defines a uint flag with specified name, default value, and usage string.
54// The argument p points to a uint variable in which to store the value of the flag.
55func UintVar(p *uint, name string, value uint, usage string) {
56 CommandLine.VarP(newUintValue(value, p), name, "", usage)
57}
58
59// UintVarP is like UintVar, but accepts a shorthand letter that can be used after a single dash.
60func UintVarP(p *uint, name, shorthand string, value uint, usage string) {
61 CommandLine.VarP(newUintValue(value, p), name, shorthand, usage)
62}
63
64// Uint defines a uint flag with specified name, default value, and usage string.
65// The return value is the address of a uint variable that stores the value of the flag.
66func (f *FlagSet) Uint(name string, value uint, usage string) *uint {
67 p := new(uint)
68 f.UintVarP(p, name, "", value, usage)
69 return p
70}
71
72// UintP is like Uint, but accepts a shorthand letter that can be used after a single dash.
73func (f *FlagSet) UintP(name, shorthand string, value uint, usage string) *uint {
74 p := new(uint)
75 f.UintVarP(p, name, shorthand, value, usage)
76 return p
77}
78
79// Uint defines a uint flag with specified name, default value, and usage string.
80// The return value is the address of a uint variable that stores the value of the flag.
81func Uint(name string, value uint, usage string) *uint {
82 return CommandLine.UintP(name, "", value, usage)
83}
84
85// UintP is like Uint, but accepts a shorthand letter that can be used after a single dash.
86func UintP(name, shorthand string, value uint, usage string) *uint {
87 return CommandLine.UintP(name, shorthand, value, usage)
88}