vendor/github.com/spf13/pflag/ipnet.go

   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 }