123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195 |
- // 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 (
- "context"
- "fmt"
- "os"
- "path/filepath"
- "strings"
-
- "code.gitea.io/gitea/models"
- repo_model "code.gitea.io/gitea/models/repo"
- "code.gitea.io/gitea/modules/git"
- "code.gitea.io/gitea/modules/log"
- repo_module "code.gitea.io/gitea/modules/repository"
- )
-
- // createTemporaryRepo creates a temporary repo with "base" for pr.BaseBranch and "tracking" for pr.HeadBranch
- // it also create a second base branch called "original_base"
- func createTemporaryRepo(ctx context.Context, pr *models.PullRequest) (string, error) {
- if err := pr.LoadHeadRepoCtx(ctx); err != nil {
- log.Error("LoadHeadRepo: %v", err)
- return "", fmt.Errorf("LoadHeadRepo: %v", err)
- } else if pr.HeadRepo == nil {
- log.Error("Pr %d HeadRepo %d does not exist", pr.ID, pr.HeadRepoID)
- return "", &repo_model.ErrRepoNotExist{
- ID: pr.HeadRepoID,
- }
- } else if err := pr.LoadBaseRepoCtx(ctx); err != nil {
- log.Error("LoadBaseRepo: %v", err)
- return "", fmt.Errorf("LoadBaseRepo: %v", err)
- } else if pr.BaseRepo == nil {
- log.Error("Pr %d BaseRepo %d does not exist", pr.ID, pr.BaseRepoID)
- return "", &repo_model.ErrRepoNotExist{
- ID: pr.BaseRepoID,
- }
- } else if err := pr.HeadRepo.GetOwner(ctx); err != nil {
- log.Error("HeadRepo.GetOwner: %v", err)
- return "", fmt.Errorf("HeadRepo.GetOwner: %v", err)
- } else if err := pr.BaseRepo.GetOwner(ctx); err != nil {
- log.Error("BaseRepo.GetOwner: %v", err)
- return "", fmt.Errorf("BaseRepo.GetOwner: %v", err)
- }
-
- // Clone base repo.
- tmpBasePath, err := repo_module.CreateTemporaryPath("pull")
- if err != nil {
- log.Error("CreateTemporaryPath: %v", err)
- return "", err
- }
-
- baseRepoPath := pr.BaseRepo.RepoPath()
- headRepoPath := pr.HeadRepo.RepoPath()
-
- if err := git.InitRepository(ctx, tmpBasePath, false); err != nil {
- log.Error("git init tmpBasePath: %v", err)
- if err := repo_module.RemoveTemporaryPath(tmpBasePath); err != nil {
- log.Error("CreateTempRepo: RemoveTemporaryPath: %s", err)
- }
- return "", err
- }
-
- remoteRepoName := "head_repo"
- baseBranch := "base"
-
- // Add head repo remote.
- addCacheRepo := func(staging, cache string) error {
- p := filepath.Join(staging, ".git", "objects", "info", "alternates")
- f, err := os.OpenFile(p, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0o600)
- if err != nil {
- log.Error("Could not create .git/objects/info/alternates file in %s: %v", staging, err)
- return err
- }
- defer f.Close()
- data := filepath.Join(cache, "objects")
- if _, err := fmt.Fprintln(f, data); err != nil {
- log.Error("Could not write to .git/objects/info/alternates file in %s: %v", staging, err)
- return err
- }
- return nil
- }
-
- if err := addCacheRepo(tmpBasePath, baseRepoPath); err != nil {
- log.Error("Unable to add base repository to temporary repo [%s -> %s]: %v", pr.BaseRepo.FullName(), tmpBasePath, err)
- if err := repo_module.RemoveTemporaryPath(tmpBasePath); err != nil {
- log.Error("CreateTempRepo: RemoveTemporaryPath: %s", err)
- }
- return "", fmt.Errorf("Unable to add base repository to temporary repo [%s -> tmpBasePath]: %v", pr.BaseRepo.FullName(), err)
- }
-
- var outbuf, errbuf strings.Builder
- if err := git.NewCommand(ctx, "remote", "add", "-t", pr.BaseBranch, "-m", pr.BaseBranch, "origin", baseRepoPath).
- Run(&git.RunOpts{
- Dir: tmpBasePath,
- Stdout: &outbuf,
- Stderr: &errbuf,
- }); err != nil {
- log.Error("Unable to add base repository as origin [%s -> %s]: %v\n%s\n%s", pr.BaseRepo.FullName(), tmpBasePath, err, outbuf.String(), errbuf.String())
- if err := repo_module.RemoveTemporaryPath(tmpBasePath); err != nil {
- log.Error("CreateTempRepo: RemoveTemporaryPath: %s", err)
- }
- return "", fmt.Errorf("Unable to add base repository as origin [%s -> tmpBasePath]: %v\n%s\n%s", pr.BaseRepo.FullName(), err, outbuf.String(), errbuf.String())
- }
- outbuf.Reset()
- errbuf.Reset()
-
- if err := git.NewCommand(ctx, "fetch", "origin", "--no-tags", "--", pr.BaseBranch+":"+baseBranch, pr.BaseBranch+":original_"+baseBranch).
- Run(&git.RunOpts{
- Dir: tmpBasePath,
- Stdout: &outbuf,
- Stderr: &errbuf,
- }); err != nil {
- log.Error("Unable to fetch origin base branch [%s:%s -> base, original_base in %s]: %v:\n%s\n%s", pr.BaseRepo.FullName(), pr.BaseBranch, tmpBasePath, err, outbuf.String(), errbuf.String())
- if err := repo_module.RemoveTemporaryPath(tmpBasePath); err != nil {
- log.Error("CreateTempRepo: RemoveTemporaryPath: %s", err)
- }
- return "", fmt.Errorf("Unable to fetch origin base branch [%s:%s -> base, original_base in tmpBasePath]: %v\n%s\n%s", pr.BaseRepo.FullName(), pr.BaseBranch, err, outbuf.String(), errbuf.String())
- }
- outbuf.Reset()
- errbuf.Reset()
-
- if err := git.NewCommand(ctx, "symbolic-ref", "HEAD", git.BranchPrefix+baseBranch).
- Run(&git.RunOpts{
- Dir: tmpBasePath,
- Stdout: &outbuf,
- Stderr: &errbuf,
- }); err != nil {
- log.Error("Unable to set HEAD as base branch [%s]: %v\n%s\n%s", tmpBasePath, err, outbuf.String(), errbuf.String())
- if err := repo_module.RemoveTemporaryPath(tmpBasePath); err != nil {
- log.Error("CreateTempRepo: RemoveTemporaryPath: %s", err)
- }
- return "", fmt.Errorf("Unable to set HEAD as base branch [tmpBasePath]: %v\n%s\n%s", err, outbuf.String(), errbuf.String())
- }
- outbuf.Reset()
- errbuf.Reset()
-
- if err := addCacheRepo(tmpBasePath, headRepoPath); err != nil {
- log.Error("Unable to add head repository to temporary repo [%s -> %s]: %v", pr.HeadRepo.FullName(), tmpBasePath, err)
- if err := repo_module.RemoveTemporaryPath(tmpBasePath); err != nil {
- log.Error("CreateTempRepo: RemoveTemporaryPath: %s", err)
- }
- return "", fmt.Errorf("Unable to head base repository to temporary repo [%s -> tmpBasePath]: %v", pr.HeadRepo.FullName(), err)
- }
-
- if err := git.NewCommand(ctx, "remote", "add", remoteRepoName, headRepoPath).
- Run(&git.RunOpts{
- Dir: tmpBasePath,
- Stdout: &outbuf,
- Stderr: &errbuf,
- }); err != nil {
- log.Error("Unable to add head repository as head_repo [%s -> %s]: %v\n%s\n%s", pr.HeadRepo.FullName(), tmpBasePath, err, outbuf.String(), errbuf.String())
- if err := repo_module.RemoveTemporaryPath(tmpBasePath); err != nil {
- log.Error("CreateTempRepo: RemoveTemporaryPath: %s", err)
- }
- return "", fmt.Errorf("Unable to add head repository as head_repo [%s -> tmpBasePath]: %v\n%s\n%s", pr.HeadRepo.FullName(), err, outbuf.String(), errbuf.String())
- }
- outbuf.Reset()
- errbuf.Reset()
-
- trackingBranch := "tracking"
- // Fetch head branch
- var headBranch string
- if pr.Flow == models.PullRequestFlowGithub {
- headBranch = git.BranchPrefix + pr.HeadBranch
- } else if len(pr.HeadCommitID) == 40 { // for not created pull request
- headBranch = pr.HeadCommitID
- } else {
- headBranch = pr.GetGitRefName()
- }
- if err := git.NewCommand(ctx, "fetch", "--no-tags", remoteRepoName, headBranch+":"+trackingBranch).
- Run(&git.RunOpts{
- Dir: tmpBasePath,
- Stdout: &outbuf,
- Stderr: &errbuf,
- }); err != nil {
- if err := repo_module.RemoveTemporaryPath(tmpBasePath); err != nil {
- log.Error("CreateTempRepo: RemoveTemporaryPath: %s", err)
- }
- if !git.IsBranchExist(ctx, pr.HeadRepo.RepoPath(), pr.HeadBranch) {
- return "", models.ErrBranchDoesNotExist{
- BranchName: pr.HeadBranch,
- }
- }
- log.Error("Unable to fetch head_repo head branch [%s:%s -> tracking in %s]: %v:\n%s\n%s", pr.HeadRepo.FullName(), pr.HeadBranch, tmpBasePath, err, outbuf.String(), errbuf.String())
- return "", fmt.Errorf("Unable to fetch head_repo head branch [%s:%s -> tracking in tmpBasePath]: %v\n%s\n%s", pr.HeadRepo.FullName(), headBranch, err, outbuf.String(), errbuf.String())
- }
- outbuf.Reset()
- errbuf.Reset()
-
- return tmpBasePath, nil
- }
|