2018-08-06 00:43:22 -04:00
// Copyright 2018 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 models
import (
2021-12-09 20:27:50 -05:00
"context"
2020-04-18 09:50:25 -04:00
"fmt"
2019-11-13 21:57:36 -05:00
"strings"
2018-08-06 00:43:22 -04:00
2021-09-19 07:49:59 -04:00
"code.gitea.io/gitea/models/db"
2022-03-29 02:29:02 -04:00
"code.gitea.io/gitea/models/organization"
2021-11-28 06:58:28 -05:00
"code.gitea.io/gitea/models/perm"
2022-05-11 06:09:36 -04:00
access_model "code.gitea.io/gitea/models/perm/access"
2021-11-09 14:57:58 -05:00
"code.gitea.io/gitea/models/unit"
2021-11-24 04:49:20 -05:00
user_model "code.gitea.io/gitea/models/user"
2020-10-12 15:55:13 -04:00
"code.gitea.io/gitea/modules/base"
2019-08-15 10:46:21 -04:00
"code.gitea.io/gitea/modules/timeutil"
2018-08-06 00:43:22 -04:00
2019-06-23 11:22:43 -04:00
"xorm.io/builder"
2018-08-06 00:43:22 -04:00
)
// ReviewType defines the sort of feedback a review gives
type ReviewType int
// ReviewTypeUnknown unknown review type
const ReviewTypeUnknown ReviewType = - 1
const (
// ReviewTypePending is a review which is not published yet
ReviewTypePending ReviewType = iota
// ReviewTypeApprove approves changes
ReviewTypeApprove
// ReviewTypeComment gives general feedback
ReviewTypeComment
// ReviewTypeReject gives feedback blocking merge
ReviewTypeReject
2020-04-06 12:33:34 -04:00
// ReviewTypeRequest request review from others
ReviewTypeRequest
2018-08-06 00:43:22 -04:00
)
// Icon returns the corresponding icon for the review type
func ( rt ReviewType ) Icon ( ) string {
switch rt {
case ReviewTypeApprove :
2020-04-03 01:12:42 -04:00
return "check"
2018-08-06 00:43:22 -04:00
case ReviewTypeReject :
2020-07-17 11:15:12 -04:00
return "diff"
2020-04-03 01:12:42 -04:00
case ReviewTypeComment :
2018-10-19 09:36:41 -04:00
return "comment"
2020-04-06 12:33:34 -04:00
case ReviewTypeRequest :
2020-07-17 11:15:12 -04:00
return "dot-fill"
2018-08-06 00:43:22 -04:00
default :
return "comment"
}
}
// Review represents collection of code comments giving feedback for a PR
type Review struct {
2020-01-23 12:28:15 -05:00
ID int64 ` xorm:"pk autoincr" `
Type ReviewType
2022-03-29 02:29:02 -04:00
Reviewer * user_model . User ` xorm:"-" `
ReviewerID int64 ` xorm:"index" `
ReviewerTeamID int64 ` xorm:"NOT NULL DEFAULT 0" `
ReviewerTeam * organization . Team ` xorm:"-" `
2020-01-23 12:28:15 -05:00
OriginalAuthor string
OriginalAuthorID int64
Issue * Issue ` xorm:"-" `
IssueID int64 ` xorm:"index" `
Content string ` xorm:"TEXT" `
2019-12-03 20:08:56 -05:00
// Official is a review made by an assigned approver (counts towards approval)
2021-02-11 12:32:25 -05:00
Official bool ` xorm:"NOT NULL DEFAULT false" `
CommitID string ` xorm:"VARCHAR(40)" `
Stale bool ` xorm:"NOT NULL DEFAULT false" `
Dismissed bool ` xorm:"NOT NULL DEFAULT false" `
2018-08-06 00:43:22 -04:00
2019-08-15 10:46:21 -04:00
CreatedUnix timeutil . TimeStamp ` xorm:"INDEX created" `
UpdatedUnix timeutil . TimeStamp ` xorm:"INDEX updated" `
2018-08-06 00:43:22 -04:00
// CodeComments are the initial code comments of the review
CodeComments CodeComments ` xorm:"-" `
2020-01-23 12:28:15 -05:00
Comments [ ] * Comment ` xorm:"-" `
2018-08-06 00:43:22 -04:00
}
2021-09-19 07:49:59 -04:00
func init ( ) {
db . RegisterModel ( new ( Review ) )
}
2022-01-19 18:26:57 -05:00
// LoadCodeComments loads CodeComments
func ( r * Review ) LoadCodeComments ( ctx context . Context ) ( err error ) {
2020-05-01 20:20:51 -04:00
if r . CodeComments != nil {
return
}
2022-05-20 10:08:52 -04:00
if err = r . loadIssue ( ctx ) ; err != nil {
2020-05-01 20:20:51 -04:00
return
2019-11-15 07:59:21 -05:00
}
2021-12-09 20:27:50 -05:00
r . CodeComments , err = fetchCodeCommentsByReview ( ctx , r . Issue , nil , r )
2018-08-06 00:43:22 -04:00
return
}
2022-05-20 10:08:52 -04:00
func ( r * Review ) loadIssue ( ctx context . Context ) ( err error ) {
2020-05-01 20:20:51 -04:00
if r . Issue != nil {
return
}
2022-05-20 10:08:52 -04:00
r . Issue , err = getIssueByID ( ctx , r . IssueID )
2018-08-06 00:43:22 -04:00
return
}
2022-05-20 10:08:52 -04:00
func ( r * Review ) loadReviewer ( ctx context . Context ) ( err error ) {
2020-10-12 15:55:13 -04:00
if r . ReviewerID == 0 || r . Reviewer != nil {
return
2018-08-06 00:43:22 -04:00
}
2022-05-20 10:08:52 -04:00
r . Reviewer , err = user_model . GetUserByIDCtx ( ctx , r . ReviewerID )
2018-08-06 00:43:22 -04:00
return
}
2022-03-29 02:29:02 -04:00
func ( r * Review ) loadReviewerTeam ( ctx context . Context ) ( err error ) {
2020-10-12 15:55:13 -04:00
if r . ReviewerTeamID == 0 || r . ReviewerTeam != nil {
return
}
2022-05-20 10:08:52 -04:00
r . ReviewerTeam , err = organization . GetTeamByID ( ctx , r . ReviewerTeamID )
2020-10-12 15:55:13 -04:00
return
}
2019-11-13 21:57:36 -05:00
// LoadReviewer loads reviewer
func ( r * Review ) LoadReviewer ( ) error {
2022-05-20 10:08:52 -04:00
return r . loadReviewer ( db . DefaultContext )
2019-11-13 21:57:36 -05:00
}
2020-10-12 15:55:13 -04:00
// LoadReviewerTeam loads reviewer team
func ( r * Review ) LoadReviewerTeam ( ) error {
2022-03-29 02:29:02 -04:00
return r . loadReviewerTeam ( db . DefaultContext )
2020-10-12 15:55:13 -04:00
}
2022-01-19 18:26:57 -05:00
// LoadAttributes loads all attributes except CodeComments
func ( r * Review ) LoadAttributes ( ctx context . Context ) ( err error ) {
2022-05-20 10:08:52 -04:00
if err = r . loadIssue ( ctx ) ; err != nil {
2018-08-06 00:43:22 -04:00
return
}
2022-01-19 18:26:57 -05:00
if err = r . LoadCodeComments ( ctx ) ; err != nil {
2020-05-01 20:20:51 -04:00
return
}
2022-05-20 10:08:52 -04:00
if err = r . loadReviewer ( ctx ) ; err != nil {
2018-08-06 00:43:22 -04:00
return
}
2022-03-29 02:29:02 -04:00
if err = r . loadReviewerTeam ( ctx ) ; err != nil {
2020-10-12 15:55:13 -04:00
return
}
2018-08-06 00:43:22 -04:00
return
}
2022-05-20 10:08:52 -04:00
// GetReviewByID returns the review by the given ID
func GetReviewByID ( ctx context . Context , id int64 ) ( * Review , error ) {
2018-08-06 00:43:22 -04:00
review := new ( Review )
2022-05-20 10:08:52 -04:00
if has , err := db . GetEngine ( ctx ) . ID ( id ) . Get ( review ) ; err != nil {
2018-08-06 00:43:22 -04:00
return nil , err
} else if ! has {
return nil , ErrReviewNotExist { ID : id }
} else {
return review , nil
}
}
// FindReviewOptions represent possible filters to find reviews
type FindReviewOptions struct {
2021-09-24 07:32:56 -04:00
db . ListOptions
2019-12-30 18:34:11 -05:00
Type ReviewType
IssueID int64
ReviewerID int64
OfficialOnly bool
2018-08-06 00:43:22 -04:00
}
func ( opts * FindReviewOptions ) toCond ( ) builder . Cond {
2021-03-14 14:52:12 -04:00
cond := builder . NewCond ( )
2018-08-06 00:43:22 -04:00
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 } )
}
2019-12-30 18:34:11 -05:00
if opts . OfficialOnly {
cond = cond . And ( builder . Eq { "official" : true } )
}
2018-08-06 00:43:22 -04:00
return cond
}
2022-05-20 10:08:52 -04:00
// FindReviews returns reviews passing FindReviewOptions
func FindReviews ( ctx context . Context , opts FindReviewOptions ) ( [ ] * Review , error ) {
2018-08-06 00:43:22 -04:00
reviews := make ( [ ] * Review , 0 , 10 )
2022-05-20 10:08:52 -04:00
sess := db . GetEngine ( ctx ) . Where ( opts . toCond ( ) )
2020-05-01 20:20:51 -04:00
if opts . Page > 0 {
2021-09-24 07:32:56 -04:00
sess = db . SetSessionPagination ( sess , & opts )
2020-05-01 20:20:51 -04:00
}
2018-08-06 00:43:22 -04:00
return reviews , sess .
Asc ( "created_unix" ) .
Asc ( "id" ) .
Find ( & reviews )
}
2021-08-12 08:43:08 -04:00
// CountReviews returns count of reviews passing FindReviewOptions
func CountReviews ( opts FindReviewOptions ) ( int64 , error ) {
2021-09-23 11:45:36 -04:00
return db . GetEngine ( db . DefaultContext ) . Where ( opts . toCond ( ) ) . Count ( & Review { } )
2021-08-12 08:43:08 -04:00
}
2018-08-06 00:43:22 -04:00
// CreateReviewOptions represent the options to create a review. Type, Issue and Reviewer are required.
type CreateReviewOptions struct {
2020-10-12 15:55:13 -04:00
Content string
Type ReviewType
Issue * Issue
2021-11-24 04:49:20 -05:00
Reviewer * user_model . User
2022-03-29 02:29:02 -04:00
ReviewerTeam * organization . Team
2020-10-12 15:55:13 -04:00
Official bool
CommitID string
Stale bool
2019-12-03 20:08:56 -05:00
}
2020-10-12 15:55:13 -04:00
// IsOfficialReviewer check if at least one of the provided reviewers can make official reviews in issue (counts towards required approvals)
2022-05-20 10:08:52 -04:00
func IsOfficialReviewer ( ctx context . Context , issue * Issue , reviewers ... * user_model . User ) ( bool , error ) {
pr , err := GetPullRequestByIssueID ( ctx , issue . ID )
2019-12-03 20:08:56 -05:00
if err != nil {
return false , err
}
2022-04-28 07:48:48 -04:00
if err = pr . LoadProtectedBranchCtx ( ctx ) ; err != nil {
2019-12-03 20:08:56 -05:00
return false , err
}
if pr . ProtectedBranch == nil {
return false , nil
}
2020-10-12 15:55:13 -04:00
for _ , reviewer := range reviewers {
2021-12-09 20:27:50 -05:00
official , err := isUserOfficialReviewer ( ctx , pr . ProtectedBranch , reviewer )
2020-10-12 15:55:13 -04:00
if official || err != nil {
return official , err
}
}
return false , nil
}
// IsOfficialReviewerTeam check if reviewer in this team can make official reviews in issue (counts towards required approvals)
2022-05-20 10:08:52 -04:00
func IsOfficialReviewerTeam ( ctx context . Context , issue * Issue , team * organization . Team ) ( bool , error ) {
pr , err := GetPullRequestByIssueID ( ctx , issue . ID )
2020-10-12 15:55:13 -04:00
if err != nil {
return false , err
}
2022-04-28 07:48:48 -04:00
if err = pr . LoadProtectedBranchCtx ( ctx ) ; err != nil {
2020-10-12 15:55:13 -04:00
return false , err
}
if pr . ProtectedBranch == nil {
return false , nil
}
if ! pr . ProtectedBranch . EnableApprovalsWhitelist {
2022-04-01 02:05:31 -04:00
return team . UnitAccessModeCtx ( ctx , unit . TypeCode ) >= perm . AccessModeWrite , nil
2020-10-12 15:55:13 -04:00
}
return base . Int64sContains ( pr . ProtectedBranch . ApprovalsWhitelistTeamIDs , team . ID ) , nil
2018-08-06 00:43:22 -04:00
}
2022-05-20 10:08:52 -04:00
// CreateReview creates a new review based on opts
func CreateReview ( ctx context . Context , opts CreateReviewOptions ) ( * Review , error ) {
2018-08-06 00:43:22 -04:00
review := & Review {
2020-10-12 15:55:13 -04:00
Type : opts . Type ,
Issue : opts . Issue ,
IssueID : opts . Issue . ID ,
Reviewer : opts . Reviewer ,
ReviewerTeam : opts . ReviewerTeam ,
Content : opts . Content ,
Official : opts . Official ,
CommitID : opts . CommitID ,
Stale : opts . Stale ,
}
if opts . Reviewer != nil {
review . ReviewerID = opts . Reviewer . ID
} else {
if review . Type != ReviewTypeRequest {
review . Type = ReviewTypeRequest
}
review . ReviewerTeamID = opts . ReviewerTeam . ID
2018-08-06 00:43:22 -04:00
}
2022-05-20 10:08:52 -04:00
return review , db . Insert ( ctx , review )
2018-08-06 00:43:22 -04:00
}
2022-05-20 10:08:52 -04:00
// GetCurrentReview returns the current pending review of reviewer for given issue
func GetCurrentReview ( ctx context . Context , reviewer * user_model . User , issue * Issue ) ( * Review , error ) {
2018-08-06 00:43:22 -04:00
if reviewer == nil {
return nil , nil
}
2022-05-20 10:08:52 -04:00
reviews , err := FindReviews ( ctx , FindReviewOptions {
2018-08-06 00:43:22 -04:00
Type : ReviewTypePending ,
IssueID : issue . ID ,
ReviewerID : reviewer . ID ,
} )
if err != nil {
return nil , err
}
if len ( reviews ) == 0 {
return nil , ErrReviewNotExist { }
}
2018-10-18 07:23:05 -04:00
reviews [ 0 ] . Reviewer = reviewer
reviews [ 0 ] . Issue = issue
2018-08-06 00:43:22 -04:00
return reviews [ 0 ] , nil
}
2019-11-24 00:46:16 -05:00
// ReviewExists returns whether a review exists for a particular line of code in the PR
func ReviewExists ( issue * Issue , treePath string , line int64 ) ( bool , error ) {
2021-09-23 11:45:36 -04:00
return db . GetEngine ( db . DefaultContext ) . Cols ( "id" ) . Exist ( & Comment { IssueID : issue . ID , TreePath : treePath , Line : line , Type : CommentTypeCode } )
2019-11-24 00:46:16 -05:00
}
2019-11-13 21:57:36 -05:00
// ContentEmptyErr represents an content empty error
2021-03-14 14:52:12 -04:00
type ContentEmptyErr struct { }
2019-11-13 21:57:36 -05:00
func ( ContentEmptyErr ) Error ( ) string {
return "Review content is empty"
}
// IsContentEmptyErr returns true if err is a ContentEmptyErr
func IsContentEmptyErr ( err error ) bool {
_ , ok := err . ( ContentEmptyErr )
return ok
}
// SubmitReview creates a review out of the existing pending review or creates a new one if no pending review exist
2021-11-24 04:49:20 -05:00
func SubmitReview ( doer * user_model . User , issue * Issue , reviewType ReviewType , content , commitID string , stale bool , attachmentUUIDs [ ] string ) ( * Review , * Comment , error ) {
2021-11-19 08:39:57 -05:00
ctx , committer , err := db . TxContext ( )
if err != nil {
2019-11-13 21:57:36 -05:00
return nil , nil , err
}
2021-11-19 08:39:57 -05:00
defer committer . Close ( )
sess := db . GetEngine ( ctx )
2019-11-13 21:57:36 -05:00
2021-03-14 14:52:12 -04:00
official := false
2019-12-03 20:08:56 -05:00
2022-05-20 10:08:52 -04:00
review , err := GetCurrentReview ( ctx , doer , issue )
2019-11-13 21:57:36 -05:00
if err != nil {
if ! IsErrReviewNotExist ( err ) {
return nil , nil , err
}
2019-11-14 15:58:01 -05:00
if reviewType != ReviewTypeApprove && len ( strings . TrimSpace ( content ) ) == 0 {
2019-11-13 21:57:36 -05:00
return nil , nil , ContentEmptyErr { }
}
2019-12-03 20:08:56 -05:00
if reviewType == ReviewTypeApprove || reviewType == ReviewTypeReject {
// Only reviewers latest review of type approve and reject shall count as "official", so existing reviews needs to be cleared
2022-05-20 10:08:52 -04:00
if _ , err := db . Exec ( ctx , "UPDATE `review` SET official=? WHERE issue_id=? AND reviewer_id=?" , false , issue . ID , doer . ID ) ; err != nil {
2019-12-03 20:08:56 -05:00
return nil , nil , err
}
2022-05-20 10:08:52 -04:00
if official , err = IsOfficialReviewer ( ctx , issue , doer ) ; err != nil {
2019-12-03 20:08:56 -05:00
return nil , nil , err
}
}
2019-11-13 21:57:36 -05:00
// No current review. Create a new one!
2022-05-20 10:08:52 -04:00
if review , err = CreateReview ( ctx , CreateReviewOptions {
2019-11-13 21:57:36 -05:00
Type : reviewType ,
Issue : issue ,
Reviewer : doer ,
Content : content ,
2019-12-03 20:08:56 -05:00
Official : official ,
2020-01-08 20:47:45 -05:00
CommitID : commitID ,
Stale : stale ,
2020-10-12 15:55:13 -04:00
} ) ; err != nil {
2019-11-13 21:57:36 -05:00
return nil , nil , err
}
} else {
2022-01-19 18:26:57 -05:00
if err := review . LoadCodeComments ( ctx ) ; err != nil {
2019-11-13 21:57:36 -05:00
return nil , nil , err
}
2019-11-14 15:58:01 -05:00
if reviewType != ReviewTypeApprove && len ( review . CodeComments ) == 0 && len ( strings . TrimSpace ( content ) ) == 0 {
2019-11-13 21:57:36 -05:00
return nil , nil , ContentEmptyErr { }
}
2019-12-03 20:08:56 -05:00
if reviewType == ReviewTypeApprove || reviewType == ReviewTypeReject {
// Only reviewers latest review of type approve and reject shall count as "official", so existing reviews needs to be cleared
2022-05-20 10:08:52 -04:00
if _ , err := db . Exec ( ctx , "UPDATE `review` SET official=? WHERE issue_id=? AND reviewer_id=?" , false , issue . ID , doer . ID ) ; err != nil {
2019-12-03 20:08:56 -05:00
return nil , nil , err
}
2022-05-20 10:08:52 -04:00
if official , err = IsOfficialReviewer ( ctx , issue , doer ) ; err != nil {
2019-12-03 20:08:56 -05:00
return nil , nil , err
}
}
review . Official = official
2019-11-13 21:57:36 -05:00
review . Issue = issue
review . Content = content
review . Type = reviewType
2020-01-08 20:47:45 -05:00
review . CommitID = commitID
review . Stale = stale
2019-12-03 20:08:56 -05:00
2020-01-08 20:47:45 -05:00
if _ , err := sess . ID ( review . ID ) . Cols ( "content, type, official, commit_id, stale" ) . Update ( review ) ; err != nil {
2019-11-13 21:57:36 -05:00
return nil , nil , err
}
2018-08-06 00:43:22 -04:00
}
2019-11-13 21:57:36 -05:00
2022-04-08 05:11:15 -04:00
comm , err := CreateCommentCtx ( ctx , & CreateCommentOptions {
2021-06-14 21:12:33 -04:00
Type : CommentTypeReview ,
Doer : doer ,
Content : review . Content ,
Issue : issue ,
Repo : issue . Repo ,
ReviewID : review . ID ,
Attachments : attachmentUUIDs ,
2019-11-13 21:57:36 -05:00
} )
if err != nil || comm == nil {
return nil , nil , err
}
2020-10-12 15:55:13 -04:00
// try to remove team review request if need
if issue . Repo . Owner . IsOrganization ( ) && ( reviewType == ReviewTypeApprove || reviewType == ReviewTypeReject ) {
teamReviewRequests := make ( [ ] * Review , 0 , 10 )
2021-10-07 16:39:59 -04:00
if err := sess . SQL ( "SELECT * FROM review WHERE issue_id = ? AND reviewer_team_id > 0 AND type = ?" , issue . ID , ReviewTypeRequest ) . Find ( & teamReviewRequests ) ; err != nil {
2020-10-12 15:55:13 -04:00
return nil , nil , err
}
for _ , teamReviewRequest := range teamReviewRequests {
2022-03-29 02:29:02 -04:00
ok , err := organization . IsTeamMember ( ctx , issue . Repo . OwnerID , teamReviewRequest . ReviewerTeamID , doer . ID )
2020-10-12 15:55:13 -04:00
if err != nil {
return nil , nil , err
} else if ! ok {
continue
}
2022-05-20 10:08:52 -04:00
if _ , err := sess . ID ( teamReviewRequest . ID ) . NoAutoCondition ( ) . Delete ( teamReviewRequest ) ; err != nil {
2020-10-12 15:55:13 -04:00
return nil , nil , err
}
}
}
2019-11-13 21:57:36 -05:00
comm . Review = review
2021-11-19 08:39:57 -05:00
return review , comm , committer . Commit ( )
2018-08-06 00:43:22 -04:00
}
2018-11-22 08:17:36 -05:00
2019-12-03 20:08:56 -05:00
// GetReviewersByIssueID gets the latest review of each reviewer for a pull request
2020-10-12 15:55:13 -04:00
func GetReviewersByIssueID ( issueID int64 ) ( [ ] * Review , error ) {
reviews := make ( [ ] * Review , 0 , 10 )
2018-11-22 08:17:36 -05:00
2021-11-21 10:41:00 -05:00
sess := db . GetEngine ( db . DefaultContext )
2018-11-22 08:17:36 -05:00
2021-07-08 07:38:13 -04:00
// Get latest review of each reviewer, sorted in order they were made
2021-02-11 12:32:25 -05:00
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 ) .
2020-10-12 15:55:13 -04:00
Find ( & reviews ) ; err != nil {
2019-12-03 20:08:56 -05:00
return nil , err
}
2020-10-12 15:55:13 -04:00
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 ... )
2018-11-22 08:17:36 -05:00
}
2019-12-03 20:08:56 -05:00
return reviews , nil
2018-11-22 08:17:36 -05:00
}
2020-01-08 20:47:45 -05:00
2020-10-14 08:11:11 -04:00
// GetReviewersFromOriginalAuthorsByIssueID gets the latest review of each original authors for a pull request
func GetReviewersFromOriginalAuthorsByIssueID ( issueID int64 ) ( [ ] * Review , error ) {
reviews := make ( [ ] * Review , 0 , 10 )
2021-07-08 07:38:13 -04:00
// Get latest review of each reviewer, sorted in order they were made
2021-09-23 11:45:36 -04:00
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" ,
2020-10-14 08:11:11 -04:00
issueID , ReviewTypeApprove , ReviewTypeReject , ReviewTypeRequest ) .
Find ( & reviews ) ; err != nil {
return nil , err
}
return reviews , nil
}
2020-10-12 15:55:13 -04:00
// GetReviewByIssueIDAndUserID get the latest review of reviewer for a pull request
2022-05-20 10:08:52 -04:00
func GetReviewByIssueIDAndUserID ( ctx context . Context , issueID , userID int64 ) ( * Review , error ) {
2020-10-12 15:55:13 -04:00
review := new ( Review )
2020-04-06 12:33:34 -04:00
2022-05-20 10:08:52 -04:00
has , err := db . GetEngine ( ctx ) . SQL ( "SELECT * FROM review WHERE id IN (SELECT max(id) as id FROM review WHERE issue_id = ? AND reviewer_id = ? AND original_author_id = 0 AND type in (?, ?, ?))" ,
2020-04-06 12:33:34 -04:00
issueID , userID , ReviewTypeApprove , ReviewTypeReject , ReviewTypeRequest ) .
2020-10-12 15:55:13 -04:00
Get ( review )
if err != nil {
return nil , err
}
if ! has {
return nil , ErrReviewNotExist { }
}
return review , nil
}
2022-01-10 04:32:37 -05:00
// GetTeamReviewerByIssueIDAndTeamID get the latest review request of reviewer team for a pull request
2022-05-20 10:08:52 -04:00
func GetTeamReviewerByIssueIDAndTeamID ( ctx context . Context , issueID , teamID int64 ) ( review * Review , err error ) {
2020-10-12 15:55:13 -04:00
review = new ( Review )
has := false
2022-05-20 10:08:52 -04:00
if has , err = db . GetEngine ( ctx ) . SQL ( "SELECT * FROM review WHERE id IN (SELECT max(id) as id FROM review WHERE issue_id = ? AND reviewer_team_id = ?)" ,
2020-10-12 15:55:13 -04:00
issueID , teamID ) .
2020-04-06 12:33:34 -04:00
Get ( review ) ; err != nil {
return nil , err
}
2020-10-12 15:55:13 -04:00
if ! has {
return nil , ErrReviewNotExist { 0 }
}
2020-04-06 12:33:34 -04:00
return
}
2020-01-08 20:47:45 -05:00
// MarkReviewsAsStale marks existing reviews as stale
func MarkReviewsAsStale ( issueID int64 ) ( err error ) {
2021-09-23 11:45:36 -04:00
_ , err = db . GetEngine ( db . DefaultContext ) . Exec ( "UPDATE `review` SET stale=? WHERE issue_id=?" , true , issueID )
2020-01-08 20:47:45 -05:00
return
}
// MarkReviewsAsNotStale marks existing reviews as not stale for a giving commit SHA
func MarkReviewsAsNotStale ( issueID int64 , commitID string ) ( err error ) {
2021-09-23 11:45:36 -04:00
_ , err = db . GetEngine ( db . DefaultContext ) . Exec ( "UPDATE `review` SET stale=? WHERE issue_id=? AND commit_id=?" , false , issueID , commitID )
2020-01-08 20:47:45 -05:00
return
}
2020-01-23 12:28:15 -05:00
2021-02-11 12:32:25 -05:00
// DismissReview change the dismiss status of a review
func DismissReview ( review * Review , isDismiss bool ) ( err error ) {
if review . Dismissed == isDismiss || ( review . Type != ReviewTypeApprove && review . Type != ReviewTypeReject ) {
return nil
}
review . Dismissed = isDismiss
2021-04-15 06:03:11 -04:00
if review . ID == 0 {
return ErrReviewNotExist { }
}
2021-09-23 11:45:36 -04:00
_ , err = db . GetEngine ( db . DefaultContext ) . ID ( review . ID ) . Cols ( "dismissed" ) . Update ( review )
2021-02-11 12:32:25 -05:00
return
}
2020-01-23 12:28:15 -05:00
// InsertReviews inserts review and review comments
func InsertReviews ( reviews [ ] * Review ) error {
2021-11-21 10:41:00 -05:00
ctx , committer , err := db . TxContext ( )
if err != nil {
2020-01-23 12:28:15 -05:00
return err
}
2021-11-21 10:41:00 -05:00
defer committer . Close ( )
sess := db . GetEngine ( ctx )
2020-01-23 12:28:15 -05:00
for _ , review := range reviews {
if _ , err := sess . NoAutoTime ( ) . Insert ( review ) ; err != nil {
return err
}
if _ , err := sess . NoAutoTime ( ) . Insert ( & Comment {
Type : CommentTypeReview ,
Content : review . Content ,
PosterID : review . ReviewerID ,
OriginalAuthor : review . OriginalAuthor ,
OriginalAuthorID : review . OriginalAuthorID ,
IssueID : review . IssueID ,
ReviewID : review . ID ,
CreatedUnix : review . CreatedUnix ,
UpdatedUnix : review . UpdatedUnix ,
} ) ; err != nil {
return err
}
for _ , c := range review . Comments {
c . ReviewID = review . ID
}
2020-04-19 23:04:08 -04:00
if len ( review . Comments ) > 0 {
if _ , err := sess . NoAutoTime ( ) . Insert ( review . Comments ) ; err != nil {
return err
}
2020-01-23 12:28:15 -05:00
}
}
2021-11-21 10:41:00 -05:00
return committer . Commit ( )
2020-01-23 12:28:15 -05:00
}
2020-04-06 12:33:34 -04:00
2020-04-30 16:24:08 -04:00
// AddReviewRequest add a review request from one reviewer
2021-11-24 04:49:20 -05:00
func AddReviewRequest ( issue * Issue , reviewer , doer * user_model . User ) ( * Comment , error ) {
2021-11-19 08:39:57 -05:00
ctx , committer , err := db . TxContext ( )
if err != nil {
2020-04-06 12:33:34 -04:00
return nil , err
}
2021-11-19 08:39:57 -05:00
defer committer . Close ( )
sess := db . GetEngine ( ctx )
2020-04-06 12:33:34 -04:00
2022-05-20 10:08:52 -04:00
review , err := GetReviewByIssueIDAndUserID ( ctx , issue . ID , reviewer . ID )
2020-10-12 15:55:13 -04:00
if err != nil && ! IsErrReviewNotExist ( err ) {
2020-04-06 12:33:34 -04:00
return nil , err
}
2020-10-12 15:55:13 -04:00
// skip it when reviewer hase been request to review
if review != nil && review . Type == ReviewTypeRequest {
return nil , nil
2020-04-06 12:33:34 -04:00
}
2022-05-20 10:08:52 -04:00
official , err := IsOfficialReviewer ( ctx , issue , reviewer , doer )
2020-10-12 15:55:13 -04:00
if err != nil {
return nil , err
} else if official {
2020-04-06 12:33:34 -04:00
if _ , err := sess . Exec ( "UPDATE `review` SET official=? WHERE issue_id=? AND reviewer_id=?" , false , issue . ID , reviewer . ID ) ; err != nil {
return nil , err
}
}
2022-05-20 10:08:52 -04:00
review , err = CreateReview ( ctx , CreateReviewOptions {
2020-04-06 12:33:34 -04:00
Type : ReviewTypeRequest ,
Issue : issue ,
Reviewer : reviewer ,
Official : official ,
Stale : false ,
2020-10-20 14:18:25 -04:00
} )
if err != nil {
2020-10-12 15:55:13 -04:00
return nil , err
2020-04-06 12:33:34 -04:00
}
2022-04-08 05:11:15 -04:00
comment , err := CreateCommentCtx ( ctx , & CreateCommentOptions {
2020-04-06 12:33:34 -04:00
Type : CommentTypeReviewRequest ,
Doer : doer ,
Repo : issue . Repo ,
Issue : issue ,
RemovedAssignee : false , // Use RemovedAssignee as !isRequest
AssigneeID : reviewer . ID , // Use AssigneeID as reviewer ID
2020-10-20 14:18:25 -04:00
ReviewID : review . ID ,
2020-04-06 12:33:34 -04:00
} )
if err != nil {
return nil , err
}
2021-11-19 08:39:57 -05:00
return comment , committer . Commit ( )
2020-04-06 12:33:34 -04:00
}
2021-03-14 14:52:12 -04:00
// RemoveReviewRequest remove a review request from one reviewer
2021-11-24 04:49:20 -05:00
func RemoveReviewRequest ( issue * Issue , reviewer , doer * user_model . User ) ( * Comment , error ) {
2021-11-19 08:39:57 -05:00
ctx , committer , err := db . TxContext ( )
if err != nil {
2020-10-12 15:55:13 -04:00
return nil , err
}
2021-11-19 08:39:57 -05:00
defer committer . Close ( )
2020-10-12 15:55:13 -04:00
2022-05-20 10:08:52 -04:00
review , err := GetReviewByIssueIDAndUserID ( ctx , issue . ID , reviewer . ID )
2020-10-12 15:55:13 -04:00
if err != nil && ! IsErrReviewNotExist ( err ) {
return nil , err
2020-04-06 12:33:34 -04:00
}
2020-10-12 15:55:13 -04:00
if review == nil || review . Type != ReviewTypeRequest {
2020-04-06 12:33:34 -04:00
return nil , nil
}
2022-05-20 10:08:52 -04:00
if _ , err = db . DeleteByBean ( ctx , review ) ; err != nil {
2020-10-12 15:55:13 -04:00
return nil , err
}
2022-05-20 10:08:52 -04:00
official , err := IsOfficialReviewer ( ctx , issue , reviewer )
2020-10-12 15:55:13 -04:00
if err != nil {
return nil , err
} else if official {
// recalculate the latest official review for reviewer
2022-05-20 10:08:52 -04:00
review , err := GetReviewByIssueIDAndUserID ( ctx , issue . ID , reviewer . ID )
2020-10-12 15:55:13 -04:00
if err != nil && ! IsErrReviewNotExist ( err ) {
return nil , err
}
if review != nil {
2022-05-20 10:08:52 -04:00
if _ , err := db . Exec ( ctx , "UPDATE `review` SET official=? WHERE id=?" , true , review . ID ) ; err != nil {
2020-10-12 15:55:13 -04:00
return nil , err
}
}
}
2022-04-08 05:11:15 -04:00
comment , err := CreateCommentCtx ( ctx , & CreateCommentOptions {
2020-10-12 15:55:13 -04:00
Type : CommentTypeReviewRequest ,
Doer : doer ,
Repo : issue . Repo ,
Issue : issue ,
RemovedAssignee : true , // Use RemovedAssignee as !isRequest
AssigneeID : reviewer . ID , // Use AssigneeID as reviewer ID
} )
if err != nil {
return nil , err
}
2021-11-19 08:39:57 -05:00
return comment , committer . Commit ( )
2020-10-12 15:55:13 -04:00
}
// AddTeamReviewRequest add a review request from one team
2022-03-29 02:29:02 -04:00
func AddTeamReviewRequest ( issue * Issue , reviewer * organization . Team , doer * user_model . User ) ( * Comment , error ) {
2021-11-19 08:39:57 -05:00
ctx , committer , err := db . TxContext ( )
if err != nil {
2020-04-06 12:33:34 -04:00
return nil , err
}
2021-11-19 08:39:57 -05:00
defer committer . Close ( )
2020-04-06 12:33:34 -04:00
2022-05-20 10:08:52 -04:00
review , err := GetTeamReviewerByIssueIDAndTeamID ( ctx , issue . ID , reviewer . ID )
2020-10-12 15:55:13 -04:00
if err != nil && ! IsErrReviewNotExist ( err ) {
2020-04-06 12:33:34 -04:00
return nil , err
}
2020-10-12 15:55:13 -04:00
// This team already has been requested to review - therefore skip this.
if review != nil {
return nil , nil
}
2022-05-20 10:08:52 -04:00
official , err := IsOfficialReviewerTeam ( ctx , issue , reviewer )
2020-04-06 12:33:34 -04:00
if err != nil {
2020-10-12 15:55:13 -04:00
return nil , fmt . Errorf ( "isOfficialReviewerTeam(): %v" , err )
} else if ! official {
2022-05-20 10:08:52 -04:00
if official , err = IsOfficialReviewer ( ctx , issue , doer ) ; err != nil {
2020-10-12 15:55:13 -04:00
return nil , fmt . Errorf ( "isOfficialReviewer(): %v" , err )
}
}
2022-05-20 10:08:52 -04:00
if review , err = CreateReview ( ctx , CreateReviewOptions {
2020-10-12 15:55:13 -04:00
Type : ReviewTypeRequest ,
Issue : issue ,
ReviewerTeam : reviewer ,
Official : official ,
Stale : false ,
} ) ; err != nil {
return nil , err
2020-04-06 12:33:34 -04:00
}
if official {
2022-05-20 10:08:52 -04:00
if _ , err := db . Exec ( ctx , "UPDATE `review` SET official=? WHERE issue_id=? AND reviewer_team_id=?" , false , issue . ID , reviewer . ID ) ; err != nil {
2020-10-12 15:55:13 -04:00
return nil , err
}
}
2022-04-08 05:11:15 -04:00
comment , err := CreateCommentCtx ( ctx , & CreateCommentOptions {
2020-10-12 15:55:13 -04:00
Type : CommentTypeReviewRequest ,
Doer : doer ,
Repo : issue . Repo ,
Issue : issue ,
RemovedAssignee : false , // Use RemovedAssignee as !isRequest
AssigneeTeamID : reviewer . ID , // Use AssigneeTeamID as reviewer team ID
2020-10-20 14:18:25 -04:00
ReviewID : review . ID ,
2020-10-12 15:55:13 -04:00
} )
if err != nil {
2022-04-08 05:11:15 -04:00
return nil , fmt . Errorf ( "CreateCommentCtx(): %v" , err )
2020-10-12 15:55:13 -04:00
}
2021-11-19 08:39:57 -05:00
return comment , committer . Commit ( )
2020-10-12 15:55:13 -04:00
}
2021-03-14 14:52:12 -04:00
// RemoveTeamReviewRequest remove a review request from one team
2022-03-29 02:29:02 -04:00
func RemoveTeamReviewRequest ( issue * Issue , reviewer * organization . Team , doer * user_model . User ) ( * Comment , error ) {
2021-11-19 08:39:57 -05:00
ctx , committer , err := db . TxContext ( )
if err != nil {
2020-10-12 15:55:13 -04:00
return nil , err
}
2021-11-19 08:39:57 -05:00
defer committer . Close ( )
2020-10-12 15:55:13 -04:00
2022-05-20 10:08:52 -04:00
review , err := GetTeamReviewerByIssueIDAndTeamID ( ctx , issue . ID , reviewer . ID )
2020-10-12 15:55:13 -04:00
if err != nil && ! IsErrReviewNotExist ( err ) {
return nil , err
}
if review == nil {
return nil , nil
}
2020-04-06 12:33:34 -04:00
2022-05-20 10:08:52 -04:00
if _ , err = db . DeleteByBean ( ctx , review ) ; err != nil {
2020-10-12 15:55:13 -04:00
return nil , err
}
2022-05-20 10:08:52 -04:00
official , err := IsOfficialReviewerTeam ( ctx , issue , reviewer )
2020-10-12 15:55:13 -04:00
if err != nil {
return nil , fmt . Errorf ( "isOfficialReviewerTeam(): %v" , err )
}
if official {
// recalculate which is the latest official review from that team
2022-05-20 10:08:52 -04:00
review , err := GetReviewByIssueIDAndUserID ( ctx , issue . ID , - reviewer . ID )
2020-10-12 15:55:13 -04:00
if err != nil && ! IsErrReviewNotExist ( err ) {
2020-04-06 12:33:34 -04:00
return nil , err
}
if review != nil {
2022-05-20 10:08:52 -04:00
if _ , err := db . Exec ( ctx , "UPDATE `review` SET official=? WHERE id=?" , true , review . ID ) ; err != nil {
2020-04-06 12:33:34 -04:00
return nil , err
}
}
}
2020-10-12 15:55:13 -04:00
if doer == nil {
2021-11-19 08:39:57 -05:00
return nil , committer . Commit ( )
2020-04-06 12:33:34 -04:00
}
2022-04-08 05:11:15 -04:00
comment , err := CreateCommentCtx ( ctx , & CreateCommentOptions {
2020-04-06 12:33:34 -04:00
Type : CommentTypeReviewRequest ,
Doer : doer ,
Repo : issue . Repo ,
Issue : issue ,
RemovedAssignee : true , // Use RemovedAssignee as !isRequest
2020-10-12 15:55:13 -04:00
AssigneeTeamID : reviewer . ID , // Use AssigneeTeamID as reviewer team ID
2020-04-06 12:33:34 -04:00
} )
if err != nil {
2022-04-08 05:11:15 -04:00
return nil , fmt . Errorf ( "CreateCommentCtx(): %v" , err )
2020-04-06 12:33:34 -04:00
}
2021-11-19 08:39:57 -05:00
return comment , committer . Commit ( )
2020-04-06 12:33:34 -04:00
}
2020-04-18 09:50:25 -04:00
// MarkConversation Add or remove Conversation mark for a code comment
2021-11-24 04:49:20 -05:00
func MarkConversation ( comment * Comment , doer * user_model . User , isResolve bool ) ( err error ) {
2020-04-18 09:50:25 -04:00
if comment . Type != CommentTypeCode {
return nil
}
if isResolve {
if comment . ResolveDoerID != 0 {
return nil
}
2021-09-23 11:45:36 -04:00
if _ , err = db . GetEngine ( db . DefaultContext ) . Exec ( "UPDATE `comment` SET resolve_doer_id=? WHERE id=?" , doer . ID , comment . ID ) ; err != nil {
2020-04-18 09:50:25 -04:00
return err
}
} else {
if comment . ResolveDoerID == 0 {
return nil
}
2021-09-23 11:45:36 -04:00
if _ , err = db . GetEngine ( db . DefaultContext ) . Exec ( "UPDATE `comment` SET resolve_doer_id=? WHERE id=?" , 0 , comment . ID ) ; err != nil {
2020-04-18 09:50:25 -04:00
return err
}
}
return nil
}
// CanMarkConversation Add or remove Conversation mark for a code comment permission check
// the PR writer , offfcial reviewer and poster can do it
2021-11-24 04:49:20 -05:00
func CanMarkConversation ( issue * Issue , doer * user_model . User ) ( permResult bool , err error ) {
2020-04-18 09:50:25 -04:00
if doer == nil || issue == nil {
return false , fmt . Errorf ( "issue or doer is nil" )
}
if doer . ID != issue . PosterID {
2022-04-08 05:11:15 -04:00
if err = issue . LoadRepo ( db . DefaultContext ) ; err != nil {
2020-04-18 09:50:25 -04:00
return false , err
}
2022-05-11 06:09:36 -04:00
p , err := access_model . GetUserRepoPermission ( db . DefaultContext , issue . Repo , doer )
2020-04-18 09:50:25 -04:00
if err != nil {
return false , err
}
2021-11-28 06:58:28 -05:00
permResult = p . CanAccess ( perm . AccessModeWrite , unit . TypePullRequests )
2020-04-18 09:50:25 -04:00
if ! permResult {
2022-05-20 10:08:52 -04:00
if permResult , err = IsOfficialReviewer ( db . DefaultContext , issue , doer ) ; err != nil {
2020-04-18 09:50:25 -04:00
return false , err
}
}
if ! permResult {
return false , nil
}
}
return true , nil
}
2020-05-01 20:20:51 -04:00
// DeleteReview delete a review and it's code comments
func DeleteReview ( r * Review ) error {
2021-11-21 10:41:00 -05:00
ctx , committer , err := db . TxContext ( )
if err != nil {
2020-05-01 20:20:51 -04:00
return err
}
2021-11-21 10:41:00 -05:00
defer committer . Close ( )
sess := db . GetEngine ( ctx )
2020-05-01 20:20:51 -04:00
if r . ID == 0 {
return fmt . Errorf ( "review is not allowed to be 0" )
}
2020-10-20 14:18:25 -04:00
if r . Type == ReviewTypeRequest {
return fmt . Errorf ( "review request can not be deleted using this method" )
}
2020-05-01 20:20:51 -04:00
opts := FindCommentsOptions {
Type : CommentTypeCode ,
IssueID : r . IssueID ,
ReviewID : r . ID ,
}
if _ , err := sess . Where ( opts . toConds ( ) ) . Delete ( new ( Comment ) ) ; err != nil {
return err
}
opts = FindCommentsOptions {
Type : CommentTypeReview ,
IssueID : r . IssueID ,
ReviewID : r . ID ,
}
if _ , err := sess . Where ( opts . toConds ( ) ) . Delete ( new ( Comment ) ) ; err != nil {
return err
}
if _ , err := sess . ID ( r . ID ) . Delete ( new ( Review ) ) ; err != nil {
return err
}
2021-11-21 10:41:00 -05:00
return committer . Commit ( )
2020-05-01 20:20:51 -04:00
}
// GetCodeCommentsCount return count of CodeComments a Review has
func ( r * Review ) GetCodeCommentsCount ( ) int {
opts := FindCommentsOptions {
Type : CommentTypeCode ,
IssueID : r . IssueID ,
ReviewID : r . ID ,
}
conds := opts . toConds ( )
if r . ID == 0 {
conds = conds . And ( builder . Eq { "invalidated" : false } )
}
2021-09-23 11:45:36 -04:00
count , err := db . GetEngine ( db . DefaultContext ) . Where ( conds ) . Count ( new ( Comment ) )
2020-05-01 20:20:51 -04:00
if err != nil {
return 0
}
return int ( count )
}
// HTMLURL formats a URL-string to the related review issue-comment
func ( r * Review ) HTMLURL ( ) string {
opts := FindCommentsOptions {
Type : CommentTypeReview ,
IssueID : r . IssueID ,
ReviewID : r . ID ,
}
comment := new ( Comment )
2021-09-23 11:45:36 -04:00
has , err := db . GetEngine ( db . DefaultContext ) . Where ( opts . toConds ( ) ) . Get ( comment )
2020-05-01 20:20:51 -04:00
if err != nil || ! has {
return ""
}
return comment . HTMLURL ( )
}
2022-02-01 13:20:28 -05:00
// RemapExternalUser ExternalUserRemappable interface
func ( r * Review ) RemapExternalUser ( externalName string , externalID , userID int64 ) error {
r . OriginalAuthor = externalName
r . OriginalAuthorID = externalID
r . ReviewerID = userID
return nil
}
// GetUserID ExternalUserRemappable interface
func ( r * Review ) GetUserID ( ) int64 { return r . ReviewerID }
// GetExternalName ExternalUserRemappable interface
func ( r * Review ) GetExternalName ( ) string { return r . OriginalAuthor }
// GetExternalID ExternalUserRemappable interface
func ( r * Review ) GetExternalID ( ) int64 { return r . OriginalAuthorID }