gitlab-ai-review 2.0.0 → 2.2.0

package/index.js

@@ -15,7 +15,7 @@ import * as DiffParser from './lib/diff-parser.js';
 export class GitLabAIReview {
   constructor(options = {}) {
     this.name = 'GitLab AI Review SDK';
-    this.version = '2.0.0';
+    this.version = '2.2.0';
 
     // 如果传入了配置，使用手动配置；否则使用自动检测
     if (options.token || options.gitlab) {
@@ -179,50 +179,132 @@ export class GitLabAIReview {
   }
 
   /**
-   * AI 审查 MR 的所有有意义的变更并自动添加行级评论
+   * AI 审查 MR 的所有有意义的变更并自动添加行级评论（按文件批量处理）
    * @param {Object} options - 选项
-   * @param {number} options.maxComments - 最大评论数量（避免过多评论）
+   * @param {number} options.maxFiles - 最大审查文件数量（默认不限制）
    * @returns {Promise<Array>} 评论结果数组
    */
   async reviewAndCommentOnLines(options = {}) {
-    const { maxComments = 10 } = options;
+    const { maxFiles = Infinity } = options;
     const changes = await this.getMergeRequestChanges();
     const results = [];
 
-    for (const change of changes) {
-      if (results.length >= maxComments) {
-        console.log(`已达到最大评论数量限制 (${maxComments})，停止审查`);
-        break;
-      }
+    const filesToReview = maxFiles === Infinity ? changes.length : Math.min(maxFiles, changes.length);
+    console.log(`共 ${changes.length} 个文件需要审查${maxFiles === Infinity ? '（不限制数量）' : `（最多审查 ${maxFiles} 个）`}`);
+
+    for (const change of changes.slice(0, filesToReview)) {
+      const fileName = change.new_path || change.old_path;
+      
+      try {
+        console.log(`\n审查文件: ${fileName}`);
+        
+        // 解析 diff，提取有意义的变更
+        const hunks = DiffParser.parseDiff(change.diff);
+        const meaningfulChanges = DiffParser.extractMeaningfulChanges(hunks);
+
+        if (meaningfulChanges.length === 0) {
+          console.log(`  跳过：没有有意义的变更`);
+          continue;
+        }
+
+        console.log(`  发现 ${meaningfulChanges.length} 个有意义的变更`);
+
+        // 调用 AI 一次性审查整个文件的所有变更（按文件批量）
+        const fileReview = await this.reviewFileChanges(change, meaningfulChanges);
+        
+        // 根据 AI 返回的结果，只对有问题的行添加评论
+        for (const review of fileReview.reviews) {
+          if (review.hasIssue) {
+            try {
+              const commentResult = await this.gitlabClient.createLineComment(
+                this.config.project.projectId,
+                this.config.project.mergeRequestIid,
+                `🤖 **AI 代码审查**\n\n${review.comment}`,
+                {
+                  filePath: fileName,
+                  oldPath: change.old_path,
+                  newLine: review.lineNumber,
+                }
+              );
 
-      // 解析 diff，提取有意义的变更
-      const hunks = DiffParser.parseDiff(change.diff);
-      const meaningfulChanges = DiffParser.extractMeaningfulChanges(hunks);
-
-      // 对每个有意义的变更进行审查
-      for (const meaningfulChange of meaningfulChanges.slice(0, maxComments - results.length)) {
-        try {
-          const result = await this.reviewAndCommentOnLine(change, meaningfulChange);
-          results.push({
-            status: 'success',
-            fileName: change.new_path || change.old_path,
-            lineNumber: meaningfulChange.lineNumber,
-            ...result,
-          });
-        } catch (error) {
-          results.push({
-            status: 'error',
-            fileName: change.new_path || change.old_path,
-            lineNumber: meaningfulChange.lineNumber,
-            error: error.message,
-          });
+              results.push({
+                status: 'success',
+                fileName,
+                lineNumber: review.lineNumber,
+                comment: review.comment,
+                commentResult,
+              });
+
+              console.log(`  ✓ 第 ${review.lineNumber} 行：已添加评论`);
+            } catch (error) {
+              results.push({
+                status: 'error',
+                fileName,
+                lineNumber: review.lineNumber,
+                error: error.message,
+              });
+              console.log(`  ✗ 第 ${review.lineNumber} 行：评论失败 - ${error.message}`);
+            }
+          } else {
+            console.log(`  ✓ 第 ${review.lineNumber} 行：代码质量良好`);
+          }
+        }
+
+        // 如果没有问题
+        if (fileReview.reviews.length === 0 || fileReview.reviews.every(r => !r.hasIssue)) {
+          console.log(`  ✓ 所有代码质量良好，无需评论`);
         }
+
+      } catch (error) {
+        results.push({
+          status: 'error',
+          fileName,
+          error: error.message,
+        });
+        console.log(`  ✗ 文件审查失败: ${error.message}`);
       }
     }
 
     return results;
   }
 
+  /**
+   * 审查单个文件的所有变更（一次 API 调用）
+   * @param {Object} change - 代码变更对象
+   * @param {Array} meaningfulChanges - 有意义的变更数组
+   * @returns {Promise<Object>} 审查结果 { reviews: [{lineNumber, hasIssue, comment}] }
+   */
+  async reviewFileChanges(change, meaningfulChanges) {
+    const aiClient = this.getAIClient();
+    const projectPrompt = this.config.ai?.guardConfig?.content || '';
+    const fileName = change.new_path || change.old_path;
+
+    // 构建整个文件的批量审查消息
+    const messages = PromptTools.buildFileReviewMessages(fileName, meaningfulChanges, projectPrompt);
+
+    // 调用 AI（一次调用审查整个文件）
+    const response = await aiClient.sendMessage(messages);
+
+    // 解析 AI 返回的 JSON
+    try {
+      // 提取 JSON（可能被包裹在 ```json ``` 中）
+      let jsonStr = response.content.trim();
+      const jsonMatch = jsonStr.match(/```json\s*([\s\S]*?)\s*```/);
+      if (jsonMatch) {
+        jsonStr = jsonMatch[1];
+      } else if (jsonStr.startsWith('```') && jsonStr.endsWith('```')) {
+        jsonStr = jsonStr.slice(3, -3).trim();
+      }
+
+      const result = JSON.parse(jsonStr);
+      return result;
+    } catch (error) {
+      console.error('解析 AI 返回的 JSON 失败:', error.message);
+      console.error('AI 原始返回:', response.content);
+      return { reviews: [] };
+    }
+  }
+
   /**
    * 测试方法
    */

package/lib/prompt-tools.js

@@ -92,10 +92,96 @@ export function buildLineReviewMessages(changeInfo, projectPrompt = '') {
   ];
 }
 
-// 只导出行级审查需要的函数
+/**
+ * 构建整个文件所有变更行的批量审查提示词
+ * @param {string} fileName - 文件名
+ * @param {Array} meaningfulChanges - 有意义的变更数组
+ * @param {string} projectPrompt - 项目配置的 prompt
+ * @returns {string} 批量审查提示词
+ */
+export function buildFileReviewPrompt(fileName, meaningfulChanges) {
+  let prompt = `请审查以下文件的代码变更，只对**有问题的行**提出审查意见，代码没有问题的行不需要评论。
+
+**文件名**: ${fileName}
+**变更数量**: ${meaningfulChanges.length} 行
+
+`;
+
+  meaningfulChanges.forEach((change, index) => {
+    prompt += `\n### 变更 ${index + 1} - 第 ${change.lineNumber} 行\n`;
+    prompt += `**类型**: ${change.type === 'addition' ? '新增' : '删除'}\n`;
+    prompt += `**内容**:\n\`\`\`\n`;
+    
+    // 上下文
+    if (change.context?.before?.length > 0) {
+      change.context.before.forEach(line => {
+        prompt += `  ${line}\n`;
+      });
+    }
+    
+    // 变更行
+    prompt += `${change.type === 'addition' ? '+' : '-'} ${change.content}\n`;
+    
+    // 下文
+    if (change.context?.after?.length > 0) {
+      change.context.after.forEach(line => {
+        prompt += `  ${line}\n`;
+      });
+    }
+    
+    prompt += `\`\`\`\n`;
+  });
+
+  prompt += `\n## 审查要求
+
+1. 仔细审查每一行变更，判断是否存在问题
+2. **只对有问题的行提出审查意见**，没有问题的行不需要评论
+3. **必须使用中文**返回审查意见
+4. 返回 JSON 格式的结果，格式如下：
+
+\`\`\`json
+{
+  "reviews": [
+    {
+      "lineNumber": 15,
+      "hasIssue": true,
+      "comment": "这里用中文描述具体的问题和改进建议"
+    },
+    {
+      "lineNumber": 20,
+      "hasIssue": false
+    }
+  ]
+}
+\`\`\`
+
+5. 如果所有代码都没有问题，返回空的 reviews 数组：\`{"reviews": []}\`
+6. 只返回 JSON，不要包含其他文字说明
+7. comment 字段必须使用中文，要简洁明了`;
+
+  return prompt;
+}
+
+/**
+ * 构建整个文件批量审查的完整消息数组
+ * @param {string} fileName - 文件名
+ * @param {Array} meaningfulChanges - 有意义的变更数组
+ * @param {string} projectPrompt - 项目配置的 prompt
+ * @returns {Array} 消息数组
+ */
+export function buildFileReviewMessages(fileName, meaningfulChanges, projectPrompt = '') {
+  return [
+    { role: 'system', content: buildSystemPrompt(projectPrompt) },
+    { role: 'user', content: buildFileReviewPrompt(fileName, meaningfulChanges) },
+  ];
+}
+
+// 导出函数
 export default {
   buildSystemPrompt,
   buildLineReviewPrompt,
   buildLineReviewMessages,
+  buildFileReviewPrompt,
+  buildFileReviewMessages,
 };
 

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "gitlab-ai-review",
-  "version": "2.0.0",
-  "description": "GitLab AI Review SDK - 行级代码审查，附带项目 Prompt",
+  "version": "2.2.0",
+  "description": "GitLab AI Review SDK - 按文件批量审查，只评论有问题的代码",
   "main": "index.js",
   "type": "module",
   "scripts": {