123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169 |
- // Copyright 2019 The Gitea Authors.
- // All rights reserved.
- // SPDX-License-Identifier: MIT
-
- package pull
-
- import (
- "context"
-
- "code.gitea.io/gitea/models/db"
- git_model "code.gitea.io/gitea/models/git"
- issues_model "code.gitea.io/gitea/models/issues"
- "code.gitea.io/gitea/modules/git"
- "code.gitea.io/gitea/modules/log"
- "code.gitea.io/gitea/modules/structs"
-
- "github.com/gobwas/glob"
- "github.com/pkg/errors"
- )
-
- // MergeRequiredContextsCommitStatus returns a commit status state for given required contexts
- func MergeRequiredContextsCommitStatus(commitStatuses []*git_model.CommitStatus, requiredContexts []string) structs.CommitStatusState {
- // matchedCount is the number of `CommitStatus.Context` that match any context of `requiredContexts`
- matchedCount := 0
- returnedStatus := structs.CommitStatusSuccess
-
- if len(requiredContexts) > 0 {
- requiredContextsGlob := make(map[string]glob.Glob, len(requiredContexts))
- for _, ctx := range requiredContexts {
- if gp, err := glob.Compile(ctx); err != nil {
- log.Error("glob.Compile %s failed. Error: %v", ctx, err)
- } else {
- requiredContextsGlob[ctx] = gp
- }
- }
-
- for _, gp := range requiredContextsGlob {
- var targetStatus structs.CommitStatusState
- for _, commitStatus := range commitStatuses {
- if gp.Match(commitStatus.Context) {
- targetStatus = commitStatus.State
- matchedCount++
- break
- }
- }
-
- // If required rule not match any action, then it is pending
- if targetStatus == "" {
- if structs.CommitStatusPending.NoBetterThan(returnedStatus) {
- returnedStatus = structs.CommitStatusPending
- }
- break
- }
-
- if targetStatus.NoBetterThan(returnedStatus) {
- returnedStatus = targetStatus
- }
- }
- }
-
- if matchedCount == 0 && returnedStatus == structs.CommitStatusSuccess {
- status := git_model.CalcCommitStatus(commitStatuses)
- if status != nil {
- return status.State
- }
- return structs.CommitStatusSuccess
- }
-
- return returnedStatus
- }
-
- // IsCommitStatusContextSuccess returns true if all required status check contexts succeed.
- func IsCommitStatusContextSuccess(commitStatuses []*git_model.CommitStatus, requiredContexts []string) bool {
- // If no specific context is required, require that last commit status is a success
- if len(requiredContexts) == 0 {
- status := git_model.CalcCommitStatus(commitStatuses)
- if status == nil || status.State != structs.CommitStatusSuccess {
- return false
- }
- return true
- }
-
- for _, ctx := range requiredContexts {
- var found bool
- for _, commitStatus := range commitStatuses {
- if commitStatus.Context == ctx {
- if commitStatus.State != structs.CommitStatusSuccess {
- return false
- }
-
- found = true
- break
- }
- }
- if !found {
- return false
- }
- }
- return true
- }
-
- // IsPullCommitStatusPass returns if all required status checks PASS
- func IsPullCommitStatusPass(ctx context.Context, pr *issues_model.PullRequest) (bool, error) {
- pb, err := git_model.GetFirstMatchProtectedBranchRule(ctx, pr.BaseRepoID, pr.BaseBranch)
- if err != nil {
- return false, errors.Wrap(err, "GetLatestCommitStatus")
- }
- if pb == nil || !pb.EnableStatusCheck {
- return true, nil
- }
-
- state, err := GetPullRequestCommitStatusState(ctx, pr)
- if err != nil {
- return false, err
- }
- return state.IsSuccess(), nil
- }
-
- // GetPullRequestCommitStatusState returns pull request merged commit status state
- func GetPullRequestCommitStatusState(ctx context.Context, pr *issues_model.PullRequest) (structs.CommitStatusState, error) {
- // Ensure HeadRepo is loaded
- if err := pr.LoadHeadRepo(ctx); err != nil {
- return "", errors.Wrap(err, "LoadHeadRepo")
- }
-
- // check if all required status checks are successful
- headGitRepo, closer, err := git.RepositoryFromContextOrOpen(ctx, pr.HeadRepo.RepoPath())
- if err != nil {
- return "", errors.Wrap(err, "OpenRepository")
- }
- defer closer.Close()
-
- if pr.Flow == issues_model.PullRequestFlowGithub && !headGitRepo.IsBranchExist(pr.HeadBranch) {
- return "", errors.New("Head branch does not exist, can not merge")
- }
- if pr.Flow == issues_model.PullRequestFlowAGit && !git.IsReferenceExist(ctx, headGitRepo.Path, pr.GetGitRefName()) {
- return "", errors.New("Head branch does not exist, can not merge")
- }
-
- var sha string
- if pr.Flow == issues_model.PullRequestFlowGithub {
- sha, err = headGitRepo.GetBranchCommitID(pr.HeadBranch)
- } else {
- sha, err = headGitRepo.GetRefCommitID(pr.GetGitRefName())
- }
- if err != nil {
- return "", err
- }
-
- if err := pr.LoadBaseRepo(ctx); err != nil {
- return "", errors.Wrap(err, "LoadBaseRepo")
- }
-
- commitStatuses, _, err := git_model.GetLatestCommitStatus(ctx, pr.BaseRepo.ID, sha, db.ListOptions{ListAll: true})
- if err != nil {
- return "", errors.Wrap(err, "GetLatestCommitStatus")
- }
-
- pb, err := git_model.GetFirstMatchProtectedBranchRule(ctx, pr.BaseRepoID, pr.BaseBranch)
- if err != nil {
- return "", errors.Wrap(err, "LoadProtectedBranch")
- }
- var requiredContexts []string
- if pb != nil {
- requiredContexts = pb.StatusCheckContexts
- }
-
- return MergeRequiredContextsCommitStatus(commitStatuses, requiredContexts), nil
- }
|