forked from runatlantis/atlantis
-
Notifications
You must be signed in to change notification settings - Fork 0
/
proxy.go
73 lines (63 loc) · 2.73 KB
/
proxy.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
// Copyright 2017 HootSuite Media Inc.
//
// Licensed under the Apache License, Version 2.0 (the License);
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
// http://www.apache.org/licenses/LICENSE-2.0
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an AS IS BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
// Modified hereafter by contributors to runatlantis/atlantis.
package vcs
import (
"github.com/runatlantis/atlantis/server/events/models"
)
// ClientProxy proxies calls to the correct VCS client depending on which
// VCS host is required.
type ClientProxy struct {
// clients maps from the vcs host type to the client that implements the
// api for that host type, ex. github -> github client.
clients map[models.VCSHostType]Client
}
func NewClientProxy(githubClient Client, gitlabClient Client, bitbucketCloudClient Client, bitbucketServerClient Client) *ClientProxy {
if githubClient == nil {
githubClient = &NotConfiguredVCSClient{}
}
if gitlabClient == nil {
gitlabClient = &NotConfiguredVCSClient{}
}
if bitbucketCloudClient == nil {
bitbucketCloudClient = &NotConfiguredVCSClient{}
}
if bitbucketServerClient == nil {
bitbucketServerClient = &NotConfiguredVCSClient{}
}
return &ClientProxy{
clients: map[models.VCSHostType]Client{
models.Github: githubClient,
models.Gitlab: gitlabClient,
models.BitbucketCloud: bitbucketCloudClient,
models.BitbucketServer: bitbucketServerClient,
},
}
}
func (d *ClientProxy) GetModifiedFiles(repo models.Repo, pull models.PullRequest) ([]string, error) {
return d.clients[repo.VCSHost.Type].GetModifiedFiles(repo, pull)
}
func (d *ClientProxy) CreateComment(repo models.Repo, pullNum int, comment string) error {
return d.clients[repo.VCSHost.Type].CreateComment(repo, pullNum, comment)
}
func (d *ClientProxy) PullIsApproved(repo models.Repo, pull models.PullRequest) (bool, error) {
return d.clients[repo.VCSHost.Type].PullIsApproved(repo, pull)
}
func (d *ClientProxy) PullIsMergeable(repo models.Repo, pull models.PullRequest) (bool, error) {
return d.clients[repo.VCSHost.Type].PullIsMergeable(repo, pull)
}
func (d *ClientProxy) UpdateStatus(repo models.Repo, pull models.PullRequest, state models.CommitStatus, src string, description string, url string) error {
return d.clients[repo.VCSHost.Type].UpdateStatus(repo, pull, state, src, description, url)
}
func (d *ClientProxy) MergePull(pull models.PullRequest) error {
return d.clients[pull.BaseRepo.VCSHost.Type].MergePull(pull)
}