From c92202e36317d9f55618b5558414e1bdac8aa63b Mon Sep 17 00:00:00 2001
From: Sebastian Sauer <sauer.sebastian@gmail.com>
Date: Fri, 20 Oct 2023 22:17:45 +0200
Subject: [PATCH] Add request review to API

---
 modules/structs/pull.go        |  4 +++-
 routers/api/v1/repo/pull.go    | 41 +++++++++++++++++++++++++++++++++-
 templates/swagger/v1_json.tmpl | 14 ++++++++++++
 3 files changed, 57 insertions(+), 2 deletions(-)

diff --git a/modules/structs/pull.go b/modules/structs/pull.go
index ab627666c9..55831e642c 100644
--- a/modules/structs/pull.go
+++ b/modules/structs/pull.go
@@ -86,7 +86,9 @@ type CreatePullRequestOption struct {
 	Milestone int64    `json:"milestone"`
 	Labels    []int64  `json:"labels"`
 	// swagger:strfmt date-time
-	Deadline *time.Time `json:"due_date"`
+	Deadline      *time.Time `json:"due_date"`
+	Reviewers     []string   `json:"reviewers"`
+	TeamReviewers []string   `json:"team_reviewers"`
 }
 
 // EditPullRequestOption options when modify pull request
diff --git a/routers/api/v1/repo/pull.go b/routers/api/v1/repo/pull.go
index 0f98507a7e..0732703f12 100644
--- a/routers/api/v1/repo/pull.go
+++ b/routers/api/v1/repo/pull.go
@@ -16,6 +16,7 @@ import (
 	activities_model "code.gitea.io/gitea/models/activities"
 	git_model "code.gitea.io/gitea/models/git"
 	issues_model "code.gitea.io/gitea/models/issues"
+	"code.gitea.io/gitea/models/organization"
 	access_model "code.gitea.io/gitea/models/perm/access"
 	pull_model "code.gitea.io/gitea/models/pull"
 	repo_model "code.gitea.io/gitea/models/repo"
@@ -553,8 +554,46 @@ func CreatePullRequest(ctx *context.APIContext) {
 			return
 		}
 	}
+	// handle reviewers
+	var reviewerIds []int64
 
-	if err := pull_service.NewPullRequest(ctx, repo, prIssue, labelIDs, []string{}, pr, assigneeIDs, []int64{}); err != nil {
+	for _, r := range form.Reviewers {
+		var reviewer *user_model.User
+		if strings.Contains(r, "@") {
+			reviewer, err = user_model.GetUserByEmail(ctx, r)
+		} else {
+			reviewer, err = user_model.GetUserByName(ctx, r)
+		}
+
+		if err != nil {
+			if user_model.IsErrUserNotExist(err) {
+				ctx.NotFound("UserNotExist", fmt.Sprintf("User with id '%s' not exist", r))
+				return
+			}
+			ctx.Error(http.StatusInternalServerError, "GetUser", err)
+			return
+		}
+		reviewerIds = append(reviewerIds, reviewer.ID)
+	}
+
+	// handle teams as reviewers
+	if ctx.Repo.Repository.Owner.IsOrganization() && len(form.TeamReviewers) > 0 {
+		for _, t := range form.TeamReviewers {
+			var teamReviewer *organization.Team
+			teamReviewer, err = organization.GetTeam(ctx, ctx.Repo.Owner.ID, t)
+			if err != nil {
+				if organization.IsErrTeamNotExist(err) {
+					ctx.NotFound("TeamNotExist", fmt.Sprintf("Team '%s' not exist", t))
+					return
+				}
+				ctx.Error(http.StatusInternalServerError, "ReviewRequest", err)
+				return
+			}
+			reviewerIds = append(reviewerIds, teamReviewer.ID)
+		}
+	}
+
+	if err := pull_service.NewPullRequest(ctx, repo, prIssue, labelIDs, []string{}, pr, assigneeIDs, reviewerIds); err != nil {
 		if repo_model.IsErrUserDoesNotHaveAccessToRepo(err) {
 			ctx.Error(http.StatusBadRequest, "UserDoesNotHaveAccessToRepo", err)
 		} else if errors.Is(err, user_model.ErrBlockedUser) {
diff --git a/templates/swagger/v1_json.tmpl b/templates/swagger/v1_json.tmpl
index 4cbf511aac..0c898eb672 100644
--- a/templates/swagger/v1_json.tmpl
+++ b/templates/swagger/v1_json.tmpl
@@ -20086,6 +20086,20 @@
           "format": "int64",
           "x-go-name": "Milestone"
         },
+        "reviewers": {
+          "type": "array",
+          "items": {
+            "type": "string"
+          },
+          "x-go-name": "Reviewers"
+        },
+        "team_reviewers": {
+          "type": "array",
+          "items": {
+            "type": "string"
+          },
+          "x-go-name": "TeamReviewers"
+        },
         "title": {
           "type": "string",
           "x-go-name": "Title"