/
whitelist.go
67 lines (55 loc) · 1.67 KB
/
whitelist.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
// Package whitelist provides methods for managing an app's whitelisted IP's.
package whitelist
import (
"encoding/json"
"fmt"
deis "github.com/deis/controller-sdk-go"
"github.com/deis/controller-sdk-go/api"
)
// List IP's whitelisted for an app.
func List(c *deis.Client, appID string) (api.Whitelist, error) {
u := fmt.Sprintf("/v2/apps/%s/whitelist/", appID)
res, reqErr := c.Request("GET", u, nil)
if reqErr != nil && !deis.IsErrAPIMismatch(reqErr) {
return api.Whitelist{}, reqErr
}
defer res.Body.Close()
whitelist := api.Whitelist{}
if err := json.NewDecoder(res.Body).Decode(&whitelist); err != nil {
return api.Whitelist{}, err
}
return whitelist, reqErr
}
// Add adds addresses to an app's whitelist.
func Add(c *deis.Client, appID string, addresses []string) (api.Whitelist, error) {
u := fmt.Sprintf("/v2/apps/%s/whitelist/", appID)
req := api.Whitelist{Addresses: addresses}
body, err := json.Marshal(req)
if err != nil {
return api.Whitelist{}, err
}
res, reqErr := c.Request("POST", u, body)
if reqErr != nil && !deis.IsErrAPIMismatch(reqErr) {
return api.Whitelist{}, reqErr
}
defer res.Body.Close()
d := api.Whitelist{}
if err = json.NewDecoder(res.Body).Decode(&d); err != nil {
return api.Whitelist{}, err
}
return d, reqErr
}
// Delete removes addresses from an app's whitelist.
func Delete(c *deis.Client, appID string, addresses []string) error {
u := fmt.Sprintf("/v2/apps/%s/whitelist/", appID)
req := api.Whitelist{Addresses: addresses}
body, err := json.Marshal(req)
if err != nil {
return err
}
_, reqErr := c.Request("DELETE", u, body)
if reqErr != nil && !deis.IsErrAPIMismatch(reqErr) {
return reqErr
}
return nil
}