Move pull request api convert to convert package (#9664)
* Move pull request api convert to convert package * Rename ToPullRequest to ToAPIPullRequestforgejo
parent
8878cfa4a1
commit
384c2b342e
@ -0,0 +1,16 @@
|
|||||||
|
// Copyright 2020 The Gitea Authors. All rights reserved.
|
||||||
|
// Use of this source code is governed by a MIT-style
|
||||||
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
|
package convert
|
||||||
|
|
||||||
|
import (
|
||||||
|
"path/filepath"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"code.gitea.io/gitea/models"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestMain(m *testing.M) {
|
||||||
|
models.MainTest(m, filepath.Join("..", ".."))
|
||||||
|
}
|
@ -0,0 +1,141 @@
|
|||||||
|
// Copyright 2020 The Gitea Authors. All rights reserved.
|
||||||
|
// Use of this source code is governed by a MIT-style
|
||||||
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
|
package convert
|
||||||
|
|
||||||
|
import (
|
||||||
|
"code.gitea.io/gitea/models"
|
||||||
|
"code.gitea.io/gitea/modules/git"
|
||||||
|
"code.gitea.io/gitea/modules/log"
|
||||||
|
api "code.gitea.io/gitea/modules/structs"
|
||||||
|
)
|
||||||
|
|
||||||
|
// ToAPIPullRequest assumes following fields have been assigned with valid values:
|
||||||
|
// Required - Issue
|
||||||
|
// Optional - Merger
|
||||||
|
func ToAPIPullRequest(pr *models.PullRequest) *api.PullRequest {
|
||||||
|
var (
|
||||||
|
baseBranch *git.Branch
|
||||||
|
headBranch *git.Branch
|
||||||
|
baseCommit *git.Commit
|
||||||
|
headCommit *git.Commit
|
||||||
|
err error
|
||||||
|
)
|
||||||
|
if err = pr.Issue.LoadRepo(); err != nil {
|
||||||
|
log.Error("loadRepo[%d]: %v", pr.ID, err)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
apiIssue := pr.Issue.APIFormat()
|
||||||
|
if pr.BaseRepo == nil {
|
||||||
|
pr.BaseRepo, err = models.GetRepositoryByID(pr.BaseRepoID)
|
||||||
|
if err != nil {
|
||||||
|
log.Error("GetRepositoryById[%d]: %v", pr.ID, err)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if pr.HeadRepo == nil {
|
||||||
|
pr.HeadRepo, err = models.GetRepositoryByID(pr.HeadRepoID)
|
||||||
|
if err != nil {
|
||||||
|
log.Error("GetRepositoryById[%d]: %v", pr.ID, err)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if err = pr.Issue.LoadRepo(); err != nil {
|
||||||
|
log.Error("pr.Issue.loadRepo[%d]: %v", pr.ID, err)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
apiPullRequest := &api.PullRequest{
|
||||||
|
ID: pr.ID,
|
||||||
|
URL: pr.Issue.HTMLURL(),
|
||||||
|
Index: pr.Index,
|
||||||
|
Poster: apiIssue.Poster,
|
||||||
|
Title: apiIssue.Title,
|
||||||
|
Body: apiIssue.Body,
|
||||||
|
Labels: apiIssue.Labels,
|
||||||
|
Milestone: apiIssue.Milestone,
|
||||||
|
Assignee: apiIssue.Assignee,
|
||||||
|
Assignees: apiIssue.Assignees,
|
||||||
|
State: apiIssue.State,
|
||||||
|
Comments: apiIssue.Comments,
|
||||||
|
HTMLURL: pr.Issue.HTMLURL(),
|
||||||
|
DiffURL: pr.Issue.DiffURL(),
|
||||||
|
PatchURL: pr.Issue.PatchURL(),
|
||||||
|
HasMerged: pr.HasMerged,
|
||||||
|
MergeBase: pr.MergeBase,
|
||||||
|
Deadline: apiIssue.Deadline,
|
||||||
|
Created: pr.Issue.CreatedUnix.AsTimePtr(),
|
||||||
|
Updated: pr.Issue.UpdatedUnix.AsTimePtr(),
|
||||||
|
}
|
||||||
|
baseBranch, err = pr.BaseRepo.GetBranch(pr.BaseBranch)
|
||||||
|
if err != nil {
|
||||||
|
if git.IsErrBranchNotExist(err) {
|
||||||
|
apiPullRequest.Base = nil
|
||||||
|
} else {
|
||||||
|
log.Error("GetBranch[%s]: %v", pr.BaseBranch, err)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
apiBaseBranchInfo := &api.PRBranchInfo{
|
||||||
|
Name: pr.BaseBranch,
|
||||||
|
Ref: pr.BaseBranch,
|
||||||
|
RepoID: pr.BaseRepoID,
|
||||||
|
Repository: pr.BaseRepo.APIFormat(models.AccessModeNone),
|
||||||
|
}
|
||||||
|
baseCommit, err = baseBranch.GetCommit()
|
||||||
|
if err != nil {
|
||||||
|
if git.IsErrNotExist(err) {
|
||||||
|
apiBaseBranchInfo.Sha = ""
|
||||||
|
} else {
|
||||||
|
log.Error("GetCommit[%s]: %v", baseBranch.Name, err)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
apiBaseBranchInfo.Sha = baseCommit.ID.String()
|
||||||
|
}
|
||||||
|
apiPullRequest.Base = apiBaseBranchInfo
|
||||||
|
}
|
||||||
|
|
||||||
|
headBranch, err = pr.HeadRepo.GetBranch(pr.HeadBranch)
|
||||||
|
if err != nil {
|
||||||
|
if git.IsErrBranchNotExist(err) {
|
||||||
|
apiPullRequest.Head = nil
|
||||||
|
} else {
|
||||||
|
log.Error("GetBranch[%s]: %v", pr.HeadBranch, err)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
apiHeadBranchInfo := &api.PRBranchInfo{
|
||||||
|
Name: pr.HeadBranch,
|
||||||
|
Ref: pr.HeadBranch,
|
||||||
|
RepoID: pr.HeadRepoID,
|
||||||
|
Repository: pr.HeadRepo.APIFormat(models.AccessModeNone),
|
||||||
|
}
|
||||||
|
headCommit, err = headBranch.GetCommit()
|
||||||
|
if err != nil {
|
||||||
|
if git.IsErrNotExist(err) {
|
||||||
|
apiHeadBranchInfo.Sha = ""
|
||||||
|
} else {
|
||||||
|
log.Error("GetCommit[%s]: %v", headBranch.Name, err)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
apiHeadBranchInfo.Sha = headCommit.ID.String()
|
||||||
|
}
|
||||||
|
apiPullRequest.Head = apiHeadBranchInfo
|
||||||
|
}
|
||||||
|
|
||||||
|
if pr.Status != models.PullRequestStatusChecking {
|
||||||
|
mergeable := !(pr.Status == models.PullRequestStatusConflict || pr.Status == models.PullRequestStatusError) && !pr.IsWorkInProgress()
|
||||||
|
apiPullRequest.Mergeable = mergeable
|
||||||
|
}
|
||||||
|
if pr.HasMerged {
|
||||||
|
apiPullRequest.Merged = pr.MergedUnix.AsTimePtr()
|
||||||
|
apiPullRequest.MergedCommitID = &pr.MergedCommitID
|
||||||
|
apiPullRequest.MergedBy = pr.Merger.APIFormat()
|
||||||
|
}
|
||||||
|
|
||||||
|
return apiPullRequest
|
||||||
|
}
|
@ -0,0 +1,23 @@
|
|||||||
|
// Copyright 2020 The Gitea Authors. All rights reserved.
|
||||||
|
// Use of this source code is governed by a MIT-style
|
||||||
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
|
package convert
|
||||||
|
|
||||||
|
import (
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"code.gitea.io/gitea/models"
|
||||||
|
|
||||||
|
"github.com/stretchr/testify/assert"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestPullRequest_APIFormat(t *testing.T) {
|
||||||
|
assert.NoError(t, models.PrepareTestDatabase())
|
||||||
|
pr := models.AssertExistsAndLoadBean(t, &models.PullRequest{ID: 1}).(*models.PullRequest)
|
||||||
|
assert.NoError(t, pr.LoadAttributes())
|
||||||
|
assert.NoError(t, pr.LoadIssue())
|
||||||
|
apiPullRequest := ToAPIPullRequest(pr)
|
||||||
|
assert.NotNil(t, apiPullRequest)
|
||||||
|
assert.Nil(t, apiPullRequest.Head)
|
||||||
|
}
|
Loading…
Reference in New Issue