gitlab-ai-review 2.5.4 → 3.0.1

package/index.js

@@ -8,6 +8,7 @@ import { GitLabClient } from './lib/gitlab-client.js';
 import { AIClient } from './lib/ai-client.js';
 import * as PromptTools from './lib/prompt-tools.js';
 import * as DiffParser from './lib/diff-parser.js';
+import * as ImpactAnalyzer from './lib/impact-analyzer.js';
 
 /**
  * GitLab AI Review SDK 主类
@@ -15,7 +16,7 @@ import * as DiffParser from './lib/diff-parser.js';
 export class GitLabAIReview {
   constructor(options = {}) {
     this.name = 'GitLab AI Review SDK';
-    this.version = '2.5.4';
+    this.version = '2.3.0';
 
     // 如果传入了配置，使用手动配置；否则使用自动检测
     if (options.token || options.gitlab) {
@@ -166,35 +167,14 @@ export class GitLabAIReview {
           continue;
         }
 
-        const additionsCount = meaningfulChanges.filter(c => c.type === 'addition').length;
-        const deletionsCount = meaningfulChanges.filter(c => c.type === 'deletion').length;
-        console.log(`  发现 ${meaningfulChanges.length} 个有意义的变更（新增 ${additionsCount} 行，删除 ${deletionsCount} 行）`);
+        console.log(`  发现 ${meaningfulChanges.length} 个有意义的变更`);
 
         // 调用 AI 一次性审查整个文件的所有变更（按文件批量）
         const fileReview = await this.reviewFileChanges(change, meaningfulChanges);
         
-        // 创建有效新增行号的集合（用于验证，只包含新增的行）
-        const validAdditionLineNumbers = new Set(
-          meaningfulChanges
-            .filter(c => c.type === 'addition')  // 只包含新增的行
-            .map(c => c.lineNumber)
-        );
-        
-        // 根据 AI 返回的结果，只对有问题的新增行添加评论
+        // 根据 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,
@@ -259,11 +239,8 @@ export class GitLabAIReview {
     const projectPrompt = this.config.ai?.guardConfig?.content || '';
     const fileName = change.new_path || change.old_path;
 
-    // 按 hunk 分组变更
-    const hunkGroups = DiffParser.groupChangesByHunk(meaningfulChanges);
-
-    // 构建整个文件的批量审查消息（按 hunk 组织）
-    const messages = PromptTools.buildFileReviewMessages(fileName, meaningfulChanges, hunkGroups, projectPrompt);
+    // 构建整个文件的批量审查消息
+    const messages = PromptTools.buildFileReviewMessages(fileName, meaningfulChanges, projectPrompt);
 
     // 调用 AI（一次调用审查整个文件）
     const response = await aiClient.sendMessage(messages);
@@ -279,9 +256,6 @@ export class GitLabAIReview {
         jsonStr = jsonStr.slice(3, -3).trim();
       }
 
-      // 清理可能的格式错误（如多余的括号）
-      jsonStr = this.cleanJsonString(jsonStr);
-
       const result = JSON.parse(jsonStr);
       return result;
     } catch (error) {
@@ -292,23 +266,159 @@ export class GitLabAIReview {
   }
 
   /**
-   * 清理 JSON 字符串中的常见格式错误
-   * @param {string} jsonStr - 待清理的 JSON 字符串
-   * @returns {string} 清理后的 JSON 字符串
+   * AI 审查 MR 的所有变更（包含影响分析）
+   * @param {Object} options - 选项
+   * @param {number} options.maxFiles - 最大审查文件数量（默认不限制）
+   * @param {number} options.maxAffectedFiles - 每个文件最多分析的受影响文件数量（默认 10）
+   * @param {boolean} options.enableImpactAnalysis - 是否启用影响分析（默认 true）
+   * @returns {Promise<Array>} 评论结果数组
    */
-  cleanJsonString(jsonStr) {
-    // 修复常见的 JSON 格式错误
+  async reviewWithImpactAnalysis(options = {}) {
+    const { 
+      maxFiles = Infinity, 
+      maxAffectedFiles = 10,
+      enableImpactAnalysis = true 
+    } = options;
     
-    // 1. 移除末尾多余的 ]}（如 ]}]} 改为 ]}）
-    jsonStr = jsonStr.replace(/\]\}\]\}$/g, ']}');
-    
-    // 2. 移除末尾多余的 ]（如 }]} 改为 }）
-    jsonStr = jsonStr.replace(/\}\]\}$/g, '}');
-    
-    // 3. 移除其他常见的多余闭合符号
-    jsonStr = jsonStr.replace(/\]\]$/g, ']');
-    
-    return jsonStr.trim();
+    const changes = await this.getMergeRequestChanges();
+    const mrInfo = await this.getMergeRequest();
+    const ref = mrInfo.target_branch || 'main';
+    const results = [];
+
+    const filesToReview = maxFiles === Infinity ? changes.length : Math.min(maxFiles, changes.length);
+    console.log(`共 ${changes.length} 个文件需要审查${maxFiles === Infinity ? '（不限制数量）' : `（最多审查 ${maxFiles} 个）`}`);
+    console.log(`影响分析: ${enableImpactAnalysis ? '已启用' : '已禁用'}`);
+
+    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} 个有意义的变更`);
+
+        // 影响分析
+        let impactAnalysis = null;
+        if (enableImpactAnalysis) {
+          impactAnalysis = await ImpactAnalyzer.analyzeImpact({
+            gitlabClient: this.gitlabClient,
+            projectId: this.config.project.projectId,
+            ref: ref,
+            change: change,
+            maxAffectedFiles: maxAffectedFiles,
+          });
+        }
+
+        // 调用 AI 审查（包含影响分析）
+        const fileReview = await this.reviewFileChangesWithImpact(
+          change, 
+          meaningfulChanges, 
+          impactAnalysis
+        );
+        
+        // 根据 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,
+                }
+              );
+
+              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;
+  }
+
+  /**
+   * 审查单个文件的所有变更（包含影响分析）
+   * @param {Object} change - 代码变更对象
+   * @param {Array} meaningfulChanges - 有意义的变更数组
+   * @param {Object} impactAnalysis - 影响分析结果
+   * @returns {Promise<Object>} 审查结果 { reviews: [{lineNumber, hasIssue, comment}] }
+   */
+  async reviewFileChangesWithImpact(change, meaningfulChanges, impactAnalysis) {
+    const aiClient = this.getAIClient();
+    const projectPrompt = this.config.ai?.guardConfig?.content || '';
+    const fileName = change.new_path || change.old_path;
+
+    // 构建包含影响分析的批量审查消息
+    const messages = impactAnalysis 
+      ? PromptTools.buildFileReviewWithImpactMessages(fileName, meaningfulChanges, impactAnalysis, projectPrompt)
+      : 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: [] };
+    }
   }
 
   /**
@@ -323,7 +433,7 @@ export class GitLabAIReview {
 export { getConfig, validateConfig } from './lib/config.js';
 export { GitLabClient } from './lib/gitlab-client.js';
 export { AIClient } from './lib/ai-client.js';
-export { PromptTools, DiffParser };
+export { PromptTools, DiffParser, ImpactAnalyzer };
 
 // 默认导出
 export default GitLabAIReview;

package/lib/diff-parser.js

@@ -187,42 +187,11 @@ 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/gitlab-client.js

@@ -145,5 +145,74 @@ export class GitLabClient {
 
     return results;
   }
+
+  /**
+   * 获取项目中的文件内容
+   * @param {string} projectId - 项目 ID
+   * @param {string} filePath - 文件路径
+   * @param {string} ref - 分支名或提交 SHA
+   * @returns {Promise<string|null>} 文件内容
+   */
+  async getProjectFile(projectId, filePath, ref) {
+    try {
+      const response = await fetch(
+        `${this.apiUrl}/projects/${encodeURIComponent(projectId)}/repository/files/${encodeURIComponent(filePath)}/raw?ref=${ref}`,
+        {
+          headers: {
+            'PRIVATE-TOKEN': this.token,
+          },
+        }
+      );
+
+      if (!response.ok) {
+        if (response.status === 404) {
+          return null;
+        }
+        throw new Error(`获取文件失败: ${response.status}`);
+      }
+
+      return await response.text();
+    } catch (error) {
+      console.warn(`获取文件 ${filePath} 失败:`, error.message);
+      return null;
+    }
+  }
+
+  /**
+   * 在项目中搜索代码
+   * @param {string} projectId - 项目 ID
+   * @param {string} search - 搜索关键词
+   * @param {string} ref - 分支名
+   * @returns {Promise<Array>} 搜索结果
+   */
+  async searchInProject(projectId, search, ref) {
+    try {
+      const response = await this.request(
+        `/projects/${encodeURIComponent(projectId)}/search?scope=blobs&search=${encodeURIComponent(search)}&ref=${ref}`
+      );
+      
+      return response || [];
+    } catch (error) {
+      console.warn(`搜索 "${search}" 失败:`, error.message);
+      return [];
+    }
+  }
+
+  /**
+   * 获取仓库树结构
+   * @param {string} projectId - 项目 ID
+   * @param {string} ref - 分支名
+   * @param {string} path - 路径（可选）
+   * @returns {Promise<Array>} 文件树
+   */
+  async getRepositoryTree(projectId, ref, path = '') {
+    try {
+      const endpoint = `/projects/${encodeURIComponent(projectId)}/repository/tree?ref=${ref}&path=${encodeURIComponent(path)}&recursive=false`;
+      return await this.request(endpoint);
+    } catch (error) {
+      console.warn(`获取仓库树失败:`, error.message);
+      return [];
+    }
+  }
 }