hong-review-cli 1.0.0

package/src/commands/mr.js

@@ -0,0 +1,270 @@
+const inquirer = require('inquirer');
+const chalk = require('chalk');
+const fs = require('fs');
+const path = require('path');
+const storage = require('../utils/storage');
+const logger = require('../utils/logger');
+const hooks = require('../utils/hooks');
+const GitLabClient = require('../core/gitlab');
+const AIClient = require('../core/ai');
+const { 
+    AGENT_SYSTEM_PROMPT, 
+    parseAgentResponse, 
+    formatFileList, 
+    formatFileContents,
+    createUserMessage,
+    createAssistantMessage
+} = require('../core/agent');
+
+module.exports = async function(mrId, options) {
+    const config = storage.getAll();
+    if (!config.gitlabToken || !config.aiKey) {
+        logger.error('缺少必要的配置，请先运行 `hong-review login` 初始化配置。');
+        return;
+    }
+
+    // 这里简化处理：因为我们需要 projectId 和 mrIid，
+    // 在真实场景中，项目 ID 通常从本地 git remote 推断或配置中读取。
+    // 为了 demo 的通用性，我们直接要求用户确认或配置 projectId。
+    let projectId = config.defaultProjectId;
+    if (!projectId) {
+         const answer = await inquirer.prompt([{
+             type: 'input',
+             name: 'pid',
+             message: '请输入当前 Git 仓库对应的 GitLab Project ID:'
+         }]);
+         projectId = answer.pid;
+         storage.set('defaultProjectId', projectId);
+    }
+
+    const mrIid = parseInt(mrId);
+    if (isNaN(mrIid)) {
+        logger.error('请输入有效的 Merge Request ID (纯数字)。');
+        return;
+    }
+
+    logger.info(`开始准备审查 Project [${projectId}] 的 MR !${mrIid}`);
+    
+    // 触发 Hook
+    await hooks.emit('onReviewStart', { projectId, mrId: mrIid });
+
+    const gitlab = new GitLabClient({ host: config.gitlabUrl, token: config.gitlabToken });
+    const ai = new AIClient({ apiKey: config.aiKey, baseURL: config.aiBaseUrl, model: config.aiModel });
+
+    try {
+        logger.startSpinner('fetching', '正在从 GitLab 拉取合并请求详情与代码变更...');
+        const detail = await gitlab.getMergeRequestDetail(projectId, mrIid);
+        const changes = await gitlab.getMergeRequestChanges(projectId, mrIid);
+        logger.stopSpinner('fetching', true, `成功拉取 MR [${detail.title}] 分析所需的 ${changes.length} 个文件变更。`);
+
+        if (changes.length === 0) {
+            logger.warn('该 MR 没有包含任何文件变更。');
+            return;
+        }
+
+        logger.startSpinner('ai-thinking', `🤖 AI (${config.aiModel}) 正在深度分析中，请稍候...`);
+        
+        let reviewResult = null;
+        let round = 1;
+        const fileOverviews = GitLabClient.getFileOverview(changes);
+        const initialMessage = formatFileList(fileOverviews);
+        const conversationHistory = [{ role: 'user', content: initialMessage }];
+
+        // 开始 Agent 轮询
+        while (true) {
+            logger.startSpinner('ai-thinking', `🤖 AI 思考中 (第 ${round} 轮)...`);
+            const aiResponse = await ai.chatWithJsonMode(AGENT_SYSTEM_PROMPT, conversationHistory);
+            const action = parseAgentResponse(aiResponse);
+
+            if (!action) {
+                throw new Error('AI 返回的数据无法解析为期望的 JSON 格式。');
+            }
+
+            conversationHistory.push({ role: 'assistant', content: aiResponse });
+
+            if (action.action === 'generate_report') {
+                reviewResult = action.result;
+                break;
+            } else if (action.action === 'request_files') {
+                logger.stopSpinner('ai-thinking', true, `AI 请求查看额外文件: ${action.files.join(', ')}`);
+                
+                const contents = await fetchFileContents(gitlab, projectId, detail.source_branch, changes, action.files, action.contentTypes);
+                const fileContentMessage = formatFileContents(contents);
+                conversationHistory.push({ role: 'user', content: fileContentMessage });
+                round++;
+            }
+        }
+
+        logger.stopSpinner('ai-thinking', true, 'AI 审查完成！');
+        
+        // 打印报告
+        printReport(mrIid, detail, reviewResult);
+
+        // 处理自动保存报告参数 -R
+        if (options.report) {
+            const reportStr = buildMarkdownReport(mrIid, detail, reviewResult);
+            const reportName = `review_report_mr_${mrIid}.md`;
+            fs.writeFileSync(path.join(process.cwd(), reportName), reportStr, 'utf-8');
+            logger.success(`审查报告已保存至当前目录: ${reportName}`);
+        }
+
+        // 判断后续状态
+        const isSuccess = reviewResult.riskLevel !== 'high';
+        if (isSuccess) {
+            await hooks.emit('onReviewSuccess', { projectId, mrId: mrIid, result: reviewResult });
+        } else {
+            await hooks.emit('onReviewFailed', { projectId, mrId: mrIid, result: reviewResult });
+        }
+
+        // 处理自动评论参数 -c
+        if (options.comment) {
+            logger.startSpinner('comment', '正在向 GitLab 发送审查总结评论...');
+            const mdBody = buildMarkdownReport(mrIid, detail, reviewResult);
+            await gitlab.addMergeRequestNote(projectId, mrIid, { body: mdBody });
+            logger.stopSpinner('comment', true, '评论提交成功。');
+        }
+
+        // 处理自动合并参数 -m
+        if (options.merge && isSuccess && reviewResult.issues.filter(i => i.severity === 'error').length === 0) {
+             logger.startSpinner('merge', '达到自动合并标准，尝试执行自动合并...');
+             try {
+                 await gitlab.merge(projectId, mrIid);
+                 logger.stopSpinner('merge', true, '合并执行成功！');
+                 await hooks.emit('onMerged', { projectId, mrId: mrIid });
+             } catch (err) {
+                 logger.stopSpinner('merge', false, `合并失败: ${err.message}`);
+             }
+        }
+
+        if (!options.merge) {
+            console.log('\n');
+            const postReviewAnswer = await inquirer.prompt([{
+                type: 'list',
+                name: 'nextAction',
+                message: '代码审查完成，请选择后续操作:',
+                choices: [
+                    { name: '✅ 合并该 MR (Merge)', value: 'merge' },
+                    { name: '❌ 暂不合并，退出 (Quit)', value: 'quit' },
+                    { name: '↩️  返回列表 (Return)', value: 'return' }
+                ]
+            }]);
+
+            if (postReviewAnswer.nextAction === 'merge') {
+                logger.startSpinner('merge', '尝试执行合并...');
+                try {
+                    await gitlab.merge(projectId, mrIid);
+                    logger.stopSpinner('merge', true, '合并执行成功！');
+                    await hooks.emit('onMerged', { projectId, mrId: mrIid });
+                } catch (err) {
+                    logger.stopSpinner('merge', false, `合并失败: ${err.message}`);
+                }
+            } else if (postReviewAnswer.nextAction === 'return') {
+                const listCmd = require('./list');
+                return listCmd();
+            } else {
+                logger.info('操作已完成，退出审查。');
+            }
+        }
+
+    } catch (e) {
+        logger.stopSpinner('fetching', false);
+        logger.stopSpinner('ai-thinking', false);
+        logger.error(`审查过程中发生异常: ${e.message}`);
+        await hooks.emit('onReviewFailed', { projectId, mrId: mrIid, error: e.message });
+    }
+};
+
+async function fetchFileContents(gitlabClient, projectId, ref, changes, requestedFiles, contentTypes) {
+    const contents = [];
+    for (let i = 0; i < requestedFiles.length; i++) {
+        const filePath = requestedFiles[i];
+        const contentType = contentTypes[i] || 'diff';
+        try {
+            if (contentType === 'full') {
+                const content = await gitlabClient.getFileContent(projectId, filePath, ref);
+                contents.push({ path: filePath, contentType: 'full', content });
+            } else {
+                const fileChange = changes.find(f => f.new_path === filePath);
+                contents.push({ 
+                    path: filePath, 
+                    contentType: 'diff', 
+                    content: fileChange ? fileChange.diff : '(文件未找到)' 
+                });
+            }
+        } catch (err) {
+            contents.push({ path: filePath, contentType, content: `(获取文件内容失败: ${err.message})` });
+        }
+    }
+    return contents;
+}
+
+function printReport(mrIid, detail, result) {
+    console.log('\n' + chalk.bold.inverse(` ====== 📋 审查报告：MR !${mrIid} (${detail.title}) ====== `) + '\n');
+    console.log(chalk.bold('📈 整体概述:'));
+    console.log(chalk.gray(result.summary) + '\n');
+
+    let errorCount = 0;
+    let warningCount = 0;
+    
+    console.log(chalk.bold('⚠️ 发现的问题:'));
+    if (!result.issues || result.issues.length === 0) {
+         console.log(chalk.green('  没有任何问题，代码很棒！\n'));
+    } else {
+         result.issues.forEach(issue => {
+             let prefix = '';
+             let colorMsg = issue.description;
+             if (issue.severity === 'error') {
+                 prefix = chalk.bgRed.white(' ERROR ');
+                 colorMsg = chalk.red(issue.description);
+                 errorCount++;
+             } else if (issue.severity === 'warning') {
+                 prefix = chalk.bgYellow.black(' WARN ');
+                 colorMsg = chalk.yellow(issue.description);
+                 warningCount++;
+             } else {
+                 prefix = chalk.bgBlue.white(' INFO ');
+                 colorMsg = chalk.blue(issue.description);
+             }
+
+             console.log(`${prefix} ${chalk.underline(issue.file)}${issue.line ? ':'+issue.line : ''} - ${chalk.bold(issue.title)}`);
+             console.log(`      💡 ${colorMsg}\n`);
+         });
+    }
+
+    if (result.suggestions && result.suggestions.length > 0) {
+        console.log(chalk.bold('💡 改进建议:'));
+        result.suggestions.forEach(s => console.log(`  - ${s}`));
+        console.log();
+    }
+
+    console.log('--------------------------------------------------');
+    const sumText = `风险等级: ${result.riskLevel.toUpperCase()} | 严重错误: ${chalk.red(errorCount)} | 优化建议: ${chalk.yellow(warningCount)}`;
+    console.log(sumText);
+    console.log('--------------------------------------------------\n');
+}
+
+function buildMarkdownReport(mrIid, detail, result) {
+    let md = `# AI 代码审查报告 (MR !${mrIid})\n\n`;
+    md += `**标题**: ${detail.title}\n`;
+    md += `**风险评级**: ${result.riskLevel.toUpperCase()}\n\n`;
+    md += `## 整体概述\n${result.summary}\n\n`;
+    
+    if (result.issues && result.issues.length > 0) {
+        md += `## 发现的问题\n\n`;
+        result.issues.forEach(issue => {
+            const icon = issue.severity === 'error' ? '🔴' : (issue.severity === 'warning' ? '🟡' : '🔵');
+            md += `### ${icon} ${issue.title} (${issue.file}${issue.line ? `:${issue.line}` : ''})\n`;
+            md += `${issue.description}\n\n`;
+        });
+    } else {
+        md += `## 发现的问题\n未发现明显问题。\n\n`;
+    }
+
+    if (result.suggestions && result.suggestions.length > 0) {
+         md += `## 改进建议\n`;
+         result.suggestions.forEach(s => md += `- ${s}\n`);
+    }
+
+    md += `\n> *本报告由 \`hong-review\` CLI 工具自动生成。*`;
+    return md;
+}

package/src/core/agent.js

@@ -0,0 +1,141 @@
+const AGENT_SYSTEM_PROMPT = `你是一个资深的代码审查专家。你的任务是分析合并请求(MR)中的代码变更，找出潜在问题并给出审查建议。
+
+## 工作流程
+
+1. 首先，你会收到一个待审查的文件列表（包含文件路径、变更类型、变更行数）
+2. 你需要根据文件列表，决定需要查看哪些文件的详细内容
+3. 你可以请求查看文件的 diff（差异）或 full（完整内容）
+4. 获取足够信息后，生成最终的审查报告
+
+## 输出格式
+
+你必须始终以 JSON 格式输出，有两种可能的动作：
+
+### 动作1: 请求查看文件
+当你需要更多信息时，输出：
+\`\`\`json
+{
+  "action": "request_files",
+  "files": ["文件路径1", "文件路径2"],
+  "contentTypes": ["diff", "full"],
+  "reasoning": "说明为什么需要查看这些文件"
+}
+\`\`\`
+
+### 动作2: 生成审查报告
+当你已获取足够信息，可以做出判断时，输出：
+\`\`\`json
+{
+  "action": "generate_report",
+  "result": {
+    "riskLevel": "low|medium|high",
+    "summary": "本次变更的整体总结",
+    "issues": [
+      {
+        "severity": "info|warning|error",
+        "file": "文件路径",
+        "line": 12,
+        "title": "问题标题",
+        "description": "问题描述"
+      }
+    ],
+    "suggestions": ["改进建议1", "改进建议2"]
+  }
+}
+\`\`\`
+
+## 注意事项
+
+- 优先查看变更行数多的文件、核心业务文件、涉及安全的文件
+- 一次不要请求太多文件（建议 3-5 个），分批请求更高效
+- 请用中文输出 reasoning 和审查结果内容`;
+
+function generateId() {
+    return `${Date.now()}-${Math.random().toString(36).slice(2, 9)}`;
+}
+
+function formatFileList(files) {
+    const statusMap = {
+        added: '新增',
+        modified: '修改',
+        deleted: '删除',
+        renamed: '重命名'
+    };
+
+    const lines = files.map(f => {
+        const status = statusMap[f.status] || f.status;
+        const changes = f.additions + f.deletions;
+        return `- [${status}] ${f.path} (${changes} 行变更)`;
+    });
+
+    return `## 待审查的代码变更
+
+以下文件发生了变更，请告诉我你需要查看哪些文件的详细内容：
+
+${lines.join('\n')}
+
+请告诉我你需要查看哪些文件的详细内容来做出准确的代码审查判断。`;
+}
+
+function formatFileContents(contents) {
+    const parts = contents.map(c => {
+        const typeLabel = c.contentType === 'diff' ? 'Diff 内容' : '完整文件内容';
+        return `### 📄 ${c.path} (${typeLabel})\n\n\`\`\`\n${c.content}\n\`\`\``;
+    });
+
+    return `已提供 ${contents.length} 个文件的变更内容：\n\n${parts.join('\n\n')}\n\n请继续分析，如果需要更多信息请继续请求，如果信息足够请生成审查报告。`;
+}
+
+function parseAgentResponse(content) {
+    try {
+        const jsonMatch = content.match(/```json\s*([\s\S]*?)\s*```/);
+        const jsonStr = jsonMatch ? jsonMatch[1] : content;
+        const cleanedJson = jsonStr.trim();
+        const parsed = JSON.parse(cleanedJson);
+
+        if (parsed.action === 'request_files' || parsed.action === 'generate_report') {
+            return parsed;
+        }
+        return null;
+    } catch (error) {
+        return null;
+    }
+}
+
+function createUserMessage(content, label, files) {
+    return {
+        id: generateId(),
+        role: 'user',
+        content,
+        timestamp: new Date(),
+        label,
+        files
+    };
+}
+
+function createAssistantMessage(content, parsedAction) {
+    let label = 'JSON';
+    if (parsedAction) {
+        if (parsedAction.action === 'request_files') label = '请求更多上下文';
+        else if (parsedAction.action === 'generate_report') label = '生成报告';
+    }
+
+    return {
+        id: generateId(),
+        role: 'assistant',
+        content,
+        timestamp: new Date(),
+        parsedAction,
+        label
+    };
+}
+
+module.exports = {
+    AGENT_SYSTEM_PROMPT,
+    generateId,
+    formatFileList,
+    formatFileContents,
+    parseAgentResponse,
+    createUserMessage,
+    createAssistantMessage
+};

package/src/core/ai.js

@@ -0,0 +1,108 @@
+const axios = require('axios');
+
+class AIClient {
+    constructor(config) {
+        this.apiKey = config.apiKey;
+        this.baseURL = config.baseURL || 'https://api.openai.com/v1';
+        this.model = config.model || 'gpt-4o';
+        this.requestTimeout = 180000;
+        
+        this.client = axios.create({
+            baseURL: this.baseURL,
+            headers: {
+                'Authorization': `Bearer ${this.apiKey}`,
+                'Content-Type': 'application/json'
+            },
+            timeout: this.requestTimeout
+        });
+    }
+
+    setTimeout(timeout) {
+        this.client.defaults.timeout = timeout;
+    }
+
+    async chatRequest(messages, jsonMode = false) {
+        const requestBody = {
+            model: this.model,
+            messages
+        };
+
+        if (jsonMode) {
+            requestBody.response_format = { type: 'json_object' };
+        }
+
+        try {
+            const response = await this.client.post('/chat/completions', requestBody);
+            return response.data.choices[0]?.message?.content || '';
+        } catch (error) {
+            if (error.response) {
+                throw new Error(`AI API error: ${error.response.status} - ${JSON.stringify(error.response.data)}`);
+            }
+            throw new Error(`AI API error: ${error.message}`);
+        }
+    }
+
+    async analyzeMergeRequest(files) {
+        try {
+            const fileChanges = files.map(f => {
+                const truncatedDiff = f.diff.length > 1500
+                    ? f.diff.slice(0, 1500) + '\n... (内容已截断)'
+                    : f.diff;
+                return `\n【文件: ${f.path}】\n${truncatedDiff}`;
+            }).join('\n---\n');
+
+            const maxLength = 15000;
+            const content = fileChanges.length > maxLength
+                ? fileChanges.slice(0, maxLength) + '\n\n... (更多文件变更已省略)'
+                : fileChanges;
+
+            const prompt = `你是一个资深的代码审查专家。请分析以下合并请求(MR)中所有文件的代码变更。
+## 代码变更内容:
+${content}
+
+## 请按以下格式输出分析报告:
+### 📋 整体概述
+### 📁 主要改动点
+### ⚠️ 需要关注
+### ✅ 审查建议
+请用中文、通俗易懂的语言回答。`;
+
+            return await this.chatRequest([{ role: 'user', content: prompt }]);
+        } catch (error) {
+            console.error('Failed to analyze MR:', error);
+            return 'AI分析服务暂时不可用，请检查API Key配置';
+        }
+    }
+
+    async chatWithJsonMode(systemPrompt, messages) {
+        const allMessages = [
+            { role: 'system', content: systemPrompt },
+            ...messages
+        ];
+        return await this.chatRequest(allMessages, true);
+    }
+
+    async testConnection() {
+        const startTime = Date.now();
+        try {
+            const response = await this.chatRequest([
+                { role: 'user', content: '请回复"OK"两个字母，不要说其他任何内容。' }
+            ]);
+            const responseTime = Date.now() - startTime;
+            
+            return {
+                success: true,
+                message: `连接成功！响应很快。${response.includes('OK')?'':'响应：'+response}`,
+                responseTime
+            };
+        } catch (error) {
+            return {
+                success: false,
+                message: error.message,
+                responseTime: Date.now() - startTime
+            };
+        }
+    }
+}
+
+module.exports = AIClient;