From 853de2cb647b46501cab081994279739e193284d Mon Sep 17 00:00:00 2001
From: Derrick Hammer <derrick@derrickhammer.com>
Date: Mon, 12 Feb 2024 16:54:59 -0500
Subject: [PATCH] feat: add update pull request

---
 api/routes_rest_api.go | 69 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 69 insertions(+)

diff --git a/api/routes_rest_api.go b/api/routes_rest_api.go
index f832c0a..cdcafb3 100644
--- a/api/routes_rest_api.go
+++ b/api/routes_rest_api.go
@@ -363,6 +363,74 @@ func (r restApi) handlerCreatePullRequest(w http.ResponseWriter, request *http.R
 	r.respond(w, http.StatusCreated, newGhPullRequest)
 }
 
+func (r restApi) handlerUpdatePullRequest(w http.ResponseWriter, request *http.Request) {
+	vars := mux.Vars(request)
+	owner := vars["owner"]
+	repo := vars["repo"]
+	pullNumber := vars["pull_number"]
+
+	parsedPullNumber, err := strconv.ParseInt(pullNumber, 10, 64)
+	if err != nil {
+		http.Error(w, "Failed to parse pull number", http.StatusBadRequest)
+		r.logger.Error("Failed to parse pull number", zap.Error(err))
+		return
+	}
+
+	client := r.getClientOrError(w)
+
+	if client == nil {
+		return
+	}
+
+	type pullRequestUpdate struct {
+		Title               *string `json:"title,omitempty"`
+		Body                *string `json:"body,omitempty"`
+		State               *string `json:"state,omitempty"`
+		Base                *string `json:"base,omitempty"`
+		MaintainerCanModify *bool   `json:"maintainer_can_modify,omitempty"`
+	}
+
+	ghPullRequest := pullRequestUpdate{}
+	pullRequest := gitea.EditPullRequestOption{}
+
+	if err := r.parseJsonBody(request, &ghPullRequest); err != nil {
+		http.Error(w, "Failed to parse request body", http.StatusBadRequest)
+		r.logger.Error("Failed to parse request body", zap.Error(err))
+	}
+
+	if ghPullRequest.State != nil {
+		if *ghPullRequest.State != "open" && *ghPullRequest.State != "closed" {
+			http.Error(w, "Invalid state", http.StatusBadRequest)
+			return
+		}
+
+		state := *ghPullRequest.State
+		stateType := gitea.StateType(state)
+
+		pullRequest.State = &stateType
+	}
+
+	if ghPullRequest.Title != nil {
+		pullRequest.Title = *ghPullRequest.Title
+	}
+
+	if ghPullRequest.Body != nil {
+		pullRequest.Body = *ghPullRequest.Body
+	}
+
+	pullRequestResponse, r2, err := client.EditPullRequest(owner, repo, parsedPullNumber, pullRequest)
+	if err != nil {
+		http.Error(w, "Failed to update pull request", http.StatusInternalServerError)
+		r.logger.Error("Failed to update pull request", zap.Error(err))
+		return
+	}
+
+	newGhPullRequest := convertPullRequest(pullRequestResponse)
+
+	r.sendPagingHeaders(w, r2)
+	r.respond(w, http.StatusOK, newGhPullRequest)
+}
+
 func (r restApi) parseJsonBody(request *http.Request, obj interface{}) error {
 	if obj == nil {
 		obj = make(map[string]interface{})
@@ -393,6 +461,7 @@ func setupRestRoutes(params RouteParams) {
 
 		// Pull Request routes
 		r.HandleFunc("/repos/{owner}/{repo}/pulls", restApi.handlerCreatePullRequest).Methods("POST")
+		r.HandleFunc("/repos/{owner}/{repo}/pulls/{pull_number}", restApi.handlerUpdatePullRequest).Methods("PATCH")
 	}
 
 	restRouter := r.PathPrefix("/api").Subrouter()