feat: 파일 관련성 추적 기능 추가 및 전역 컨텍스트 처리 개선

kim-the9 · kim-the9 · commit 3b21a2ed0f61 · 2026-01-07T20:50:19.000+09:00
diff --git a/src/bot.ts b/src/bot.ts
@@ -10,11 +10,7 @@ const MAX_PATCH_COUNT = process.env.MAX_PATCH_LENGTH
   : Infinity;
 const MAX_FILE_COUNT = 30;
 
-// 파일 본문을 가져올 때 허용하는 최대 크기 (바이트 단위)
-// 너무 큰 파일은 그대로 두되, 이후 prompt 구성 단계에서 잘라냅니다.
-const MAX_FILE_BYTES = process.env.MAX_FILE_BYTES
-  ? +process.env.MAX_FILE_BYTES
-  : 64 * 1024; // 기본 64KB
+
 
 // 프롬프트에 포함할 수 있는 최대 문자 수 (가드레일)
 // 이 값을 넘어가면 중간 부분을 잘라내고 앞/뒤 일부만 남깁니다.
@@ -27,10 +23,7 @@ const MAX_GLOBAL_CONTEXT_CHARS = process.env.MAX_GLOBAL_CONTEXT_CHARS
   ? +process.env.MAX_GLOBAL_CONTEXT_CHARS
   : 40000;
 
-// 전역 컨텍스트 제공 모드
-// - "meta": PR 메타 정보(제목/커밋/파일목록)만 제공 (권장)
-// - "full": 메타 + 전체 diff 발췌 제공 (레거시)
-const GLOBAL_CONTEXT_MODE = process.env.GLOBAL_CONTEXT_MODE || "meta"; // meta | full
+
 
 // Fetch full file content (at a specific ref) to give the model context
 const fetchFileContent = async (
@@ -71,13 +64,149 @@ const allocateBudget = (limit: number, ratio: number): number => {
   return Math.max(0, Math.floor(limit * ratio));
 };
 
-// 📌 리뷰 입력을 만드는 함수 (전역 컨텍스트 + 파일 전체 내용 + diff)
-// - globalContext: PR 메타 정보를 주로 제공 (필요 시 관련 파일 일부 패치만 첨부)
+// 파일 경로에서 디렉토리 추출
+const getDirname = (filepath: string): string => {
+  const lastSlash = filepath.lastIndexOf("/");
+  return lastSlash === -1 ? "" : filepath.slice(0, lastSlash);
+};
+
+// 파일 내용에서 import/require 경로 추출
+const extractImports = (content: string, currentFile: string): string[] => {
+  const imports: string[] = [];
+  const currentDir = getDirname(currentFile);
+
+  // ES6 import: import ... from '...' or import '...'
+  const esImportRegex = /import\s+(?:[\s\S]*?\s+from\s+)?['"]([^'"]+)['"]/g;
+  // CommonJS require: require('...')
+  const requireRegex = /require\s*\(\s*['"]([^'"]+)['"]\s*\)/g;
+
+  let match;
+  while ((match = esImportRegex.exec(content)) !== null) {
+    imports.push(match[1]);
+  }
+  while ((match = requireRegex.exec(content)) !== null) {
+    imports.push(match[1]);
+  }
+
+  return imports
+    .filter((imp) => imp.startsWith("."))
+    .map((imp) => {
+      let resolved = imp;
+      if (imp.startsWith("./")) {
+        resolved = currentDir ? `${currentDir}/${imp.slice(2)}` : imp.slice(2);
+      } else if (imp.startsWith("../")) {
+        const parts = currentDir.split("/");
+        let impParts = imp.split("/");
+        while (impParts[0] === "..") {
+          parts.pop();
+          impParts.shift();
+        }
+        resolved = [...parts, ...impParts].join("/");
+      }
+      // 확장자 정규화
+      if (!resolved.match(/\.(ts|tsx|js|jsx|mjs|cjs)$/)) {
+        return resolved; // 확장자 없으면 그대로 (매칭 시 startsWith로 처리)
+      }
+      return resolved;
+    });
+};
+
+// 파일 경로가 import 경로와 매칭되는지 확인
+const matchesImportPath = (filePath: string, importPath: string): boolean => {
+  const fileWithoutExt = filePath.replace(/\.(ts|tsx|js|jsx|mjs|cjs)$/, "");
+  return (
+    fileWithoutExt === importPath ||
+    fileWithoutExt.startsWith(importPath + "/") ||
+    filePath.startsWith(importPath)
+  );
+};
+
+// 현재 파일과 관련된 파일들 찾기
+// - 정방향: 현재 파일이 import하는 파일
+// - 역방향: 현재 파일을 import하는 파일
+// - 같은 디렉토리
+const findRelatedFiles = (
+  currentFile: string,
+  currentContent: string | null,
+  allChangedFiles: Array<{ filename: string; patch?: string; status?: string; content?: string | null }>
+): Array<{ filename: string; patch: string; relation: "imports" | "imported-by" | "same-dir" }> => {
+  const related: Array<{ filename: string; patch: string; relation: "imports" | "imported-by" | "same-dir" }> = [];
+  const currentDir = getDirname(currentFile);
+  const currentFileWithoutExt = currentFile.replace(/\.(ts|tsx|js|jsx|mjs|cjs)$/, "");
+
+  const importedByCurrentFile = currentContent
+    ? extractImports(currentContent, currentFile)
+    : [];
+
+  for (const file of allChangedFiles) {
+    if (file.filename === currentFile || !file.patch) continue;
+
+    const fileDir = getDirname(file.filename);
+
+    // 정방향: 현재 파일이 이 파일을 import함
+    const isImportedByCurrentFile = importedByCurrentFile.some((imp) =>
+      matchesImportPath(file.filename, imp)
+    );
+
+    // 역방향: 이 파일이 현재 파일을 import함
+    let importsCurrentFile = false;
+    if (file.content) {
+      const fileImports = extractImports(file.content, file.filename);
+      importsCurrentFile = fileImports.some((imp) =>
+        matchesImportPath(currentFile, imp) || currentFileWithoutExt === imp
+      );
+    }
+
+    const isSameDir = fileDir === currentDir;
+
+    if (isImportedByCurrentFile) {
+      related.push({ filename: file.filename, patch: file.patch, relation: "imports" });
+    } else if (importsCurrentFile) {
+      related.push({ filename: file.filename, patch: file.patch, relation: "imported-by" });
+    } else if (isSameDir) {
+      related.push({ filename: file.filename, patch: file.patch, relation: "same-dir" });
+    }
+  }
+
+  return related;
+};
+
+const RELATION_LABELS: Record<string, string> = {
+  "imports": "이 파일이 import함",
+  "imported-by": "이 파일을 import함",
+  "same-dir": "같은 디렉토리",
+};
+
+const buildRelatedFilesContext = (
+  relatedFiles: Array<{ filename: string; patch: string; relation: string }>,
+  budget: number
+): string => {
+  if (relatedFiles.length === 0) return "";
+
+  const parts: string[] = ["## 관련 파일 변경사항 (참고용)"];
+  let usedChars = parts[0].length;
+  const perFileBudget = Math.floor(budget / relatedFiles.length);
+
+  for (const file of relatedFiles) {
+    const relationLabel = RELATION_LABELS[file.relation] || file.relation;
+    const header = `\n### ${file.filename} (${relationLabel})`;
+    const patchContent = truncateForPrompt(file.patch, perFileBudget - header.length - 20);
+    const section = `${header}\n\`\`\`diff\n${patchContent}\n\`\`\``;
+
+    if (usedChars + section.length > budget) break;
+    parts.push(section);
+    usedChars += section.length;
+  }
+
+  return parts.join("\n");
+};
+
 const buildReviewInput = (
   filename: string,
   content: string | null,
   patch: string,
-  globalContext: string | null
+  prMeta: string | null,
+  relatedFilesContext: string | null
 ): string => {
   const parts: string[] = [];
 
@@ -118,13 +247,15 @@ const buildReviewInput = (
     parts.push("```");
   }
 
-  // (선택) PR 전역 컨텍스트는 맨 아래에 위치시켜 영향 최소화
-  if (globalContext) {
+  if (relatedFilesContext) {
     parts.push("\n---\n");
-    parts.push("# PR 전역 컨텍스트 (참고용)");
-    parts.push(
-      truncateForPrompt(globalContext, allocateBudget(MAX_CONTEXT_CHARS, 0.4))
-    );
+    parts.push(relatedFilesContext);
+  }
+
+  if (prMeta) {
+    parts.push("\n---\n");
+    parts.push("# PR 메타 정보 (참고용)");
+    parts.push(truncateForPrompt(prMeta, allocateBudget(MAX_CONTEXT_CHARS, 0.2)));
   }
 
   return parts.join("\n");
@@ -302,54 +433,19 @@ export const robot = (app: Probot) => {
           overallReview = tmp ?? null;
         }
 
-        // 1) PR 제목/커밋 메시지/변경 파일 목록을 제공하고
-        // 2) summarizeChanges 결과(있다면)를 먼저 보여주며
-        // 3) 전체 diff의 발췌본을 함께 전달합니다.
-        let globalContext: string | null = null;
-        {
-          const prTitle = pull_request.title || "(no title)";
-          const changedFileList = changedFiles
-            .map((f) => `- ${f.filename} (${f.status})`)
-            .join("\n");
-          const summary = overallReview?.summary
-            ? `\n\n## 변경 요약\n${overallReview.summary}`
-            : "";
-
-          // meta 모드: 메타데이터만 제공 (제목/커밋 메시지/파일 목록)
-          if (GLOBAL_CONTEXT_MODE === "meta") {
-            globalContext = [
-              `## PR 제목\n${prTitle}`,
-              `\n## 커밋 메시지(집계)\n${truncateForPrompt(
-                aggregatedCommitMessages,
-                allocateBudget(MAX_GLOBAL_CONTEXT_CHARS, 0.4)
-              )}`,
-              `\n## 변경 파일 목록\n${changedFileList}`,
-              summary,
-            ].join("\n");
-          } else {
-            // full 모드: 전체 diff 발췌를 소량만 포함 (하위 호환)
-            const diffDigest = aggregatedPatch
-              ? truncateForPrompt(
-                  aggregatedPatch,
-                  allocateBudget(MAX_GLOBAL_CONTEXT_CHARS, 0.3)
-                )
-              : "";
-            globalContext = [
-              `## PR 제목\n${prTitle}`,
-              `\n## 커밋 메시지(집계)\n${truncateForPrompt(
-                aggregatedCommitMessages,
-                allocateBudget(MAX_GLOBAL_CONTEXT_CHARS, 0.3)
-              )}`,
-              `\n## 변경 파일 목록\n${changedFileList}`,
-              summary,
-              diffDigest
-                ? `\n\n## 전체 diff (일부 발췌)\n\n\`\`\`diff\n${diffDigest}\n\`\`\``
-                : "",
-            ].join("\n");
-          }
-        }
+        const prTitle = pull_request.title || "(no title)";
+        const summary = overallReview?.summary
+          ? `\n## 변경 요약\n${overallReview.summary}`
+          : "";
+        const prMeta = [
+          `## PR 제목\n${prTitle}`,
+          `\n## 커밋 메시지\n${truncateForPrompt(
+            aggregatedCommitMessages,
+            allocateBudget(MAX_GLOBAL_CONTEXT_CHARS, 0.3)
+          )}`,
+          summary,
+        ].join("\n");
 
-        // Fetch PR-wide file patches to compute accurate review comment positions
         let prFilePatchByPath = new Map<string, string>();
         let commentablePaths = new Set<string>();
         try {
@@ -369,8 +465,31 @@ export const robot = (app: Probot) => {
           log.debug("Failed to fetch PR files for position mapping", e);
         }
 
-        for (let i = 0; i < changedFiles.length; i++) {
-          const file = changedFiles[i];
+        type ChangedFileWithContent = {
+          filename: string;
+          patch?: string;
+          status?: string;
+          content?: string | null;
+        };
+        const filesWithContent: ChangedFileWithContent[] = await Promise.all(
+          changedFiles.map(async (file) => {
+            if (file.status !== "modified" && file.status !== "added") {
+              return { ...file, content: null };
+            }
+            try {
+              const content = await fetchFileContent(
+                context,
+                file.filename,
+                context.payload.pull_request.head.sha
+              );
+              return { ...file, content };
+            } catch {
+              return { ...file, content: null };
+            }
+          })
+        );
+
+        for (const file of filesWithContent) {
           const patch = file.patch || "";
 
           if (file.status !== "modified" && file.status !== "added") {
@@ -384,29 +503,19 @@ export const robot = (app: Probot) => {
             continue;
           }
 
-          let fullContent: string | null = null;
-          try {
-            // Only fetch content for reasonably sized files (octokit returns size in metadata, but we guard by prompt length later)
-            fullContent = await fetchFileContent(
-              context,
-              file.filename,
-              context.payload.pull_request.head.sha
-            );
-            if (
-              fullContent &&
-              Buffer.byteLength(fullContent, "utf-8") > MAX_FILE_BYTES
-            ) {
-              // Keep but it will be truncated by buildReviewInput/truncateForPrompt
-            }
-          } catch (e) {
-            log.debug(`content fetch failed for ${file.filename}`, e);
-          }
+          const fullContent = file.content ?? null;
+          const relatedFiles = findRelatedFiles(file.filename, fullContent, filesWithContent);
+          const relatedContext = buildRelatedFilesContext(
+            relatedFiles,
+            allocateBudget(MAX_CONTEXT_CHARS, 0.3)
+          );
 
           const reviewInput = buildReviewInput(
             file.filename,
             fullContent,
             patch,
-            globalContext
+            prMeta,
+            relatedContext || null
           );
 
           try {
