2015-10-19 05:00:39 +05:30
// Copyright 2015 The Gogs Authors. All rights reserved.
2019-04-19 17:47:27 +05:30
// Copyright 2019 The Gitea Authors. All rights reserved.
2015-10-19 05:00:39 +05:30
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file.
package models
import (
2021-12-10 06:57:50 +05:30
"context"
2015-10-19 05:00:39 +05:30
"fmt"
2019-12-31 05:04:11 +05:30
"io"
2015-10-19 05:00:39 +05:30
"strings"
2021-09-19 17:19:59 +05:30
"code.gitea.io/gitea/models/db"
2022-06-12 21:21:54 +05:30
git_model "code.gitea.io/gitea/models/git"
2022-05-08 19:16:34 +05:30
pull_model "code.gitea.io/gitea/models/pull"
2021-12-10 06:57:50 +05:30
repo_model "code.gitea.io/gitea/models/repo"
2021-11-24 15:19:20 +05:30
user_model "code.gitea.io/gitea/models/user"
2021-12-02 12:58:08 +05:30
"code.gitea.io/gitea/modules/git"
2016-11-10 21:54:48 +05:30
"code.gitea.io/gitea/modules/log"
"code.gitea.io/gitea/modules/setting"
2019-08-15 20:16:21 +05:30
"code.gitea.io/gitea/modules/timeutil"
2021-02-18 08:15:49 +05:30
"code.gitea.io/gitea/modules/util"
2022-05-07 22:35:52 +05:30
"xorm.io/builder"
2015-10-19 05:00:39 +05:30
)
2016-11-28 21:01:06 +05:30
// PullRequestType defines pull request type
2015-10-19 05:00:39 +05:30
type PullRequestType int
2016-11-28 21:01:06 +05:30
// Enumerate all the pull request types
2015-10-19 05:00:39 +05:30
const (
2016-11-07 21:07:32 +05:30
PullRequestGitea PullRequestType = iota
PullRequestGit
2015-10-19 05:00:39 +05:30
)
2016-11-28 21:01:06 +05:30
// PullRequestStatus defines pull request status
2015-10-19 05:00:39 +05:30
type PullRequestStatus int
2016-11-28 21:01:06 +05:30
// Enumerate all the pull request status
2015-10-19 05:00:39 +05:30
const (
2016-11-07 21:07:32 +05:30
PullRequestStatusConflict PullRequestStatus = iota
PullRequestStatusChecking
PullRequestStatusMergeable
2017-02-05 18:37:44 +05:30
PullRequestStatusManuallyMerged
2020-01-10 05:44:14 +05:30
PullRequestStatusError
2021-03-04 09:11:23 +05:30
PullRequestStatusEmpty
2015-10-19 05:00:39 +05:30
)
2021-07-28 15:12:56 +05:30
// PullRequestFlow the flow of pull request
type PullRequestFlow int
const (
// PullRequestFlowGithub github flow from head branch to base branch
PullRequestFlowGithub PullRequestFlow = iota
// PullRequestFlowAGit Agit flow pull request, head branch is not exist
PullRequestFlowAGit
)
2015-10-19 05:00:39 +05:30
// PullRequest represents relation between pull request and repositories.
type PullRequest struct {
2019-02-05 17:24:49 +05:30
ID int64 ` xorm:"pk autoincr" `
Type PullRequestType
Status PullRequestStatus
ConflictedFiles [ ] string ` xorm:"TEXT JSON" `
2020-04-14 19:23:34 +05:30
CommitsAhead int
CommitsBehind int
2015-10-19 05:00:39 +05:30
2020-10-14 00:20:57 +05:30
ChangedProtectedFiles [ ] string ` xorm:"TEXT JSON" `
2015-10-19 05:00:39 +05:30
IssueID int64 ` xorm:"INDEX" `
Issue * Issue ` xorm:"-" `
Index int64
2022-04-28 21:15:33 +05:30
HeadRepoID int64 ` xorm:"INDEX" `
HeadRepo * repo_model . Repository ` xorm:"-" `
BaseRepoID int64 ` xorm:"INDEX" `
BaseRepo * repo_model . Repository ` xorm:"-" `
HeadBranch string
HeadCommitID string ` xorm:"-" `
BaseBranch string
2022-06-12 21:21:54 +05:30
ProtectedBranch * git_model . ProtectedBranch ` xorm:"-" `
MergeBase string ` xorm:"VARCHAR(40)" `
AllowMaintainerEdit bool ` xorm:"NOT NULL DEFAULT false" `
2015-10-19 05:00:39 +05:30
2019-08-15 20:16:21 +05:30
HasMerged bool ` xorm:"INDEX" `
MergedCommitID string ` xorm:"VARCHAR(40)" `
MergerID int64 ` xorm:"INDEX" `
2021-11-24 15:19:20 +05:30
Merger * user_model . User ` xorm:"-" `
2019-08-15 20:16:21 +05:30
MergedUnix timeutil . TimeStamp ` xorm:"updated INDEX" `
2020-03-03 04:01:55 +05:30
isHeadRepoLoaded bool ` xorm:"-" `
2021-07-28 15:12:56 +05:30
Flow PullRequestFlow ` xorm:"NOT NULL DEFAULT 0" `
2015-10-19 05:00:39 +05:30
}
2021-09-19 17:19:59 +05:30
func init ( ) {
db . RegisterModel ( new ( PullRequest ) )
}
2022-05-20 19:38:52 +05:30
func deletePullsByBaseRepoID ( ctx context . Context , repoID int64 ) error {
2022-05-08 19:16:34 +05:30
deleteCond := builder . Select ( "id" ) . From ( "pull_request" ) . Where ( builder . Eq { "pull_request.base_repo_id" : repoID } )
// Delete scheduled auto merges
2022-05-20 19:38:52 +05:30
if _ , err := db . GetEngine ( ctx ) . In ( "pull_id" , deleteCond ) .
2022-05-08 19:16:34 +05:30
Delete ( & pull_model . AutoMerge { } ) ; err != nil {
return err
}
// Delete review states
2022-05-20 19:38:52 +05:30
if _ , err := db . GetEngine ( ctx ) . In ( "pull_id" , deleteCond ) .
2022-05-08 19:16:34 +05:30
Delete ( & pull_model . ReviewState { } ) ; err != nil {
return err
}
2022-05-20 19:38:52 +05:30
_ , err := db . DeleteByBean ( ctx , & PullRequest { BaseRepoID : repoID } )
2022-05-08 19:16:34 +05:30
return err
}
2019-10-18 16:43:31 +05:30
// MustHeadUserName returns the HeadRepo's username if failed return blank
func ( pr * PullRequest ) MustHeadUserName ( ) string {
if err := pr . LoadHeadRepo ( ) ; err != nil {
2021-12-10 06:57:50 +05:30
if ! repo_model . IsErrRepoNotExist ( err ) {
2020-01-25 08:18:22 +05:30
log . Error ( "LoadHeadRepo: %v" , err )
} else {
log . Warn ( "LoadHeadRepo %d but repository does not exist: %v" , pr . HeadRepoID , err )
}
2019-10-18 16:43:31 +05:30
return ""
}
2020-03-03 04:01:55 +05:30
if pr . HeadRepo == nil {
return ""
}
2020-01-12 15:06:21 +05:30
return pr . HeadRepo . OwnerName
2019-10-18 16:43:31 +05:30
}
2016-08-14 16:02:24 +05:30
// Note: don't try to get Issue because will end up recursive querying.
2022-05-20 19:38:52 +05:30
func ( pr * PullRequest ) loadAttributes ( ctx context . Context ) ( err error ) {
2016-08-14 16:02:24 +05:30
if pr . HasMerged && pr . Merger == nil {
2022-05-20 19:38:52 +05:30
pr . Merger , err = user_model . GetUserByIDCtx ( ctx , pr . MergerID )
2021-11-24 15:19:20 +05:30
if user_model . IsErrUserNotExist ( err ) {
2016-08-14 16:02:24 +05:30
pr . MergerID = - 1
2021-11-24 15:19:20 +05:30
pr . Merger = user_model . NewGhostUser ( )
2016-08-14 16:02:24 +05:30
} else if err != nil {
return fmt . Errorf ( "getUserByID [%d]: %v" , pr . MergerID , err )
}
}
return nil
}
2016-11-28 21:01:06 +05:30
// LoadAttributes loads pull request attributes from database
2016-08-14 16:02:24 +05:30
func ( pr * PullRequest ) LoadAttributes ( ) error {
2022-05-20 19:38:52 +05:30
return pr . loadAttributes ( db . DefaultContext )
2016-08-14 16:02:24 +05:30
}
2022-04-28 17:18:48 +05:30
// LoadHeadRepoCtx loads the head repository
func ( pr * PullRequest ) LoadHeadRepoCtx ( ctx context . Context ) ( err error ) {
2020-03-03 04:01:55 +05:30
if ! pr . isHeadRepoLoaded && pr . HeadRepo == nil && pr . HeadRepoID > 0 {
if pr . HeadRepoID == pr . BaseRepoID {
if pr . BaseRepo != nil {
pr . HeadRepo = pr . BaseRepo
return nil
} else if pr . Issue != nil && pr . Issue . Repo != nil {
pr . HeadRepo = pr . Issue . Repo
return nil
}
2019-10-18 16:43:31 +05:30
}
2020-03-03 04:01:55 +05:30
2021-12-10 06:57:50 +05:30
pr . HeadRepo , err = repo_model . GetRepositoryByIDCtx ( ctx , pr . HeadRepoID )
if err != nil && ! repo_model . IsErrRepoNotExist ( err ) { // Head repo maybe deleted, but it should still work
2020-03-03 04:01:55 +05:30
return fmt . Errorf ( "getRepositoryByID(head): %v" , err )
2019-09-18 11:09:45 +05:30
}
2020-03-03 04:01:55 +05:30
pr . isHeadRepoLoaded = true
2019-09-18 11:09:45 +05:30
}
return nil
}
2020-03-03 04:01:55 +05:30
// LoadHeadRepo loads the head repository
2019-10-18 16:43:31 +05:30
func ( pr * PullRequest ) LoadHeadRepo ( ) error {
2022-04-28 17:18:48 +05:30
return pr . LoadHeadRepoCtx ( db . DefaultContext )
2020-03-03 04:01:55 +05:30
}
// LoadBaseRepo loads the target repository
func ( pr * PullRequest ) LoadBaseRepo ( ) error {
2022-04-28 17:18:48 +05:30
return pr . LoadBaseRepoCtx ( db . DefaultContext )
2020-03-03 04:01:55 +05:30
}
2022-04-28 17:18:48 +05:30
// LoadBaseRepoCtx loads the target repository
func ( pr * PullRequest ) LoadBaseRepoCtx ( ctx context . Context ) ( err error ) {
2020-03-03 04:01:55 +05:30
if pr . BaseRepo != nil {
return nil
}
if pr . HeadRepoID == pr . BaseRepoID && pr . HeadRepo != nil {
pr . BaseRepo = pr . HeadRepo
return nil
}
if pr . Issue != nil && pr . Issue . Repo != nil {
pr . BaseRepo = pr . Issue . Repo
return nil
}
2021-12-10 06:57:50 +05:30
pr . BaseRepo , err = repo_model . GetRepositoryByIDCtx ( ctx , pr . BaseRepoID )
2020-03-03 04:01:55 +05:30
if err != nil {
2021-12-10 06:57:50 +05:30
return fmt . Errorf ( "repo_model.GetRepositoryByID(base): %v" , err )
2019-10-18 16:43:31 +05:30
}
return nil
}
2016-11-28 21:01:06 +05:30
// LoadIssue loads issue information from database
2016-08-14 16:02:24 +05:30
func ( pr * PullRequest ) LoadIssue ( ) ( err error ) {
2022-04-28 17:18:48 +05:30
return pr . LoadIssueCtx ( db . DefaultContext )
2017-02-27 06:12:55 +05:30
}
2022-04-28 17:18:48 +05:30
// LoadIssueCtx loads issue information from database
func ( pr * PullRequest ) LoadIssueCtx ( ctx context . Context ) ( err error ) {
2016-08-16 22:49:09 +05:30
if pr . Issue != nil {
return nil
}
2022-05-20 19:38:52 +05:30
pr . Issue , err = getIssueByID ( ctx , pr . IssueID )
2019-11-21 22:38:42 +05:30
if err == nil {
pr . Issue . PullRequest = pr
}
2016-08-14 16:02:24 +05:30
return err
}
2018-12-11 16:58:37 +05:30
// LoadProtectedBranch loads the protected branch of the base branch
func ( pr * PullRequest ) LoadProtectedBranch ( ) ( err error ) {
2022-04-28 17:18:48 +05:30
return pr . LoadProtectedBranchCtx ( db . DefaultContext )
2019-12-04 06:38:56 +05:30
}
2022-04-28 17:18:48 +05:30
// LoadProtectedBranchCtx loads the protected branch of the base branch
func ( pr * PullRequest ) LoadProtectedBranchCtx ( ctx context . Context ) ( err error ) {
2020-01-15 14:02:57 +05:30
if pr . ProtectedBranch == nil {
if pr . BaseRepo == nil {
if pr . BaseRepoID == 0 {
return nil
}
2021-12-10 06:57:50 +05:30
pr . BaseRepo , err = repo_model . GetRepositoryByIDCtx ( ctx , pr . BaseRepoID )
2020-01-15 14:02:57 +05:30
if err != nil {
return
}
2018-12-17 23:05:21 +05:30
}
2022-06-12 21:21:54 +05:30
pr . ProtectedBranch , err = git_model . GetProtectedBranchBy ( ctx , pr . BaseRepo . ID , pr . BaseBranch )
2018-12-17 23:05:21 +05:30
}
2018-12-11 16:58:37 +05:30
return
}
2020-03-06 09:14:06 +05:30
// ReviewCount represents a count of Reviews
type ReviewCount struct {
IssueID int64
Type ReviewType
Count int64
}
// GetApprovalCounts returns the approval counts by type
// FIXME: Only returns official counts due to double counting of non-official counts
2022-05-20 19:38:52 +05:30
func ( pr * PullRequest ) GetApprovalCounts ( ctx context . Context ) ( [ ] * ReviewCount , error ) {
2020-03-06 09:14:06 +05:30
rCounts := make ( [ ] * ReviewCount , 0 , 6 )
2022-05-20 19:38:52 +05:30
sess := db . GetEngine ( ctx ) . Where ( "issue_id = ?" , pr . IssueID )
2021-02-11 23:02:25 +05:30
return rCounts , sess . Select ( "issue_id, type, count(id) as `count`" ) . Where ( "official = ? AND dismissed = ?" , true , false ) . GroupBy ( "issue_id, type" ) . Table ( "review" ) . Find ( & rCounts )
2020-03-06 09:14:06 +05:30
}
2019-12-31 05:04:11 +05:30
// GetApprovers returns the approvers of the pull request
func ( pr * PullRequest ) GetApprovers ( ) string {
stringBuilder := strings . Builder { }
if err := pr . getReviewedByLines ( & stringBuilder ) ; err != nil {
log . Error ( "Unable to getReviewedByLines: Error: %v" , err )
return ""
}
return stringBuilder . String ( )
}
func ( pr * PullRequest ) getReviewedByLines ( writer io . Writer ) error {
maxReviewers := setting . Repository . PullRequest . DefaultMergeMessageMaxApprovers
if maxReviewers == 0 {
return nil
}
2021-11-21 21:11:00 +05:30
ctx , committer , err := db . TxContext ( )
if err != nil {
2019-12-31 05:04:11 +05:30
return err
}
2021-11-21 21:11:00 +05:30
defer committer . Close ( )
2019-12-31 05:04:11 +05:30
// Note: This doesn't page as we only expect a very limited number of reviews
2022-05-20 19:38:52 +05:30
reviews , err := FindReviews ( ctx , FindReviewOptions {
2019-12-31 05:04:11 +05:30
Type : ReviewTypeApprove ,
IssueID : pr . IssueID ,
OfficialOnly : setting . Repository . PullRequest . DefaultMergeMessageOfficialApproversOnly ,
} )
if err != nil {
log . Error ( "Unable to FindReviews for PR ID %d: %v" , pr . ID , err )
return err
}
reviewersWritten := 0
for _ , review := range reviews {
if maxReviewers > 0 && reviewersWritten > maxReviewers {
break
}
2022-05-20 19:38:52 +05:30
if err := review . loadReviewer ( ctx ) ; err != nil && ! user_model . IsErrUserNotExist ( err ) {
2019-12-31 05:04:11 +05:30
log . Error ( "Unable to LoadReviewer[%d] for PR ID %d : %v" , review . ReviewerID , pr . ID , err )
return err
} else if review . Reviewer == nil {
continue
}
if _ , err := writer . Write ( [ ] byte ( "Reviewed-by: " ) ) ; err != nil {
return err
}
if _ , err := writer . Write ( [ ] byte ( review . Reviewer . NewGitSig ( ) . String ( ) ) ) ; err != nil {
return err
}
if _ , err := writer . Write ( [ ] byte { '\n' } ) ; err != nil {
return err
}
reviewersWritten ++
}
2021-11-21 21:11:00 +05:30
return committer . Commit ( )
2019-12-31 05:04:11 +05:30
}
2018-01-19 11:48:51 +05:30
// GetGitRefName returns git ref for hidden pull request branch
func ( pr * PullRequest ) GetGitRefName ( ) string {
2021-12-03 01:06:50 +05:30
return fmt . Sprintf ( "%s%d/head" , git . PullPrefix , pr . Index )
2018-01-19 11:48:51 +05:30
}
2015-10-24 13:06:47 +05:30
// IsChecking returns true if this pull request is still checking conflict.
func ( pr * PullRequest ) IsChecking ( ) bool {
2016-11-07 21:07:32 +05:30
return pr . Status == PullRequestStatusChecking
2015-10-24 13:06:47 +05:30
}
2015-10-19 05:00:39 +05:30
// CanAutoMerge returns true if this pull request can be merged automatically.
func ( pr * PullRequest ) CanAutoMerge ( ) bool {
2016-11-07 21:07:32 +05:30
return pr . Status == PullRequestStatusMergeable
2015-10-19 05:00:39 +05:30
}
2021-03-04 09:11:23 +05:30
// IsEmpty returns true if this pull request is empty.
func ( pr * PullRequest ) IsEmpty ( ) bool {
return pr . Status == PullRequestStatusEmpty
}
2019-06-22 23:05:34 +05:30
// SetMerged sets a pull request to merged and closes the corresponding issue
2022-05-04 01:16:28 +05:30
func ( pr * PullRequest ) SetMerged ( ctx context . Context ) ( bool , error ) {
2017-02-05 18:37:44 +05:30
if pr . HasMerged {
2020-02-10 04:39:31 +05:30
return false , fmt . Errorf ( "PullRequest[%d] already merged" , pr . Index )
2017-02-05 18:37:44 +05:30
}
2017-12-11 10:07:04 +05:30
if pr . MergedCommitID == "" || pr . MergedUnix == 0 || pr . Merger == nil {
2020-02-10 04:39:31 +05:30
return false , fmt . Errorf ( "Unable to merge PullRequest[%d], some required fields are empty" , pr . Index )
2017-02-05 18:37:44 +05:30
}
pr . HasMerged = true
2021-11-19 19:09:57 +05:30
sess := db . GetEngine ( ctx )
2017-02-05 18:37:44 +05:30
2020-02-10 04:39:31 +05:30
if _ , err := sess . Exec ( "UPDATE `issue` SET `repo_id` = `repo_id` WHERE `id` = ?" , pr . IssueID ) ; err != nil {
return false , err
2017-02-05 18:37:44 +05:30
}
2020-02-10 04:39:31 +05:30
if _ , err := sess . Exec ( "UPDATE `pull_request` SET `issue_id` = `issue_id` WHERE `id` = ?" , pr . ID ) ; err != nil {
return false , err
2017-02-27 06:12:55 +05:30
}
2020-02-10 04:39:31 +05:30
pr . Issue = nil
2022-04-28 17:18:48 +05:30
if err := pr . LoadIssueCtx ( ctx ) ; err != nil {
2020-02-10 04:39:31 +05:30
return false , err
2017-02-05 18:37:44 +05:30
}
2022-05-20 19:38:52 +05:30
if tmpPr , err := GetPullRequestByID ( ctx , pr . ID ) ; err != nil {
2020-02-10 04:39:31 +05:30
return false , err
} else if tmpPr . HasMerged {
if pr . Issue . IsClosed {
return false , nil
}
return false , fmt . Errorf ( "PullRequest[%d] already merged but it's associated issue [%d] is not closed" , pr . Index , pr . IssueID )
} else if pr . Issue . IsClosed {
return false , fmt . Errorf ( "PullRequest[%d] already closed" , pr . Index )
2017-02-05 18:37:44 +05:30
}
2020-02-10 04:39:31 +05:30
2022-04-08 14:41:15 +05:30
if err := pr . Issue . LoadRepo ( ctx ) ; err != nil {
2020-02-10 04:39:31 +05:30
return false , err
2017-02-05 18:37:44 +05:30
}
2021-12-10 06:57:50 +05:30
if err := pr . Issue . Repo . GetOwner ( ctx ) ; err != nil {
2020-02-10 04:39:31 +05:30
return false , err
2017-02-05 18:37:44 +05:30
}
2020-02-10 04:39:31 +05:30
2022-03-29 20:27:33 +05:30
if _ , err := changeIssueStatus ( ctx , pr . Issue , pr . Merger , true , true ) ; err != nil {
2020-02-10 04:39:31 +05:30
return false , fmt . Errorf ( "Issue.changeStatus: %v" , err )
}
2022-03-29 22:12:34 +05:30
// reset the conflicted files as there cannot be any if we're merged
pr . ConflictedFiles = [ ] string { }
2021-04-10 13:57:29 +05:30
// We need to save all of the data used to compute this merge as it may have already been changed by TestPatch. FIXME: need to set some state to prevent TestPatch from running whilst we are merging.
2022-03-29 22:12:34 +05:30
if _ , err := sess . Where ( "id = ?" , pr . ID ) . Cols ( "has_merged, status, merge_base, merged_commit_id, merger_id, merged_unix, conflicted_files" ) . Update ( pr ) ; err != nil {
2020-02-10 04:39:31 +05:30
return false , fmt . Errorf ( "Failed to update pr[%d]: %v" , pr . ID , err )
}
return true , nil
2017-02-05 18:37:44 +05:30
}
2015-10-19 05:00:39 +05:30
// NewPullRequest creates new pull request with labels for repository.
2022-01-20 04:56:57 +05:30
func NewPullRequest ( outerCtx context . Context , repo * repo_model . Repository , issue * Issue , labelIDs [ ] int64 , uuids [ ] string , pr * PullRequest ) ( err error ) {
2021-09-19 17:19:59 +05:30
idx , err := db . GetNextResourceIndex ( "issue_index" , repo . ID )
2021-06-14 07:52:55 +05:30
if err != nil {
2021-09-23 16:20:06 +05:30
return fmt . Errorf ( "generate pull request index failed: %v" , err )
2019-10-03 03:58:30 +05:30
}
2021-06-14 07:52:55 +05:30
issue . Index = idx
2021-11-19 19:09:57 +05:30
ctx , committer , err := db . TxContext ( )
if err != nil {
2015-10-19 05:00:39 +05:30
return err
}
2021-11-19 19:09:57 +05:30
defer committer . Close ( )
2022-01-20 04:56:57 +05:30
ctx . WithContext ( outerCtx )
2015-10-19 05:00:39 +05:30
2021-11-19 19:09:57 +05:30
if err = newIssue ( ctx , issue . Poster , NewIssueOptions {
2016-08-16 07:10:32 +05:30
Repo : repo ,
2021-06-14 07:52:55 +05:30
Issue : issue ,
2017-03-01 06:38:45 +05:30
LabelIDs : labelIDs ,
2016-08-16 07:10:32 +05:30
Attachments : uuids ,
IsPull : true ,
} ) ; err != nil {
2019-10-03 03:58:30 +05:30
if IsErrUserDoesNotHaveAccessToRepo ( err ) || IsErrNewIssueInsert ( err ) {
2018-05-09 21:59:04 +05:30
return err
}
2015-10-19 05:00:39 +05:30
return fmt . Errorf ( "newIssue: %v" , err )
}
2021-06-14 07:52:55 +05:30
pr . Index = issue . Index
2015-10-24 13:06:47 +05:30
pr . BaseRepo = repo
2021-06-14 07:52:55 +05:30
pr . IssueID = issue . ID
2021-11-19 19:09:57 +05:30
if err = db . Insert ( ctx , pr ) ; err != nil {
2015-10-19 05:00:39 +05:30
return fmt . Errorf ( "insert pull repo: %v" , err )
}
2021-11-19 19:09:57 +05:30
if err = committer . Commit ( ) ; err != nil {
2016-07-15 22:06:39 +05:30
return fmt . Errorf ( "Commit: %v" , err )
}
return nil
2015-10-19 05:00:39 +05:30
}
2017-01-05 06:20:34 +05:30
// GetUnmergedPullRequest returns a pull request that is open and has not been merged
2015-10-19 05:00:39 +05:30
// by given head/base and repo/branch.
2021-07-28 15:12:56 +05:30
func GetUnmergedPullRequest ( headRepoID , baseRepoID int64 , headBranch , baseBranch string , flow PullRequestFlow ) ( * PullRequest , error ) {
2015-10-19 05:00:39 +05:30
pr := new ( PullRequest )
2021-09-23 21:15:36 +05:30
has , err := db . GetEngine ( db . DefaultContext ) .
2021-07-28 15:12:56 +05:30
Where ( "head_repo_id=? AND head_branch=? AND base_repo_id=? AND base_branch=? AND has_merged=? AND flow = ? AND issue.is_closed=?" ,
headRepoID , headBranch , baseRepoID , baseBranch , false , flow , false ) .
2016-11-10 20:46:32 +05:30
Join ( "INNER" , "issue" , "issue.id=pull_request.issue_id" ) .
Get ( pr )
2015-10-19 05:00:39 +05:30
if err != nil {
return nil , err
} else if ! has {
return nil , ErrPullRequestNotExist { 0 , 0 , headRepoID , baseRepoID , headBranch , baseBranch }
}
return pr , nil
}
2019-06-27 19:45:30 +05:30
// GetLatestPullRequestByHeadInfo returns the latest pull request (regardless of its status)
// by given head information (repo and branch).
func GetLatestPullRequestByHeadInfo ( repoID int64 , branch string ) ( * PullRequest , error ) {
pr := new ( PullRequest )
2021-09-23 21:15:36 +05:30
has , err := db . GetEngine ( db . DefaultContext ) .
2021-07-28 15:12:56 +05:30
Where ( "head_repo_id = ? AND head_branch = ? AND flow = ?" , repoID , branch , PullRequestFlowGithub ) .
2019-06-27 19:45:30 +05:30
OrderBy ( "id DESC" ) .
Get ( pr )
if ! has {
return nil , err
}
return pr , err
}
2016-12-02 16:40:39 +05:30
// GetPullRequestByIndex returns a pull request by the given index
2022-05-20 19:38:52 +05:30
func GetPullRequestByIndex ( ctx context . Context , repoID , index int64 ) ( * PullRequest , error ) {
2021-10-05 20:11:48 +05:30
if index < 1 {
return nil , ErrPullRequestNotExist { }
}
2016-12-02 16:40:39 +05:30
pr := & PullRequest {
BaseRepoID : repoID ,
Index : index ,
}
2022-04-28 17:18:48 +05:30
has , err := db . GetEngine ( ctx ) . Get ( pr )
2016-12-02 16:40:39 +05:30
if err != nil {
return nil , err
} else if ! has {
2017-01-01 23:45:09 +05:30
return nil , ErrPullRequestNotExist { 0 , 0 , 0 , repoID , "" , "" }
2016-12-02 16:40:39 +05:30
}
2022-05-20 19:38:52 +05:30
if err = pr . loadAttributes ( ctx ) ; err != nil {
2016-12-05 16:47:39 +05:30
return nil , err
}
2022-04-28 17:18:48 +05:30
if err = pr . LoadIssueCtx ( ctx ) ; err != nil {
2016-12-05 16:47:39 +05:30
return nil , err
}
2016-12-02 16:40:39 +05:30
return pr , nil
}
2022-05-20 19:38:52 +05:30
// GetPullRequestByID returns a pull request by given ID.
func GetPullRequestByID ( ctx context . Context , id int64 ) ( * PullRequest , error ) {
2015-10-24 13:06:47 +05:30
pr := new ( PullRequest )
2022-05-20 19:38:52 +05:30
has , err := db . GetEngine ( ctx ) . ID ( id ) . Get ( pr )
2015-10-24 13:06:47 +05:30
if err != nil {
return nil , err
} else if ! has {
return nil , ErrPullRequestNotExist { id , 0 , 0 , 0 , "" , "" }
}
2022-05-20 19:38:52 +05:30
return pr , pr . loadAttributes ( ctx )
2016-08-14 16:02:24 +05:30
}
2020-01-23 22:58:15 +05:30
// GetPullRequestByIssueIDWithNoAttributes returns pull request with no attributes loaded by given issue ID.
func GetPullRequestByIssueIDWithNoAttributes ( issueID int64 ) ( * PullRequest , error ) {
var pr PullRequest
2021-09-23 21:15:36 +05:30
has , err := db . GetEngine ( db . DefaultContext ) . Where ( "issue_id = ?" , issueID ) . Get ( & pr )
2020-01-23 22:58:15 +05:30
if err != nil {
return nil , err
}
if ! has {
return nil , ErrPullRequestNotExist { 0 , issueID , 0 , 0 , "" , "" }
}
return & pr , nil
}
2022-05-20 19:38:52 +05:30
// GetPullRequestByIssueID returns pull request by given issue ID.
func GetPullRequestByIssueID ( ctx context . Context , issueID int64 ) ( * PullRequest , error ) {
2015-10-23 00:17:24 +05:30
pr := & PullRequest {
IssueID : issueID ,
}
2022-05-20 19:38:52 +05:30
has , err := db . GetByBean ( ctx , pr )
2015-10-19 05:00:39 +05:30
if err != nil {
return nil , err
} else if ! has {
2015-10-23 00:17:24 +05:30
return nil , ErrPullRequestNotExist { 0 , issueID , 0 , 0 , "" , "" }
2015-10-19 05:00:39 +05:30
}
2022-05-20 19:38:52 +05:30
return pr , pr . loadAttributes ( ctx )
2016-08-14 16:02:24 +05:30
}
2021-07-28 15:12:56 +05:30
// GetAllUnmergedAgitPullRequestByPoster get all unmerged agit flow pull request
// By poster id.
func GetAllUnmergedAgitPullRequestByPoster ( uid int64 ) ( [ ] * PullRequest , error ) {
pulls := make ( [ ] * PullRequest , 0 , 10 )
2021-09-23 21:15:36 +05:30
err := db . GetEngine ( db . DefaultContext ) .
2021-07-28 15:12:56 +05:30
Where ( "has_merged=? AND flow = ? AND issue.is_closed=? AND issue.poster_id=?" ,
false , PullRequestFlowAGit , false , uid ) .
Join ( "INNER" , "issue" , "issue.id=pull_request.issue_id" ) .
Find ( & pulls )
return pulls , err
}
2015-10-24 13:06:47 +05:30
// Update updates all fields of pull request.
func ( pr * PullRequest ) Update ( ) error {
2021-09-23 21:15:36 +05:30
_ , err := db . GetEngine ( db . DefaultContext ) . ID ( pr . ID ) . AllCols ( ) . Update ( pr )
2015-10-24 13:06:47 +05:30
return err
}
2016-11-28 21:01:06 +05:30
// UpdateCols updates specific fields of pull request.
2015-10-24 13:06:47 +05:30
func ( pr * PullRequest ) UpdateCols ( cols ... string ) error {
2021-09-23 21:15:36 +05:30
_ , err := db . GetEngine ( db . DefaultContext ) . ID ( pr . ID ) . Cols ( cols ... ) . Update ( pr )
2015-10-24 13:06:47 +05:30
return err
2020-02-10 04:39:31 +05:30
}
// UpdateColsIfNotMerged updates specific fields of a pull request if it has not been merged
func ( pr * PullRequest ) UpdateColsIfNotMerged ( cols ... string ) error {
2021-09-23 21:15:36 +05:30
_ , err := db . GetEngine ( db . DefaultContext ) . Where ( "id = ? AND has_merged = ?" , pr . ID , false ) . Cols ( cols ... ) . Update ( pr )
2020-02-10 04:39:31 +05:30
return err
2015-10-24 13:06:47 +05:30
}
2018-08-14 00:34:39 +05:30
// IsWorkInProgress determine if the Pull Request is a Work In Progress by its title
func ( pr * PullRequest ) IsWorkInProgress ( ) bool {
if err := pr . LoadIssue ( ) ; err != nil {
2019-04-02 13:18:31 +05:30
log . Error ( "LoadIssue: %v" , err )
2018-08-14 00:34:39 +05:30
return false
}
2021-06-23 09:44:22 +05:30
return HasWorkInProgressPrefix ( pr . Issue . Title )
}
2018-08-14 00:34:39 +05:30
2021-06-23 09:44:22 +05:30
// HasWorkInProgressPrefix determines if the given PR title has a Work In Progress prefix
func HasWorkInProgressPrefix ( title string ) bool {
2018-08-14 00:34:39 +05:30
for _ , prefix := range setting . Repository . PullRequest . WorkInProgressPrefixes {
Make WIP prefixes case insensitive, e.g. allow `Draft` as a WIP prefix (#19780)
The issue was that only the actual title was converted to uppercase, but
not the prefix as specified in `WORK_IN_PROGRESS_PREFIXES`. As a result,
the following did not work:
WORK_IN_PROGRESS_PREFIXES=Draft:,[Draft],WIP:,[WIP]
One possible workaround was:
WORK_IN_PROGRESS_PREFIXES=DRAFT:,[DRAFT],WIP:,[WIP]
Then indeed one could use `Draft` (as well as `DRAFT`) in the title.
However, the link `Start the title with DRAFT: to prevent the pull request
from being merged accidentally.` showed the suggestion in uppercase; so
it is not possible to show it as `Draft`. This PR fixes it, and allows
to use `Draft` in `WORK_IN_PROGRESS_PREFIXES`.
Fixes #19779.
Co-authored-by: zeripath <art27@cantab.net>
2022-05-26 14:49:24 +05:30
if strings . HasPrefix ( strings . ToUpper ( title ) , strings . ToUpper ( prefix ) ) {
2018-08-14 00:34:39 +05:30
return true
}
}
return false
}
2019-02-05 17:24:49 +05:30
// IsFilesConflicted determines if the Pull Request has changes conflicting with the target branch.
func ( pr * PullRequest ) IsFilesConflicted ( ) bool {
return len ( pr . ConflictedFiles ) > 0
}
2018-08-14 00:34:39 +05:30
// GetWorkInProgressPrefix returns the prefix used to mark the pull request as a work in progress.
// It returns an empty string when none were found
func ( pr * PullRequest ) GetWorkInProgressPrefix ( ) string {
if err := pr . LoadIssue ( ) ; err != nil {
2019-04-02 13:18:31 +05:30
log . Error ( "LoadIssue: %v" , err )
2018-08-14 00:34:39 +05:30
return ""
}
for _ , prefix := range setting . Repository . PullRequest . WorkInProgressPrefixes {
Make WIP prefixes case insensitive, e.g. allow `Draft` as a WIP prefix (#19780)
The issue was that only the actual title was converted to uppercase, but
not the prefix as specified in `WORK_IN_PROGRESS_PREFIXES`. As a result,
the following did not work:
WORK_IN_PROGRESS_PREFIXES=Draft:,[Draft],WIP:,[WIP]
One possible workaround was:
WORK_IN_PROGRESS_PREFIXES=DRAFT:,[DRAFT],WIP:,[WIP]
Then indeed one could use `Draft` (as well as `DRAFT`) in the title.
However, the link `Start the title with DRAFT: to prevent the pull request
from being merged accidentally.` showed the suggestion in uppercase; so
it is not possible to show it as `Draft`. This PR fixes it, and allows
to use `Draft` in `WORK_IN_PROGRESS_PREFIXES`.
Fixes #19779.
Co-authored-by: zeripath <art27@cantab.net>
2022-05-26 14:49:24 +05:30
if strings . HasPrefix ( strings . ToUpper ( pr . Issue . Title ) , strings . ToUpper ( prefix ) ) {
2018-08-14 00:34:39 +05:30
return pr . Issue . Title [ 0 : len ( prefix ) ]
}
}
return ""
}
2019-12-16 11:50:25 +05:30
2020-04-14 19:23:34 +05:30
// UpdateCommitDivergence update Divergence of a pull request
2022-05-20 19:38:52 +05:30
func ( pr * PullRequest ) UpdateCommitDivergence ( ctx context . Context , ahead , behind int ) error {
2020-04-14 19:23:34 +05:30
if pr . ID == 0 {
return fmt . Errorf ( "pull ID is 0" )
}
pr . CommitsAhead = ahead
pr . CommitsBehind = behind
2022-05-20 19:38:52 +05:30
_ , err := db . GetEngine ( ctx ) . ID ( pr . ID ) . Cols ( "commits_ahead" , "commits_behind" ) . Update ( pr )
2020-04-14 19:23:34 +05:30
return err
}
2020-01-17 11:33:40 +05:30
// IsSameRepo returns true if base repo and head repo is the same
func ( pr * PullRequest ) IsSameRepo ( ) bool {
return pr . BaseRepoID == pr . HeadRepoID
}
2021-02-18 08:15:49 +05:30
2022-05-07 22:35:52 +05:30
// GetPullRequestsByHeadBranch returns all prs by head branch
// Since there could be multiple prs with the same head branch, this function returns a slice of prs
func GetPullRequestsByHeadBranch ( ctx context . Context , headBranch string , headRepoID int64 ) ( [ ] * PullRequest , error ) {
log . Trace ( "GetPullRequestsByHeadBranch: headBranch: '%s', headRepoID: '%d'" , headBranch , headRepoID )
prs := make ( [ ] * PullRequest , 0 , 2 )
if err := db . GetEngine ( ctx ) . Where ( builder . Eq { "head_branch" : headBranch , "head_repo_id" : headRepoID } ) .
Find ( & prs ) ; err != nil {
return nil , err
}
return prs , nil
}
2021-02-18 08:15:49 +05:30
// GetBaseBranchHTMLURL returns the HTML URL of the base branch
func ( pr * PullRequest ) GetBaseBranchHTMLURL ( ) string {
if err := pr . LoadBaseRepo ( ) ; err != nil {
log . Error ( "LoadBaseRepo: %v" , err )
return ""
}
if pr . BaseRepo == nil {
return ""
}
return pr . BaseRepo . HTMLURL ( ) + "/src/branch/" + util . PathEscapeSegments ( pr . BaseBranch )
}
// GetHeadBranchHTMLURL returns the HTML URL of the head branch
func ( pr * PullRequest ) GetHeadBranchHTMLURL ( ) string {
2021-07-28 15:12:56 +05:30
if pr . Flow == PullRequestFlowAGit {
return ""
}
2021-02-18 08:15:49 +05:30
if err := pr . LoadHeadRepo ( ) ; err != nil {
log . Error ( "LoadHeadRepo: %v" , err )
return ""
}
if pr . HeadRepo == nil {
return ""
}
return pr . HeadRepo . HTMLURL ( ) + "/src/branch/" + util . PathEscapeSegments ( pr . HeadBranch )
}
2022-04-22 03:25:45 +05:30
2022-04-28 21:15:33 +05:30
// UpdateAllowEdits update if PR can be edited from maintainers
func UpdateAllowEdits ( ctx context . Context , pr * PullRequest ) error {
if _ , err := db . GetEngine ( ctx ) . ID ( pr . ID ) . Cols ( "allow_maintainer_edit" ) . Update ( pr ) ; err != nil {
return err
}
return nil
}
2022-04-22 03:25:45 +05:30
// Mergeable returns if the pullrequest is mergeable.
func ( pr * PullRequest ) Mergeable ( ) bool {
// If a pull request isn't mergable if it's:
// - Being conflict checked.
// - Has a conflict.
// - Received a error while being conflict checked.
// - Is a work-in-progress pull request.
return pr . Status != PullRequestStatusChecking && pr . Status != PullRequestStatusConflict &&
pr . Status != PullRequestStatusError && ! pr . IsWorkInProgress ( )
}