mirror of
https://git.vectorsigma.ru/public/atlantis.git
synced 2026-07-29 10:48:27 +00:00
d425de95fa
* set version in one place * rename ExecutionContext to CommandContext * delete unneeded PullRequestContext since it just duplicated what is in CommandContext * split fields inside CommandContext into Repo, PullRequest, and User models * clean up unused fields * remove base_executor stuff which was trying to do inheritance in go and didn't get us anywhere * clean up unused code and assets * created a locking.Client that creates the Keys that are used on the front-end. This allows the backends to store the locks any way they like as long as they support the interface
74 lines
2.5 KiB
Go
74 lines
2.5 KiB
Go
package server
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"github.com/google/go-github/github"
|
|
"github.com/hootsuite/atlantis/models"
|
|
)
|
|
|
|
type GithubClient struct {
|
|
client *github.Client
|
|
ctx context.Context
|
|
}
|
|
|
|
const (
|
|
statusContext = "Atlantis"
|
|
PendingStatus = "pending"
|
|
SuccessStatus = "success"
|
|
ErrorStatus = "error"
|
|
FailureStatus = "failure"
|
|
)
|
|
|
|
func (g *GithubClient) UpdateStatus(repo models.Repo, pull models.PullRequest, status string, description string) {
|
|
repoStatus := github.RepoStatus{State: github.String(status), Description: github.String(description), Context: github.String(statusContext)}
|
|
g.client.Repositories.CreateStatus(g.ctx, repo.Owner, repo.Name, pull.HeadCommit, &repoStatus)
|
|
// todo: deal with error updating status
|
|
}
|
|
|
|
// GetModifiedFiles returns the names of files that were modified in the pull request.
|
|
// The names include the path to the file from the repo root, ex. parent/child/file.txt
|
|
func (g *GithubClient) GetModifiedFiles(repo models.Repo, pull models.PullRequest) ([]string, error) {
|
|
var files []string
|
|
comparison, _, err := g.client.Repositories.CompareCommits(g.ctx, repo.Owner, repo.Name, pull.BaseCommit, pull.HeadCommit)
|
|
if err != nil {
|
|
return files, err
|
|
}
|
|
for _, file := range comparison.Files {
|
|
files = append(files, *file.Filename)
|
|
}
|
|
return files, nil
|
|
}
|
|
|
|
func (g *GithubClient) CreateComment(ctx *CommandContext, comment string) error {
|
|
_, _, err := g.client.Issues.CreateComment(g.ctx, ctx.Repo.Owner, ctx.Repo.Name, ctx.Pull.Num, &github.IssueComment{Body: &comment})
|
|
return err
|
|
}
|
|
|
|
func (g *GithubClient) PullIsApproved(repo models.Repo, pull models.PullRequest) (bool, error) {
|
|
// todo: move back to using g.client.PullRequests.ListReviews when we update our GitHub enterprise version
|
|
// to where we don't need to include the custom accept header
|
|
u := fmt.Sprintf("repos/%v/%v/pulls/%d/reviews", repo.Owner, repo.Name, pull.Num)
|
|
req, err := g.client.NewRequest("GET", u, nil)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
req.Header.Set("Accept", "application/vnd.github.black-cat-preview+json")
|
|
|
|
var reviews []*github.PullRequestReview
|
|
_, err = g.client.Do(g.ctx, req, &reviews)
|
|
if err != nil {
|
|
return false, fmt.Errorf("failed to retrieve reviews: %v", err)
|
|
}
|
|
for _, review := range reviews {
|
|
if review != nil && review.State != nil && *review.State == "APPROVED" {
|
|
return true, nil
|
|
}
|
|
}
|
|
return false, nil
|
|
}
|
|
|
|
func (g *GithubClient) GetPullRequest(repo models.Repo, num int) (*github.PullRequest, *github.Response, error) {
|
|
return g.client.PullRequests.Get(g.ctx, repo.Owner, repo.Name, num)
|
|
}
|