// 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 (
"context"
"fmt"
"strings"
"code.gitea.io/gitea/models/db"
"code.gitea.io/gitea/models/organization"
"code.gitea.io/gitea/models/perm"
"code.gitea.io/gitea/models/unit"
user_model "code.gitea.io/gitea/models/user"
"code.gitea.io/gitea/modules/base"
"code.gitea.io/gitea/modules/timeutil"
"xorm.io/builder"
)
// 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
// ReviewTypeRequest request review from others
ReviewTypeRequest
)
// Icon returns the corresponding icon for the review type
func ( rt ReviewType ) Icon ( ) string {
switch rt {
case ReviewTypeApprove :
return "check"
case ReviewTypeReject :
return "diff"
case ReviewTypeComment :
return "comment"
case ReviewTypeRequest :
return "dot-fill"
default :
return "comment"
}
}
// Review represents collection of code comments giving feedback for a PR
type Review struct {
ID int64 ` xorm:"pk autoincr" `
Type ReviewType
Reviewer * user_model . User ` xorm:"-" `
ReviewerID int64 ` xorm:"index" `
ReviewerTeamID int64 ` xorm:"NOT NULL DEFAULT 0" `
ReviewerTeam * organization . Team ` xorm:"-" `
OriginalAuthor string
OriginalAuthorID int64
Issue * Issue ` xorm:"-" `
IssueID int64 ` xorm:"index" `
Content string ` xorm:"TEXT" `
// Official is a review made by an assigned approver (counts towards approval)
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" `
CreatedUnix timeutil . TimeStamp ` xorm:"INDEX created" `
UpdatedUnix timeutil . TimeStamp ` xorm:"INDEX updated" `
// CodeComments are the initial code comments of the review
CodeComments CodeComments ` xorm:"-" `
Comments [ ] * Comment ` xorm:"-" `
}
func init ( ) {
db . RegisterModel ( new ( Review ) )
}
// LoadCodeComments loads CodeComments
func ( r * Review ) LoadCodeComments ( ctx context . Context ) ( err error ) {
if r . CodeComments != nil {
return
}
if err = r . loadIssue ( db . GetEngine ( ctx ) ) ; err != nil {
return
}
r . CodeComments , err = fetchCodeCommentsByReview ( ctx , r . Issue , nil , r )
return
}
func ( r * Review ) loadIssue ( e db . Engine ) ( err error ) {
if r . Issue != nil {
return
}
r . Issue , err = getIssueByID ( e , r . IssueID )
return
}
func ( r * Review ) loadReviewer ( e db . Engine ) ( err error ) {
if r . ReviewerID == 0 || r . Reviewer != nil {
return
}
r . Reviewer , err = user_model . GetUserByIDEngine ( e , r . ReviewerID )
return
}
func ( r * Review ) loadReviewerTeam ( ctx context . Context ) ( err error ) {
if r . ReviewerTeamID == 0 || r . ReviewerTeam != nil {
return
}
r . ReviewerTeam , err = organization . GetTeamByIDCtx ( ctx , r . ReviewerTeamID )
return
}
// LoadReviewer loads reviewer
func ( r * Review ) LoadReviewer ( ) error {
return r . loadReviewer ( db . GetEngine ( db . DefaultContext ) )
}
// LoadReviewerTeam loads reviewer team
func ( r * Review ) LoadReviewerTeam ( ) error {
return r . loadReviewerTeam ( db . DefaultContext )
}
// LoadAttributes loads all attributes except CodeComments
func ( r * Review ) LoadAttributes ( ctx context . Context ) ( err error ) {
e := db . GetEngine ( ctx )
if err = r . loadIssue ( e ) ; err != nil {
return
}
if err = r . LoadCodeComments ( ctx ) ; err != nil {
return
}
if err = r . loadReviewer ( e ) ; err != nil {
return
}
if err = r . loadReviewerTeam ( ctx ) ; err != nil {
return
}
return
}
func getReviewByID ( e db . Engine , id int64 ) ( * Review , error ) {
review := new ( Review )
if has , err := e . ID ( id ) . Get ( review ) ; err != nil {
return nil , err
} else if ! has {
return nil , ErrReviewNotExist { ID : id }
} else {
return review , nil
}
}
// GetReviewByID returns the review by the given ID
func GetReviewByID ( id int64 ) ( * Review , error ) {
return getReviewByID ( db . GetEngine ( db . DefaultContext ) , id )
}
// FindReviewOptions represent possible filters to find reviews
type FindReviewOptions struct {
db . ListOptions
Type ReviewType
IssueID int64
ReviewerID int64
OfficialOnly bool
}
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 } )
}
return cond
}
func findReviews ( e db . Engine , opts FindReviewOptions ) ( [ ] * Review , error ) {
reviews := make ( [ ] * Review , 0 , 10 )
sess := e . Where ( opts . toCond ( ) )
if opts . Page > 0 {
sess = db . SetSessionPagination ( sess , & opts )
}
return reviews , sess .
Asc ( "created_unix" ) .
Asc ( "id" ) .
Find ( & reviews )
}
// FindReviews returns reviews passing FindReviewOptions
func FindReviews ( opts FindReviewOptions ) ( [ ] * Review , error ) {
return findReviews ( db . GetEngine ( db . DefaultContext ) , opts )
}
// CountReviews returns count of reviews passing FindReviewOptions
func CountReviews ( opts FindReviewOptions ) ( int64 , error ) {
return db . GetEngine ( db . DefaultContext ) . Where ( opts . toCond ( ) ) . Count ( & Review { } )
}
// CreateReviewOptions represent the options to create a review. Type, Issue and Reviewer are required.
type CreateReviewOptions struct {
Content string
Type ReviewType
Issue * Issue
Reviewer * user_model . User
ReviewerTeam * organization . Team
Official bool
CommitID string
Stale bool
}
// IsOfficialReviewer check if at least one of the provided reviewers can make official reviews in issue (counts towards required approvals)
func IsOfficialReviewer ( issue * Issue , reviewers ... * user_model . User ) ( bool , error ) {
return isOfficialReviewer ( db . DefaultContext , issue , reviewers ... )
}
func isOfficialReviewer ( ctx context . Context , issue * Issue , reviewers ... * user_model . User ) ( bool , error ) {
pr , err := getPullRequestByIssueID ( db . GetEngine ( ctx ) , issue . ID )
if err != nil {
return false , err
}
if err = pr . loadProtectedBranch ( ctx ) ; err != nil {
return false , err
}
if pr . ProtectedBranch == nil {
return false , nil
}
for _ , reviewer := range reviewers {
official , err := isUserOfficialReviewer ( ctx , pr . ProtectedBranch , reviewer )
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)
func IsOfficialReviewerTeam ( issue * Issue , team * organization . Team ) ( bool , error ) {
return isOfficialReviewerTeam ( db . DefaultContext , issue , team )
}
func isOfficialReviewerTeam ( ctx context . Context , issue * Issue , team * organization . Team ) ( bool , error ) {
pr , err := getPullRequestByIssueID ( db . GetEngine ( ctx ) , issue . ID )
if err != nil {
return false , err
}
if err = pr . loadProtectedBranch ( ctx ) ; err != nil {
return false , err
}
if pr . ProtectedBranch == nil {
return false , nil
}
if ! pr . ProtectedBranch . EnableApprovalsWhitelist {
return team . UnitAccessMode ( ctx , unit . TypeCode ) >= perm . AccessModeWrite , nil
}
return base . Int64sContains ( pr . ProtectedBranch . ApprovalsWhitelistTeamIDs , team . ID ) , nil
}
func createReview ( e db . Engine , opts CreateReviewOptions ) ( * Review , error ) {
review := & Review {
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
}
if _ , err := e . Insert ( review ) ; err != nil {
return nil , err
}
return review , nil
}
// CreateReview creates a new review based on opts
func CreateReview ( opts CreateReviewOptions ) ( * Review , error ) {
return createReview ( db . GetEngine ( db . DefaultContext ) , opts )
}
func getCurrentReview ( e db . Engine , reviewer * user_model . User , issue * Issue ) ( * Review , error ) {
if reviewer == nil {
return nil , nil
}
reviews , err := findReviews ( e , FindReviewOptions {
Type : ReviewTypePending ,
IssueID : issue . ID ,
ReviewerID : reviewer . ID ,
} )
if err != nil {
return nil , err
}
if len ( reviews ) == 0 {
return nil , ErrReviewNotExist { }
}
reviews [ 0 ] . Reviewer = reviewer
reviews [ 0 ] . Issue = issue
return reviews [ 0 ] , nil
}
// 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 ) {
return db . GetEngine ( db . DefaultContext ) . Cols ( "id" ) . Exist ( & Comment { IssueID : issue . ID , TreePath : treePath , Line : line , Type : CommentTypeCode } )
}
// GetCurrentReview returns the current pending review of reviewer for given issue
func GetCurrentReview ( reviewer * user_model . User , issue * Issue ) ( * Review , error ) {
return getCurrentReview ( db . GetEngine ( db . DefaultContext ) , reviewer , issue )
}
// ContentEmptyErr represents an content empty error
type ContentEmptyErr struct { }
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
func SubmitReview ( doer * user_model . User , issue * Issue , reviewType ReviewType , content , commitID string , stale bool , attachmentUUIDs [ ] string ) ( * Review , * Comment , error ) {
ctx , committer , err := db . TxContext ( )
if err != nil {
return nil , nil , err
}
defer committer . Close ( )
sess := db . GetEngine ( ctx )
official := false
review , err := getCurrentReview ( sess , doer , issue )
if err != nil {
if ! IsErrReviewNotExist ( err ) {
return nil , nil , err
}
if reviewType != ReviewTypeApprove && len ( strings . TrimSpace ( content ) ) == 0 {
return nil , nil , ContentEmptyErr { }
}
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
if _ , err := sess . Exec ( "UPDATE `review` SET official=? WHERE issue_id=? AND reviewer_id=?" , false , issue . ID , doer . ID ) ; err != nil {
return nil , nil , err
}
if official , err = isOfficialReviewer ( ctx , issue , doer ) ; err != nil {
return nil , nil , err
}
}
// No current review. Create a new one!
if review , err = createReview ( sess , CreateReviewOptions {
Type : reviewType ,
Issue : issue ,
Reviewer : doer ,
Content : content ,
Official : official ,
CommitID : commitID ,
Stale : stale ,
} ) ; err != nil {
return nil , nil , err
}
} else {
if err := review . LoadCodeComments ( ctx ) ; err != nil {
return nil , nil , err
}
if reviewType != ReviewTypeApprove && len ( review . CodeComments ) == 0 && len ( strings . TrimSpace ( content ) ) == 0 {
return nil , nil , ContentEmptyErr { }
}
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
if _ , err := sess . Exec ( "UPDATE `review` SET official=? WHERE issue_id=? AND reviewer_id=?" , false , issue . ID , doer . ID ) ; err != nil {
return nil , nil , err
}
if official , err = isOfficialReviewer ( ctx , issue , doer ) ; err != nil {
return nil , nil , err
}
}
review . Official = official
review . Issue = issue
review . Content = content
review . Type = reviewType
review . CommitID = commitID
review . Stale = stale
if _ , err := sess . ID ( review . ID ) . Cols ( "content, type, official, commit_id, stale" ) . Update ( review ) ; err != nil {
return nil , nil , err
}
}
comm , err := createComment ( ctx , & CreateCommentOptions {
Type : CommentTypeReview ,
Doer : doer ,
Content : review . Content ,
Issue : issue ,
Repo : issue . Repo ,
ReviewID : review . ID ,
Attachments : attachmentUUIDs ,
} )
if err != nil || comm == nil {
return nil , nil , err
}
// try to remove team review request if need
if issue . Repo . Owner . IsOrganization ( ) && ( reviewType == ReviewTypeApprove || reviewType == ReviewTypeReject ) {
teamReviewRequests := make ( [ ] * Review , 0 , 10 )
if err := sess . SQL ( "SELECT * FROM review WHERE issue_id = ? AND reviewer_team_id > 0 AND type = ?" , issue . ID , ReviewTypeRequest ) . Find ( & teamReviewRequests ) ; err != nil {
return nil , nil , err
}
for _ , teamReviewRequest := range teamReviewRequests {
ok , err := organization . IsTeamMember ( ctx , issue . Repo . OwnerID , teamReviewRequest . ReviewerTeamID , doer . ID )
if err != nil {
return nil , nil , err
} else if ! ok {
continue
}
if _ , err := sess . Delete ( teamReviewRequest ) ; err != nil {
return nil , nil , err
}
}
}
comm . Review = review
return review , comm , committer . Commit ( )
}
// GetReviewersByIssueID gets the latest review of each reviewer for a pull request
func GetReviewersByIssueID ( issueID int64 ) ( [ ] * Review , 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
}
// GetReviewersFromOriginalAuthorsByIssueID gets the latest review of each original authors for a pull request
func GetReviewersFromOriginalAuthorsByIssueID ( issueID int64 ) ( [ ] * Review , 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
}
// GetReviewByIssueIDAndUserID get the latest review of reviewer for a pull request
func GetReviewByIssueIDAndUserID ( issueID , userID int64 ) ( * Review , error ) {
return getReviewByIssueIDAndUserID ( db . GetEngine ( db . DefaultContext ) , issueID , userID )
}
func getReviewByIssueIDAndUserID ( e db . Engine , issueID , userID int64 ) ( * Review , error ) {
review := new ( Review )
has , err := e . 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 (?, ?, ?))" ,
issueID , userID , ReviewTypeApprove , ReviewTypeReject , ReviewTypeRequest ) .
Get ( review )
if err != nil {
return nil , err
}
if ! has {
return nil , ErrReviewNotExist { }
}
return review , nil
}
// GetTeamReviewerByIssueIDAndTeamID get the latest review request of reviewer team for a pull request
func GetTeamReviewerByIssueIDAndTeamID ( issueID , teamID int64 ) ( review * Review , err error ) {
return getTeamReviewerByIssueIDAndTeamID ( db . GetEngine ( db . DefaultContext ) , issueID , teamID )
}
func getTeamReviewerByIssueIDAndTeamID ( e db . Engine , issueID , teamID int64 ) ( review * Review , err error ) {
review = new ( Review )
has := false
if has , err = e . SQL ( "SELECT * FROM review WHERE id IN (SELECT max(id) as id FROM review WHERE issue_id = ? AND reviewer_team_id = ?)" ,
issueID , teamID ) .
Get ( review ) ; err != nil {
return nil , err
}
if ! has {
return nil , ErrReviewNotExist { 0 }
}
return
}
// MarkReviewsAsStale marks existing reviews as stale
func MarkReviewsAsStale ( issueID int64 ) ( err error ) {
_ , err = db . GetEngine ( db . DefaultContext ) . Exec ( "UPDATE `review` SET stale=? WHERE issue_id=?" , true , issueID )
return
}
// MarkReviewsAsNotStale marks existing reviews as not stale for a giving commit SHA
func MarkReviewsAsNotStale ( issueID int64 , commitID string ) ( err error ) {
_ , err = db . GetEngine ( db . DefaultContext ) . Exec ( "UPDATE `review` SET stale=? WHERE issue_id=? AND commit_id=?" , false , issueID , commitID )
return
}
// 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
if review . ID == 0 {
return ErrReviewNotExist { }
}
_ , err = db . GetEngine ( db . DefaultContext ) . ID ( review . ID ) . Cols ( "dismissed" ) . Update ( review )
return
}
// InsertReviews inserts review and review comments
func InsertReviews ( reviews [ ] * Review ) error {
ctx , committer , err := db . TxContext ( )
if err != nil {
return err
}
defer committer . Close ( )
sess := db . GetEngine ( ctx )
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
}
if len ( review . Comments ) > 0 {
if _ , err := sess . NoAutoTime ( ) . Insert ( review . Comments ) ; err != nil {
return err
}
}
}
return committer . Commit ( )
}
// AddReviewRequest add a review request from one reviewer
func AddReviewRequest ( issue * Issue , reviewer , doer * user_model . User ) ( * Comment , error ) {
ctx , committer , err := db . TxContext ( )
if err != nil {
return nil , err
}
defer committer . Close ( )
sess := db . GetEngine ( ctx )
review , err := getReviewByIssueIDAndUserID ( sess , issue . ID , reviewer . ID )
if err != nil && ! IsErrReviewNotExist ( err ) {
return nil , err
}
// skip it when reviewer hase been request to review
if review != nil && review . Type == ReviewTypeRequest {
return nil , nil
}
official , err := isOfficialReviewer ( ctx , issue , reviewer , doer )
if err != nil {
return nil , err
} else if official {
if _ , err := sess . Exec ( "UPDATE `review` SET official=? WHERE issue_id=? AND reviewer_id=?" , false , issue . ID , reviewer . ID ) ; err != nil {
return nil , err
}
}
review , err = createReview ( sess , CreateReviewOptions {
Type : ReviewTypeRequest ,
Issue : issue ,
Reviewer : reviewer ,
Official : official ,
Stale : false ,
} )
if err != nil {
return nil , err
}
comment , err := createComment ( ctx , & CreateCommentOptions {
Type : CommentTypeReviewRequest ,
Doer : doer ,
Repo : issue . Repo ,
Issue : issue ,
RemovedAssignee : false , // Use RemovedAssignee as !isRequest
AssigneeID : reviewer . ID , // Use AssigneeID as reviewer ID
ReviewID : review . ID ,
} )
if err != nil {
return nil , err
}
return comment , committer . Commit ( )
}
// RemoveReviewRequest remove a review request from one reviewer
func RemoveReviewRequest ( issue * Issue , reviewer , doer * user_model . User ) ( * Comment , error ) {
ctx , committer , err := db . TxContext ( )
if err != nil {
return nil , err
}
defer committer . Close ( )
sess := db . GetEngine ( ctx )
review , err := getReviewByIssueIDAndUserID ( sess , issue . ID , reviewer . ID )
if err != nil && ! IsErrReviewNotExist ( err ) {
return nil , err
}
if review == nil || review . Type != ReviewTypeRequest {
return nil , nil
}
if _ , err = sess . Delete ( review ) ; err != nil {
return nil , err
}
official , err := isOfficialReviewer ( ctx , issue , reviewer )
if err != nil {
return nil , err
} else if official {
// recalculate the latest official review for reviewer
review , err := getReviewByIssueIDAndUserID ( sess , issue . ID , reviewer . ID )
if err != nil && ! IsErrReviewNotExist ( err ) {
return nil , err
}
if review != nil {
if _ , err := sess . Exec ( "UPDATE `review` SET official=? WHERE id=?" , true , review . ID ) ; err != nil {
return nil , err
}
}
}
comment , err := createComment ( ctx , & CreateCommentOptions {
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
}
return comment , committer . Commit ( )
}
// AddTeamReviewRequest add a review request from one team
func AddTeamReviewRequest ( issue * Issue , reviewer * organization . Team , doer * user_model . User ) ( * Comment , error ) {
ctx , committer , err := db . TxContext ( )
if err != nil {
return nil , err
}
defer committer . Close ( )
sess := db . GetEngine ( ctx )
review , err := getTeamReviewerByIssueIDAndTeamID ( sess , issue . ID , reviewer . ID )
if err != nil && ! IsErrReviewNotExist ( err ) {
return nil , err
}
// This team already has been requested to review - therefore skip this.
if review != nil {
return nil , nil
}
official , err := isOfficialReviewerTeam ( ctx , issue , reviewer )
if err != nil {
return nil , fmt . Errorf ( "isOfficialReviewerTeam(): %v" , err )
} else if ! official {
if official , err = isOfficialReviewer ( ctx , issue , doer ) ; err != nil {
return nil , fmt . Errorf ( "isOfficialReviewer(): %v" , err )
}
}
if review , err = createReview ( sess , CreateReviewOptions {
Type : ReviewTypeRequest ,
Issue : issue ,
ReviewerTeam : reviewer ,
Official : official ,
Stale : false ,
} ) ; err != nil {
return nil , err
}
if official {
if _ , err := sess . Exec ( "UPDATE `review` SET official=? WHERE issue_id=? AND reviewer_team_id=?" , false , issue . ID , reviewer . ID ) ; err != nil {
return nil , err
}
}
comment , err := createComment ( ctx , & CreateCommentOptions {
Type : CommentTypeReviewRequest ,
Doer : doer ,
Repo : issue . Repo ,
Issue : issue ,
RemovedAssignee : false , // Use RemovedAssignee as !isRequest
AssigneeTeamID : reviewer . ID , // Use AssigneeTeamID as reviewer team ID
ReviewID : review . ID ,
} )
if err != nil {
return nil , fmt . Errorf ( "createComment(): %v" , err )
}
return comment , committer . Commit ( )
}
// RemoveTeamReviewRequest remove a review request from one team
func RemoveTeamReviewRequest ( issue * Issue , reviewer * organization . Team , doer * user_model . User ) ( * Comment , error ) {
ctx , committer , err := db . TxContext ( )
if err != nil {
return nil , err
}
defer committer . Close ( )
sess := db . GetEngine ( ctx )
review , err := getTeamReviewerByIssueIDAndTeamID ( sess , issue . ID , reviewer . ID )
if err != nil && ! IsErrReviewNotExist ( err ) {
return nil , err
}
if review == nil {
return nil , nil
}
if _ , err = sess . Delete ( review ) ; err != nil {
return nil , err
}
official , err := isOfficialReviewerTeam ( ctx , issue , reviewer )
if err != nil {
return nil , fmt . Errorf ( "isOfficialReviewerTeam(): %v" , err )
}
if official {
// recalculate which is the latest official review from that team
review , err := getReviewByIssueIDAndUserID ( sess , issue . ID , - reviewer . ID )
if err != nil && ! IsErrReviewNotExist ( err ) {
return nil , err
}
if review != nil {
if _ , err := sess . Exec ( "UPDATE `review` SET official=? WHERE id=?" , true , review . ID ) ; err != nil {
return nil , err
}
}
}
if doer == nil {
return nil , committer . Commit ( )
}
comment , err := createComment ( ctx , & CreateCommentOptions {
Type : CommentTypeReviewRequest ,
Doer : doer ,
Repo : issue . Repo ,
Issue : issue ,
RemovedAssignee : true , // Use RemovedAssignee as !isRequest
AssigneeTeamID : reviewer . ID , // Use AssigneeTeamID as reviewer team ID
} )
if err != nil {
return nil , fmt . Errorf ( "createComment(): %v" , err )
}
return comment , committer . Commit ( )
}
// MarkConversation Add or remove Conversation mark for a code comment
func MarkConversation ( comment * Comment , doer * user_model . User , isResolve bool ) ( err error ) {
if comment . Type != CommentTypeCode {
return nil
}
if isResolve {
if comment . ResolveDoerID != 0 {
return nil
}
if _ , err = db . GetEngine ( db . DefaultContext ) . Exec ( "UPDATE `comment` SET resolve_doer_id=? WHERE id=?" , doer . ID , comment . ID ) ; err != nil {
return err
}
} else {
if comment . ResolveDoerID == 0 {
return nil
}
if _ , err = db . GetEngine ( db . DefaultContext ) . Exec ( "UPDATE `comment` SET resolve_doer_id=? WHERE id=?" , 0 , comment . ID ) ; err != nil {
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
func CanMarkConversation ( issue * Issue , doer * user_model . User ) ( permResult bool , err error ) {
if doer == nil || issue == nil {
return false , fmt . Errorf ( "issue or doer is nil" )
}
if doer . ID != issue . PosterID {
if err = issue . LoadRepo ( ) ; err != nil {
return false , err
}
p , err := GetUserRepoPermission ( issue . Repo , doer )
if err != nil {
return false , err
}
permResult = p . CanAccess ( perm . AccessModeWrite , unit . TypePullRequests )
if ! permResult {
if permResult , err = IsOfficialReviewer ( issue , doer ) ; err != nil {
return false , err
}
}
if ! permResult {
return false , nil
}
}
return true , nil
}
// DeleteReview delete a review and it's code comments
func DeleteReview ( r * Review ) error {
ctx , committer , err := db . TxContext ( )
if err != nil {
return err
}
defer committer . Close ( )
sess := db . GetEngine ( ctx )
if r . ID == 0 {
return fmt . Errorf ( "review is not allowed to be 0" )
}
if r . Type == ReviewTypeRequest {
return fmt . Errorf ( "review request can not be deleted using this method" )
}
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
}
return committer . Commit ( )
}
// 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 } )
}
count , err := db . GetEngine ( db . DefaultContext ) . Where ( conds ) . Count ( new ( Comment ) )
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 )
has , err := db . GetEngine ( db . DefaultContext ) . Where ( opts . toConds ( ) ) . Get ( comment )
if err != nil || ! has {
return ""
}
return comment . HTMLURL ( )
}
// 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 }