rebase

ashwin-ant · ashwin-ant · commit 4486e5c0ea73 · 2025-04-08T07:49:09.000-07:00
diff --git a/github-mcp-server b/github-mcp-server
diff --git a/pkg/github/pullrequests.go b/pkg/github/pullrequests.go
@@ -558,7 +558,7 @@ func AddPullRequestReviewComment(client *github.Client, t translations.Translati
 			if err != nil {
 				return mcp.NewToolResultError(err.Error()), nil
 			}
-			pullNumber, err := requiredInt(request, "pull_number")
+			pullNumber, err := RequiredInt(request, "pull_number")
 			if err != nil {
 				return mcp.NewToolResultError(err.Error()), nil
 			}
@@ -586,12 +586,13 @@ func AddPullRequestReviewComment(client *github.Client, t translations.Translati
 				comment.InReplyTo = github.Ptr(int64(replyToFloat))
 			} else {
 				// Handle subject_type parameter
-				subjectType, err := optionalParam[string](request, "subject_type")
+				subjectType, err := OptionalParam[string](request, "subject_type")
 				if err != nil {
 					return mcp.NewToolResultError(err.Error()), nil
 				}
 				if subjectType == "file" {
 					// When commenting on a file, no line/position fields are needed
+					// No action needed
 				} else {
 					// Handle line or position-based comments
 					line, lineExists := request.Params.Arguments["line"].(float64)
@@ -684,11 +685,11 @@ func ReplyToPullRequestReviewComment(client *github.Client, t translations.Trans
 			if err != nil {
 				return mcp.NewToolResultError(err.Error()), nil
 			}
-			pullNumber, err := requiredInt(request, "pull_number")
+			pullNumber, err := RequiredInt(request, "pull_number")
 			if err != nil {
 				return mcp.NewToolResultError(err.Error()), nil
 			}
-			commentID, err := requiredInt(request, "comment_id")
+			commentID, err := RequiredInt(request, "comment_id")
 			if err != nil {
 				return mcp.NewToolResultError(err.Error()), nil
 			}
diff --git a/pkg/github/pullrequests_test.go b/pkg/github/pullrequests_test.go
@@ -1575,7 +1575,11 @@ func Test_AddPullRequestReviewComment(t *testing.T) {
 					mock.PostReposPullsCommentsByOwnerByRepoByPullNumber,
 					http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
 						w.WriteHeader(http.StatusCreated)
-						json.NewEncoder(w).Encode(mockComment)
+						err := json.NewEncoder(w).Encode(mockComment)
+						if err != nil {
+							http.Error(w, err.Error(), http.StatusInternalServerError)
+							return
+						}
 					}),
 				),
 			),
@@ -1686,7 +1690,11 @@ func Test_ReplyToPullRequestReviewComment(t *testing.T) {
 					mock.PostReposPullsCommentsByOwnerByRepoByPullNumber,
 					http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
 						w.WriteHeader(http.StatusCreated)
-						json.NewEncoder(w).Encode(mockReply)
+						err := json.NewEncoder(w).Encode(mockReply)
+						if err != nil {
+							http.Error(w, err.Error(), http.StatusInternalServerError)
+							return
+						}
 					}),
 				),
 			),
@@ -1728,7 +1736,7 @@ func Test_ReplyToPullRequestReviewComment(t *testing.T) {
 		t.Run(tc.name, func(t *testing.T) {
 			mockClient := github.NewClient(tc.mockedClient)
 
-			_, handler := replyToPullRequestReviewComment(mockClient, translations.NullTranslationHelper)
+			_, handler := ReplyToPullRequestReviewComment(mockClient, translations.NullTranslationHelper)
 
 			request := createMCPRequest(tc.requestArgs)
 
