2019-06-23 01:35:34 +08:00
// Copyright 2019 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 pull
import (
"bufio"
"bytes"
2022-01-20 07:26:57 +08:00
"context"
2019-06-23 01:35:34 +08:00
"fmt"
"os"
"path/filepath"
2019-11-01 08:30:02 +08:00
"regexp"
2022-05-08 20:32:45 +08:00
"strconv"
2019-06-23 01:35:34 +08:00
"strings"
2019-10-16 21:42:42 +08:00
"time"
2019-06-23 01:35:34 +08:00
"code.gitea.io/gitea/models"
2022-05-04 03:46:28 +08:00
"code.gitea.io/gitea/models/db"
2022-05-11 18:09:36 +08:00
access_model "code.gitea.io/gitea/models/perm/access"
2022-05-08 01:05:52 +08:00
pull_model "code.gitea.io/gitea/models/pull"
2021-12-10 09:27:50 +08:00
repo_model "code.gitea.io/gitea/models/repo"
2021-11-10 03:57:58 +08:00
"code.gitea.io/gitea/models/unit"
2021-11-24 17:49:20 +08:00
user_model "code.gitea.io/gitea/models/user"
2019-06-23 01:35:34 +08:00
"code.gitea.io/gitea/modules/cache"
"code.gitea.io/gitea/modules/git"
"code.gitea.io/gitea/modules/log"
2019-11-05 19:04:08 +08:00
"code.gitea.io/gitea/modules/notification"
2019-11-18 21:13:07 +08:00
"code.gitea.io/gitea/modules/references"
2022-05-09 00:46:32 +08:00
repo_module "code.gitea.io/gitea/modules/repository"
2019-06-23 01:35:34 +08:00
"code.gitea.io/gitea/modules/setting"
2019-08-15 22:46:21 +08:00
"code.gitea.io/gitea/modules/timeutil"
2021-12-10 16:14:24 +08:00
asymkey_service "code.gitea.io/gitea/services/asymkey"
2019-11-18 21:13:07 +08:00
issue_service "code.gitea.io/gitea/services/issue"
2019-06-23 01:35:34 +08:00
)
2022-05-08 20:32:45 +08:00
// GetDefaultMergeMessage returns default message used when merging pull request
func GetDefaultMergeMessage ( baseGitRepo * git . Repository , pr * models . PullRequest , mergeStyle repo_model . MergeStyle ) ( string , error ) {
if err := pr . LoadHeadRepo ( ) ; err != nil {
return "" , err
}
if err := pr . LoadBaseRepo ( ) ; err != nil {
return "" , err
}
if pr . BaseRepo == nil {
return "" , repo_model . ErrRepoNotExist { ID : pr . BaseRepoID }
}
if err := pr . LoadIssue ( ) ; err != nil {
return "" , err
}
isExternalTracker := pr . BaseRepo . UnitEnabled ( unit . TypeExternalTracker )
issueReference := "#"
if isExternalTracker {
issueReference = "!"
}
if mergeStyle != "" {
templateFilepath := fmt . Sprintf ( ".gitea/default_merge_message/%s_TEMPLATE.md" , strings . ToUpper ( string ( mergeStyle ) ) )
commit , err := baseGitRepo . GetBranchCommit ( pr . BaseRepo . DefaultBranch )
if err != nil {
return "" , err
}
templateContent , err := commit . GetFileContent ( templateFilepath , setting . Repository . PullRequest . DefaultMergeMessageSize )
if err != nil {
if ! git . IsErrNotExist ( err ) {
return "" , err
}
} else {
vars := map [ string ] string {
"BaseRepoOwnerName" : pr . BaseRepo . OwnerName ,
"BaseRepoName" : pr . BaseRepo . Name ,
"BaseBranch" : pr . BaseBranch ,
"HeadRepoOwnerName" : "" ,
"HeadRepoName" : "" ,
"HeadBranch" : pr . HeadBranch ,
"PullRequestTitle" : pr . Issue . Title ,
"PullRequestDescription" : pr . Issue . Content ,
"PullRequestPosterName" : pr . Issue . Poster . Name ,
"PullRequestIndex" : strconv . FormatInt ( pr . Index , 10 ) ,
"PullRequestReference" : fmt . Sprintf ( "%s%d" , issueReference , pr . Index ) ,
}
if pr . HeadRepo != nil {
vars [ "HeadRepoOwnerName" ] = pr . HeadRepo . OwnerName
vars [ "HeadRepoName" ] = pr . HeadRepo . Name
}
refs , err := pr . ResolveCrossReferences ( baseGitRepo . Ctx )
if err == nil {
closeIssueIndexes := make ( [ ] string , 0 , len ( refs ) )
closeWord := "close"
if len ( setting . Repository . PullRequest . CloseKeywords ) > 0 {
closeWord = setting . Repository . PullRequest . CloseKeywords [ 0 ]
}
for _ , ref := range refs {
if ref . RefAction == references . XRefActionCloses {
closeIssueIndexes = append ( closeIssueIndexes , fmt . Sprintf ( "%s %s%d" , closeWord , issueReference , ref . Issue . Index ) )
}
}
if len ( closeIssueIndexes ) > 0 {
vars [ "ClosingIssues" ] = strings . Join ( closeIssueIndexes , ", " )
} else {
vars [ "ClosingIssues" ] = ""
}
}
return os . Expand ( templateContent , func ( s string ) string {
return vars [ s ]
} ) , nil
}
}
// Squash merge has a different from other styles.
if mergeStyle == repo_model . MergeStyleSquash {
return fmt . Sprintf ( "%s (%s%d)" , pr . Issue . Title , issueReference , pr . Issue . Index ) , nil
}
if pr . BaseRepoID == pr . HeadRepoID {
return fmt . Sprintf ( "Merge pull request '%s' (%s%d) from %s into %s" , pr . Issue . Title , issueReference , pr . Issue . Index , pr . HeadBranch , pr . BaseBranch ) , nil
}
if pr . HeadRepo == nil {
return fmt . Sprintf ( "Merge pull request '%s' (%s%d) from <deleted>:%s into %s" , pr . Issue . Title , issueReference , pr . Issue . Index , pr . HeadBranch , pr . BaseBranch ) , nil
}
return fmt . Sprintf ( "Merge pull request '%s' (%s%d) from %s:%s into %s" , pr . Issue . Title , issueReference , pr . Issue . Index , pr . HeadRepo . FullName ( ) , pr . HeadBranch , pr . BaseBranch ) , nil
}
2019-06-23 01:35:34 +08:00
// Merge merges pull request to base repository.
2020-01-11 15:29:34 +08:00
// Caller should check PR is ready to be merged (review and status checks)
2022-05-08 20:32:45 +08:00
func Merge ( ctx context . Context , pr * models . PullRequest , doer * user_model . User , baseGitRepo * git . Repository , mergeStyle repo_model . MergeStyle , expectedHeadCommitID , message string ) error {
2022-05-04 03:46:28 +08:00
if err := pr . LoadHeadRepo ( ) ; err != nil {
2020-03-03 06:31:55 +08:00
log . Error ( "LoadHeadRepo: %v" , err )
return fmt . Errorf ( "LoadHeadRepo: %v" , err )
2022-05-04 03:46:28 +08:00
} else if err := pr . LoadBaseRepo ( ) ; err != nil {
2020-03-03 06:31:55 +08:00
log . Error ( "LoadBaseRepo: %v" , err )
return fmt . Errorf ( "LoadBaseRepo: %v" , err )
2019-06-23 01:35:34 +08:00
}
2022-05-05 00:06:23 +08:00
pullWorkingPool . CheckIn ( fmt . Sprint ( pr . ID ) )
defer pullWorkingPool . CheckOut ( fmt . Sprint ( pr . ID ) )
2022-05-08 01:05:52 +08:00
// Removing an auto merge pull and ignore if not exist
2022-05-08 21:46:34 +08:00
if err := pull_model . DeleteScheduledAutoMerge ( db . DefaultContext , pr . ID ) ; err != nil && ! db . IsErrNotExist ( err ) {
2022-05-08 01:05:52 +08:00
return err
}
2021-11-10 03:57:58 +08:00
prUnit , err := pr . BaseRepo . GetUnit ( unit . TypePullRequests )
2019-06-23 01:35:34 +08:00
if err != nil {
2021-11-10 03:57:58 +08:00
log . Error ( "pr.BaseRepo.GetUnit(unit.TypePullRequests): %v" , err )
2019-06-23 01:35:34 +08:00
return err
}
prConfig := prUnit . PullRequestsConfig ( )
// Check if merge style is correct and allowed
if ! prConfig . IsMergeStyleAllowed ( mergeStyle ) {
return models . ErrInvalidMergeStyle { ID : pr . BaseRepo . ID , Style : mergeStyle }
}
defer func ( ) {
2020-01-09 09:47:45 +08:00
go AddTestPullRequestTask ( doer , pr . BaseRepo . ID , pr . BaseBranch , false , "" , "" )
2019-06-23 01:35:34 +08:00
} ( )
2022-05-04 03:46:28 +08:00
// TODO: make it able to do this in a database session
mergeCtx := context . Background ( )
pr . MergedCommitID , err = rawMerge ( mergeCtx , pr , doer , mergeStyle , expectedHeadCommitID , message )
2020-01-17 14:03:40 +08:00
if err != nil {
2020-02-10 07:09:31 +08:00
return err
2020-01-17 14:03:40 +08:00
}
pr . MergedUnix = timeutil . TimeStampNow ( )
pr . Merger = doer
pr . MergerID = doer . ID
2022-05-08 20:32:45 +08:00
if _ , err := pr . SetMerged ( ctx ) ; err != nil {
2020-01-17 14:03:40 +08:00
log . Error ( "setMerged [%d]: %v" , pr . ID , err )
}
2022-05-08 20:32:45 +08:00
if err := pr . LoadIssueCtx ( ctx ) ; err != nil {
2020-02-10 07:09:31 +08:00
log . Error ( "loadIssue [%d]: %v" , pr . ID , err )
}
2022-05-08 20:32:45 +08:00
if err := pr . Issue . LoadRepo ( ctx ) ; err != nil {
2020-02-10 07:09:31 +08:00
log . Error ( "loadRepo for issue [%d]: %v" , pr . ID , err )
}
2022-05-08 20:32:45 +08:00
if err := pr . Issue . Repo . GetOwner ( ctx ) ; err != nil {
2020-02-10 07:09:31 +08:00
log . Error ( "GetOwner for issue repo [%d]: %v" , pr . ID , err )
}
2020-01-17 14:03:40 +08:00
notification . NotifyMergePullRequest ( pr , doer )
// Reset cached commit count
cache . Remove ( pr . Issue . Repo . GetCommitsCountCacheKey ( pr . BaseBranch , true ) )
// Resolve cross references
2022-05-08 20:32:45 +08:00
refs , err := pr . ResolveCrossReferences ( ctx )
2020-01-17 14:03:40 +08:00
if err != nil {
log . Error ( "ResolveCrossReferences: %v" , err )
return nil
}
for _ , ref := range refs {
2022-05-08 20:32:45 +08:00
if err = ref . LoadIssueCtx ( ctx ) ; err != nil {
2020-01-17 14:03:40 +08:00
return err
}
2022-05-08 20:32:45 +08:00
if err = ref . Issue . LoadRepo ( ctx ) ; err != nil {
2020-01-17 14:03:40 +08:00
return err
}
2021-04-09 15:40:34 +08:00
close := ref . RefAction == references . XRefActionCloses
2020-01-17 14:03:40 +08:00
if close != ref . Issue . IsClosed {
if err = issue_service . ChangeStatus ( ref . Issue , doer , close ) ; err != nil {
2022-01-19 07:26:42 +08:00
// Allow ErrDependenciesLeft
if ! models . IsErrDependenciesLeft ( err ) {
return err
}
2020-01-17 14:03:40 +08:00
}
}
}
return nil
}
// rawMerge perform the merge operation without changing any pull information in database
2022-01-20 07:26:57 +08:00
func rawMerge ( ctx context . Context , pr * models . PullRequest , doer * user_model . User , mergeStyle repo_model . MergeStyle , expectedHeadCommitID , message string ) ( string , error ) {
2019-06-23 01:35:34 +08:00
// Clone base repo.
2022-01-20 07:26:57 +08:00
tmpBasePath , err := createTemporaryRepo ( ctx , pr )
2019-06-23 01:35:34 +08:00
if err != nil {
2019-11-10 16:42:51 +08:00
log . Error ( "CreateTemporaryPath: %v" , err )
2020-02-10 07:09:31 +08:00
return "" , err
2019-06-23 01:35:34 +08:00
}
defer func ( ) {
2022-05-09 00:46:32 +08:00
if err := repo_module . RemoveTemporaryPath ( tmpBasePath ) ; err != nil {
2019-06-23 01:35:34 +08:00
log . Error ( "Merge: RemoveTemporaryPath: %s" , err )
}
} ( )
2019-10-12 08:13:27 +08:00
baseBranch := "base"
trackingBranch := "tracking"
stagingBranch := "staging"
2019-06-23 01:35:34 +08:00
2021-12-20 08:32:54 +08:00
if expectedHeadCommitID != "" {
2022-04-01 10:55:30 +08:00
trackingCommitID , _ , err := git . NewCommand ( ctx , "show-ref" , "--hash" , git . BranchPrefix + trackingBranch ) . RunStdString ( & git . RunOpts { Dir : tmpBasePath } )
2021-12-20 08:32:54 +08:00
if err != nil {
log . Error ( "show-ref[%s] --hash refs/heads/trackingn: %v" , tmpBasePath , git . BranchPrefix + trackingBranch , err )
return "" , fmt . Errorf ( "getDiffTree: %v" , err )
}
if strings . TrimSpace ( trackingCommitID ) != expectedHeadCommitID {
return "" , models . ErrSHADoesNotMatch {
GivenSHA : expectedHeadCommitID ,
CurrentSHA : trackingCommitID ,
}
}
}
2019-12-14 06:21:06 +08:00
var outbuf , errbuf strings . Builder
2019-06-23 01:35:34 +08:00
// Enable sparse-checkout
2022-01-20 07:26:57 +08:00
sparseCheckoutList , err := getDiffTree ( ctx , tmpBasePath , baseBranch , trackingBranch )
2019-06-23 01:35:34 +08:00
if err != nil {
2019-11-10 16:42:51 +08:00
log . Error ( "getDiffTree(%s, %s, %s): %v" , tmpBasePath , baseBranch , trackingBranch , err )
2020-02-10 07:09:31 +08:00
return "" , fmt . Errorf ( "getDiffTree: %v" , err )
2019-06-23 01:35:34 +08:00
}
infoPath := filepath . Join ( tmpBasePath , ".git" , "info" )
2022-01-21 01:46:10 +08:00
if err := os . MkdirAll ( infoPath , 0 o700 ) ; err != nil {
2019-11-10 16:42:51 +08:00
log . Error ( "Unable to create .git/info in %s: %v" , tmpBasePath , err )
2020-02-10 07:09:31 +08:00
return "" , fmt . Errorf ( "Unable to create .git/info in tmpBasePath: %v" , err )
2019-06-23 01:35:34 +08:00
}
2019-11-10 16:42:51 +08:00
2019-06-23 01:35:34 +08:00
sparseCheckoutListPath := filepath . Join ( infoPath , "sparse-checkout" )
2022-01-21 01:46:10 +08:00
if err := os . WriteFile ( sparseCheckoutListPath , [ ] byte ( sparseCheckoutList ) , 0 o600 ) ; err != nil {
2019-11-10 16:42:51 +08:00
log . Error ( "Unable to write .git/info/sparse-checkout file in %s: %v" , tmpBasePath , err )
2020-02-10 07:09:31 +08:00
return "" , fmt . Errorf ( "Unable to write .git/info/sparse-checkout file in tmpBasePath: %v" , err )
2019-06-23 01:35:34 +08:00
}
2019-11-10 16:42:51 +08:00
var gitConfigCommand func ( ) * git . Command
2020-10-21 23:42:08 +08:00
if git . CheckGitVersionAtLeast ( "1.8.0" ) == nil {
2019-11-10 16:42:51 +08:00
gitConfigCommand = func ( ) * git . Command {
2022-02-07 03:01:47 +08:00
return git . NewCommand ( ctx , "config" , "--local" )
2019-10-12 08:13:27 +08:00
}
2019-11-10 16:42:51 +08:00
} else {
gitConfigCommand = func ( ) * git . Command {
2022-02-07 03:01:47 +08:00
return git . NewCommand ( ctx , "config" )
2019-10-12 08:13:27 +08:00
}
2019-11-10 16:42:51 +08:00
}
2019-10-12 08:13:27 +08:00
2019-06-23 01:35:34 +08:00
// Switch off LFS process (set required, clean and smudge here also)
2022-02-11 20:47:22 +08:00
if err := gitConfigCommand ( ) . AddArguments ( "filter.lfs.process" , "" ) .
2022-04-01 10:55:30 +08:00
Run ( & git . RunOpts {
Dir : tmpBasePath ,
Stdout : & outbuf ,
Stderr : & errbuf ,
2022-02-11 20:47:22 +08:00
} ) ; err != nil {
2019-11-10 16:42:51 +08:00
log . Error ( "git config [filter.lfs.process -> <> ]: %v\n%s\n%s" , err , outbuf . String ( ) , errbuf . String ( ) )
2020-02-10 07:09:31 +08:00
return "" , fmt . Errorf ( "git config [filter.lfs.process -> <> ]: %v\n%s\n%s" , err , outbuf . String ( ) , errbuf . String ( ) )
2019-06-23 01:35:34 +08:00
}
2019-11-10 16:42:51 +08:00
outbuf . Reset ( )
errbuf . Reset ( )
2022-02-11 20:47:22 +08:00
if err := gitConfigCommand ( ) . AddArguments ( "filter.lfs.required" , "false" ) .
2022-04-01 10:55:30 +08:00
Run ( & git . RunOpts {
Dir : tmpBasePath ,
Stdout : & outbuf ,
Stderr : & errbuf ,
2022-02-11 20:47:22 +08:00
} ) ; err != nil {
2019-11-10 16:42:51 +08:00
log . Error ( "git config [filter.lfs.required -> <false> ]: %v\n%s\n%s" , err , outbuf . String ( ) , errbuf . String ( ) )
2020-02-10 07:09:31 +08:00
return "" , fmt . Errorf ( "git config [filter.lfs.required -> <false> ]: %v\n%s\n%s" , err , outbuf . String ( ) , errbuf . String ( ) )
2019-06-23 01:35:34 +08:00
}
2019-11-10 16:42:51 +08:00
outbuf . Reset ( )
errbuf . Reset ( )
2022-02-11 20:47:22 +08:00
if err := gitConfigCommand ( ) . AddArguments ( "filter.lfs.clean" , "" ) .
2022-04-01 10:55:30 +08:00
Run ( & git . RunOpts {
Dir : tmpBasePath ,
Stdout : & outbuf ,
Stderr : & errbuf ,
2022-02-11 20:47:22 +08:00
} ) ; err != nil {
2019-11-10 16:42:51 +08:00
log . Error ( "git config [filter.lfs.clean -> <> ]: %v\n%s\n%s" , err , outbuf . String ( ) , errbuf . String ( ) )
2020-02-10 07:09:31 +08:00
return "" , fmt . Errorf ( "git config [filter.lfs.clean -> <> ]: %v\n%s\n%s" , err , outbuf . String ( ) , errbuf . String ( ) )
2019-06-23 01:35:34 +08:00
}
2019-11-10 16:42:51 +08:00
outbuf . Reset ( )
errbuf . Reset ( )
2022-02-11 20:47:22 +08:00
if err := gitConfigCommand ( ) . AddArguments ( "filter.lfs.smudge" , "" ) .
2022-04-01 10:55:30 +08:00
Run ( & git . RunOpts {
Dir : tmpBasePath ,
Stdout : & outbuf ,
Stderr : & errbuf ,
2022-02-11 20:47:22 +08:00
} ) ; err != nil {
2019-11-10 16:42:51 +08:00
log . Error ( "git config [filter.lfs.smudge -> <> ]: %v\n%s\n%s" , err , outbuf . String ( ) , errbuf . String ( ) )
2020-02-10 07:09:31 +08:00
return "" , fmt . Errorf ( "git config [filter.lfs.smudge -> <> ]: %v\n%s\n%s" , err , outbuf . String ( ) , errbuf . String ( ) )
2019-06-23 01:35:34 +08:00
}
2019-11-10 16:42:51 +08:00
outbuf . Reset ( )
errbuf . Reset ( )
2019-06-23 01:35:34 +08:00
2022-02-11 20:47:22 +08:00
if err := gitConfigCommand ( ) . AddArguments ( "core.sparseCheckout" , "true" ) .
2022-04-01 10:55:30 +08:00
Run ( & git . RunOpts {
Dir : tmpBasePath ,
Stdout : & outbuf ,
Stderr : & errbuf ,
2022-02-11 20:47:22 +08:00
} ) ; err != nil {
2019-11-10 16:42:51 +08:00
log . Error ( "git config [core.sparseCheckout -> true ]: %v\n%s\n%s" , err , outbuf . String ( ) , errbuf . String ( ) )
2020-02-10 07:09:31 +08:00
return "" , fmt . Errorf ( "git config [core.sparsecheckout -> true]: %v\n%s\n%s" , err , outbuf . String ( ) , errbuf . String ( ) )
2019-06-23 01:35:34 +08:00
}
2019-11-10 16:42:51 +08:00
outbuf . Reset ( )
errbuf . Reset ( )
2019-06-23 01:35:34 +08:00
// Read base branch index
2022-02-11 20:47:22 +08:00
if err := git . NewCommand ( ctx , "read-tree" , "HEAD" ) .
2022-04-01 10:55:30 +08:00
Run ( & git . RunOpts {
Dir : tmpBasePath ,
Stdout : & outbuf ,
Stderr : & errbuf ,
2022-02-11 20:47:22 +08:00
} ) ; err != nil {
2019-11-10 16:42:51 +08:00
log . Error ( "git read-tree HEAD: %v\n%s\n%s" , err , outbuf . String ( ) , errbuf . String ( ) )
2020-02-10 07:09:31 +08:00
return "" , fmt . Errorf ( "Unable to read base branch in to the index: %v\n%s\n%s" , err , outbuf . String ( ) , errbuf . String ( ) )
2019-06-23 01:35:34 +08:00
}
2019-11-10 16:42:51 +08:00
outbuf . Reset ( )
errbuf . Reset ( )
2019-06-23 01:35:34 +08:00
2020-09-20 00:44:55 +08:00
sig := doer . NewGitSig ( )
committer := sig
2019-10-16 21:42:42 +08:00
// Determine if we should sign
signArg := ""
2020-10-21 23:42:08 +08:00
if git . CheckGitVersionAtLeast ( "1.7.9" ) == nil {
2022-01-20 07:26:57 +08:00
sign , keyID , signer , _ := asymkey_service . SignMerge ( ctx , pr , doer , tmpBasePath , "HEAD" , trackingBranch )
2019-10-16 21:42:42 +08:00
if sign {
signArg = "-S" + keyID
2021-12-10 09:27:50 +08:00
if pr . BaseRepo . GetTrustModel ( ) == repo_model . CommitterTrustModel || pr . BaseRepo . GetTrustModel ( ) == repo_model . CollaboratorCommitterTrustModel {
2020-09-20 00:44:55 +08:00
committer = signer
}
2020-10-21 23:42:08 +08:00
} else if git . CheckGitVersionAtLeast ( "2.0.0" ) == nil {
2019-10-16 21:42:42 +08:00
signArg = "--no-gpg-sign"
}
}
commitTimeStr := time . Now ( ) . Format ( time . RFC3339 )
// Because this may call hooks we should pass in the environment
env := append ( os . Environ ( ) ,
"GIT_AUTHOR_NAME=" + sig . Name ,
"GIT_AUTHOR_EMAIL=" + sig . Email ,
"GIT_AUTHOR_DATE=" + commitTimeStr ,
2020-09-20 00:44:55 +08:00
"GIT_COMMITTER_NAME=" + committer . Name ,
"GIT_COMMITTER_EMAIL=" + committer . Email ,
2019-10-16 21:42:42 +08:00
"GIT_COMMITTER_DATE=" + commitTimeStr ,
)
2019-06-23 01:35:34 +08:00
// Merge commits.
switch mergeStyle {
2021-12-10 09:27:50 +08:00
case repo_model . MergeStyleMerge :
2022-02-07 03:01:47 +08:00
cmd := git . NewCommand ( ctx , "merge" , "--no-ff" , "--no-commit" , trackingBranch )
2019-11-10 16:42:51 +08:00
if err := runMergeCommand ( pr , mergeStyle , cmd , tmpBasePath ) ; err != nil {
log . Error ( "Unable to merge tracking into base: %v" , err )
2020-02-10 07:09:31 +08:00
return "" , err
2019-06-23 01:35:34 +08:00
}
2022-01-20 07:26:57 +08:00
if err := commitAndSignNoAuthor ( ctx , pr , message , signArg , tmpBasePath , env ) ; err != nil {
2019-11-10 16:42:51 +08:00
log . Error ( "Unable to make final commit: %v" , err )
2020-02-10 07:09:31 +08:00
return "" , err
2019-06-23 01:35:34 +08:00
}
2021-12-10 09:27:50 +08:00
case repo_model . MergeStyleRebase :
2019-11-10 16:42:51 +08:00
fallthrough
2021-12-10 09:27:50 +08:00
case repo_model . MergeStyleRebaseUpdate :
2021-08-31 22:03:45 +08:00
fallthrough
2021-12-10 09:27:50 +08:00
case repo_model . MergeStyleRebaseMerge :
2019-06-23 01:35:34 +08:00
// Checkout head branch
2022-02-11 20:47:22 +08:00
if err := git . NewCommand ( ctx , "checkout" , "-b" , stagingBranch , trackingBranch ) .
2022-04-01 10:55:30 +08:00
Run ( & git . RunOpts {
Dir : tmpBasePath ,
Stdout : & outbuf ,
Stderr : & errbuf ,
2022-02-11 20:47:22 +08:00
} ) ; err != nil {
2019-11-10 16:42:51 +08:00
log . Error ( "git checkout base prior to merge post staging rebase [%s:%s -> %s:%s]: %v\n%s\n%s" , pr . HeadRepo . FullName ( ) , pr . HeadBranch , pr . BaseRepo . FullName ( ) , pr . BaseBranch , err , outbuf . String ( ) , errbuf . String ( ) )
2020-02-10 07:09:31 +08:00
return "" , fmt . Errorf ( "git checkout base prior to merge post staging rebase [%s:%s -> %s:%s]: %v\n%s\n%s" , pr . HeadRepo . FullName ( ) , pr . HeadBranch , pr . BaseRepo . FullName ( ) , pr . BaseBranch , err , outbuf . String ( ) , errbuf . String ( ) )
2019-06-23 01:35:34 +08:00
}
2019-11-10 16:42:51 +08:00
outbuf . Reset ( )
errbuf . Reset ( )
2019-06-23 01:35:34 +08:00
// Rebase before merging
2022-02-11 20:47:22 +08:00
if err := git . NewCommand ( ctx , "rebase" , baseBranch ) .
2022-04-01 10:55:30 +08:00
Run ( & git . RunOpts {
Dir : tmpBasePath ,
Stdout : & outbuf ,
Stderr : & errbuf ,
2022-02-11 20:47:22 +08:00
} ) ; err != nil {
2019-11-10 16:42:51 +08:00
// Rebase will leave a REBASE_HEAD file in .git if there is a conflict
if _ , statErr := os . Stat ( filepath . Join ( tmpBasePath , ".git" , "REBASE_HEAD" ) ) ; statErr == nil {
2020-04-04 00:00:41 +08:00
var commitSha string
ok := false
failingCommitPaths := [ ] string {
filepath . Join ( tmpBasePath , ".git" , "rebase-apply" , "original-commit" ) , // Git < 2.26
filepath . Join ( tmpBasePath , ".git" , "rebase-merge" , "stopped-sha" ) , // Git >= 2.26
}
for _ , failingCommitPath := range failingCommitPaths {
2021-11-15 14:02:53 +08:00
if _ , statErr := os . Stat ( failingCommitPath ) ; statErr == nil {
commitShaBytes , readErr := os . ReadFile ( failingCommitPath )
2020-04-04 00:00:41 +08:00
if readErr != nil {
// Abandon this attempt to handle the error
log . Error ( "git rebase staging on to base [%s:%s -> %s:%s]: %v\n%s\n%s" , pr . HeadRepo . FullName ( ) , pr . HeadBranch , pr . BaseRepo . FullName ( ) , pr . BaseBranch , err , outbuf . String ( ) , errbuf . String ( ) )
return "" , fmt . Errorf ( "git rebase staging on to base [%s:%s -> %s:%s]: %v\n%s\n%s" , pr . HeadRepo . FullName ( ) , pr . HeadBranch , pr . BaseRepo . FullName ( ) , pr . BaseBranch , err , outbuf . String ( ) , errbuf . String ( ) )
}
commitSha = strings . TrimSpace ( string ( commitShaBytes ) )
ok = true
break
}
}
if ! ok {
log . Error ( "Unable to determine failing commit sha for this rebase message. Cannot cast as models.ErrRebaseConflicts." )
2019-11-10 16:42:51 +08:00
log . Error ( "git rebase staging on to base [%s:%s -> %s:%s]: %v\n%s\n%s" , pr . HeadRepo . FullName ( ) , pr . HeadBranch , pr . BaseRepo . FullName ( ) , pr . BaseBranch , err , outbuf . String ( ) , errbuf . String ( ) )
2020-02-10 07:09:31 +08:00
return "" , fmt . Errorf ( "git rebase staging on to base [%s:%s -> %s:%s]: %v\n%s\n%s" , pr . HeadRepo . FullName ( ) , pr . HeadBranch , pr . BaseRepo . FullName ( ) , pr . BaseBranch , err , outbuf . String ( ) , errbuf . String ( ) )
2019-11-10 16:42:51 +08:00
}
2020-04-04 00:00:41 +08:00
log . Debug ( "RebaseConflict at %s [%s:%s -> %s:%s]: %v\n%s\n%s" , commitSha , pr . HeadRepo . FullName ( ) , pr . HeadBranch , pr . BaseRepo . FullName ( ) , pr . BaseBranch , err , outbuf . String ( ) , errbuf . String ( ) )
2020-02-10 07:09:31 +08:00
return "" , models . ErrRebaseConflicts {
2019-11-10 16:42:51 +08:00
Style : mergeStyle ,
2020-04-04 00:00:41 +08:00
CommitSHA : commitSha ,
2019-11-10 16:42:51 +08:00
StdOut : outbuf . String ( ) ,
StdErr : errbuf . String ( ) ,
Err : err ,
}
}
log . Error ( "git rebase staging on to base [%s:%s -> %s:%s]: %v\n%s\n%s" , pr . HeadRepo . FullName ( ) , pr . HeadBranch , pr . BaseRepo . FullName ( ) , pr . BaseBranch , err , outbuf . String ( ) , errbuf . String ( ) )
2020-02-10 07:09:31 +08:00
return "" , fmt . Errorf ( "git rebase staging on to base [%s:%s -> %s:%s]: %v\n%s\n%s" , pr . HeadRepo . FullName ( ) , pr . HeadBranch , pr . BaseRepo . FullName ( ) , pr . BaseBranch , err , outbuf . String ( ) , errbuf . String ( ) )
2019-06-23 01:35:34 +08:00
}
2019-11-10 16:42:51 +08:00
outbuf . Reset ( )
errbuf . Reset ( )
2021-08-31 22:03:45 +08:00
// not need merge, just update by rebase. so skip
2021-12-10 09:27:50 +08:00
if mergeStyle == repo_model . MergeStyleRebaseUpdate {
2021-08-31 22:03:45 +08:00
break
}
2019-06-23 01:35:34 +08:00
// Checkout base branch again
2022-02-11 20:47:22 +08:00
if err := git . NewCommand ( ctx , "checkout" , baseBranch ) .
2022-04-01 10:55:30 +08:00
Run ( & git . RunOpts {
Dir : tmpBasePath ,
Stdout : & outbuf ,
Stderr : & errbuf ,
2022-02-11 20:47:22 +08:00
} ) ; err != nil {
2019-11-10 16:42:51 +08:00
log . Error ( "git checkout base prior to merge post staging rebase [%s:%s -> %s:%s]: %v\n%s\n%s" , pr . HeadRepo . FullName ( ) , pr . HeadBranch , pr . BaseRepo . FullName ( ) , pr . BaseBranch , err , outbuf . String ( ) , errbuf . String ( ) )
2020-02-10 07:09:31 +08:00
return "" , fmt . Errorf ( "git checkout base prior to merge post staging rebase [%s:%s -> %s:%s]: %v\n%s\n%s" , pr . HeadRepo . FullName ( ) , pr . HeadBranch , pr . BaseRepo . FullName ( ) , pr . BaseBranch , err , outbuf . String ( ) , errbuf . String ( ) )
2019-06-23 01:35:34 +08:00
}
2019-11-10 16:42:51 +08:00
outbuf . Reset ( )
errbuf . Reset ( )
2019-06-23 01:35:34 +08:00
2022-02-07 03:01:47 +08:00
cmd := git . NewCommand ( ctx , "merge" )
2021-12-10 09:27:50 +08:00
if mergeStyle == repo_model . MergeStyleRebase {
2019-11-10 16:42:51 +08:00
cmd . AddArguments ( "--ff-only" )
2019-10-16 21:42:42 +08:00
} else {
2019-11-10 16:42:51 +08:00
cmd . AddArguments ( "--no-ff" , "--no-commit" )
2019-06-23 01:35:34 +08:00
}
2019-11-10 16:42:51 +08:00
cmd . AddArguments ( stagingBranch )
2019-06-23 01:35:34 +08:00
2019-11-10 16:42:51 +08:00
// Prepare merge with commit
if err := runMergeCommand ( pr , mergeStyle , cmd , tmpBasePath ) ; err != nil {
log . Error ( "Unable to merge staging into base: %v" , err )
2020-02-10 07:09:31 +08:00
return "" , err
2019-11-10 16:42:51 +08:00
}
2021-12-10 09:27:50 +08:00
if mergeStyle == repo_model . MergeStyleRebaseMerge {
2022-01-20 07:26:57 +08:00
if err := commitAndSignNoAuthor ( ctx , pr , message , signArg , tmpBasePath , env ) ; err != nil {
2019-11-10 16:42:51 +08:00
log . Error ( "Unable to make final commit: %v" , err )
2020-02-10 07:09:31 +08:00
return "" , err
2019-11-10 16:42:51 +08:00
}
}
2021-12-10 09:27:50 +08:00
case repo_model . MergeStyleSquash :
2019-06-23 01:35:34 +08:00
// Merge with squash
2022-02-07 03:01:47 +08:00
cmd := git . NewCommand ( ctx , "merge" , "--squash" , trackingBranch )
2019-11-10 16:42:51 +08:00
if err := runMergeCommand ( pr , mergeStyle , cmd , tmpBasePath ) ; err != nil {
log . Error ( "Unable to merge --squash tracking into base: %v" , err )
2020-02-10 07:09:31 +08:00
return "" , err
2019-06-23 01:35:34 +08:00
}
2019-11-10 16:42:51 +08:00
2020-04-10 18:40:36 +08:00
if err = pr . Issue . LoadPoster ( ) ; err != nil {
log . Error ( "LoadPoster: %v" , err )
return "" , fmt . Errorf ( "LoadPoster: %v" , err )
}
2019-06-23 01:35:34 +08:00
sig := pr . Issue . Poster . NewGitSig ( )
2019-10-16 21:42:42 +08:00
if signArg == "" {
2022-02-11 20:47:22 +08:00
if err := git . NewCommand ( ctx , "commit" , fmt . Sprintf ( "--author='%s <%s>'" , sig . Name , sig . Email ) , "-m" , message ) .
2022-04-01 10:55:30 +08:00
Run ( & git . RunOpts {
Env : env ,
Dir : tmpBasePath ,
Stdout : & outbuf ,
Stderr : & errbuf ,
2022-02-11 20:47:22 +08:00
} ) ; err != nil {
2019-11-10 16:42:51 +08:00
log . Error ( "git commit [%s:%s -> %s:%s]: %v\n%s\n%s" , pr . HeadRepo . FullName ( ) , pr . HeadBranch , pr . BaseRepo . FullName ( ) , pr . BaseBranch , err , outbuf . String ( ) , errbuf . String ( ) )
2020-02-10 07:09:31 +08:00
return "" , fmt . Errorf ( "git commit [%s:%s -> %s:%s]: %v\n%s\n%s" , pr . HeadRepo . FullName ( ) , pr . HeadBranch , pr . BaseRepo . FullName ( ) , pr . BaseBranch , err , outbuf . String ( ) , errbuf . String ( ) )
2019-10-16 21:42:42 +08:00
}
} else {
2021-11-29 15:09:55 +08:00
if setting . Repository . PullRequest . AddCoCommitterTrailers && committer . String ( ) != sig . String ( ) {
2020-09-20 00:44:55 +08:00
// add trailer
2020-12-22 10:19:33 +08:00
message += fmt . Sprintf ( "\nCo-authored-by: %s\nCo-committed-by: %s\n" , sig . String ( ) , sig . String ( ) )
2020-09-20 00:44:55 +08:00
}
2022-02-11 20:47:22 +08:00
if err := git . NewCommand ( ctx , "commit" , signArg , fmt . Sprintf ( "--author='%s <%s>'" , sig . Name , sig . Email ) , "-m" , message ) .
2022-04-01 10:55:30 +08:00
Run ( & git . RunOpts {
Env : env ,
Dir : tmpBasePath ,
Stdout : & outbuf ,
Stderr : & errbuf ,
2022-02-11 20:47:22 +08:00
} ) ; err != nil {
2019-11-10 16:42:51 +08:00
log . Error ( "git commit [%s:%s -> %s:%s]: %v\n%s\n%s" , pr . HeadRepo . FullName ( ) , pr . HeadBranch , pr . BaseRepo . FullName ( ) , pr . BaseBranch , err , outbuf . String ( ) , errbuf . String ( ) )
2020-02-10 07:09:31 +08:00
return "" , fmt . Errorf ( "git commit [%s:%s -> %s:%s]: %v\n%s\n%s" , pr . HeadRepo . FullName ( ) , pr . HeadBranch , pr . BaseRepo . FullName ( ) , pr . BaseBranch , err , outbuf . String ( ) , errbuf . String ( ) )
2019-10-16 21:42:42 +08:00
}
2019-06-23 01:35:34 +08:00
}
2019-11-10 16:42:51 +08:00
outbuf . Reset ( )
errbuf . Reset ( )
2019-06-23 01:35:34 +08:00
default :
2020-02-10 07:09:31 +08:00
return "" , models . ErrInvalidMergeStyle { ID : pr . BaseRepo . ID , Style : mergeStyle }
2019-06-23 01:35:34 +08:00
}
// OK we should cache our current head and origin/headbranch
2022-01-20 07:26:57 +08:00
mergeHeadSHA , err := git . GetFullCommitID ( ctx , tmpBasePath , "HEAD" )
2019-06-23 01:35:34 +08:00
if err != nil {
2020-02-10 07:09:31 +08:00
return "" , fmt . Errorf ( "Failed to get full commit id for HEAD: %v" , err )
2019-06-23 01:35:34 +08:00
}
2022-01-20 07:26:57 +08:00
mergeBaseSHA , err := git . GetFullCommitID ( ctx , tmpBasePath , "original_" + baseBranch )
2019-06-23 01:35:34 +08:00
if err != nil {
2020-02-10 07:09:31 +08:00
return "" , fmt . Errorf ( "Failed to get full commit id for origin/%s: %v" , pr . BaseBranch , err )
}
2022-01-20 07:26:57 +08:00
mergeCommitID , err := git . GetFullCommitID ( ctx , tmpBasePath , baseBranch )
2020-02-10 07:09:31 +08:00
if err != nil {
return "" , fmt . Errorf ( "Failed to get full commit id for the new merge: %v" , err )
2019-06-23 01:35:34 +08:00
}
// Now it's questionable about where this should go - either after or before the push
// I think in the interests of data safety - failures to push to the lfs should prevent
// the merge as you can always remerge.
if setting . LFS . StartServer {
2022-01-20 07:26:57 +08:00
if err := LFSPush ( ctx , tmpBasePath , mergeHeadSHA , mergeBaseSHA , pr ) ; err != nil {
2020-02-10 07:09:31 +08:00
return "" , err
2019-06-23 01:35:34 +08:00
}
}
2021-11-24 17:49:20 +08:00
var headUser * user_model . User
2022-03-22 23:22:54 +08:00
err = pr . HeadRepo . GetOwner ( ctx )
2019-07-01 09:18:13 +08:00
if err != nil {
2021-11-24 17:49:20 +08:00
if ! user_model . IsErrUserNotExist ( err ) {
2019-10-18 19:13:31 +08:00
log . Error ( "Can't find user: %d for head repository - %v" , pr . HeadRepo . OwnerID , err )
2020-02-10 07:09:31 +08:00
return "" , err
2019-07-01 09:18:13 +08:00
}
2019-10-18 19:13:31 +08:00
log . Error ( "Can't find user: %d for head repository - defaulting to doer: %s - %v" , pr . HeadRepo . OwnerID , doer . Name , err )
2019-07-01 09:18:13 +08:00
headUser = doer
2019-10-18 19:13:31 +08:00
} else {
headUser = pr . HeadRepo . Owner
2019-07-01 09:18:13 +08:00
}
2022-05-09 00:46:32 +08:00
env = repo_module . FullPushingEnvironment (
2019-07-26 05:50:20 +08:00
headUser ,
doer ,
pr . BaseRepo ,
pr . BaseRepo . Name ,
pr . ID ,
)
2019-06-23 01:35:34 +08:00
2021-08-31 22:03:45 +08:00
var pushCmd * git . Command
2021-12-10 09:27:50 +08:00
if mergeStyle == repo_model . MergeStyleRebaseUpdate {
2022-01-10 17:32:37 +08:00
// force push the rebase result to head branch
2022-02-07 03:01:47 +08:00
pushCmd = git . NewCommand ( ctx , "push" , "-f" , "head_repo" , stagingBranch + ":" + git . BranchPrefix + pr . HeadBranch )
2021-08-31 22:03:45 +08:00
} else {
2022-02-07 03:01:47 +08:00
pushCmd = git . NewCommand ( ctx , "push" , "origin" , baseBranch + ":" + git . BranchPrefix + pr . BaseBranch )
2021-08-31 22:03:45 +08:00
}
2019-06-23 01:35:34 +08:00
// Push back to upstream.
2022-05-04 03:46:28 +08:00
// TODO: this cause an api call to "/api/internal/hook/post-receive/...",
// that prevents us from doint the whole merge in one db transaction
2022-04-01 10:55:30 +08:00
if err := pushCmd . Run ( & git . RunOpts {
Env : env ,
Dir : tmpBasePath ,
Stdout : & outbuf ,
Stderr : & errbuf ,
2022-02-11 20:47:22 +08:00
} ) ; err != nil {
2019-11-10 16:42:51 +08:00
if strings . Contains ( errbuf . String ( ) , "non-fast-forward" ) {
2020-03-28 12:13:18 +08:00
return "" , & git . ErrPushOutOfDate {
2019-11-10 16:42:51 +08:00
StdOut : outbuf . String ( ) ,
StdErr : errbuf . String ( ) ,
Err : err ,
}
2020-02-22 21:08:48 +08:00
} else if strings . Contains ( errbuf . String ( ) , "! [remote rejected]" ) {
2020-03-28 12:13:18 +08:00
err := & git . ErrPushRejected {
2020-02-22 21:08:48 +08:00
StdOut : outbuf . String ( ) ,
StdErr : errbuf . String ( ) ,
Err : err ,
}
err . GenerateMessage ( )
return "" , err
2019-11-10 16:42:51 +08:00
}
2020-02-10 07:09:31 +08:00
return "" , fmt . Errorf ( "git push: %s" , errbuf . String ( ) )
2019-06-23 01:35:34 +08:00
}
2019-11-10 16:42:51 +08:00
outbuf . Reset ( )
errbuf . Reset ( )
2019-06-23 01:35:34 +08:00
2020-02-10 07:09:31 +08:00
return mergeCommitID , nil
2019-06-23 01:35:34 +08:00
}
2022-01-20 07:26:57 +08:00
func commitAndSignNoAuthor ( ctx context . Context , pr * models . PullRequest , message , signArg , tmpBasePath string , env [ ] string ) error {
2019-11-10 16:42:51 +08:00
var outbuf , errbuf strings . Builder
if signArg == "" {
2022-02-11 20:47:22 +08:00
if err := git . NewCommand ( ctx , "commit" , "-m" , message ) .
2022-04-01 10:55:30 +08:00
Run ( & git . RunOpts {
Env : env ,
Dir : tmpBasePath ,
Stdout : & outbuf ,
Stderr : & errbuf ,
2022-02-11 20:47:22 +08:00
} ) ; err != nil {
2019-11-10 16:42:51 +08:00
log . Error ( "git commit [%s:%s -> %s:%s]: %v\n%s\n%s" , pr . HeadRepo . FullName ( ) , pr . HeadBranch , pr . BaseRepo . FullName ( ) , pr . BaseBranch , err , outbuf . String ( ) , errbuf . String ( ) )
return fmt . Errorf ( "git commit [%s:%s -> %s:%s]: %v\n%s\n%s" , pr . HeadRepo . FullName ( ) , pr . HeadBranch , pr . BaseRepo . FullName ( ) , pr . BaseBranch , err , outbuf . String ( ) , errbuf . String ( ) )
}
} else {
2022-02-11 20:47:22 +08:00
if err := git . NewCommand ( ctx , "commit" , signArg , "-m" , message ) .
2022-04-01 10:55:30 +08:00
Run ( & git . RunOpts {
Env : env ,
Dir : tmpBasePath ,
Stdout : & outbuf ,
Stderr : & errbuf ,
2022-02-11 20:47:22 +08:00
} ) ; err != nil {
2019-11-10 16:42:51 +08:00
log . Error ( "git commit [%s:%s -> %s:%s]: %v\n%s\n%s" , pr . HeadRepo . FullName ( ) , pr . HeadBranch , pr . BaseRepo . FullName ( ) , pr . BaseBranch , err , outbuf . String ( ) , errbuf . String ( ) )
return fmt . Errorf ( "git commit [%s:%s -> %s:%s]: %v\n%s\n%s" , pr . HeadRepo . FullName ( ) , pr . HeadBranch , pr . BaseRepo . FullName ( ) , pr . BaseBranch , err , outbuf . String ( ) , errbuf . String ( ) )
}
}
return nil
}
2021-12-10 09:27:50 +08:00
func runMergeCommand ( pr * models . PullRequest , mergeStyle repo_model . MergeStyle , cmd * git . Command , tmpBasePath string ) error {
2019-11-10 16:42:51 +08:00
var outbuf , errbuf strings . Builder
2022-04-01 10:55:30 +08:00
if err := cmd . Run ( & git . RunOpts {
Dir : tmpBasePath ,
Stdout : & outbuf ,
Stderr : & errbuf ,
2022-02-11 20:47:22 +08:00
} ) ; err != nil {
2019-11-10 16:42:51 +08:00
// Merge will leave a MERGE_HEAD file in the .git folder if there is a conflict
if _ , statErr := os . Stat ( filepath . Join ( tmpBasePath , ".git" , "MERGE_HEAD" ) ) ; statErr == nil {
// We have a merge conflict error
log . Debug ( "MergeConflict [%s:%s -> %s:%s]: %v\n%s\n%s" , pr . HeadRepo . FullName ( ) , pr . HeadBranch , pr . BaseRepo . FullName ( ) , pr . BaseBranch , err , outbuf . String ( ) , errbuf . String ( ) )
return models . ErrMergeConflicts {
Style : mergeStyle ,
StdOut : outbuf . String ( ) ,
StdErr : errbuf . String ( ) ,
Err : err ,
}
} else if strings . Contains ( errbuf . String ( ) , "refusing to merge unrelated histories" ) {
log . Debug ( "MergeUnrelatedHistories [%s:%s -> %s:%s]: %v\n%s\n%s" , pr . HeadRepo . FullName ( ) , pr . HeadBranch , pr . BaseRepo . FullName ( ) , pr . BaseBranch , err , outbuf . String ( ) , errbuf . String ( ) )
return models . ErrMergeUnrelatedHistories {
Style : mergeStyle ,
StdOut : outbuf . String ( ) ,
StdErr : errbuf . String ( ) ,
Err : err ,
}
}
log . Error ( "git merge [%s:%s -> %s:%s]: %v\n%s\n%s" , pr . HeadRepo . FullName ( ) , pr . HeadBranch , pr . BaseRepo . FullName ( ) , pr . BaseBranch , err , outbuf . String ( ) , errbuf . String ( ) )
return fmt . Errorf ( "git merge [%s:%s -> %s:%s]: %v\n%s\n%s" , pr . HeadRepo . FullName ( ) , pr . HeadBranch , pr . BaseRepo . FullName ( ) , pr . BaseBranch , err , outbuf . String ( ) , errbuf . String ( ) )
}
return nil
}
2019-11-01 08:30:02 +08:00
var escapedSymbols = regexp . MustCompile ( ` ([*[?! \\]) ` )
2022-01-20 07:26:57 +08:00
func getDiffTree ( ctx context . Context , repoPath , baseBranch , headBranch string ) ( string , error ) {
2019-06-23 01:35:34 +08:00
getDiffTreeFromBranch := func ( repoPath , baseBranch , headBranch string ) ( string , error ) {
var outbuf , errbuf strings . Builder
// Compute the diff-tree for sparse-checkout
2022-02-11 20:47:22 +08:00
if err := git . NewCommand ( ctx , "diff-tree" , "--no-commit-id" , "--name-only" , "-r" , "-z" , "--root" , baseBranch , headBranch , "--" ) .
2022-04-01 10:55:30 +08:00
Run ( & git . RunOpts {
Dir : repoPath ,
Stdout : & outbuf ,
Stderr : & errbuf ,
2022-02-11 20:47:22 +08:00
} ) ; err != nil {
2019-06-23 01:35:34 +08:00
return "" , fmt . Errorf ( "git diff-tree [%s base:%s head:%s]: %s" , repoPath , baseBranch , headBranch , errbuf . String ( ) )
}
return outbuf . String ( ) , nil
}
2019-11-01 08:30:02 +08:00
scanNullTerminatedStrings := func ( data [ ] byte , atEOF bool ) ( advance int , token [ ] byte , err error ) {
if atEOF && len ( data ) == 0 {
return 0 , nil , nil
}
if i := bytes . IndexByte ( data , '\x00' ) ; i >= 0 {
return i + 1 , data [ 0 : i ] , nil
}
if atEOF {
return len ( data ) , data , nil
}
return 0 , nil , nil
}
2019-06-23 01:35:34 +08:00
list , err := getDiffTreeFromBranch ( repoPath , baseBranch , headBranch )
if err != nil {
return "" , err
}
// Prefixing '/' for each entry, otherwise all files with the same name in subdirectories would be matched.
out := bytes . Buffer { }
scanner := bufio . NewScanner ( strings . NewReader ( list ) )
2019-11-01 08:30:02 +08:00
scanner . Split ( scanNullTerminatedStrings )
2019-06-23 01:35:34 +08:00
for scanner . Scan ( ) {
2019-11-01 08:30:02 +08:00
filepath := scanner . Text ( )
// escape '*', '?', '[', spaces and '!' prefix
filepath = escapedSymbols . ReplaceAllString ( filepath , ` \$1 ` )
// no necessary to escape the first '#' symbol because the first symbol is '/'
fmt . Fprintf ( & out , "/%s\n" , filepath )
2019-06-23 01:35:34 +08:00
}
2019-11-01 08:30:02 +08:00
2019-06-23 01:35:34 +08:00
return out . String ( ) , nil
}
2020-01-11 15:29:34 +08:00
// IsUserAllowedToMerge check if user is allowed to merge PR with given permissions and branch protections
2022-05-11 18:09:36 +08:00
func IsUserAllowedToMerge ( ctx context . Context , pr * models . PullRequest , p access_model . Permission , user * user_model . User ) ( bool , error ) {
2020-04-15 00:29:31 +08:00
if user == nil {
return false , nil
}
2020-01-11 15:29:34 +08:00
2022-05-04 03:46:28 +08:00
err := pr . LoadProtectedBranchCtx ( ctx )
2020-01-11 15:29:34 +08:00
if err != nil {
return false , err
}
2022-05-04 03:46:28 +08:00
if ( p . CanWrite ( unit . TypeCode ) && pr . ProtectedBranch == nil ) || ( pr . ProtectedBranch != nil && models . IsUserMergeWhitelisted ( ctx , pr . ProtectedBranch , user . ID , p ) ) {
2020-01-11 15:29:34 +08:00
return true , nil
}
return false , nil
}
2022-05-02 07:54:44 +08:00
// CheckPullBranchProtections checks whether the PR is ready to be merged (reviews and status checks)
func CheckPullBranchProtections ( ctx context . Context , pr * models . PullRequest , skipProtectedFilesCheck bool ) ( err error ) {
2022-04-28 19:48:48 +08:00
if err = pr . LoadBaseRepoCtx ( ctx ) ; err != nil {
2020-03-03 06:31:55 +08:00
return fmt . Errorf ( "LoadBaseRepo: %v" , err )
}
2022-05-04 03:46:28 +08:00
if err = pr . LoadProtectedBranchCtx ( ctx ) ; err != nil {
2020-03-03 06:31:55 +08:00
return fmt . Errorf ( "LoadProtectedBranch: %v" , err )
2020-01-11 15:29:34 +08:00
}
if pr . ProtectedBranch == nil {
2020-03-03 06:31:55 +08:00
return nil
2020-01-11 15:29:34 +08:00
}
2022-01-20 07:26:57 +08:00
isPass , err := IsPullCommitStatusPass ( ctx , pr )
2020-01-11 15:29:34 +08:00
if err != nil {
return err
}
if ! isPass {
2022-03-31 22:53:08 +08:00
return models . ErrDisallowedToMerge {
2020-01-11 15:29:34 +08:00
Reason : "Not all required status checks successful" ,
}
}
2022-05-04 03:46:28 +08:00
if ! pr . ProtectedBranch . HasEnoughApprovals ( ctx , pr ) {
2022-03-31 22:53:08 +08:00
return models . ErrDisallowedToMerge {
2020-01-11 15:29:34 +08:00
Reason : "Does not have enough approvals" ,
}
}
2022-05-04 03:46:28 +08:00
if pr . ProtectedBranch . MergeBlockedByRejectedReview ( ctx , pr ) {
2022-03-31 22:53:08 +08:00
return models . ErrDisallowedToMerge {
2020-01-11 15:29:34 +08:00
Reason : "There are requested changes" ,
}
}
2022-05-04 03:46:28 +08:00
if pr . ProtectedBranch . MergeBlockedByOfficialReviewRequests ( ctx , pr ) {
2022-03-31 22:53:08 +08:00
return models . ErrDisallowedToMerge {
2020-11-29 03:30:46 +08:00
Reason : "There are official review requests" ,
}
}
2020-01-11 15:29:34 +08:00
2020-04-17 09:00:36 +08:00
if pr . ProtectedBranch . MergeBlockedByOutdatedBranch ( pr ) {
2022-03-31 22:53:08 +08:00
return models . ErrDisallowedToMerge {
2020-04-17 09:00:36 +08:00
Reason : "The head branch is behind the base branch" ,
}
}
2020-10-14 02:50:57 +08:00
if skipProtectedFilesCheck {
return nil
}
if pr . ProtectedBranch . MergeBlockedByProtectedFiles ( pr ) {
2022-03-31 22:53:08 +08:00
return models . ErrDisallowedToMerge {
2020-10-14 02:50:57 +08:00
Reason : "Changed protected files" ,
}
}
2020-01-11 15:29:34 +08:00
return nil
}
2021-03-04 11:41:23 +08:00
// MergedManually mark pr as merged manually
2022-05-04 03:46:28 +08:00
func MergedManually ( pr * models . PullRequest , doer * user_model . User , baseGitRepo * git . Repository , commitID string ) error {
2022-05-05 00:06:23 +08:00
pullWorkingPool . CheckIn ( fmt . Sprint ( pr . ID ) )
defer pullWorkingPool . CheckOut ( fmt . Sprint ( pr . ID ) )
2022-05-04 03:46:28 +08:00
if err := db . WithTx ( func ( ctx context . Context ) error {
prUnit , err := pr . BaseRepo . GetUnitCtx ( ctx , unit . TypePullRequests )
if err != nil {
return err
}
prConfig := prUnit . PullRequestsConfig ( )
2021-03-04 11:41:23 +08:00
2022-05-04 03:46:28 +08:00
// Check if merge style is correct and allowed
if ! prConfig . IsMergeStyleAllowed ( repo_model . MergeStyleManuallyMerged ) {
return models . ErrInvalidMergeStyle { ID : pr . BaseRepo . ID , Style : repo_model . MergeStyleManuallyMerged }
}
2021-03-04 11:41:23 +08:00
2022-05-04 03:46:28 +08:00
if len ( commitID ) < 40 {
2021-03-04 11:41:23 +08:00
return fmt . Errorf ( "Wrong commit ID" )
}
2022-05-04 03:46:28 +08:00
commit , err := baseGitRepo . GetCommit ( commitID )
if err != nil {
if git . IsErrNotExist ( err ) {
return fmt . Errorf ( "Wrong commit ID" )
}
return err
}
commitID = commit . ID . String ( )
2021-03-04 11:41:23 +08:00
2022-05-04 03:46:28 +08:00
ok , err := baseGitRepo . IsCommitInBranch ( commitID , pr . BaseBranch )
if err != nil {
return err
}
if ! ok {
return fmt . Errorf ( "Wrong commit ID" )
}
2021-03-04 11:41:23 +08:00
2022-05-04 03:46:28 +08:00
pr . MergedCommitID = commitID
pr . MergedUnix = timeutil . TimeStamp ( commit . Author . When . Unix ( ) )
pr . Status = models . PullRequestStatusManuallyMerged
pr . Merger = doer
pr . MergerID = doer . ID
merged := false
if merged , err = pr . SetMerged ( ctx ) ; err != nil {
return err
} else if ! merged {
return fmt . Errorf ( "SetMerged failed" )
}
return nil
} ) ; err != nil {
return err
2021-03-04 11:41:23 +08:00
}
notification . NotifyMergePullRequest ( pr , doer )
2022-05-04 03:46:28 +08:00
log . Info ( "manuallyMerged[%d]: Marked as manually merged into %s/%s by commit id: %s" , pr . ID , pr . BaseRepo . Name , pr . BaseBranch , commitID )
2021-03-04 11:41:23 +08:00
return nil
}