gitlab-ai-review 2.2.0 → 2.4.0

package/cli.js

@@ -0,0 +1,66 @@
+#!/usr/bin/env node
+
+/**
+ * GitLab AI Review CLI
+ * 支持通过 npx 直接执行代码审查
+ */
+
+import { GitLabAIReview } from './index.js';
+
+async function main() {
+  console.log('🚀 GitLab AI Review CLI 启动...\n');
+
+  try {
+    // 1. 创建 SDK 实例（自动从环境变量读取配置）
+    const sdk = new GitLabAIReview();
+
+    // 2. 验证配置
+    console.log('📋 配置信息:');
+    console.log(`  - GitLab Host: ${sdk.config.gitlab.host}`);
+    console.log(`  - Project ID: ${sdk.config.project.projectId}`);
+    console.log(`  - MR IID: ${sdk.config.project.mergeRequestIid}`);
+    console.log(`  - AI API Key: ${sdk.config.ai?.arkApiKey ? '✅ 已配置' : '❌ 未配置'}`);
+    console.log(`  - Review Guard: ${sdk.config.ai?.guardConfig?.content ? '✅ 已加载' : '⚠️ 未找到'}`);
+    console.log();
+
+    // 3. 获取 MR 变更信息
+    console.log('📥 获取 MR 变更信息...');
+    const changes = await sdk.getMergeRequestChanges();
+    console.log(`✅ 共发现 ${changes.length} 个文件变更\n`);
+
+    // 4. 执行 AI 代码审查
+    if (sdk.config.ai?.arkApiKey && changes.length > 0) {
+      console.log('🤖 开始 AI 代码审查...\n');
+      
+      const results = await sdk.reviewAndCommentOnLines();
+      
+      console.log('\n📊 审查结果汇总:');
+      console.log(`  - 审查文件数: ${results.length}`);
+      const successCount = results.filter(r => r.success).length;
+      const commentCount = results.reduce((sum, r) => sum + (r.commentCount || 0), 0);
+      console.log(`  - 成功: ${successCount}`);
+      console.log(`  - 添加评论数: ${commentCount}`);
+      
+      if (results.some(r => !r.success)) {
+        console.log('\n⚠️ 部分文件审查失败，请检查上方日志');
+        process.exit(1);
+      } else {
+        console.log('\n✅ 代码审查完成！');
+      }
+    } else if (!sdk.config.ai?.arkApiKey) {
+      console.error('❌ 错误: 未配置 ARK_API_KEY 环境变量');
+      process.exit(1);
+    } else {
+      console.log('ℹ️ 没有文件变更，跳过审查');
+    }
+
+  } catch (error) {
+    console.error('\n❌ 执行失败:', error.message);
+    console.error(error.stack);
+    process.exit(1);
+  }
+}
+
+// 执行主函数
+main();
+

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.2.0';
+    this.version = '2.3.0';
 
     // 如果传入了配置，使用手动配置；否则使用自动检测
     if (options.token || options.gitlab) {
@@ -137,47 +137,6 @@ export class GitLabAIReview {
     return this.aiClient;
   }
 
-  /**
-   * AI 审查特定行的变更并在该行添加评论（附带项目 prompt）
-   * @param {Object} change - 代码变更对象
-   * @param {Object} changeInfo - 具体的行变更信息
-   * @returns {Promise<Object>} 评论结果
-   */
-  async reviewAndCommentOnLine(change, changeInfo) {
-    const aiClient = this.getAIClient();
-    
-    // 获取项目配置的 prompt（来自 reviewguard.md）
-    const projectPrompt = this.config.ai?.guardConfig?.content || '';
-
-    // 构建针对特定行的审查消息（附带项目 prompt）
-    const messages = PromptTools.buildLineReviewMessages({
-      ...changeInfo,
-      fileName: change.new_path || change.old_path,
-    }, projectPrompt);
-
-    // 调用 AI 审查
-    const review = await aiClient.sendMessage(messages);
-
-    // 在该行添加评论
-    const lineInfo = {
-      filePath: change.new_path || change.old_path,
-      oldPath: change.old_path,
-      newLine: changeInfo.lineNumber,
-    };
-
-    const commentResult = await this.gitlabClient.createLineComment(
-      this.config.project.projectId,
-      this.config.project.mergeRequestIid,
-      `🤖 **AI 代码审查**\n\n${review.content}`,
-      lineInfo
-    );
-
-    return {
-      review,
-      comment: commentResult,
-    };
-  }
-
   /**
    * AI 审查 MR 的所有有意义的变更并自动添加行级评论（按文件批量处理）
    * @param {Object} options - 选项

package/lib/prompt-tools.js

@@ -27,71 +27,6 @@ export function buildSystemPrompt(projectPrompt = '') {
   return prompt;
 }
 
-/**
- * 构建针对特定行变更的审查提示词（附带项目 prompt）
- * @param {Object} changeInfo - 变更信息
- * @param {string} changeInfo.content - 变更内容
- * @param {string} changeInfo.type - 变更类型（addition/deletion）
- * @param {number} changeInfo.lineNumber - 行号
- * @param {string} changeInfo.fileName - 文件名
- * @param {Object} changeInfo.context - 上下文
- * @returns {string} 审查提示词
- */
-export function buildLineReviewPrompt(changeInfo) {
-  const { content, type, lineNumber, fileName, context } = changeInfo;
-  
-  let prompt = `请审查以下代码变更：
-
-**文件**: ${fileName}
-**行号**: ${lineNumber}
-**变更类型**: ${type === 'addition' ? '新增' : '删除'}
-
-**变更内容**:
-\`\`\`
-${type === 'addition' ? '+' : '-'} ${content}
-\`\`\`
-`;
-
-  if (context && (context.before.length > 0 || context.after.length > 0)) {
-    prompt += '\n**上下文**:\n```\n';
-    if (context.before.length > 0) {
-      context.before.forEach(line => {
-        prompt += `  ${line}\n`;
-      });
-    }
-    prompt += `${type === 'addition' ? '+' : '-'} ${content}\n`;
-    if (context.after.length > 0) {
-      context.after.forEach(line => {
-        prompt += `  ${line}\n`;
-      });
-    }
-    prompt += '```\n';
-  }
-
-  prompt += `
-请提供简洁的审查意见：
-1. 如果有问题，指出具体问题
-2. 如果有改进建议，提供具体的建议
-3. 如果代码没有问题，简单说明"代码质量良好"
-
-请直接给出审查意见，不要重复代码内容。`;
-
-  return prompt;
-}
-
-/**
- * 构建针对特定行变更的完整消息数组（附带项目 prompt）
- * @param {Object} changeInfo - 变更信息
- * @param {string} projectPrompt - 项目配置的 prompt（来自 reviewguard.md）
- * @returns {Array} 消息数组
- */
-export function buildLineReviewMessages(changeInfo, projectPrompt = '') {
-  return [
-    { role: 'system', content: buildSystemPrompt(projectPrompt) },
-    { role: 'user', content: buildLineReviewPrompt(changeInfo) },
-  ];
-}
-
 /**
  * 构建整个文件所有变更行的批量审查提示词
  * @param {string} fileName - 文件名
@@ -179,8 +114,6 @@ export function buildFileReviewMessages(fileName, meaningfulChanges, projectProm
 // 导出函数
 export default {
   buildSystemPrompt,
-  buildLineReviewPrompt,
-  buildLineReviewMessages,
   buildFileReviewPrompt,
   buildFileReviewMessages,
 };

package/package.json

@@ -1,15 +1,19 @@
 {
   "name": "gitlab-ai-review",
-  "version": "2.2.0",
-  "description": "GitLab AI Review SDK - 按文件批量审查，只评论有问题的代码",
+  "version": "2.4.0",
+  "description": "GitLab AI Review SDK - 支持 SDK 调用和 CLI 执行",
   "main": "index.js",
   "type": "module",
+  "bin": {
+    "gitlab-ai-review": "./cli.js"
+  },
   "scripts": {
     "test": "node test-info.js",
     "test:info": "node test-info.js"
   },
   "files": [
     "index.js",
+    "cli.js",
     "lib/",
     "test-info.js",
     "README.md"