1package 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}
89