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
|
package main
import (
"context"
"encoding/json"
"errors"
"fmt"
"net/http"
"os"
"inet.af/netaddr"
)
type device struct {
Hostname string `json:"hostname"`
ID string `json:"id"`
Addresses []string `json:"addresses"`
}
const (
TS_NAME = "franck.cuny@gmail.com"
TS_API_DOMAIN = "api.tailscale.com"
)
func getTsDevice(ctx context.Context, deviceName string) (*device, error) {
apiKey, found := os.LookupEnv("TS_API_KEY")
if !found {
return nil, errors.New("the environment variable TS_API_KEY is not set")
}
url := fmt.Sprintf("https://%s/api/v2/tailnet/%s/devices", TS_API_DOMAIN, TS_NAME)
req, err := http.NewRequestWithContext(ctx, "GET", url, nil)
if err != nil {
return nil, err
}
req.SetBasicAuth(apiKey, "")
resp, err := http.DefaultClient.Do(req)
if err != nil {
return nil, err
}
defer resp.Body.Close()
if resp.StatusCode != http.StatusOK {
return nil, fmt.Errorf("non-ok status code %d returned from tailscale api: %s", resp.StatusCode, resp.Status)
}
var buf struct {
Devices []device `json:"devices"`
}
if err := json.NewDecoder(resp.Body).Decode(&buf); err != nil {
return nil, err
}
for _, d := range buf.Devices {
if d.Hostname == deviceName {
return &d, nil
}
}
return nil, fmt.Errorf("could not find the tailscale device named %s", deviceName)
}
// Get the Tailscale IPv4 and IPv6 addresses associated with the given device.
func getTsIpsDevice(ctx context.Context, device string) ([]string, []string, error) {
ts_device, err := getTsDevice(ctx, device)
if err != nil {
return nil, nil, fmt.Errorf("failed to get Tailscale device information: %v", err)
}
var (
tsIpV4Addresses = []string{}
tsIpV6Addresses = []string{}
)
for _, ipString := range ts_device.Addresses {
// we convert the string to a netaddr.IP so we can check if
// it's an IP v4 or v6. We need to know what's the version in
// order to use it properly when creating/updating the
// record. Then we convert it back as a string, since this is
// what the DNS API expect.
ip := netaddr.MustParseIP(ipString)
if ip.Is4() {
tsIpV4Addresses = append(tsIpV4Addresses, ip.String())
} else {
tsIpV6Addresses = append(tsIpV6Addresses, ip.String())
}
}
return tsIpV4Addresses, tsIpV6Addresses, nil
}
|
