Enhance AI review functionality by adding user request handling and reaction to comments; refactor comment processing for better clarity and structure.

DaleSeo · DaleSeo · commit 578225fb362f · 2025-11-07T00:12:02.000-05:00
diff --git a/handlers/webhooks.js b/handlers/webhooks.js
@@ -19,7 +19,7 @@ import {
   isClosedPR,
 } from "../utils/validation.js";
 import { ALLOWED_REPO } from "../utils/constants.js";
-import { performAIReview } from "../utils/prReview.js";
+import { performAIReview, addReactionToComment } from "../utils/prReview.js";
 
 /**
  * GitHub webhook 이벤트 처리
@@ -249,6 +249,26 @@ async function handlePullRequestEvent(payload, env) {
   });
 }
 
+/**
+ * 댓글에서 @dalestudy 멘션과 사용자 요청 추출
+ *
+ * @param {string} commentBody - 댓글 내용
+ * @returns {Object|null} { isMentioned, userRequest } 또는 null
+ */
+function extractMentionAndRequest(commentBody) {
+  const lowerBody = commentBody.toLowerCase();
+  const isMentioned = lowerBody.includes("@dalestudy");
+
+  if (!isMentioned) {
+    return null;
+  }
+
+  const mentionMatch = commentBody.match(/@dalestudy\s*(.*)/i);
+  const userRequest = mentionMatch && mentionMatch[1].trim() ? mentionMatch[1].trim() : null;
+
+  return { isMentioned: true, userRequest };
+}
+
 /**
  * Issue Comment 이벤트 처리 (AI 코드 리뷰 요청)
  */
@@ -276,21 +296,14 @@ async function handleIssueCommentEvent(payload, env) {
 
   const prNumber = issue.number;
 
-  // 멘션 감지: @dalestudy만 체크
-  const commentBody = comment.body;
-  const lowerBody = commentBody.toLowerCase();
-  const isMentioned = lowerBody.includes("@dalestudy");
-
-  if (!isMentioned) {
+  // 멘션 감지 및 사용자 요청 추출
+  const mention = extractMentionAndRequest(comment.body);
+  if (!mention) {
     console.log("Ignoring: bot not mentioned");
     return corsResponse({ message: "Ignored: not mentioned" });
   }
 
-  // 멘션 뒤의 텍스트 추출
-  const mentionMatch = commentBody.match(/@dalestudy\s*(.*)/i);
-  const userRequest = mentionMatch && mentionMatch[1].trim() ? mentionMatch[1].trim() : null;
-
-  console.log(`AI review requested for PR #${prNumber}${userRequest ? ` - Request: ${userRequest}` : ""}`);
+  console.log(`AI review requested for PR #${prNumber}${mention.userRequest ? ` - Request: ${mention.userRequest}` : ""}`);
 
   // OPENAI_API_KEY 확인
   if (!env.OPENAI_API_KEY) {
@@ -303,13 +316,13 @@ async function handleIssueCommentEvent(payload, env) {
     const appToken = await generateGitHubAppToken(env);
 
     // 👀 reaction 추가 (리뷰 시작 알림)
-    await fetch(
-      `https://api.github.com/repos/${repoOwner}/${repoName}/issues/comments/${comment.id}/reactions`,
-      {
-        method: "POST",
-        headers: getGitHubHeaders(appToken),
-        body: JSON.stringify({ content: "eyes" }),
-      }
+    await addReactionToComment(
+      repoOwner,
+      repoName,
+      comment.id,
+      "issue",
+      "eyes",
+      appToken
     );
 
     await performAIReview(
@@ -319,7 +332,8 @@ async function handleIssueCommentEvent(payload, env) {
       issue.title,
       issue.body,
       appToken,
-      env.OPENAI_API_KEY
+      env.OPENAI_API_KEY,
+      mention.userRequest
     );
 
     console.log(`AI review completed for PR #${prNumber}`);
@@ -354,16 +368,14 @@ async function handlePullRequestReviewCommentEvent(payload, env) {
   const repoName = payload.repository.name;
   const prNumber = pullRequest.number;
 
-  // 멘션 감지: @dalestudy만 체크
-  const commentBody = comment.body.toLowerCase();
-  const isMentioned = commentBody.includes("@dalestudy");
-
-  if (!isMentioned) {
+  // 멘션 감지 및 사용자 요청 추출
+  const mention = extractMentionAndRequest(comment.body);
+  if (!mention) {
     console.log("Ignoring: bot not mentioned");
     return corsResponse({ message: "Ignored: not mentioned" });
   }
 
-  console.log(`AI review requested for PR #${prNumber} (review comment)`);
+  console.log(`AI review requested for PR #${prNumber} (review comment)${mention.userRequest ? ` - Request: ${mention.userRequest}` : ""}`);
 
   // OPENAI_API_KEY 확인
   if (!env.OPENAI_API_KEY) {
@@ -376,43 +388,26 @@ async function handlePullRequestReviewCommentEvent(payload, env) {
     const appToken = await generateGitHubAppToken(env);
 
     // 👀 reaction 추가
-    await fetch(
-      `https://api.github.com/repos/${repoOwner}/${repoName}/pulls/comments/${comment.id}/reactions`,
-      {
-        method: "POST",
-        headers: getGitHubHeaders(appToken),
-        body: JSON.stringify({ content: "eyes" }),
-      }
+    await addReactionToComment(
+      repoOwner,
+      repoName,
+      comment.id,
+      "pull",
+      "eyes",
+      appToken
     );
 
-    // AI 리뷰 생성
-    const { generateCodeReview } = await import("../utils/openai.js");
-    const { getPRDiff } = await import("../utils/prReview.js");
-
-    const prDiff = await getPRDiff(repoOwner, repoName, prNumber, appToken);
-
-    // diff가 너무 크면 스킵
-    const diffLines = prDiff.split("\n").length;
-    if (diffLines > 1000) {
-      console.log(`Skipping AI review: diff too large (${diffLines} lines)`);
-      return corsResponse({ message: "Diff too large" });
-    }
-
-    const reviewContent = await generateCodeReview(
-      prDiff,
+    // performAIReview 사용 (스레드 답변 모드)
+    await performAIReview(
+      repoOwner,
+      repoName,
+      prNumber,
       pullRequest.title,
       pullRequest.body,
-      env.OPENAI_API_KEY
-    );
-
-    // 스레드 답변으로 작성
-    await fetch(
-      `https://api.github.com/repos/${repoOwner}/${repoName}/pulls/${prNumber}/comments/${comment.id}/replies`,
-      {
-        method: "POST",
-        headers: getGitHubHeaders(appToken),
-        body: JSON.stringify({ body: reviewContent }),
-      }
+      appToken,
+      env.OPENAI_API_KEY,
+      mention.userRequest,
+      comment.id  // 스레드 답변으로 작성
     );
 
     console.log(`AI review completed for PR #${prNumber} (thread reply)`);
diff --git a/utils/openai.js b/utils/openai.js
@@ -24,13 +24,6 @@ export async function generateCodeReview(
     ? `당신은 리트코드 스터디 그룹의 AI 코치입니다.
 사용자가 PR의 코드에 대해 구체적인 질문을 했습니다.
 PR의 코드 변경 사항을 참고하여 사용자의 질문에 명확하고 도움이 되는 답변을 제공하세요.
-
-답변 시:
-  •	사용자의 질문에 직접적으로 답변하세요
-  •	코드의 해당 부분을 구체적으로 언급하세요
-  •	필요하면 예시나 개선 방법을 제안하세요
-  •	격려와 학습이 되는 피드백을 함께 주세요
-
 300 글자를 초과하지 말아주세요.`
     : `당신은 리트코드 스터디 그룹의 AI 코치입니다.
 아래 코드 변경 사항을 리뷰하고 건설적인 피드백을 제공하세요.
@@ -40,29 +33,29 @@ PR의 코드 변경 사항을 참고하여 사용자의 질문에 명확하고 
   •	시간/공간 복잡도 분석이 정확한지 평가
   •	더 나은 접근법이나 알고리즘이 있는지 제안
   •	코드의 가독성 및 스타일, 베스트 프랙티스 준수 여부
-	•	잠재적인 버그 또는 개선 가능성
+	•	불필요한 nickpick은 피하고, 꼭 필요한 피드백만 주세요.
 
 단순히 지적만 하지 말고, 격려와 학습이 되는 피드백을 함께 주세요.
 해당 사항없는 항목은 생략하고 자연스럽게 작성하세요.
-300 글자를 초과하지 말아주세요.
+500 글자를 초과하지 말아주세요.
 `;
 
-  let userPrompt = `# PR 제목
+  let userPrompt = `# PR Title
 ${prTitle}
 
-# PR 설명
-${prBody || "설명 없음"}
+# PR Description
+${prBody || "No description provided"}
 
-# 코드 변경사항
+# Code Changes
 \`\`\`diff
 ${prDiff}
 \`\`\`
 `;
 
   if (userRequest) {
-    userPrompt += `\n# 사용자 질문\n${userRequest}`;
+    userPrompt += `\n# User's Question\n${userRequest}`;
   } else {
-    userPrompt += `\n이 PR을 리뷰해주세요.`;
+    userPrompt += `\nPlease review this pull request.`;
   }
 
   const response = await fetch("https://api.openai.com/v1/chat/completions", {
diff --git a/utils/prReview.js b/utils/prReview.js
@@ -48,18 +48,73 @@ async function postReviewComment(
   reviewContent,
   githubToken
 ) {
-  const commentBody = `${reviewContent}`;
-
   await fetch(
     `https://api.github.com/repos/${repoOwner}/${repoName}/issues/${prNumber}/comments`,
     {
       method: "POST",
       headers: getGitHubHeaders(githubToken),
-      body: JSON.stringify({ body: commentBody }),
+      body: JSON.stringify({ body: reviewContent }),
     }
   );
 }
 
+/**
+ * 스레드 답변으로 AI 코드 리뷰 작성
+ *
+ * @param {string} repoOwner - 저장소 소유자
+ * @param {string} repoName - 저장소 이름
+ * @param {number} prNumber - PR 번호
+ * @param {number} commentId - 원본 댓글 ID
+ * @param {string} reviewContent - 리뷰 내용 (마크다운)
+ * @param {string} githubToken - GitHub 토큰
+ */
+async function postThreadReply(
+  repoOwner,
+  repoName,
+  prNumber,
+  commentId,
+  reviewContent,
+  githubToken
+) {
+  await fetch(
+    `https://api.github.com/repos/${repoOwner}/${repoName}/pulls/${prNumber}/comments/${commentId}/replies`,
+    {
+      method: "POST",
+      headers: getGitHubHeaders(githubToken),
+      body: JSON.stringify({ body: reviewContent }),
+    }
+  );
+}
+
+/**
+ * 댓글에 reaction 추가
+ *
+ * @param {string} repoOwner - 저장소 소유자
+ * @param {string} repoName - 저장소 이름
+ * @param {number} commentId - 댓글 ID
+ * @param {string} commentType - 댓글 타입 ('issue' 또는 'pull')
+ * @param {string} reaction - reaction 타입 (예: 'eyes')
+ * @param {string} githubToken - GitHub 토큰
+ */
+export async function addReactionToComment(
+  repoOwner,
+  repoName,
+  commentId,
+  commentType,
+  reaction,
+  githubToken
+) {
+  const endpoint = commentType === "issue"
+    ? `https://api.github.com/repos/${repoOwner}/${repoName}/issues/comments/${commentId}/reactions`
+    : `https://api.github.com/repos/${repoOwner}/${repoName}/pulls/comments/${commentId}/reactions`;
+
+  await fetch(endpoint, {
+    method: "POST",
+    headers: getGitHubHeaders(githubToken),
+    body: JSON.stringify({ content: reaction }),
+  });
+}
+
 /**
  * PR에 대해 AI 리뷰 수행
  *
@@ -71,6 +126,7 @@ async function postReviewComment(
  * @param {string} githubToken - GitHub 토큰
  * @param {string} openaiApiKey - OpenAI API 키
  * @param {string} userRequest - 사용자의 구체적인 요청 (선택사항)
+ * @param {number} replyToCommentId - 스레드 답변으로 작성할 댓글 ID (선택사항)
  */
 export async function performAIReview(
   repoOwner,
@@ -80,9 +136,10 @@ export async function performAIReview(
   prBody,
   githubToken,
   openaiApiKey,
-  userRequest = null
+  userRequest = null,
+  replyToCommentId = null
 ) {
-  console.log(`Starting AI review for PR #${prNumber}`);
+  console.log(`Starting AI review for PR #${prNumber}${userRequest ? ` - Request: ${userRequest}` : ""}`);
 
   // PR diff 가져오기
   const prDiff = await getPRDiff(repoOwner, repoName, prNumber, githubToken);
@@ -94,22 +151,34 @@ export async function performAIReview(
     return;
   }
 
-  // AI 리뷰 생성
+  // AI 리뷰 생성 (userRequest 전달)
   const reviewContent = await generateCodeReview(
     prDiff,
     prTitle,
     prBody,
-    openaiApiKey
-  );
-
-  // 리뷰 댓글 작성
-  await postReviewComment(
-    repoOwner,
-    repoName,
-    prNumber,
-    reviewContent,
-    githubToken
+    openaiApiKey,
+    userRequest
   );
 
-  console.log(`AI review posted for PR #${prNumber}`);
+  // 리뷰 댓글 작성 (스레드 답변 또는 일반 댓글)
+  if (replyToCommentId) {
+    await postThreadReply(
+      repoOwner,
+      repoName,
+      prNumber,
+      replyToCommentId,
+      reviewContent,
+      githubToken
+    );
+    console.log(`AI review posted as thread reply for PR #${prNumber}`);
+  } else {
+    await postReviewComment(
+      repoOwner,
+      repoName,
+      prNumber,
+      reviewContent,
+      githubToken
+    );
+    console.log(`AI review posted for PR #${prNumber}`);
+  }
 }
