You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
197 lines
5.5 KiB
Go
197 lines
5.5 KiB
Go
// Copyright 2017 Drone.IO Inc. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package github
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
"git.awesome-for.me/liuzhiguo/go-scm/scm"
|
|
"git.awesome-for.me/liuzhiguo/go-scm/scm/driver/internal/null"
|
|
)
|
|
|
|
type pullService struct {
|
|
*issueService
|
|
}
|
|
|
|
func (s *pullService) Find(ctx context.Context, repo string, number int) (*scm.PullRequest, *scm.Response, error) {
|
|
path := fmt.Sprintf("repos/%s/pulls/%d", repo, number)
|
|
out := new(pr)
|
|
res, err := s.client.do(ctx, "GET", path, nil, out)
|
|
return convertPullRequest(out), res, err
|
|
}
|
|
|
|
func (s *pullService) List(ctx context.Context, repo string, opts scm.PullRequestListOptions) ([]*scm.PullRequest, *scm.Response, error) {
|
|
path := fmt.Sprintf("repos/%s/pulls?%s", repo, encodePullRequestListOptions(opts))
|
|
out := []*pr{}
|
|
res, err := s.client.do(ctx, "GET", path, nil, &out)
|
|
return convertPullRequestList(out), res, err
|
|
}
|
|
|
|
func (s *pullService) ListChanges(ctx context.Context, repo string, number int, opts scm.ListOptions) ([]*scm.Change, *scm.Response, error) {
|
|
path := fmt.Sprintf("repos/%s/pulls/%d/files?%s", repo, number, encodeListOptions(opts))
|
|
out := []*file{}
|
|
res, err := s.client.do(ctx, "GET", path, nil, &out)
|
|
return convertChangeList(out), res, err
|
|
}
|
|
|
|
func (s *pullService) ListCommits(ctx context.Context, repo string, number int, opts scm.ListOptions) ([]*scm.Commit, *scm.Response, error) {
|
|
path := fmt.Sprintf("repos/%s/pulls/%d/commits?%s", repo, number, encodeListOptions(opts))
|
|
out := []*commit{}
|
|
res, err := s.client.do(ctx, "GET", path, nil, &out)
|
|
return convertCommitList(out), res, err
|
|
}
|
|
|
|
func (s *pullService) Merge(ctx context.Context, repo string, number int) (*scm.Response, error) {
|
|
path := fmt.Sprintf("repos/%s/pulls/%d/merge", repo, number)
|
|
res, err := s.client.do(ctx, "PUT", path, nil, nil)
|
|
return res, err
|
|
}
|
|
|
|
func (s *pullService) Close(ctx context.Context, repo string, number int) (*scm.Response, error) {
|
|
path := fmt.Sprintf("repos/%s/pulls/%d", repo, number)
|
|
data := map[string]string{"state": "closed"}
|
|
res, err := s.client.do(ctx, "PATCH", path, &data, nil)
|
|
return res, err
|
|
}
|
|
|
|
func (s *pullService) Create(ctx context.Context, repo string, input *scm.PullRequestInput) (*scm.PullRequest, *scm.Response, error) {
|
|
path := fmt.Sprintf("repos/%s/pulls", repo)
|
|
in := &prInput{
|
|
Title: input.Title,
|
|
Body: input.Body,
|
|
Head: input.Source,
|
|
Base: input.Target,
|
|
}
|
|
out := new(pr)
|
|
res, err := s.client.do(ctx, "POST", path, in, out)
|
|
return convertPullRequest(out), res, err
|
|
}
|
|
|
|
type pr struct {
|
|
Number int `json:"number"`
|
|
State string `json:"state"`
|
|
Title string `json:"title"`
|
|
Body string `json:"body"`
|
|
DiffURL string `json:"diff_url"`
|
|
HTMLURL string `json:"html_url"`
|
|
User struct {
|
|
Login string `json:"login"`
|
|
AvatarURL string `json:"avatar_url"`
|
|
} `json:"user"`
|
|
Head struct {
|
|
Ref string `json:"ref"`
|
|
Sha string `json:"sha"`
|
|
User struct {
|
|
Login string `json:"login"`
|
|
AvatarURL string `json:"avatar_url"`
|
|
}
|
|
Repo struct {
|
|
FullName string `json:"full_name"`
|
|
} `json:"repo"`
|
|
} `json:"head"`
|
|
Base struct {
|
|
Ref string `json:"ref"`
|
|
Sha string `json:"sha"`
|
|
User struct {
|
|
Login string `json:"login"`
|
|
AvatarURL string `json:"avatar_url"`
|
|
}
|
|
} `json:"base"`
|
|
MergedAt null.String `json:"merged_at"`
|
|
CreatedAt time.Time `json:"created_at"`
|
|
UpdatedAt time.Time `json:"updated_at"`
|
|
Labels []struct {
|
|
Name string `json:"name"`
|
|
Color string `json:"color"`
|
|
} `json:"labels"`
|
|
}
|
|
|
|
type prInput struct {
|
|
Title string `json:"title"`
|
|
Body string `json:"body"`
|
|
Head string `json:"head"`
|
|
Base string `json:"base"`
|
|
}
|
|
|
|
type file struct {
|
|
BlobID string `json:"sha"`
|
|
Filename string `json:"filename"`
|
|
Status string `json:"status"`
|
|
Additions int `json:"additions"`
|
|
Deletions int `json:"deletions"`
|
|
Changes int `json:"changes"`
|
|
PreviousFilename string `json:"previous_filename"`
|
|
}
|
|
|
|
func convertPullRequestList(from []*pr) []*scm.PullRequest {
|
|
to := []*scm.PullRequest{}
|
|
for _, v := range from {
|
|
to = append(to, convertPullRequest(v))
|
|
}
|
|
return to
|
|
}
|
|
|
|
func convertPullRequest(from *pr) *scm.PullRequest {
|
|
var labels []scm.Label
|
|
for _, label := range from.Labels {
|
|
labels = append(labels, scm.Label{
|
|
Name: label.Name,
|
|
Color: label.Color,
|
|
})
|
|
}
|
|
return &scm.PullRequest{
|
|
Number: from.Number,
|
|
Title: from.Title,
|
|
Body: from.Body,
|
|
Sha: from.Head.Sha,
|
|
Ref: fmt.Sprintf("refs/pull/%d/head", from.Number),
|
|
Source: from.Head.Ref,
|
|
Target: from.Base.Ref,
|
|
Fork: from.Head.Repo.FullName,
|
|
Link: from.HTMLURL,
|
|
Diff: from.DiffURL,
|
|
Closed: from.State != "open",
|
|
Merged: from.MergedAt.String != "",
|
|
Head: scm.Reference{
|
|
Name: from.Head.Ref,
|
|
Path: scm.ExpandRef(from.Head.Ref, "refs/heads"),
|
|
Sha: from.Head.Sha,
|
|
},
|
|
Base: scm.Reference{
|
|
Name: from.Base.Ref,
|
|
Path: scm.ExpandRef(from.Base.Ref, "refs/heads"),
|
|
Sha: from.Base.Sha,
|
|
},
|
|
Author: scm.User{
|
|
Login: from.User.Login,
|
|
Avatar: from.User.AvatarURL,
|
|
},
|
|
Created: from.CreatedAt,
|
|
Updated: from.UpdatedAt,
|
|
Labels: labels,
|
|
}
|
|
}
|
|
|
|
func convertChangeList(from []*file) []*scm.Change {
|
|
to := []*scm.Change{}
|
|
for _, v := range from {
|
|
to = append(to, convertChange(v))
|
|
}
|
|
return to
|
|
}
|
|
|
|
func convertChange(from *file) *scm.Change {
|
|
return &scm.Change{
|
|
Path: from.Filename,
|
|
Added: from.Status == "added",
|
|
Deleted: from.Status == "removed",
|
|
Renamed: from.Status == "renamed",
|
|
BlobID: from.BlobID,
|
|
PrevFilePath: from.PreviousFilename,
|
|
}
|
|
}
|