From 20403f75fbf043747eb0c9e571a8ff7c4594ad2a Mon Sep 17 00:00:00 2001 From: Martin Hartkorn Date: Sat, 30 Jan 2016 23:56:38 +0100 Subject: Enable a way to checkout Pull Requests from remote refs --- models/pull.go | 39 +++++++++++++++++++++++++++++++++++++++ 1 file changed, 39 insertions(+) (limited to 'models/pull.go') diff --git a/models/pull.go b/models/pull.go index 47d80473..3935c130 100644 --- a/models/pull.go +++ b/models/pull.go @@ -5,6 +5,7 @@ package models import ( + "errors" "fmt" "os" "path" @@ -395,6 +396,44 @@ func GetUnmergedPullRequestsByBaseInfo(repoID int64, branch string) ([]*PullRequ Join("INNER", "issue", "issue.id=pull_request.issue_id").Find(&prs) } +// Gets a Pull Request by the path of the forked repo and the branch from where the PR +// got submitted. +func GetUnmergedPullRequestByRepoPathAndHeadBranch(user, repo, branch string) (*PullRequest, error) { + userLower := strings.ToLower(user) + repoLower := strings.ToLower(repo) + + pr := new(PullRequest) + if x == nil { + return nil, errors.New("Fail") + } + has, err := x. + Where("head_user_name=? AND head_branch=? AND has_merged=? AND issue.is_closed=? AND repository.lower_name=?", userLower, branch, 0, 0, repoLower). + Join("INNER", "repository", "repository.id=pull_request.head_repo_id"). + Join("INNER", "issue", "issue.id=pull_request.issue_id"). + Get(pr) + + if err != nil { + return nil, err + } else if !has { + return nil, ErrPullRequestNotExist{0, 0, 0, 0, branch, ""} + } + + baseRepo := new(Repository) + has, err = x.Where("repository.id=?", pr.BaseRepoID). + Join("LEFT", "user", "user.id=repository.owner_id"). + Get(baseRepo) + + if err != nil { + return nil, err + } else if !has { + return nil, ErrRepoNotExist{pr.BaseRepoID, 0, ""} + } + + pr.BaseRepo = baseRepo + + return pr, nil +} + // GetPullRequestByID returns a pull request by given ID. func GetPullRequestByID(id int64) (*PullRequest, error) { pr := new(PullRequest) -- cgit v1.2.3