forked from facebookarchive/flag.addrs
-
Notifications
You must be signed in to change notification settings - Fork 1
/
addrs.go
125 lines (112 loc) · 2.9 KB
/
addrs.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
// Package addrs provides flags to define one or an array of net.Addr.
package addrs
import (
"bytes"
"flag"
"fmt"
"net"
"strings"
)
type flagOne struct {
addr *net.Addr
}
func (f *flagOne) Set(addr string) error {
a, err := resolveAddr(addr)
if err != nil {
return err
}
*f.addr = a
return nil
}
func (f *flagOne) String() string {
if f.addr == nil || (*f.addr) == nil {
return ""
}
return (*f.addr).Network() + ":" + (*f.addr).String()
}
// FlagOneVar sets a single net.Addr by a flag. The value is expected to be a
// colon separated net:host:port.
func FlagOneVar(dest *net.Addr, name string, addr string, usage string) {
FlagSetOneVar(flag.CommandLine, dest, name, addr, usage)
}
// FlagSetOneVar sets a single net.Addr by a flag in a flag.FlagSet. The value
// is expected to be a colon separated net:host:port.
func FlagSetOneVar(set *flag.FlagSet, dest *net.Addr, name string, addr string, usage string) {
f := &flagOne{addr: dest}
if addr != "" {
if err := f.Set(addr); err != nil {
panic(err)
}
}
set.Var(f, name, usage)
}
type flagMany struct {
addrs *[]net.Addr
}
func (f *flagMany) Set(val string) error {
var resolved []net.Addr
for _, addr := range strings.Split(val, ",") {
a, err := resolveAddr(addr)
if err != nil {
return err
}
resolved = append(resolved, a)
}
*f.addrs = resolved
return nil
}
func (f *flagMany) String() string {
if f.addrs == nil || (*f.addrs) == nil {
return ""
}
var buf bytes.Buffer
first := true
for _, a := range *f.addrs {
if !first {
buf.WriteString(",")
}
first = false
buf.WriteString(a.Network())
buf.WriteString(":")
buf.WriteString(a.String())
}
return buf.String()
}
// FlagManyVar sets a slice of net.Addr by a flag. The values are expected to
// be comma separated list of net:host:port triples.
func FlagManyVar(dest *[]net.Addr, name string, addrs string, usage string) {
FlagSetManyVar(flag.CommandLine, dest, name, addrs, usage)
}
// FlagSetManyVar sets a slice of net.Addr by a flag in a flag.FlagSet. The
// values are expected to be comma separated list of net:host:port triples.
func FlagSetManyVar(set *flag.FlagSet, dest *[]net.Addr, name string, addrs string, usage string) {
f := &flagMany{addrs: dest}
if addrs != "" {
if err := f.Set(addrs); err != nil {
panic(err)
}
}
set.Var(f, name, usage)
}
func resolveAddr(addr string) (net.Addr, error) {
parts := strings.Split(addr, ":")
if len(parts) != 3 {
return nil, fmt.Errorf(
`invalid address format, must be "net:host:port": %s`,
addr,
)
}
hp := parts[1] + ":" + parts[2]
switch parts[0] {
default:
return nil, net.UnknownNetworkError(parts[0])
case "ip", "ip4", "ip6":
return net.ResolveIPAddr(parts[0], hp)
case "tcp", "tcp4", "tcp6":
return net.ResolveTCPAddr(parts[0], hp)
case "udp", "udp4", "udp6":
return net.ResolveUDPAddr(parts[0], hp)
case "unix", "unixgram", "unixpacket":
return net.ResolveUnixAddr(parts[0], hp)
}
}