1
package pflag
2

3
import (
4
	"fmt"
5
	"net"
6
	"strings"
7
)
8

9
// IPNet adapts net.IPNet for use as a flag.
10
type ipNetValue net.IPNet
11

12
func (ipnet ipNetValue) String() string {
13
	n := net.IPNet(ipnet)
14
	return n.String()
15
}
16

17
func (ipnet *ipNetValue) Set(value string) error {
18
	_, n, err := net.ParseCIDR(strings.TrimSpace(value))
19
	if err != nil {
20
		return err
21
	}
22
	*ipnet = ipNetValue(*n)
23
	return nil
24
}
25

26
func (*ipNetValue) Type() string {
27
	return "ipNet"
28
}
29

30
func newIPNetValue(val net.IPNet, p *net.IPNet) *ipNetValue {
31
	*p = val
32
	return (*ipNetValue)(p)
33
}
34

35
func ipNetConv(sval string) (interface{}, error) {
36
	_, n, err := net.ParseCIDR(strings.TrimSpace(sval))
37
	if err == nil {
38
		return *n, nil
39
	}
40
	return nil, fmt.Errorf("invalid string being converted to IPNet: %s", sval)
41
}
42

43
// GetIPNet return the net.IPNet value of a flag with the given name
44
func (f *FlagSet) GetIPNet(name string) (net.IPNet, error) {
45
	val, err := f.getFlagType(name, "ipNet", ipNetConv)
46
	if err != nil {
47
		return net.IPNet{}, err
48
	}
49
	return val.(net.IPNet), nil
50
}
51

52
// IPNetVar defines an net.IPNet flag with specified name, default value, and usage string.
53
// The argument p points to an net.IPNet variable in which to store the value of the flag.
54
func (f *FlagSet) IPNetVar(p *net.IPNet, name string, value net.IPNet, usage string) {
55
	f.VarP(newIPNetValue(value, p), name, "", usage)
56
}
57

58
// IPNetVarP is like IPNetVar, but accepts a shorthand letter that can be used after a single dash.
59
func (f *FlagSet) IPNetVarP(p *net.IPNet, name, shorthand string, value net.IPNet, usage string) {
60
	f.VarP(newIPNetValue(value, p), name, shorthand, usage)
61
}
62

63
// IPNetVar defines an net.IPNet flag with specified name, default value, and usage string.
64
// The argument p points to an net.IPNet variable in which to store the value of the flag.
65
func IPNetVar(p *net.IPNet, name string, value net.IPNet, usage string) {
66
	CommandLine.VarP(newIPNetValue(value, p), name, "", usage)
67
}
68

69
// IPNetVarP is like IPNetVar, but accepts a shorthand letter that can be used after a single dash.
70
func IPNetVarP(p *net.IPNet, name, shorthand string, value net.IPNet, usage string) {
71
	CommandLine.VarP(newIPNetValue(value, p), name, shorthand, usage)
72
}
73

74
// IPNet defines an net.IPNet flag with specified name, default value, and usage string.
75
// The return value is the address of an net.IPNet variable that stores the value of the flag.
76
func (f *FlagSet) IPNet(name string, value net.IPNet, usage string) *net.IPNet {
77
	p := new(net.IPNet)
78
	f.IPNetVarP(p, name, "", value, usage)
79
	return p
80
}
81

82
// IPNetP is like IPNet, but accepts a shorthand letter that can be used after a single dash.
83
func (f *FlagSet) IPNetP(name, shorthand string, value net.IPNet, usage string) *net.IPNet {
84
	p := new(net.IPNet)
85
	f.IPNetVarP(p, name, shorthand, value, usage)
86
	return p
87
}
88

89
// IPNet defines an net.IPNet flag with specified name, default value, and usage string.
90
// The return value is the address of an net.IPNet variable that stores the value of the flag.
91
func IPNet(name string, value net.IPNet, usage string) *net.IPNet {
92
	return CommandLine.IPNetP(name, "", value, usage)
93
}
94

95
// IPNetP is like IPNet, but accepts a shorthand letter that can be used after a single dash.
96
func IPNetP(name, shorthand string, value net.IPNet, usage string) *net.IPNet {
97
	return CommandLine.IPNetP(name, shorthand, value, usage)
98
}
99

100