-
Notifications
You must be signed in to change notification settings - Fork 0
/
provider_client.go
115 lines (95 loc) · 3.99 KB
/
provider_client.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
package circleci
import (
circleciapi "github.com/andrewstucki/terraform-provider-circleci/circleci/client"
"github.com/cenkalti/backoff"
)
// ProviderClient is a thin commodity wrapper on top of circleciapi
type ProviderClient struct {
client *circleciapi.Client
vcsType string
organization string
}
// NewConfig initialize circleci API client and returns a new config object
func NewConfig(token, vscType, organization string) *ProviderClient {
return &ProviderClient{
client: &circleciapi.Client{
Token: token,
},
vcsType: vscType,
organization: organization,
}
}
// GetEnvVar get the environment variable with given name
// It returns an empty structure if no environment variable exists with that name
func (pv *ProviderClient) GetEnvVar(projectName, envVarName string) (*circleciapi.EnvVar, error) {
return pv.client.GetEnvVar(pv.vcsType, pv.organization, projectName, envVarName)
}
// EnvVarExists check if environment variable exists with given name
func (pv *ProviderClient) EnvVarExists(projectName, envVarName string) (bool, error) {
envVar, err := pv.client.GetEnvVar(pv.vcsType, pv.organization, projectName, envVarName)
if err != nil {
return false, err
}
return bool(envVar.Name != ""), nil
}
// AddEnvVar create an environment variable with given name and value
func (pv *ProviderClient) AddEnvVar(projectName, envVarName, envVarValue string) (*circleciapi.EnvVar, error) {
return pv.client.AddEnvVar(pv.vcsType, pv.organization, projectName, envVarName, envVarValue)
}
// DeleteEnvVar delete the environment variable with given name
func (pv *ProviderClient) DeleteEnvVar(projectName, envVarName string) error {
retry := backoff.WithMaxRetries(backoff.NewExponentialBackOff(), 3)
return backoff.Retry(func() error {
return pv.client.DeleteEnvVar(pv.vcsType, pv.organization, projectName, envVarName)
}, retry)
}
// GetProject reads the project with given name
func (pv *ProviderClient) GetProject(projectName string) (*circleciapi.Project, error) {
retry := backoff.WithMaxRetries(backoff.NewExponentialBackOff(), 3)
var err error
var project *circleciapi.Project
err = backoff.Retry(func() error {
project, err = pv.client.GetProject(pv.organization, projectName)
return err
}, retry)
return project, err
}
// EnableProject enables the project with given name
func (pv *ProviderClient) EnableProject(projectName string) error {
retry := backoff.WithMaxRetries(backoff.NewExponentialBackOff(), 3)
return backoff.Retry(func() error {
return pv.client.EnableProject(pv.vcsType, pv.organization, projectName)
}, retry)
}
// FollowProject follows the project with given name
func (pv *ProviderClient) FollowProject(projectName string) (*circleciapi.Project, error) {
retry := backoff.WithMaxRetries(backoff.NewExponentialBackOff(), 3)
var err error
var project *circleciapi.Project
err = backoff.Retry(func() error {
project, err = pv.client.FollowProject(pv.vcsType, pv.organization, projectName)
return err
}, retry)
return project, err
}
// DisableProject disables the project with given name
func (pv *ProviderClient) DisableProject(projectName string) error {
retry := backoff.WithMaxRetries(backoff.NewExponentialBackOff(), 3)
return backoff.Retry(func() error {
return pv.client.DisableProject(pv.vcsType, pv.organization, projectName)
}, retry)
}
// AddSSHKey adds an ssh private key to the project
func (pv *ProviderClient) AddSSHKey(projectName, hostname, privateKey string) error {
retry := backoff.WithMaxRetries(backoff.NewExponentialBackOff(), 3)
return backoff.Retry(func() error {
return pv.client.AddSSHKey(pv.vcsType, pv.organization, projectName, hostname, privateKey)
}, retry)
}
// DeleteSSHKey deletes an ssh private key from the project
func (pv *ProviderClient) DeleteSSHKey(projectName, hostname, fingerprint string) error {
retry := backoff.WithMaxRetries(backoff.NewExponentialBackOff(), 3)
return backoff.Retry(func() error {
return pv.client.DeleteSSHKey(pv.vcsType, pv.organization, projectName, hostname, fingerprint)
}, retry)
}