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
|
// Package digitalocean provides node discovery for DigitalOcean.
package digitalocean
import (
"context"
"fmt"
"io/ioutil"
"log"
"github.com/digitalocean/godo"
"golang.org/x/oauth2"
)
type Provider struct {
userAgent string
}
func (p *Provider) SetUserAgent(s string) {
p.userAgent = s
}
func (p *Provider) Help() string {
return `DigitalOcean:
provider: "digitalocean"
region: The DigitalOcean region to filter on
tag_name: The tag name to filter on
api_token: The DigitalOcean API token to use
`
}
type TokenSource struct {
AccessToken string
}
func (t *TokenSource) Token() (*oauth2.Token, error) {
token := &oauth2.Token{
AccessToken: t.AccessToken,
}
return token, nil
}
func listDropletsByTag(c *godo.Client, tagName string) ([]godo.Droplet, error) {
dropletList := []godo.Droplet{}
pageOpt := &godo.ListOptions{
Page: 1,
PerPage: 200,
}
for {
droplets, resp, err := c.Droplets.ListByTag(context.TODO(), tagName, pageOpt)
if err != nil {
return nil, err
}
for _, d := range droplets {
dropletList = append(dropletList, d)
}
if resp.Links == nil || resp.Links.IsLastPage() {
break
}
page, err := resp.Links.CurrentPage()
if err != nil {
return nil, err
}
pageOpt.Page = page + 1
}
return dropletList, nil
}
func (p *Provider) Addrs(args map[string]string, l *log.Logger) ([]string, error) {
if args["provider"] != "digitalocean" {
return nil, fmt.Errorf("discover-digitalocean: invalid provider " + args["provider"])
}
if l == nil {
l = log.New(ioutil.Discard, "", 0)
}
region := args["region"]
tagName := args["tag_name"]
apiToken := args["api_token"]
l.Printf("[DEBUG] discover-digitalocean: Using region=%s tag_name=%s", region, tagName)
tokenSource := &TokenSource{
AccessToken: apiToken,
}
oauthClient := oauth2.NewClient(context.TODO(), tokenSource)
client := godo.NewClient(oauthClient)
if p.userAgent != "" {
client.UserAgent = p.userAgent
}
droplets, err := listDropletsByTag(client, tagName)
if err != nil {
return nil, fmt.Errorf("discover-digitalocean: %s", err)
}
var addrs []string
for _, d := range droplets {
if d.Region.Slug == region || region == "" {
privateIP, err := d.PrivateIPv4()
if err != nil {
return nil, fmt.Errorf("discover-digitalocean: %s", err)
}
if privateIP != "" {
l.Printf("[INFO] discover-digitalocean: Found instance %s (%d) with private IP: %s", d.Name, d.ID, privateIP)
addrs = append(addrs, privateIP)
}
}
}
l.Printf("[DEBUG] discover-digitalocean: Found ip addresses: %v", addrs)
return addrs, nil
}
|
