forked from github/multus-cni
ac21a96804
gomodule is still in progress to migrate for now, hence multus team decide to keep vendor directory to support build without gomodule.
95 lines
3.0 KiB
Go
95 lines
3.0 KiB
Go
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)
|
|
}
|