mirror of
https://codeberg.org/forgejo/forgejo
synced 2024-12-12 22:41:59 +01:00
173 lines
5.1 KiB
Go
173 lines
5.1 KiB
Go
|
// Copyright 2023 The Gitea Authors. All rights reserved.
|
||
|
// SPDX-License-Identifier: MIT
|
||
|
|
||
|
package issues
|
||
|
|
||
|
import (
|
||
|
"context"
|
||
|
|
||
|
"code.gitea.io/gitea/models/db"
|
||
|
user_model "code.gitea.io/gitea/models/user"
|
||
|
"code.gitea.io/gitea/modules/container"
|
||
|
"code.gitea.io/gitea/modules/util"
|
||
|
|
||
|
"xorm.io/builder"
|
||
|
)
|
||
|
|
||
|
type ReviewList []*Review
|
||
|
|
||
|
// LoadReviewers loads reviewers
|
||
|
func (reviews ReviewList) LoadReviewers(ctx context.Context) error {
|
||
|
reviewerIds := make([]int64, len(reviews))
|
||
|
for i := 0; i < len(reviews); i++ {
|
||
|
reviewerIds[i] = reviews[i].ReviewerID
|
||
|
}
|
||
|
reviewers, err := user_model.GetPossibleUserByIDs(ctx, reviewerIds)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
userMap := make(map[int64]*user_model.User, len(reviewers))
|
||
|
for _, reviewer := range reviewers {
|
||
|
userMap[reviewer.ID] = reviewer
|
||
|
}
|
||
|
for _, review := range reviews {
|
||
|
review.Reviewer = userMap[review.ReviewerID]
|
||
|
}
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
func (reviews ReviewList) LoadIssues(ctx context.Context) error {
|
||
|
issueIds := container.Set[int64]{}
|
||
|
for i := 0; i < len(reviews); i++ {
|
||
|
issueIds.Add(reviews[i].IssueID)
|
||
|
}
|
||
|
|
||
|
issues, err := GetIssuesByIDs(ctx, issueIds.Values())
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
if _, err := issues.LoadRepositories(ctx); err != nil {
|
||
|
return err
|
||
|
}
|
||
|
issueMap := make(map[int64]*Issue, len(issues))
|
||
|
for _, issue := range issues {
|
||
|
issueMap[issue.ID] = issue
|
||
|
}
|
||
|
|
||
|
for _, review := range reviews {
|
||
|
review.Issue = issueMap[review.IssueID]
|
||
|
}
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
// FindReviewOptions represent possible filters to find reviews
|
||
|
type FindReviewOptions struct {
|
||
|
db.ListOptions
|
||
|
Type ReviewType
|
||
|
IssueID int64
|
||
|
ReviewerID int64
|
||
|
OfficialOnly bool
|
||
|
Dismissed util.OptionalBool
|
||
|
}
|
||
|
|
||
|
func (opts *FindReviewOptions) toCond() builder.Cond {
|
||
|
cond := builder.NewCond()
|
||
|
if opts.IssueID > 0 {
|
||
|
cond = cond.And(builder.Eq{"issue_id": opts.IssueID})
|
||
|
}
|
||
|
if opts.ReviewerID > 0 {
|
||
|
cond = cond.And(builder.Eq{"reviewer_id": opts.ReviewerID})
|
||
|
}
|
||
|
if opts.Type != ReviewTypeUnknown {
|
||
|
cond = cond.And(builder.Eq{"type": opts.Type})
|
||
|
}
|
||
|
if opts.OfficialOnly {
|
||
|
cond = cond.And(builder.Eq{"official": true})
|
||
|
}
|
||
|
if !opts.Dismissed.IsNone() {
|
||
|
cond = cond.And(builder.Eq{"dismissed": opts.Dismissed.IsTrue()})
|
||
|
}
|
||
|
return cond
|
||
|
}
|
||
|
|
||
|
// FindReviews returns reviews passing FindReviewOptions
|
||
|
func FindReviews(ctx context.Context, opts FindReviewOptions) (ReviewList, error) {
|
||
|
reviews := make([]*Review, 0, 10)
|
||
|
sess := db.GetEngine(ctx).Where(opts.toCond())
|
||
|
if opts.Page > 0 && !opts.IsListAll() {
|
||
|
sess = db.SetSessionPagination(sess, &opts)
|
||
|
}
|
||
|
return reviews, sess.
|
||
|
Asc("created_unix").
|
||
|
Asc("id").
|
||
|
Find(&reviews)
|
||
|
}
|
||
|
|
||
|
// FindLatestReviews returns only latest reviews per user, passing FindReviewOptions
|
||
|
func FindLatestReviews(ctx context.Context, opts FindReviewOptions) (ReviewList, error) {
|
||
|
reviews := make([]*Review, 0, 10)
|
||
|
cond := opts.toCond()
|
||
|
sess := db.GetEngine(ctx).Where(cond)
|
||
|
if opts.Page > 0 {
|
||
|
sess = db.SetSessionPagination(sess, &opts)
|
||
|
}
|
||
|
|
||
|
sess.In("id", builder.
|
||
|
Select("max ( id ) ").
|
||
|
From("review").
|
||
|
Where(cond).
|
||
|
GroupBy("reviewer_id"))
|
||
|
|
||
|
return reviews, sess.
|
||
|
Asc("created_unix").
|
||
|
Asc("id").
|
||
|
Find(&reviews)
|
||
|
}
|
||
|
|
||
|
// CountReviews returns count of reviews passing FindReviewOptions
|
||
|
func CountReviews(opts FindReviewOptions) (int64, error) {
|
||
|
return db.GetEngine(db.DefaultContext).Where(opts.toCond()).Count(&Review{})
|
||
|
}
|
||
|
|
||
|
// GetReviewersFromOriginalAuthorsByIssueID gets the latest review of each original authors for a pull request
|
||
|
func GetReviewersFromOriginalAuthorsByIssueID(issueID int64) (ReviewList, error) {
|
||
|
reviews := make([]*Review, 0, 10)
|
||
|
|
||
|
// Get latest review of each reviewer, sorted in order they were made
|
||
|
if err := db.GetEngine(db.DefaultContext).SQL("SELECT * FROM review WHERE id IN (SELECT max(id) as id FROM review WHERE issue_id = ? AND reviewer_team_id = 0 AND type in (?, ?, ?) AND original_author_id <> 0 GROUP BY issue_id, original_author_id) ORDER BY review.updated_unix ASC",
|
||
|
issueID, ReviewTypeApprove, ReviewTypeReject, ReviewTypeRequest).
|
||
|
Find(&reviews); err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
return reviews, nil
|
||
|
}
|
||
|
|
||
|
// GetReviewsByIssueID gets the latest review of each reviewer for a pull request
|
||
|
func GetReviewsByIssueID(issueID int64) (ReviewList, error) {
|
||
|
reviews := make([]*Review, 0, 10)
|
||
|
|
||
|
sess := db.GetEngine(db.DefaultContext)
|
||
|
|
||
|
// Get latest review of each reviewer, sorted in order they were made
|
||
|
if err := sess.SQL("SELECT * FROM review WHERE id IN (SELECT max(id) as id FROM review WHERE issue_id = ? AND reviewer_team_id = 0 AND type in (?, ?, ?) AND dismissed = ? AND original_author_id = 0 GROUP BY issue_id, reviewer_id) ORDER BY review.updated_unix ASC",
|
||
|
issueID, ReviewTypeApprove, ReviewTypeReject, ReviewTypeRequest, false).
|
||
|
Find(&reviews); err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
teamReviewRequests := make([]*Review, 0, 5)
|
||
|
if err := sess.SQL("SELECT * FROM review WHERE id IN (SELECT max(id) as id FROM review WHERE issue_id = ? AND reviewer_team_id <> 0 AND original_author_id = 0 GROUP BY issue_id, reviewer_team_id) ORDER BY review.updated_unix ASC",
|
||
|
issueID).
|
||
|
Find(&teamReviewRequests); err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
if len(teamReviewRequests) > 0 {
|
||
|
reviews = append(reviews, teamReviewRequests...)
|
||
|
}
|
||
|
|
||
|
return reviews, nil
|
||
|
}
|