gitlab-ai-review 2.4.0 → 2.5.1

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.3.0';
+    this.version = '2.5.1';
 
     // 如果传入了配置，使用手动配置；否则使用自动检测
     if (options.token || options.gitlab) {
@@ -166,14 +166,35 @@ export class GitLabAIReview {
           continue;
         }
 
-        console.log(`  发现 ${meaningfulChanges.length} 个有意义的变更`);
+        const additionsCount = meaningfulChanges.filter(c => c.type === 'addition').length;
+        const deletionsCount = meaningfulChanges.filter(c => c.type === 'deletion').length;
+        console.log(`  发现 ${meaningfulChanges.length} 个有意义的变更（新增 ${additionsCount} 行，删除 ${deletionsCount} 行）`);
 
         // 调用 AI 一次性审查整个文件的所有变更（按文件批量）
         const fileReview = await this.reviewFileChanges(change, meaningfulChanges);
         
-        // 根据 AI 返回的结果，只对有问题的行添加评论
+        // 创建有效新增行号的集合（用于验证，只包含新增的行）
+        const validAdditionLineNumbers = new Set(
+          meaningfulChanges
+            .filter(c => c.type === 'addition')  // 只包含新增的行
+            .map(c => c.lineNumber)
+        );
+        
+        // 根据 AI 返回的结果，只对有问题的新增行添加评论
         for (const review of fileReview.reviews) {
           if (review.hasIssue) {
+            // 验证行号是否是新增行
+            if (!validAdditionLineNumbers.has(review.lineNumber)) {
+              console.log(`  ⚠ 第 ${review.lineNumber} 行：不是新增行或不在变更范围内，跳过评论`);
+              results.push({
+                status: 'skipped',
+                fileName,
+                lineNumber: review.lineNumber,
+                reason: '不是新增行或行号不在 diff 变更范围内',
+              });
+              continue;
+            }
+            
             try {
               const commentResult = await this.gitlabClient.createLineComment(
                 this.config.project.projectId,
@@ -238,8 +259,11 @@ export class GitLabAIReview {
     const projectPrompt = this.config.ai?.guardConfig?.content || '';
     const fileName = change.new_path || change.old_path;
 
-    // 构建整个文件的批量审查消息
-    const messages = PromptTools.buildFileReviewMessages(fileName, meaningfulChanges, projectPrompt);
+    // 按 hunk 分组变更
+    const hunkGroups = DiffParser.groupChangesByHunk(meaningfulChanges);
+
+    // 构建整个文件的批量审查消息（按 hunk 组织）
+    const messages = PromptTools.buildFileReviewMessages(fileName, meaningfulChanges, hunkGroups, projectPrompt);
 
     // 调用 AI（一次调用审查整个文件）
     const response = await aiClient.sendMessage(messages);
@@ -255,6 +279,9 @@ export class GitLabAIReview {
         jsonStr = jsonStr.slice(3, -3).trim();
       }
 
+      // 清理可能的格式错误（如多余的括号）
+      jsonStr = this.cleanJsonString(jsonStr);
+
       const result = JSON.parse(jsonStr);
       return result;
     } catch (error) {
@@ -264,6 +291,26 @@ export class GitLabAIReview {
     }
   }
 
+  /**
+   * 清理 JSON 字符串中的常见格式错误
+   * @param {string} jsonStr - 待清理的 JSON 字符串
+   * @returns {string} 清理后的 JSON 字符串
+   */
+  cleanJsonString(jsonStr) {
+    // 修复常见的 JSON 格式错误
+    
+    // 1. 移除末尾多余的 ]}（如 ]}]} 改为 ]}）
+    jsonStr = jsonStr.replace(/\]\}\]\}$/g, ']}');
+    
+    // 2. 移除末尾多余的 ]（如 }]} 改为 }）
+    jsonStr = jsonStr.replace(/\}\]\}$/g, '}');
+    
+    // 3. 移除其他常见的多余闭合符号
+    jsonStr = jsonStr.replace(/\]\]$/g, ']');
+    
+    return jsonStr.trim();
+  }
+
   /**
    * 测试方法
    */

package/lib/diff-parser.js

@@ -187,11 +187,42 @@ export function generateDiffSummary(diff) {
   };
 }
 
+/**
+ * 将有意义的变更按 hunk 分组
+ * @param {Array} meaningfulChanges - 有意义的变更数组
+ * @returns {Array} 按 hunk 分组的变更数组
+ */
+export function groupChangesByHunk(meaningfulChanges) {
+  const hunkMap = new Map();
+  
+  meaningfulChanges.forEach(change => {
+    const hunkKey = change.hunk; // 使用 hunk header 作为 key
+    
+    if (!hunkMap.has(hunkKey)) {
+      hunkMap.set(hunkKey, {
+        header: hunkKey,
+        additions: [],
+        deletions: [],
+      });
+    }
+    
+    const hunkGroup = hunkMap.get(hunkKey);
+    if (change.type === 'addition') {
+      hunkGroup.additions.push(change);
+    } else if (change.type === 'deletion') {
+      hunkGroup.deletions.push(change);
+    }
+  });
+  
+  return Array.from(hunkMap.values());
+}
+
 export default {
   parseHunkHeader,
   parseDiff,
   calculateNewLineNumber,
   extractMeaningfulChanges,
   generateDiffSummary,
+  groupChangesByHunk,
 };
 

package/lib/prompt-tools.js

@@ -28,51 +28,90 @@ export function buildSystemPrompt(projectPrompt = '') {
 }
 
 /**
- * 构建整个文件所有变更行的批量审查提示词
+ * 构建整个文件所有变更行的批量审查提示词（按 hunk 组织）
  * @param {string} fileName - 文件名
  * @param {Array} meaningfulChanges - 有意义的变更数组
+ * @param {Array} hunkGroups - 按 hunk 分组的变更数组
  * @param {string} projectPrompt - 项目配置的 prompt
  * @returns {string} 批量审查提示词
  */
-export function buildFileReviewPrompt(fileName, meaningfulChanges) {
-  let prompt = `请审查以下文件的代码变更，只对**有问题的行**提出审查意见，代码没有问题的行不需要评论。
+export function buildFileReviewPrompt(fileName, meaningfulChanges, hunkGroups) {
+  const totalAdditions = meaningfulChanges.filter(c => c.type === 'addition').length;
+  const totalDeletions = meaningfulChanges.filter(c => c.type === 'deletion').length;
+  
+  let prompt = `请审查以下文件的代码变更，只对**有问题的新增行**提出审查意见。
 
 **文件名**: ${fileName}
-**变更数量**: ${meaningfulChanges.length} 行
+**变更统计**: 
+- 共 ${hunkGroups.length} 个代码块（hunk）
+- 删除了 ${totalDeletions} 行（旧代码）
+- 新增了 ${totalAdditions} 行（新代码）
+
+## 重要说明
+- **每个代码块都明确标注了删除内容和新增内容**
+- **删除的行**：标注为【删除】，是被移除的旧代码，**无需审查**
+- **新增的行**：标注为【新增】，是新加入的代码，**只审查这些行**
+- 行号是指新增行在新文件中的位置
+
+---
 
 `;
 
-  meaningfulChanges.forEach((change, index) => {
-    prompt += `\n### 变更 ${index + 1} - 第 ${change.lineNumber} 行\n`;
-    prompt += `**类型**: ${change.type === 'addition' ? '新增' : '删除'}\n`;
-    prompt += `**内容**:\n\`\`\`\n`;
+  // 按 hunk 展示变更
+  hunkGroups.forEach((hunk, hunkIndex) => {
+    prompt += `## 代码块 ${hunkIndex + 1}：${hunk.header}\n\n`;
     
-    // 上下文
-    if (change.context?.before?.length > 0) {
-      change.context.before.forEach(line => {
-        prompt += `  ${line}\n`;
+    // 展示删除的内容
+    if (hunk.deletions.length > 0) {
+      prompt += `### 📝 该代码块中删除的内容（无需审查）\n\n`;
+      hunk.deletions.forEach((change, index) => {
+        prompt += `\`\`\`\n`;
+        // 上下文
+        if (change.context?.before?.length > 0 && index === 0) {
+          change.context.before.forEach(line => {
+            prompt += `  ${line}\n`;
+          });
+        }
+        // 删除的行
+        prompt += `【删除】${change.content}\n`;
+        prompt += `\`\`\`\n\n`;
       });
     }
     
-    // 变更行
-    prompt += `${change.type === 'addition' ? '+' : '-'} ${change.content}\n`;
-    
-    // 下文
-    if (change.context?.after?.length > 0) {
-      change.context.after.forEach(line => {
-        prompt += `  ${line}\n`;
+    // 展示新增的内容
+    if (hunk.additions.length > 0) {
+      prompt += `### ✨ 该代码块中新增的内容（**请审查**）\n\n`;
+      hunk.additions.forEach((change, index) => {
+        prompt += `\`\`\`\n`;
+        // 上下文
+        if (change.context?.before?.length > 0 && index === 0) {
+          change.context.before.forEach(line => {
+            prompt += `  ${line}\n`;
+          });
+        }
+        // 新增的行
+        prompt += `【新增 - 第 ${change.lineNumber} 行】${change.content}\n`;
+        prompt += `\`\`\`\n\n`;
       });
     }
     
-    prompt += `\`\`\`\n`;
+    // 如果该 hunk 没有删除也没有新增
+    if (hunk.deletions.length === 0 && hunk.additions.length === 0) {
+      prompt += `*该代码块无实质性变更*\n\n`;
+    }
+    
+    prompt += `---\n\n`;
   });
 
-  prompt += `\n## 审查要求
+  prompt += `## 审查要求
 
-1. 仔细审查每一行变更，判断是否存在问题
-2. **只对有问题的行提出审查意见**，没有问题的行不需要评论
-3. **必须使用中文**返回审查意见
-4. 返回 JSON 格式的结果，格式如下：
+1. **只审查上面标注为【新增】的代码行**
+2. **不要审查标注为【删除】的代码行**
+3. 每个代码块的删除和新增内容已明确标注，请关注新增内容的质量
+4. 判断新增的代码是否存在问题（安全、性能、逻辑、最佳实践等）
+5. **只对有问题的新增行提出审查意见**，没有问题的行不需要评论
+6. **必须使用中文**返回审查意见
+7. 返回 JSON 格式的结果，格式如下：
 
 \`\`\`json
 {
@@ -90,9 +129,10 @@ export function buildFileReviewPrompt(fileName, meaningfulChanges) {
 }
 \`\`\`
 
-5. 如果所有代码都没有问题，返回空的 reviews 数组：\`{"reviews": []}\`
-6. 只返回 JSON，不要包含其他文字说明
-7. comment 字段必须使用中文，要简洁明了`;
+8. lineNumber 必须是新增行的行号（在【新增 - 第 X 行】中标注的行号）
+9. 如果所有新增代码都没有问题，返回空的 reviews 数组：\`{"reviews": []}\`
+10. 只返回 JSON，不要包含其他文字说明
+11. comment 字段必须使用中文，要简洁明了`;
 
   return prompt;
 }
@@ -101,13 +141,14 @@ export function buildFileReviewPrompt(fileName, meaningfulChanges) {
  * 构建整个文件批量审查的完整消息数组
  * @param {string} fileName - 文件名
  * @param {Array} meaningfulChanges - 有意义的变更数组
+ * @param {Array} hunkGroups - 按 hunk 分组的变更数组
  * @param {string} projectPrompt - 项目配置的 prompt
  * @returns {Array} 消息数组
  */
-export function buildFileReviewMessages(fileName, meaningfulChanges, projectPrompt = '') {
+export function buildFileReviewMessages(fileName, meaningfulChanges, hunkGroups, projectPrompt = '') {
   return [
     { role: 'system', content: buildSystemPrompt(projectPrompt) },
-    { role: 'user', content: buildFileReviewPrompt(fileName, meaningfulChanges) },
+    { role: 'user', content: buildFileReviewPrompt(fileName, meaningfulChanges, hunkGroups) },
   ];
 }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "gitlab-ai-review",
-  "version": "2.4.0",
+  "version": "2.5.1",
   "description": "GitLab AI Review SDK - 支持 SDK 调用和 CLI 执行",
   "main": "index.js",
   "type": "module",