153 lines · 4.1 KB · Go
RawDownload
// Copyright 2024 The Gitea Authors. All rights reserved.// SPDX-License-Identifier: MIT package issue import (	"context"	"fmt"	"time" 	issues_model "forgejo.org/models/issues"	org_model "forgejo.org/models/organization"	access_model "forgejo.org/models/perm/access"	"forgejo.org/models/unit"	user_model "forgejo.org/models/user"	"forgejo.org/modules/git"	"forgejo.org/modules/gitrepo"	"forgejo.org/modules/log"	"forgejo.org/modules/setting") func getMergeBase(repo *git.Repository, pr *issues_model.PullRequest, baseBranch, headBranch string) (string, error) {	// Add a temporary remote	tmpRemote := fmt.Sprintf("mergebase-%d-%d", pr.ID, time.Now().UnixNano())	if err := repo.AddRemote(tmpRemote, repo.Path, false); err != nil {		return "", fmt.Errorf("AddRemote: %w", err)	}	defer func() {		if err := repo.RemoveRemote(tmpRemote); err != nil {			log.Error("getMergeBase: RemoveRemote: %v", err)		}	}() 	mergeBase, _, err := repo.GetMergeBase(tmpRemote, baseBranch, headBranch)	return mergeBase, err} type ReviewRequestNotifier struct {	Comment    *issues_model.Comment	IsAdd      bool	Reviewer   *user_model.User	ReviewTeam *org_model.Team} func PullRequestCodeOwnersReview(ctx context.Context, issue *issues_model.Issue, pr *issues_model.PullRequest) ([]*ReviewRequestNotifier, error) {	files := []string{"CODEOWNERS", "docs/CODEOWNERS", ".gitea/CODEOWNERS"} 	if pr.IsWorkInProgress(ctx) {		return nil, nil	} 	if err := pr.LoadHeadRepo(ctx); err != nil {		return nil, err	} 	if err := pr.LoadBaseRepo(ctx); err != nil {		return nil, err	} 	if pr.BaseRepo.IsFork {		return nil, nil	} 	repo, err := gitrepo.OpenRepository(ctx, pr.BaseRepo)	if err != nil {		return nil, err	}	defer repo.Close() 	commit, err := repo.GetBranchCommit(pr.BaseRepo.DefaultBranch)	if err != nil {		return nil, err	} 	var data string	for _, file := range files {		if blob, err := commit.GetBlobByPath(file); err == nil {			data, err = blob.GetBlobContent(setting.UI.MaxDisplayFileSize)			if err == nil {				break			}		}	} 	rules, _ := issues_model.GetCodeOwnersFromContent(ctx, data) 	// get the mergebase	mergeBase, err := getMergeBase(repo, pr, git.BranchPrefix+pr.BaseBranch, pr.GetGitRefName())	if err != nil {		return nil, err	} 	// https://github.com/go-gitea/gitea/issues/29763, we need to get the files changed	// between the merge base and the head commit but not the base branch and the head commit	changedFiles, err := repo.GetFilesChangedBetween(mergeBase, pr.GetGitRefName())	if err != nil {		return nil, err	} 	uniqUsers := make(map[int64]*user_model.User)	uniqTeams := make(map[string]*org_model.Team)	for _, rule := range rules {		for _, f := range changedFiles {			if (rule.Rule.MatchString(f) && !rule.Negative) || (!rule.Rule.MatchString(f) && rule.Negative) {				for _, u := range rule.Users {					uniqUsers[u.ID] = u				}				for _, t := range rule.Teams {					uniqTeams[fmt.Sprintf("%d/%d", t.OrgID, t.ID)] = t				}			}		}	} 	notifiers := make([]*ReviewRequestNotifier, 0, len(uniqUsers)+len(uniqTeams)) 	if err := issue.LoadPoster(ctx); err != nil {		return nil, err	} 	for _, u := range uniqUsers {		permission, err := access_model.GetUserRepoPermission(ctx, issue.Repo, u)		if err != nil {			return nil, fmt.Errorf("GetUserRepoPermission: %w", err)		}		if u.ID != issue.Poster.ID && permission.CanRead(unit.TypePullRequests) {			comment, err := issues_model.AddReviewRequest(ctx, issue, u, issue.Poster)			if err != nil {				log.Warn("Failed add assignee user: %s to PR review: %s#%d, error: %s", u.Name, pr.BaseRepo.Name, pr.ID, err)				return nil, err			}			notifiers = append(notifiers, &ReviewRequestNotifier{				Comment:  comment,				IsAdd:    true,				Reviewer: u,			})		}	}	for _, t := range uniqTeams {		comment, err := issues_model.AddTeamReviewRequest(ctx, issue, t, issue.Poster)		if err != nil {			log.Warn("Failed add assignee team: %s to PR review: %s#%d, error: %s", t.Name, pr.BaseRepo.Name, pr.ID, err)			return nil, err		}		notifiers = append(notifiers, &ReviewRequestNotifier{			Comment:    comment,			IsAdd:      true,			ReviewTeam: t,		})	} 	return notifiers, nil}