2019-09-27 05:52:36 +05:30
// Copyright 2019 The Gitea Authors. All rights reserved.
2022-11-27 23:50:29 +05:30
// SPDX-License-Identifier: MIT
2019-09-27 05:52:36 +05:30
package pull
import (
2020-01-09 07:17:45 +05:30
"bufio"
"bytes"
2019-12-15 15:21:28 +05:30
"context"
2019-09-27 05:52:36 +05:30
"fmt"
2022-01-20 04:56:57 +05:30
"io"
2021-06-25 22:31:43 +05:30
"regexp"
2020-01-09 07:17:45 +05:30
"strings"
"time"
2019-09-27 05:52:36 +05:30
"code.gitea.io/gitea/models"
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-06-13 15:07:59 +05:30
issues_model "code.gitea.io/gitea/models/issues"
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"
2022-10-12 10:48:26 +05:30
"code.gitea.io/gitea/modules/container"
2019-10-15 08:58:40 +05:30
"code.gitea.io/gitea/modules/git"
2019-12-15 15:21:28 +05:30
"code.gitea.io/gitea/modules/graceful"
2021-07-24 21:33:58 +05:30
"code.gitea.io/gitea/modules/json"
2019-09-27 05:52:36 +05:30
"code.gitea.io/gitea/modules/log"
2019-11-04 02:29:09 +05:30
"code.gitea.io/gitea/modules/notification"
2022-01-20 04:56:57 +05:30
"code.gitea.io/gitea/modules/process"
2022-05-08 22:16:32 +05:30
repo_module "code.gitea.io/gitea/modules/repository"
2020-04-10 16:56:37 +05:30
"code.gitea.io/gitea/modules/setting"
2022-05-04 21:36:23 +05:30
"code.gitea.io/gitea/modules/sync"
2019-10-28 22:15:43 +05:30
issue_service "code.gitea.io/gitea/services/issue"
2019-09-27 05:52:36 +05:30
)
2022-05-04 21:36:23 +05:30
// TODO: use clustered lock (unique queue? or *abuse* cache)
var pullWorkingPool = sync . NewExclusivePool ( )
2019-09-27 05:52:36 +05:30
// NewPullRequest creates new pull request with labels for repository.
2022-06-13 15:07:59 +05:30
func NewPullRequest ( ctx context . Context , repo * repo_model . Repository , pull * issues_model . Issue , labelIDs [ ] int64 , uuids [ ] string , pr * issues_model . PullRequest , assigneeIDs [ ] int64 ) error {
2019-12-14 03:51:06 +05:30
if err := TestPatch ( pr ) ; err != nil {
return err
}
2022-01-20 04:56:57 +05:30
divergence , err := GetDiverging ( ctx , pr )
2020-04-14 19:23:34 +05:30
if err != nil {
return err
}
pr . CommitsAhead = divergence . Ahead
pr . CommitsBehind = divergence . Behind
2022-06-13 15:07:59 +05:30
if err := issues_model . NewPullRequest ( ctx , repo , pull , labelIDs , uuids , pr ) ; err != nil {
2019-09-27 05:52:36 +05:30
return err
}
2019-10-28 22:15:43 +05:30
for _ , assigneeID := range assigneeIDs {
if err := issue_service . AddAssigneeIfNotAssigned ( pull , pull . Poster , assigneeID ) ; err != nil {
return err
}
}
2019-09-27 05:52:36 +05:30
pr . Issue = pull
pull . PullRequest = pr
2019-11-04 02:29:09 +05:30
2022-01-20 04:56:57 +05:30
// Now - even if the request context has been cancelled as the PR has been created
// in the db and there is no way to cancel that transaction we have to proceed - therefore
// create new context and work from there
prCtx , _ , finished := process . GetManager ( ) . AddContext ( graceful . GetManager ( ) . HammerContext ( ) , fmt . Sprintf ( "NewPullRequest: %s:%d" , repo . FullName ( ) , pr . Index ) )
defer finished ( )
2022-06-13 15:07:59 +05:30
if pr . Flow == issues_model . PullRequestFlowGithub {
2022-01-20 04:56:57 +05:30
err = PushToBaseRepo ( prCtx , pr )
2021-07-28 15:12:56 +05:30
} else {
2022-01-20 04:56:57 +05:30
err = UpdateRef ( prCtx , pr )
2021-07-28 15:12:56 +05:30
}
if err != nil {
2019-12-15 08:58:51 +05:30
return err
}
2022-06-13 15:07:59 +05:30
mentions , err := issues_model . FindAndUpdateIssueMentions ( ctx , pull , pull . Poster , pull . Content )
2021-01-02 22:34:02 +05:30
if err != nil {
return err
}
2022-11-19 13:42:33 +05:30
notification . NotifyNewPullRequest ( prCtx , pr , mentions )
2021-01-17 19:45:57 +05:30
if len ( pull . Labels ) > 0 {
2022-11-19 13:42:33 +05:30
notification . NotifyIssueChangeLabels ( prCtx , pull . Poster , pull , pull . Labels , nil )
2021-01-17 19:45:57 +05:30
}
if pull . Milestone != nil {
2022-11-19 13:42:33 +05:30
notification . NotifyIssueChangeMilestone ( prCtx , pull . Poster , pull , 0 )
2021-01-17 19:45:57 +05:30
}
2019-09-27 05:52:36 +05:30
2020-05-20 18:17:24 +05:30
// add first push codes comment
2022-03-30 00:43:41 +05:30
baseGitRepo , err := git . OpenRepository ( prCtx , pr . BaseRepo . RepoPath ( ) )
2020-05-20 18:17:24 +05:30
if err != nil {
return err
}
defer baseGitRepo . Close ( )
compareInfo , err := baseGitRepo . GetCompareInfo ( pr . BaseRepo . RepoPath ( ) ,
2022-01-18 13:15:43 +05:30
git . BranchPrefix + pr . BaseBranch , pr . GetGitRefName ( ) , false , false )
2020-05-20 18:17:24 +05:30
if err != nil {
return err
}
2021-08-09 23:38:51 +05:30
if len ( compareInfo . Commits ) > 0 {
2022-06-13 15:07:59 +05:30
data := issues_model . PushActionContent { IsForcePush : false }
2021-08-09 23:38:51 +05:30
data . CommitIDs = make ( [ ] string , 0 , len ( compareInfo . Commits ) )
for i := len ( compareInfo . Commits ) - 1 ; i >= 0 ; i -- {
data . CommitIDs = append ( data . CommitIDs , compareInfo . Commits [ i ] . ID . String ( ) )
2020-05-20 18:17:24 +05:30
}
dataJSON , err := json . Marshal ( data )
if err != nil {
return err
}
2022-06-13 15:07:59 +05:30
ops := & issues_model . CreateCommentOptions {
Type : issues_model . CommentTypePullRequestPush ,
2020-05-20 18:17:24 +05:30
Doer : pull . Poster ,
Repo : repo ,
Issue : pr . Issue ,
IsForcePush : false ,
Content : string ( dataJSON ) ,
}
2022-06-13 15:07:59 +05:30
_ , _ = issues_model . CreateComment ( ops )
2020-05-20 18:17:24 +05:30
}
2019-09-27 05:52:36 +05:30
return nil
}
2019-10-15 08:58:40 +05:30
2019-12-16 11:50:25 +05:30
// ChangeTargetBranch changes the target branch of this pull request, as the given user.
2022-06-13 15:07:59 +05:30
func ChangeTargetBranch ( ctx context . Context , pr * issues_model . PullRequest , doer * user_model . User , targetBranch string ) ( err error ) {
2022-05-04 21:36:23 +05:30
pullWorkingPool . CheckIn ( fmt . Sprint ( pr . ID ) )
defer pullWorkingPool . CheckOut ( fmt . Sprint ( pr . ID ) )
2019-12-16 11:50:25 +05:30
// Current target branch is already the same
if pr . BaseBranch == targetBranch {
return nil
}
if pr . Issue . IsClosed {
2022-06-13 15:07:59 +05:30
return issues_model . ErrIssueIsClosed {
2019-12-16 11:50:25 +05:30
ID : pr . Issue . ID ,
RepoID : pr . Issue . RepoID ,
Index : pr . Issue . Index ,
}
}
if pr . HasMerged {
return models . ErrPullRequestHasMerged {
ID : pr . ID ,
IssueID : pr . Index ,
HeadRepoID : pr . HeadRepoID ,
BaseRepoID : pr . BaseRepoID ,
HeadBranch : pr . HeadBranch ,
BaseBranch : pr . BaseBranch ,
}
}
// Check if branches are equal
2022-01-20 04:56:57 +05:30
branchesEqual , err := IsHeadEqualWithBranch ( ctx , pr , targetBranch )
2019-12-16 11:50:25 +05:30
if err != nil {
return err
}
if branchesEqual {
return models . ErrBranchesEqual {
HeadBranchName : pr . HeadBranch ,
BaseBranchName : targetBranch ,
}
}
// Check if pull request for the new target branch already exists
2022-11-19 13:42:33 +05:30
existingPr , err := issues_model . GetUnmergedPullRequest ( ctx , pr . HeadRepoID , pr . BaseRepoID , pr . HeadBranch , targetBranch , issues_model . PullRequestFlowGithub )
2019-12-16 11:50:25 +05:30
if existingPr != nil {
2022-06-13 15:07:59 +05:30
return issues_model . ErrPullRequestAlreadyExists {
2019-12-16 11:50:25 +05:30
ID : existingPr . ID ,
IssueID : existingPr . Index ,
HeadRepoID : existingPr . HeadRepoID ,
BaseRepoID : existingPr . BaseRepoID ,
HeadBranch : existingPr . HeadBranch ,
BaseBranch : existingPr . BaseBranch ,
}
}
2022-06-13 15:07:59 +05:30
if err != nil && ! issues_model . IsErrPullRequestNotExist ( err ) {
2019-12-16 11:50:25 +05:30
return err
}
// Set new target branch
oldBranch := pr . BaseBranch
pr . BaseBranch = targetBranch
// Refresh patch
if err := TestPatch ( pr ) ; err != nil {
return err
}
// Update target branch, PR diff and status
// This is the same as checkAndUpdateStatus in check service, but also updates base_branch
2022-06-13 15:07:59 +05:30
if pr . Status == issues_model . PullRequestStatusChecking {
pr . Status = issues_model . PullRequestStatusMergeable
2019-12-16 11:50:25 +05:30
}
2020-06-16 23:22:33 +05:30
// Update Commit Divergence
2022-01-20 04:56:57 +05:30
divergence , err := GetDiverging ( ctx , pr )
2020-06-16 23:22:33 +05:30
if err != nil {
return err
}
pr . CommitsAhead = divergence . Ahead
pr . CommitsBehind = divergence . Behind
2022-11-19 13:42:33 +05:30
if err := pr . UpdateColsIfNotMerged ( ctx , "merge_base" , "status" , "conflicted_files" , "changed_protected_files" , "base_branch" , "commits_ahead" , "commits_behind" ) ; err != nil {
2019-12-16 11:50:25 +05:30
return err
}
// Create comment
2022-06-13 15:07:59 +05:30
options := & issues_model . CreateCommentOptions {
Type : issues_model . CommentTypeChangeTargetBranch ,
2019-12-16 11:50:25 +05:30
Doer : doer ,
Repo : pr . Issue . Repo ,
Issue : pr . Issue ,
OldRef : oldBranch ,
NewRef : targetBranch ,
}
2022-06-13 15:07:59 +05:30
if _ , err = issues_model . CreateComment ( options ) ; err != nil {
2022-10-25 00:59:17 +05:30
return fmt . Errorf ( "CreateChangeTargetBranchComment: %w" , err )
2019-12-16 11:50:25 +05:30
}
return nil
}
2022-06-13 15:07:59 +05:30
func checkForInvalidation ( ctx context . Context , requests issues_model . PullRequestList , repoID int64 , doer * user_model . User , branch string ) error {
2022-12-03 08:18:26 +05:30
repo , err := repo_model . GetRepositoryByID ( ctx , repoID )
2019-10-15 08:58:40 +05:30
if err != nil {
2022-11-19 13:42:33 +05:30
return fmt . Errorf ( "GetRepositoryByIDCtx: %w" , err )
2019-10-15 08:58:40 +05:30
}
2022-03-30 00:43:41 +05:30
gitRepo , err := git . OpenRepository ( ctx , repo . RepoPath ( ) )
2019-10-15 08:58:40 +05:30
if err != nil {
2022-10-25 00:59:17 +05:30
return fmt . Errorf ( "git.OpenRepository: %w" , err )
2019-10-15 08:58:40 +05:30
}
go func ( ) {
2019-12-15 15:21:28 +05:30
// FIXME: graceful: We need to tell the manager we're doing something...
2022-04-28 17:18:48 +05:30
err := requests . InvalidateCodeComments ( ctx , doer , gitRepo , branch )
2019-10-15 08:58:40 +05:30
if err != nil {
log . Error ( "PullRequestList.InvalidateCodeComments: %v" , err )
}
2019-11-13 12:31:19 +05:30
gitRepo . Close ( )
2019-10-15 08:58:40 +05:30
} ( )
return nil
}
// AddTestPullRequestTask adds new test tasks by given head/base repository and head/base branch,
// and generate new patch for testing as needed.
2021-11-24 15:19:20 +05:30
func AddTestPullRequestTask ( doer * user_model . User , repoID int64 , branch string , isSync bool , oldCommitID , newCommitID string ) {
2019-10-15 08:58:40 +05:30
log . Trace ( "AddTestPullRequestTask [head_repo_id: %d, head_branch: %s]: finding pull requests" , repoID , branch )
2019-12-15 15:21:28 +05:30
graceful . GetManager ( ) . RunWithShutdownContext ( func ( ctx context . Context ) {
// There is no sensible way to shut this down ":-("
// If you don't let it run all the way then you will lose data
2022-05-07 22:35:52 +05:30
// TODO: graceful: AddTestPullRequestTask needs to become a queue!
2019-10-15 08:58:40 +05:30
2022-06-13 15:07:59 +05:30
prs , err := issues_model . GetUnmergedPullRequestsByHeadInfo ( repoID , branch )
2019-12-15 15:21:28 +05:30
if err != nil {
log . Error ( "Find pull requests [head_repo_id: %d, head_branch: %s]: %v" , repoID , branch , err )
return
2019-10-15 08:58:40 +05:30
}
2019-12-15 15:21:28 +05:30
if isSync {
2022-06-13 15:07:59 +05:30
requests := issues_model . PullRequestList ( prs )
2019-12-15 15:21:28 +05:30
if err = requests . LoadAttributes ( ) ; err != nil {
log . Error ( "PullRequestList.LoadAttributes: %v" , err )
}
2022-01-20 04:56:57 +05:30
if invalidationErr := checkForInvalidation ( ctx , requests , repoID , doer , branch ) ; invalidationErr != nil {
2019-12-15 15:21:28 +05:30
log . Error ( "checkForInvalidation: %v" , invalidationErr )
}
if err == nil {
for _ , pr := range prs {
2020-01-09 07:17:45 +05:30
if newCommitID != "" && newCommitID != git . EmptySHA {
2022-01-20 04:56:57 +05:30
changed , err := checkIfPRContentChanged ( ctx , pr , oldCommitID , newCommitID )
2020-01-09 07:17:45 +05:30
if err != nil {
log . Error ( "checkIfPRContentChanged: %v" , err )
}
if changed {
// Mark old reviews as stale if diff to mergebase has changed
2022-06-13 15:07:59 +05:30
if err := issues_model . MarkReviewsAsStale ( pr . IssueID ) ; err != nil {
2020-01-09 07:17:45 +05:30
log . Error ( "MarkReviewsAsStale: %v" , err )
}
}
2022-06-13 15:07:59 +05:30
if err := issues_model . MarkReviewsAsNotStale ( pr . IssueID , newCommitID ) ; err != nil {
2020-01-09 07:17:45 +05:30
log . Error ( "MarkReviewsAsNotStale: %v" , err )
}
2022-01-20 04:56:57 +05:30
divergence , err := GetDiverging ( ctx , pr )
2020-04-14 19:23:34 +05:30
if err != nil {
log . Error ( "GetDiverging: %v" , err )
} else {
2022-05-20 19:38:52 +05:30
err = pr . UpdateCommitDivergence ( ctx , divergence . Ahead , divergence . Behind )
2020-04-14 19:23:34 +05:30
if err != nil {
log . Error ( "UpdateCommitDivergence: %v" , err )
}
}
2020-01-09 07:17:45 +05:30
}
2019-12-15 15:21:28 +05:30
pr . Issue . PullRequest = pr
2022-11-19 13:42:33 +05:30
notification . NotifyPullRequestSynchronized ( ctx , doer , pr )
2019-12-15 15:21:28 +05:30
}
2019-10-15 08:58:40 +05:30
}
}
2020-05-20 18:17:24 +05:30
for _ , pr := range prs {
2020-09-15 09:02:31 +05:30
log . Trace ( "Updating PR[%d]: composing new test task" , pr . ID )
2022-06-13 15:07:59 +05:30
if pr . Flow == issues_model . PullRequestFlowGithub {
2022-01-20 04:56:57 +05:30
if err := PushToBaseRepo ( ctx , pr ) ; err != nil {
2021-07-28 15:12:56 +05:30
log . Error ( "PushToBaseRepo: %v" , err )
continue
}
} else {
2020-09-15 09:02:31 +05:30
continue
}
AddToTaskQueue ( pr )
2022-06-13 15:07:59 +05:30
comment , err := issues_model . CreatePushPullComment ( ctx , doer , pr , oldCommitID , newCommitID )
2020-05-20 18:17:24 +05:30
if err == nil && comment != nil {
2022-11-19 13:42:33 +05:30
notification . NotifyPullRequestPushCommits ( ctx , doer , pr , comment )
2020-05-20 18:17:24 +05:30
}
}
2019-10-15 08:58:40 +05:30
2019-12-15 15:21:28 +05:30
log . Trace ( "AddTestPullRequestTask [base_repo_id: %d, base_branch: %s]: finding pull requests" , repoID , branch )
2022-06-13 15:07:59 +05:30
prs , err = issues_model . GetUnmergedPullRequestsByBaseInfo ( repoID , branch )
2019-12-15 15:21:28 +05:30
if err != nil {
log . Error ( "Find pull requests [base_repo_id: %d, base_branch: %s]: %v" , repoID , branch , err )
return
}
for _ , pr := range prs {
2022-01-20 04:56:57 +05:30
divergence , err := GetDiverging ( ctx , pr )
2020-04-14 19:23:34 +05:30
if err != nil {
2021-12-01 01:36:32 +05:30
if models . IsErrBranchDoesNotExist ( err ) && ! git . IsBranchExist ( ctx , pr . HeadRepo . RepoPath ( ) , pr . HeadBranch ) {
2021-07-13 04:56:25 +05:30
log . Warn ( "Cannot test PR %s/%d: head_branch %s no longer exists" , pr . BaseRepo . Name , pr . IssueID , pr . HeadBranch )
} else {
log . Error ( "GetDiverging: %v" , err )
}
2020-04-14 19:23:34 +05:30
} else {
2022-05-20 19:38:52 +05:30
err = pr . UpdateCommitDivergence ( ctx , divergence . Ahead , divergence . Behind )
2020-04-14 19:23:34 +05:30
if err != nil {
log . Error ( "UpdateCommitDivergence: %v" , err )
}
}
2019-12-15 15:21:28 +05:30
AddToTaskQueue ( pr )
}
} )
2019-10-15 08:58:40 +05:30
}
2019-12-15 08:58:51 +05:30
2020-01-09 07:17:45 +05:30
// checkIfPRContentChanged checks if diff to target branch has changed by push
// A commit can be considered to leave the PR untouched if the patch/diff with its merge base is unchanged
2022-06-13 15:07:59 +05:30
func checkIfPRContentChanged ( ctx context . Context , pr * issues_model . PullRequest , oldCommitID , newCommitID string ) ( hasChanged bool , err error ) {
2022-11-19 13:42:33 +05:30
if err = pr . LoadHeadRepo ( ctx ) ; err != nil {
2022-10-25 00:59:17 +05:30
return false , fmt . Errorf ( "LoadHeadRepo: %w" , err )
2020-01-09 07:17:45 +05:30
} else if pr . HeadRepo == nil {
// corrupt data assumed changed
return true , nil
}
2022-11-19 13:42:33 +05:30
if err = pr . LoadBaseRepo ( ctx ) ; err != nil {
2022-10-25 00:59:17 +05:30
return false , fmt . Errorf ( "LoadBaseRepo: %w" , err )
2020-01-09 07:17:45 +05:30
}
2022-03-30 00:43:41 +05:30
headGitRepo , err := git . OpenRepository ( ctx , pr . HeadRepo . RepoPath ( ) )
2020-01-09 07:17:45 +05:30
if err != nil {
2022-10-25 00:59:17 +05:30
return false , fmt . Errorf ( "OpenRepository: %w" , err )
2020-01-09 07:17:45 +05:30
}
defer headGitRepo . Close ( )
// Add a temporary remote.
2020-12-25 15:29:32 +05:30
tmpRemote := "checkIfPRContentChanged-" + fmt . Sprint ( time . Now ( ) . UnixNano ( ) )
2020-01-12 15:06:21 +05:30
if err = headGitRepo . AddRemote ( tmpRemote , pr . BaseRepo . RepoPath ( ) , true ) ; err != nil {
2022-10-25 00:59:17 +05:30
return false , fmt . Errorf ( "AddRemote: %s/%s-%s: %w" , pr . HeadRepo . OwnerName , pr . HeadRepo . Name , tmpRemote , err )
2020-01-09 07:17:45 +05:30
}
defer func ( ) {
if err := headGitRepo . RemoveRemote ( tmpRemote ) ; err != nil {
log . Error ( "checkIfPRContentChanged: RemoveRemote: %s/%s-%s: %v" , pr . HeadRepo . OwnerName , pr . HeadRepo . Name , tmpRemote , err )
}
} ( )
// To synchronize repo and get a base ref
_ , base , err := headGitRepo . GetMergeBase ( tmpRemote , pr . BaseBranch , pr . HeadBranch )
if err != nil {
2022-10-25 00:59:17 +05:30
return false , fmt . Errorf ( "GetMergeBase: %w" , err )
2020-01-09 07:17:45 +05:30
}
diffBefore := & bytes . Buffer { }
diffAfter := & bytes . Buffer { }
if err := headGitRepo . GetDiffFromMergeBase ( base , oldCommitID , diffBefore ) ; err != nil {
// If old commit not found, assume changed.
log . Debug ( "GetDiffFromMergeBase: %v" , err )
return true , nil
}
if err := headGitRepo . GetDiffFromMergeBase ( base , newCommitID , diffAfter ) ; err != nil {
// New commit should be found
2022-10-25 00:59:17 +05:30
return false , fmt . Errorf ( "GetDiffFromMergeBase: %w" , err )
2020-01-09 07:17:45 +05:30
}
diffBeforeLines := bufio . NewScanner ( diffBefore )
diffAfterLines := bufio . NewScanner ( diffAfter )
for diffBeforeLines . Scan ( ) && diffAfterLines . Scan ( ) {
if strings . HasPrefix ( diffBeforeLines . Text ( ) , "index" ) && strings . HasPrefix ( diffAfterLines . Text ( ) , "index" ) {
// file hashes can change without the diff changing
continue
} else if strings . HasPrefix ( diffBeforeLines . Text ( ) , "@@" ) && strings . HasPrefix ( diffAfterLines . Text ( ) , "@@" ) {
// the location of the difference may change
continue
} else if ! bytes . Equal ( diffBeforeLines . Bytes ( ) , diffAfterLines . Bytes ( ) ) {
return true , nil
}
}
if diffBeforeLines . Scan ( ) || diffAfterLines . Scan ( ) {
// Diffs not of equal length
return true , nil
}
return false , nil
}
2019-12-15 08:58:51 +05:30
// PushToBaseRepo pushes commits from branches of head repository to
// corresponding branches of base repository.
// FIXME: Only push branches that are actually updates?
2022-06-13 15:07:59 +05:30
func PushToBaseRepo ( ctx context . Context , pr * issues_model . PullRequest ) ( err error ) {
2022-01-20 04:56:57 +05:30
return pushToBaseRepoHelper ( ctx , pr , "" )
2021-06-24 02:38:26 +05:30
}
2022-06-13 15:07:59 +05:30
func pushToBaseRepoHelper ( ctx context . Context , pr * issues_model . PullRequest , prefixHeadBranch string ) ( err error ) {
2019-12-15 08:58:51 +05:30
log . Trace ( "PushToBaseRepo[%d]: pushing commits to base repo '%s'" , pr . BaseRepoID , pr . GetGitRefName ( ) )
2022-11-19 13:42:33 +05:30
if err := pr . LoadHeadRepo ( ctx ) ; err != nil {
2020-02-21 23:48:13 +05:30
log . Error ( "Unable to load head repository for PR[%d] Error: %v" , pr . ID , err )
return err
}
2019-12-15 08:58:51 +05:30
headRepoPath := pr . HeadRepo . RepoPath ( )
2020-01-28 15:53:58 +05:30
2022-11-19 13:42:33 +05:30
if err := pr . LoadBaseRepo ( ctx ) ; err != nil {
2020-02-21 23:48:13 +05:30
log . Error ( "Unable to load base repository for PR[%d] Error: %v" , pr . ID , err )
return err
}
2020-09-15 09:02:31 +05:30
baseRepoPath := pr . BaseRepo . RepoPath ( )
2019-12-15 08:58:51 +05:30
2022-11-19 13:42:33 +05:30
if err = pr . LoadIssue ( ctx ) ; err != nil {
2022-10-25 00:59:17 +05:30
return fmt . Errorf ( "unable to load issue %d for pr %d: %w" , pr . IssueID , pr . ID , err )
2019-12-28 02:45:04 +05:30
}
2022-11-19 13:42:33 +05:30
if err = pr . Issue . LoadPoster ( ctx ) ; err != nil {
2022-10-25 00:59:17 +05:30
return fmt . Errorf ( "unable to load poster %d for pr %d: %w" , pr . Issue . PosterID , pr . ID , err )
2019-12-28 02:45:04 +05:30
}
2020-09-15 09:02:31 +05:30
gitRefName := pr . GetGitRefName ( )
2022-01-20 04:56:57 +05:30
if err := git . Push ( ctx , headRepoPath , git . PushOptions {
2020-09-15 09:02:31 +05:30
Remote : baseRepoPath ,
2021-06-24 02:38:26 +05:30
Branch : prefixHeadBranch + pr . HeadBranch + ":" + gitRefName ,
2019-12-15 08:58:51 +05:30
Force : true ,
2019-12-28 02:45:04 +05:30
// Use InternalPushingEnvironment here because we know that pre-receive and post-receive do not run on a refs/pulls/...
2022-05-08 22:16:32 +05:30
Env : repo_module . InternalPushingEnvironment ( pr . Issue . Poster , pr . BaseRepo ) ,
2019-12-15 08:58:51 +05:30
} ) ; err != nil {
2020-06-08 23:37:41 +05:30
if git . IsErrPushOutOfDate ( err ) {
// This should not happen as we're using force!
2020-09-15 09:02:31 +05:30
log . Error ( "Unable to push PR head for %s#%d (%-v:%s) due to ErrPushOfDate: %v" , pr . BaseRepo . FullName ( ) , pr . Index , pr . BaseRepo , gitRefName , err )
2020-06-08 23:37:41 +05:30
return err
} else if git . IsErrPushRejected ( err ) {
rejectErr := err . ( * git . ErrPushRejected )
2020-09-15 09:02:31 +05:30
log . Info ( "Unable to push PR head for %s#%d (%-v:%s) due to rejection:\nStdout: %s\nStderr: %s\nError: %v" , pr . BaseRepo . FullName ( ) , pr . Index , pr . BaseRepo , gitRefName , rejectErr . StdOut , rejectErr . StdErr , rejectErr . Err )
2020-06-08 23:37:41 +05:30
return err
2021-06-24 02:38:26 +05:30
} else if git . IsErrMoreThanOne ( err ) {
if prefixHeadBranch != "" {
log . Info ( "Can't push with %s%s" , prefixHeadBranch , pr . HeadBranch )
return err
}
2022-01-20 04:56:57 +05:30
log . Info ( "Retrying to push with %s%s" , git . BranchPrefix , pr . HeadBranch )
err = pushToBaseRepoHelper ( ctx , pr , git . BranchPrefix )
2021-06-24 02:38:26 +05:30
return err
2020-06-08 23:37:41 +05:30
}
2020-09-15 09:02:31 +05:30
log . Error ( "Unable to push PR head for %s#%d (%-v:%s) due to Error: %v" , pr . BaseRepo . FullName ( ) , pr . Index , pr . BaseRepo , gitRefName , err )
2022-10-25 00:59:17 +05:30
return fmt . Errorf ( "Push: %s:%s %s:%s %w" , pr . HeadRepo . FullName ( ) , pr . HeadBranch , pr . BaseRepo . FullName ( ) , gitRefName , err )
2019-12-15 08:58:51 +05:30
}
return nil
}
2020-01-25 08:18:22 +05:30
2021-07-28 15:12:56 +05:30
// UpdateRef update refs/pull/id/head directly for agit flow pull request
2022-06-13 15:07:59 +05:30
func UpdateRef ( ctx context . Context , pr * issues_model . PullRequest ) ( err error ) {
2021-07-28 15:12:56 +05:30
log . Trace ( "UpdateRef[%d]: upgate pull request ref in base repo '%s'" , pr . ID , pr . GetGitRefName ( ) )
2022-11-19 13:42:33 +05:30
if err := pr . LoadBaseRepo ( ctx ) ; err != nil {
2021-07-28 15:12:56 +05:30
log . Error ( "Unable to load base repository for PR[%d] Error: %v" , pr . ID , err )
return err
}
2022-10-23 20:14:45 +05:30
_ , _ , err = git . NewCommand ( ctx , "update-ref" ) . AddDynamicArguments ( pr . GetGitRefName ( ) , pr . HeadCommitID ) . RunStdString ( & git . RunOpts { Dir : pr . BaseRepo . RepoPath ( ) } )
2021-07-28 15:12:56 +05:30
if err != nil {
log . Error ( "Unable to update ref in base repository for PR[%d] Error: %v" , pr . ID , err )
}
return err
}
2020-01-25 08:18:22 +05:30
type errlist [ ] error
func ( errs errlist ) Error ( ) string {
if len ( errs ) > 0 {
var buf strings . Builder
for i , err := range errs {
if i > 0 {
buf . WriteString ( ", " )
}
buf . WriteString ( err . Error ( ) )
}
return buf . String ( )
}
return ""
}
// CloseBranchPulls close all the pull requests who's head branch is the branch
2021-11-24 15:19:20 +05:30
func CloseBranchPulls ( doer * user_model . User , repoID int64 , branch string ) error {
2022-06-13 15:07:59 +05:30
prs , err := issues_model . GetUnmergedPullRequestsByHeadInfo ( repoID , branch )
2020-01-25 08:18:22 +05:30
if err != nil {
return err
}
2022-06-13 15:07:59 +05:30
prs2 , err := issues_model . GetUnmergedPullRequestsByBaseInfo ( repoID , branch )
2020-01-25 08:18:22 +05:30
if err != nil {
return err
}
prs = append ( prs , prs2 ... )
2022-06-13 15:07:59 +05:30
if err := issues_model . PullRequestList ( prs ) . LoadAttributes ( ) ; err != nil {
2020-01-25 08:18:22 +05:30
return err
}
var errs errlist
for _ , pr := range prs {
2022-06-13 15:07:59 +05:30
if err = issue_service . ChangeStatus ( pr . Issue , doer , true ) ; err != nil && ! issues_model . IsErrPullWasClosed ( err ) && ! issues_model . IsErrDependenciesLeft ( err ) {
2020-01-25 08:18:22 +05:30
errs = append ( errs , err )
}
}
if len ( errs ) > 0 {
return errs
}
return nil
}
2021-03-01 23:09:44 +05:30
// CloseRepoBranchesPulls close all pull requests which head branches are in the given repository, but only whose base repo is not in the given repository
2022-01-20 04:56:57 +05:30
func CloseRepoBranchesPulls ( ctx context . Context , doer * user_model . User , repo * repo_model . Repository ) error {
branches , _ , err := git . GetBranchesByPath ( ctx , repo . RepoPath ( ) , 0 , 0 )
2020-01-25 08:18:22 +05:30
if err != nil {
return err
}
var errs errlist
for _ , branch := range branches {
2022-06-13 15:07:59 +05:30
prs , err := issues_model . GetUnmergedPullRequestsByHeadInfo ( repo . ID , branch . Name )
2020-01-25 08:18:22 +05:30
if err != nil {
return err
}
2022-06-13 15:07:59 +05:30
if err = issues_model . PullRequestList ( prs ) . LoadAttributes ( ) ; err != nil {
2020-01-25 08:18:22 +05:30
return err
}
for _ , pr := range prs {
2021-03-01 23:09:44 +05:30
// If the base repository for this pr is this repository there is no need to close it
// as it is going to be deleted anyway
if pr . BaseRepoID == repo . ID {
continue
}
2022-06-13 15:07:59 +05:30
if err = issue_service . ChangeStatus ( pr . Issue , doer , true ) ; err != nil && ! issues_model . IsErrPullWasClosed ( err ) {
2020-01-25 08:18:22 +05:30
errs = append ( errs , err )
}
}
}
if len ( errs ) > 0 {
return errs
}
return nil
}
2020-04-10 16:56:37 +05:30
2021-06-25 22:31:43 +05:30
var commitMessageTrailersPattern = regexp . MustCompile ( ` (?:^|\n\n)(?:[\w-]+[ \t]*:[^\n]+\n*(?:[ \t]+[^\n]+\n*)*)+$ ` )
2020-12-21 22:16:14 +05:30
// GetSquashMergeCommitMessages returns the commit messages between head and merge base (if there is one)
2022-06-13 15:07:59 +05:30
func GetSquashMergeCommitMessages ( ctx context . Context , pr * issues_model . PullRequest ) string {
2022-11-19 13:42:33 +05:30
if err := pr . LoadIssue ( ctx ) ; err != nil {
2020-04-10 16:56:37 +05:30
log . Error ( "Cannot load issue %d for PR id %d: Error: %v" , pr . IssueID , pr . ID , err )
return ""
}
2022-11-19 13:42:33 +05:30
if err := pr . Issue . LoadPoster ( ctx ) ; err != nil {
2020-04-10 16:56:37 +05:30
log . Error ( "Cannot load poster %d for pr id %d, index %d Error: %v" , pr . Issue . PosterID , pr . ID , pr . Index , err )
return ""
}
if pr . HeadRepo == nil {
var err error
2022-12-03 08:18:26 +05:30
pr . HeadRepo , err = repo_model . GetRepositoryByID ( ctx , pr . HeadRepoID )
2020-04-10 16:56:37 +05:30
if err != nil {
2022-11-19 13:42:33 +05:30
log . Error ( "GetRepositoryByIdCtx[%d]: %v" , pr . HeadRepoID , err )
2020-04-10 16:56:37 +05:30
return ""
}
}
2022-01-20 04:56:57 +05:30
gitRepo , closer , err := git . RepositoryFromContextOrOpen ( ctx , pr . HeadRepo . RepoPath ( ) )
2020-04-10 16:56:37 +05:30
if err != nil {
log . Error ( "Unable to open head repository: Error: %v" , err )
return ""
}
2022-01-20 04:56:57 +05:30
defer closer . Close ( )
2020-04-10 16:56:37 +05:30
2021-07-28 15:12:56 +05:30
var headCommit * git . Commit
2022-06-13 15:07:59 +05:30
if pr . Flow == issues_model . PullRequestFlowGithub {
2021-07-28 15:12:56 +05:30
headCommit , err = gitRepo . GetBranchCommit ( pr . HeadBranch )
} else {
pr . HeadCommitID , err = gitRepo . GetRefCommitID ( pr . GetGitRefName ( ) )
if err != nil {
log . Error ( "Unable to get head commit: %s Error: %v" , pr . GetGitRefName ( ) , err )
return ""
}
headCommit , err = gitRepo . GetCommit ( pr . HeadCommitID )
}
2020-04-10 16:56:37 +05:30
if err != nil {
log . Error ( "Unable to get head commit: %s Error: %v" , pr . HeadBranch , err )
return ""
}
mergeBase , err := gitRepo . GetCommit ( pr . MergeBase )
if err != nil {
log . Error ( "Unable to get merge base commit: %s Error: %v" , pr . MergeBase , err )
return ""
}
limit := setting . Repository . PullRequest . DefaultMergeMessageCommitsLimit
2021-08-09 23:38:51 +05:30
commits , err := gitRepo . CommitsBetweenLimit ( headCommit , mergeBase , limit , 0 )
2020-04-10 16:56:37 +05:30
if err != nil {
log . Error ( "Unable to get commits between: %s %s Error: %v" , pr . HeadBranch , pr . MergeBase , err )
return ""
}
posterSig := pr . Issue . Poster . NewGitSig ( ) . String ( )
2022-10-12 10:48:26 +05:30
uniqueAuthors := make ( container . Set [ string ] )
2021-08-09 23:38:51 +05:30
authors := make ( [ ] string , 0 , len ( commits ) )
2020-04-10 16:56:37 +05:30
stringBuilder := strings . Builder { }
2020-12-21 22:16:14 +05:30
2021-06-19 03:38:22 +05:30
if ! setting . Repository . PullRequest . PopulateSquashCommentWithCommitMessages {
2021-06-25 22:31:43 +05:30
message := strings . TrimSpace ( pr . Issue . Content )
stringBuilder . WriteString ( message )
2021-06-19 03:38:22 +05:30
if stringBuilder . Len ( ) > 0 {
stringBuilder . WriteRune ( '\n' )
2021-06-25 22:31:43 +05:30
if ! commitMessageTrailersPattern . MatchString ( message ) {
stringBuilder . WriteRune ( '\n' )
}
2021-06-19 03:38:22 +05:30
}
2020-12-21 22:16:14 +05:30
}
2020-11-26 01:38:17 +05:30
// commits list is in reverse chronological order
2021-08-09 23:38:51 +05:30
first := true
for i := len ( commits ) - 1 ; i >= 0 ; i -- {
commit := commits [ i ]
2021-06-19 03:38:22 +05:30
if setting . Repository . PullRequest . PopulateSquashCommentWithCommitMessages {
maxSize := setting . Repository . PullRequest . DefaultMergeMessageSize
if maxSize < 0 || stringBuilder . Len ( ) < maxSize {
var toWrite [ ] byte
2021-08-09 23:38:51 +05:30
if first {
first = false
2021-06-19 03:38:22 +05:30
toWrite = [ ] byte ( strings . TrimPrefix ( commit . CommitMessage , pr . Issue . Title ) )
} else {
toWrite = [ ] byte ( commit . CommitMessage )
}
if len ( toWrite ) > maxSize - stringBuilder . Len ( ) && maxSize > - 1 {
toWrite = append ( toWrite [ : maxSize - stringBuilder . Len ( ) ] , "..." ... )
}
if _ , err := stringBuilder . Write ( toWrite ) ; err != nil {
log . Error ( "Unable to write commit message Error: %v" , err )
return ""
}
if _ , err := stringBuilder . WriteRune ( '\n' ) ; err != nil {
log . Error ( "Unable to write commit message Error: %v" , err )
return ""
}
}
}
2020-04-10 16:56:37 +05:30
authorString := commit . Author . String ( )
2022-10-12 10:48:26 +05:30
if uniqueAuthors . Add ( authorString ) && authorString != posterSig {
2020-04-10 16:56:37 +05:30
authors = append ( authors , authorString )
}
}
// Consider collecting the remaining authors
if limit >= 0 && setting . Repository . PullRequest . DefaultMergeMessageAllAuthors {
skip := limit
limit = 30
for {
2021-08-09 23:38:51 +05:30
commits , err := gitRepo . CommitsBetweenLimit ( headCommit , mergeBase , limit , skip )
2020-04-10 16:56:37 +05:30
if err != nil {
log . Error ( "Unable to get commits between: %s %s Error: %v" , pr . HeadBranch , pr . MergeBase , err )
return ""
}
2021-08-09 23:38:51 +05:30
if len ( commits ) == 0 {
2020-04-10 16:56:37 +05:30
break
}
2021-08-09 23:38:51 +05:30
for _ , commit := range commits {
2020-04-10 16:56:37 +05:30
authorString := commit . Author . String ( )
2022-10-12 10:48:26 +05:30
if uniqueAuthors . Add ( authorString ) && authorString != posterSig {
2020-04-10 16:56:37 +05:30
authors = append ( authors , authorString )
}
}
2020-11-28 01:30:52 +05:30
skip += limit
2020-04-10 16:56:37 +05:30
}
}
for _ , author := range authors {
if _ , err := stringBuilder . Write ( [ ] byte ( "Co-authored-by: " ) ) ; err != nil {
log . Error ( "Unable to write to string builder Error: %v" , err )
return ""
}
if _ , err := stringBuilder . Write ( [ ] byte ( author ) ) ; err != nil {
log . Error ( "Unable to write to string builder Error: %v" , err )
return ""
}
if _ , err := stringBuilder . WriteRune ( '\n' ) ; err != nil {
log . Error ( "Unable to write to string builder Error: %v" , err )
return ""
}
}
return stringBuilder . String ( )
}
2022-04-27 04:10:01 +05:30
// GetIssuesLastCommitStatus returns a map of issue ID to the most recent commit's latest status
2022-06-13 15:07:59 +05:30
func GetIssuesLastCommitStatus ( ctx context . Context , issues issues_model . IssueList ) ( map [ int64 ] * git_model . CommitStatus , error ) {
2022-04-27 04:10:01 +05:30
_ , lastStatus , err := GetIssuesAllCommitStatus ( ctx , issues )
return lastStatus , err
}
// GetIssuesAllCommitStatus returns a map of issue ID to a list of all statuses for the most recent commit as well as a map of issue ID to only the commit's latest status
2022-06-13 15:07:59 +05:30
func GetIssuesAllCommitStatus ( ctx context . Context , issues issues_model . IssueList ) ( map [ int64 ] [ ] * git_model . CommitStatus , map [ int64 ] * git_model . CommitStatus , error ) {
2022-11-19 13:42:33 +05:30
if err := issues . LoadPullRequests ( ctx ) ; err != nil {
2022-04-27 04:10:01 +05:30
return nil , nil , err
2021-04-15 23:04:43 +05:30
}
2022-11-19 13:42:33 +05:30
if _ , err := issues . LoadRepositories ( ctx ) ; err != nil {
2022-04-27 04:10:01 +05:30
return nil , nil , err
2021-04-15 23:04:43 +05:30
}
var (
gitRepos = make ( map [ int64 ] * git . Repository )
2022-06-12 21:21:54 +05:30
res = make ( map [ int64 ] [ ] * git_model . CommitStatus )
lastRes = make ( map [ int64 ] * git_model . CommitStatus )
2021-04-15 23:04:43 +05:30
err error
)
defer func ( ) {
for _ , gitRepo := range gitRepos {
gitRepo . Close ( )
}
} ( )
for _ , issue := range issues {
if ! issue . IsPull {
continue
}
gitRepo , ok := gitRepos [ issue . RepoID ]
if ! ok {
2022-03-30 00:43:41 +05:30
gitRepo , err = git . OpenRepository ( ctx , issue . Repo . RepoPath ( ) )
2021-04-15 23:04:43 +05:30
if err != nil {
2021-12-17 00:31:14 +05:30
log . Error ( "Cannot open git repository %-v for issue #%d[%d]. Error: %v" , issue . Repo , issue . Index , issue . ID , err )
continue
2021-04-15 23:04:43 +05:30
}
gitRepos [ issue . RepoID ] = gitRepo
}
2022-04-27 04:10:01 +05:30
statuses , lastStatus , err := getAllCommitStatus ( gitRepo , issue . PullRequest )
2021-04-15 23:04:43 +05:30
if err != nil {
2022-04-27 04:10:01 +05:30
log . Error ( "getAllCommitStatus: cant get commit statuses of pull [%d]: %v" , issue . PullRequest . ID , err )
2021-05-04 17:33:02 +05:30
continue
2021-04-15 23:04:43 +05:30
}
2022-04-27 04:10:01 +05:30
res [ issue . PullRequest . ID ] = statuses
lastRes [ issue . PullRequest . ID ] = lastStatus
2021-04-15 23:04:43 +05:30
}
2022-04-27 04:10:01 +05:30
return res , lastRes , nil
2021-04-15 23:04:43 +05:30
}
2022-04-27 04:10:01 +05:30
// getAllCommitStatus get pr's commit statuses.
2022-06-13 15:07:59 +05:30
func getAllCommitStatus ( gitRepo * git . Repository , pr * issues_model . PullRequest ) ( statuses [ ] * git_model . CommitStatus , lastStatus * git_model . CommitStatus , err error ) {
2022-04-27 04:10:01 +05:30
sha , shaErr := gitRepo . GetRefCommitID ( pr . GetGitRefName ( ) )
if shaErr != nil {
return nil , nil , shaErr
2020-04-10 16:56:37 +05:30
}
2022-06-12 21:21:54 +05:30
statuses , _ , err = git_model . GetLatestCommitStatus ( db . DefaultContext , pr . BaseRepo . ID , sha , db . ListOptions { } )
lastStatus = git_model . CalcCommitStatus ( statuses )
2022-04-27 04:10:01 +05:30
return statuses , lastStatus , err
2020-04-10 16:56:37 +05:30
}
// IsHeadEqualWithBranch returns if the commits of branchName are available in pull request head
2022-06-13 15:07:59 +05:30
func IsHeadEqualWithBranch ( ctx context . Context , pr * issues_model . PullRequest , branchName string ) ( bool , error ) {
2020-04-10 16:56:37 +05:30
var err error
2022-11-19 13:42:33 +05:30
if err = pr . LoadBaseRepo ( ctx ) ; err != nil {
2020-04-10 16:56:37 +05:30
return false , err
}
2022-01-20 04:56:57 +05:30
baseGitRepo , closer , err := git . RepositoryFromContextOrOpen ( ctx , pr . BaseRepo . RepoPath ( ) )
2020-04-10 16:56:37 +05:30
if err != nil {
return false , err
}
2022-01-20 04:56:57 +05:30
defer closer . Close ( )
2021-01-07 00:53:57 +05:30
2020-04-10 16:56:37 +05:30
baseCommit , err := baseGitRepo . GetBranchCommit ( branchName )
if err != nil {
return false , err
}
2022-11-19 13:42:33 +05:30
if err = pr . LoadHeadRepo ( ctx ) ; err != nil {
2020-04-10 16:56:37 +05:30
return false , err
}
2022-01-20 04:56:57 +05:30
var headGitRepo * git . Repository
if pr . HeadRepoID == pr . BaseRepoID {
headGitRepo = baseGitRepo
} else {
var closer io . Closer
headGitRepo , closer , err = git . RepositoryFromContextOrOpen ( ctx , pr . HeadRepo . RepoPath ( ) )
if err != nil {
return false , err
}
defer closer . Close ( )
2020-04-10 16:56:37 +05:30
}
2021-01-07 00:53:57 +05:30
2021-07-28 15:12:56 +05:30
var headCommit * git . Commit
2022-06-13 15:07:59 +05:30
if pr . Flow == issues_model . PullRequestFlowGithub {
2021-07-28 15:12:56 +05:30
headCommit , err = headGitRepo . GetBranchCommit ( pr . HeadBranch )
if err != nil {
return false , err
}
} else {
pr . HeadCommitID , err = baseGitRepo . GetRefCommitID ( pr . GetGitRefName ( ) )
if err != nil {
return false , err
}
if headCommit , err = baseGitRepo . GetCommit ( pr . HeadCommitID ) ; err != nil {
return false , err
}
2020-04-10 16:56:37 +05:30
}
return baseCommit . HasPreviousCommit ( headCommit . ID )
}