| package pflag |
| |
| import ( |
| "fmt" |
| "net" |
| "strings" |
| ) |
| |
| // -- net.IP value |
| type ipValue net.IP |
| |
| func newIPValue(val net.IP, p *net.IP) *ipValue { |
| *p = val |
| return (*ipValue)(p) |
| } |
| |
| func (i *ipValue) String() string { return net.IP(*i).String() } |
| func (i *ipValue) Set(s string) error { |
| ip := net.ParseIP(strings.TrimSpace(s)) |
| if ip == nil { |
| return fmt.Errorf("failed to parse IP: %q", s) |
| } |
| *i = ipValue(ip) |
| return nil |
| } |
| |
| func (i *ipValue) Type() string { |
| return "ip" |
| } |
| |
| func ipConv(sval string) (interface{}, error) { |
| ip := net.ParseIP(sval) |
| if ip != nil { |
| return ip, nil |
| } |
| return nil, fmt.Errorf("invalid string being converted to IP address: %s", sval) |
| } |
| |
| // GetIP return the net.IP value of a flag with the given name |
| func (f *FlagSet) GetIP(name string) (net.IP, error) { |
| val, err := f.getFlagType(name, "ip", ipConv) |
| if err != nil { |
| return nil, err |
| } |
| return val.(net.IP), nil |
| } |
| |
| // IPVar defines an net.IP flag with specified name, default value, and usage string. |
| // The argument p points to an net.IP variable in which to store the value of the flag. |
| func (f *FlagSet) IPVar(p *net.IP, name string, value net.IP, usage string) { |
| f.VarP(newIPValue(value, p), name, "", usage) |
| } |
| |
| // IPVarP is like IPVar, but accepts a shorthand letter that can be used after a single dash. |
| func (f *FlagSet) IPVarP(p *net.IP, name, shorthand string, value net.IP, usage string) { |
| f.VarP(newIPValue(value, p), name, shorthand, usage) |
| } |
| |
| // IPVar defines an net.IP flag with specified name, default value, and usage string. |
| // The argument p points to an net.IP variable in which to store the value of the flag. |
| func IPVar(p *net.IP, name string, value net.IP, usage string) { |
| CommandLine.VarP(newIPValue(value, p), name, "", usage) |
| } |
| |
| // IPVarP is like IPVar, but accepts a shorthand letter that can be used after a single dash. |
| func IPVarP(p *net.IP, name, shorthand string, value net.IP, usage string) { |
| CommandLine.VarP(newIPValue(value, p), name, shorthand, usage) |
| } |
| |
| // IP defines an net.IP flag with specified name, default value, and usage string. |
| // The return value is the address of an net.IP variable that stores the value of the flag. |
| func (f *FlagSet) IP(name string, value net.IP, usage string) *net.IP { |
| p := new(net.IP) |
| f.IPVarP(p, name, "", value, usage) |
| return p |
| } |
| |
| // IPP is like IP, but accepts a shorthand letter that can be used after a single dash. |
| func (f *FlagSet) IPP(name, shorthand string, value net.IP, usage string) *net.IP { |
| p := new(net.IP) |
| f.IPVarP(p, name, shorthand, value, usage) |
| return p |
| } |
| |
| // IP defines an net.IP flag with specified name, default value, and usage string. |
| // The return value is the address of an net.IP variable that stores the value of the flag. |
| func IP(name string, value net.IP, usage string) *net.IP { |
| return CommandLine.IPP(name, "", value, usage) |
| } |
| |
| // IPP is like IP, but accepts a shorthand letter that can be used after a single dash. |
| func IPP(name, shorthand string, value net.IP, usage string) *net.IP { |
| return CommandLine.IPP(name, shorthand, value, usage) |
| } |