git-ai-shen 1.0.0

package/lib/commands/commit.js

@@ -0,0 +1,256 @@
+const { GitOperations } = require('../core/git-operations');
+const { AIReviewer } = require('../core/ai-reviewer');
+const { SimpleConfigManager } = require('../core/simple-config');
+const chalk = require('chalk');
+const inquirer = require('inquirer');
+
+class CommitCommand {
+  constructor() {
+    this.config = new SimpleConfigManager();
+    this.gitOps = new GitOperations();
+    this.aiReviewer = new AIReviewer(this.config);
+    this.sshPassword = null; // 存储 SSH 密码
+  }
+
+  async run(message, options) {
+    console.log(chalk.cyan('\n🔍 Git AI 智能提交工具\n'));
+
+    try {
+      // 检查是否是git仓库
+      const isRepo = await this.gitOps.isGitRepo();
+      if (!isRepo) {
+        throw new Error('当前目录不是Git仓库');
+      }
+
+      // 获取变更文件
+      console.log(chalk.cyan('📝 检测文件变更...'));
+      const changedFiles = await this.gitOps.getChangedFiles();
+
+      if (changedFiles.length === 0) {
+        console.log(chalk.yellow('⚠️ 没有检测到文件变更'));
+        return;
+      }
+
+      console.log(chalk.green(`✅ 检测到 ${changedFiles.length} 个变更文件:`));
+      changedFiles.forEach(file => {
+        console.log(`   ${chalk.gray('-')} ${file}`);
+      });
+
+      // AI审查 - 使用基于 diff 的精确审查
+      if (!options.skipReview && this.config.get('review.enabled')) {
+        console.log(chalk.cyan('\n🔍 AI审查中（基于行级变更）...'));
+
+        // 获取详细的 diff 信息
+        console.log(chalk.gray('   正在分析文件变更...'));
+        const diffs = await this.gitOps.getDiffs(changedFiles, false); // false 表示未暂存的变更
+
+        // 使用基于 diff 的审查
+        const review = await this.aiReviewer.reviewWithDiff(changedFiles, diffs);
+
+        if (review.risks.length > 0) {
+          console.log(chalk.yellow('\n⚠️ 检测到风险项:'));
+          review.risks.forEach(risk => {
+            console.log(`   ${chalk.red('[风险]')} ${risk}`);
+          });
+        }
+
+        if (review.suggestions.length > 0) {
+          console.log(chalk.cyan('\n💡 优化建议:'));
+          review.suggestions.forEach(suggestion => {
+            console.log(`   ${chalk.yellow('[建议]')} ${suggestion}`);
+          });
+        }
+
+        // 如果检测到风险，询问是否继续
+        if (review.risks.length > 0) {
+          const { confirm } = await inquirer.prompt([
+            {
+              type: 'confirm',
+              name: 'confirm',
+              message: '存在风险项，是否继续提交？',
+              default: false
+            }
+          ]);
+
+          if (!confirm) {
+            console.log(chalk.yellow('⚠️ 提交已取消'));
+            return;
+          }
+        }
+      }
+
+      // 获取提交信息 - 优先使用AI生成
+      // 获取提交信息 - 优先使用AI生成
+      let commitMessage = message;
+      if (!commitMessage) {
+        // 检查是否有API密钥，有则尝试AI生成
+        const apiKey = this.config.get('ai.apiKey');
+        if (apiKey && this.aiReviewer.isAvailable()) {
+          console.log(chalk.cyan('\n🤖 AI正在基于代码变更生成提交信息...'));
+          try {
+            // 获取 diffs 用于生成提交信息
+            const diffs = await this.gitOps.getDiffs(changedFiles, false);
+            commitMessage = await this.aiReviewer.generateCommitMessageWithDiff(changedFiles, diffs, this.gitOps);
+            console.log(chalk.green(`✅ AI生成提交信息: ${commitMessage}`));
+
+            // 让用户确认或修改
+            const { action } = await inquirer.prompt([
+              {
+                type: 'list',
+                name: 'action',
+                message: '请选择操作:',
+                choices: [
+                  { name: '使用此信息', value: 'use' },
+                  { name: '重新生成', value: 'regenerate' },
+                  { name: '手动输入', value: 'manual' }
+                ]
+              }
+            ]);
+
+            if (action === 'regenerate') {
+              // 递归调用重新生成
+              return this.run(message, options);
+            } else if (action === 'manual') {
+              const { customMessage } = await inquirer.prompt([
+                {
+                  type: 'input',
+                  name: 'customMessage',
+                  message: '请输入提交信息:',
+                  validate: input => input ? true : '提交信息不能为空'
+                }
+              ]);
+              commitMessage = customMessage;
+            }
+            // 如果选择 'use'，保持生成的 message
+
+          } catch (error) {
+            console.log(chalk.yellow(`⚠️ AI生成失败: ${error.message}`));
+            // AI生成失败时回退到手动输入
+            const { customMessage } = await inquirer.prompt([
+              {
+                type: 'input',
+                name: 'customMessage',
+                message: '请输入提交信息:',
+                validate: input => input ? true : '提交信息不能为空'
+              }
+            ]);
+            commitMessage = customMessage;
+          }
+        } else {
+          // 没有API密钥，手动输入
+          console.log(chalk.yellow('\n⚠️ 未配置AI API密钥，使用手动输入'));
+          const { customMessage } = await inquirer.prompt([
+            {
+              type: 'input',
+              name: 'customMessage',
+              message: '请输入提交信息:',
+              validate: input => input ? true : '提交信息不能为空'
+            }
+          ]);
+          commitMessage = customMessage;
+        }
+      }
+
+      // 添加文件
+      if (options.autoAdd !== false && this.config.get('git.autoAdd')) {
+        console.log(chalk.cyan('\n📦 添加文件到暂存区...'));
+        await this.gitOps.addFiles();
+        console.log(chalk.green('✅ 添加成功'));
+      }
+
+      // 提交
+      console.log(chalk.cyan('\n💾 提交变更...'));
+      await this.gitOps.commit(commitMessage);
+      console.log(chalk.green('✅ 提交成功'));
+
+      // 修改拉取部分
+      if (this.config.get('git.autoPull')) {
+        console.log(chalk.cyan('\n⬇️ 拉取远程更新...'));
+
+        let pullSuccess = false;
+        let needPassword = false;
+
+        do {
+          const pullResult = await this.gitOps.pull('origin', null, this.sshPassword);
+
+          if (pullResult.success) {
+            console.log(chalk.green('✅ 拉取成功'));
+            pullSuccess = true;
+          } else {
+            if (pullResult.needPassword) {
+              console.log(chalk.yellow('\n⚠️ 需要 SSH 密钥密码'));
+
+              // 询问密码
+              const { password } = await inquirer.prompt([
+                {
+                  type: 'password',
+                  name: 'password',
+                  message: '请输入 SSH 密钥密码:',
+                  mask: '*',
+                  validate: input => input ? true : '密码不能为空'
+                }
+              ]);
+
+              this.sshPassword = password;
+              needPassword = true;
+            } else {
+              console.log(chalk.yellow(`⚠️ 拉取失败: ${pullResult.error}`));
+              break;
+            }
+          }
+        } while (needPassword && !pullSuccess);
+      }
+
+      // 修改推送部分
+      if (!options.noPush && this.config.get('git.autoPush')) {
+        console.log(chalk.cyan('\n⬆️ 推送到远程...'));
+
+        let pushSuccess = false;
+        let needPassword = false;
+
+        do {
+          const pushResult = await this.gitOps.push('origin', null, this.sshPassword);
+
+          if (pushResult.success) {
+            console.log(chalk.green('✅ 推送成功'));
+            pushSuccess = true;
+          } else {
+            if (pushResult.needPassword) {
+              if (!this.sshPassword) {
+                console.log(chalk.yellow('\n⚠️ 需要 SSH 密钥密码'));
+
+                const { password } = await inquirer.prompt([
+                  {
+                    type: 'password',
+                    name: 'password',
+                    message: '请输入 SSH 密钥密码:',
+                    mask: '*',
+                    validate: input => input ? true : '密码不能为空'
+                  }
+                ]);
+
+                this.sshPassword = password;
+                needPassword = true;
+              }
+            } else {
+              console.log(chalk.red(`❌ 推送失败: ${pushResult.error}`));
+              break;
+            }
+          }
+        } while (needPassword && !pushSuccess);
+
+        if (!pushSuccess) {
+          console.log(chalk.red('❌ 推送失败'));
+          return;
+        }
+      }
+
+      console.log(chalk.green(`\n✨ 完成！提交信息: ${commitMessage}`));
+
+    } catch (error) {
+      console.log(chalk.red(`\n❌ 错误: ${error.message}`));
+    }
+  }
+}
+
+module.exports = { CommitCommand };

package/lib/core/ai-reviewer.js

@@ -0,0 +1,321 @@
+const path = require('path');
+const ignore = require('ignore');
+const chalk = require('chalk');  // 添加这一行
+const { OpenAI } = require('openai');
+class AIReviewer {
+  constructor(config) {
+    this.config = config;
+    this.openai = null;
+    this.initOpenAI();
+  }
+  /**
+ * 基于 diff 进行精确的代码审查
+ * @param {Array<string>} files 文件列表
+ * @param {Object} diffs diff 信息
+ * @returns {Object} 审查结果
+ */
+  async reviewWithDiff(files, diffs) {
+    const risks = [];
+    const suggestions = [];
+
+    for (const file of files) {
+      const diff = diffs[file];
+      if (!diff || !diff.changes || diff.changes.length === 0) {
+        continue;
+      }
+
+      // 只审查新增和修改的行
+      for (const hunk of diff.changes) {
+        for (const change of hunk.changes) {
+          if (change.type === 'add' || change.type === 'delete') {
+            const lineContent = change.content;
+            const lineNumber = change.lineNumber || '?';
+
+            // 检查硬编码密钥（只检查新增的行）
+            if (change.type === 'add') {
+              if (/password\s*[:=]\s*['"][^'"]+['"]/i.test(lineContent)) {
+                risks.push(`${file}:${lineNumber} 行可能包含密码硬编码`);
+              }
+
+              if (/api[_-]?key\s*[:=]\s*['"][^'"]+['"]/i.test(lineContent)) {
+                risks.push(`${file}:${lineNumber} 行可能包含API密钥硬编码`);
+              }
+
+              if (/token\s*[:=]\s*['"][^'"]+['"]/i.test(lineContent)) {
+                risks.push(`${file}:${lineNumber} 行可能包含令牌硬编码`);
+              }
+            }
+
+            // 检查调试代码
+            if (change.type === 'add') {
+              if (/console\.log/.test(lineContent)) {
+                suggestions.push(`${file}:${lineNumber} 行添加了console.log，考虑移除`);
+              }
+              if (/debugger;/.test(lineContent)) {
+                risks.push(`${file}:${lineNumber} 行添加了debugger语句`);
+              }
+              if (/TODO/.test(lineContent)) {
+                suggestions.push(`${file}:${lineNumber} 行添加了TODO注释`);
+              }
+            }
+          }
+        }
+      }
+
+      // 生成变更摘要
+      suggestions.push(`${file}: ${diff.summary}`);
+    }
+
+    return { risks, suggestions };
+  }
+  initOpenAI() {
+    const apiKey = this.config.get('ai.apiKey');
+    const apiUrl = this.config.get('ai.apiUrl');
+
+    if (apiKey) {
+      try {
+        this.openai = new OpenAI({
+          apiKey: apiKey,
+          baseURL: apiUrl  // 使用你配置的国产API地址
+        });
+        console.log(chalk.green('✅ AI客户端初始化成功'));
+      } catch (error) {
+        console.log(chalk.yellow(`⚠️ AI客户端初始化失败: ${error.message}`));
+      }
+    }
+  }
+  async basicReview(files, gitOps) {
+    const risks = [];
+    const suggestions = [];
+
+    // 确保 files 是数组
+    if (!Array.isArray(files)) {
+      console.log('警告: files 参数不是数组');
+      return { risks, suggestions };
+    }
+
+    for (const file of files) {
+      // 确保 file 是字符串
+      if (typeof file !== 'string') {
+        console.log(`跳过非字符串文件: ${file}`);
+        continue;
+      }
+
+      try {
+        const content = await gitOps.getFileContent(file);
+        if (!content) continue;
+
+        // 获取文件扩展名
+        const ext = path.extname(file).toLowerCase();
+
+        // 检查硬编码密钥
+        if (this.config.get('review.checkSecrets')) {
+          if (/password\s*[:=]\s*['"][^'"]+['"]/i.test(content)) {
+            risks.push(`${file}: 可能包含密码硬编码`);
+          }
+
+          if (/api[_-]?key\s*[:=]\s*['"][^'"]+['"]/i.test(content)) {
+            risks.push(`${file}: 可能包含API密钥硬编码`);
+          }
+
+          if (/token\s*[:=]\s*['"][^'"]+['"]/i.test(content)) {
+            risks.push(`${file}: 可能包含令牌硬编码`);
+          }
+
+          if (/secret\s*[:=]\s*['"][^'"]+['"]/i.test(content)) {
+            risks.push(`${file}: 可能包含密钥硬编码`);
+          }
+        }
+
+        // 检查TODO和FIXME
+        if (this.config.get('review.checkTodos')) {
+          if (/TODO/i.test(content)) {
+            suggestions.push(`${file}: 包含待办事项(TODO)`);
+          }
+          if (/FIXME/i.test(content)) {
+            suggestions.push(`${file}: 包含待修复项(FIXME)`);
+          }
+        }
+
+        // 检查调试代码
+        if (this.config.get('review.checkDebug')) {
+          // JavaScript/TypeScript 文件
+          if (['.js', '.jsx', '.ts', '.tsx'].includes(ext)) {
+            if (/console\.log/.test(content)) {
+              suggestions.push(`${file}: 包含console.log，考虑移除`);
+            }
+            if (/debugger;/.test(content)) {
+              risks.push(`${file}: 包含debugger语句`);
+            }
+            if (/alert\(/.test(content)) {
+              suggestions.push(`${file}: 包含alert语句，考虑移除`);
+            }
+          }
+
+          // Python 文件
+          if (['.py'].includes(ext)) {
+            if (/print\(/.test(content)) {
+              suggestions.push(`${file}: 包含print语句，考虑使用日志记录`);
+            }
+            if (/import pdb;/.test(content)) {
+              risks.push(`${file}: 包含pdb调试代码`);
+            }
+          }
+        }
+
+        // TODO: 调用AI模型进行深度审查
+        if (this.config.get('ai.apiKey') && content.length < 10000) {
+          // AI审查逻辑...
+        }
+      } catch (error) {
+        console.log(`审查文件 ${file} 时出错: ${error.message}`);
+      }
+    }
+
+    return { risks, suggestions };
+  }
+
+  // 确保 isAvailable 方法存在
+  isAvailable() {
+    return !!this.config.get('ai.apiKey');
+  }
+  async generateCommitMessage(files, gitOps) {
+    if (!this.isAvailable()) {
+      return '自动提交';
+    }
+
+    try {
+      // 获取文件变更的简要信息
+      const fileList = files.slice(0, 5).map(f => `- ${f}`).join('\n');
+      const fileCount = files.length > 5 ? `\n- ... 等${files.length}个文件` : '';
+
+      // 获取部分文件内容作为上下文
+      let fileContents = '';
+      for (const file of files.slice(0, 3)) {
+        const content = await gitOps.getFileContent(file);
+        if (content) {
+          fileContents += `\n文件 ${file} 的部分内容:\n${content.substring(0, 200)}...\n`;
+        }
+      }
+
+      const response = await this.openai.chat.completions.create({
+        model: this.config.get('ai.model') || 'gpt-3.5-turbo',
+        messages: [
+          {
+            role: 'system',
+            content: '你是一个Git提交信息生成助手。根据变更文件生成符合常规提交格式的提交信息。'
+          },
+          {
+            role: 'user',
+            content: `根据以下变更文件生成一个简洁的Git提交信息：
+变更文件:
+${fileList}${fileCount}
+${fileContents}
+
+请返回一个符合常规提交格式的提交信息，例如：
+- feat: 添加用户登录功能
+- fix: 修复空指针异常
+- docs: 更新README文档
+- refactor: 重构用户验证逻辑
+- test: 添加单元测试
+
+只需要返回提交信息本身，不要有其他内容。`
+          }
+        ],
+        temperature: 0.3,
+        max_tokens: 50
+      });
+      return response.choices[0].message.content.trim();
+    } catch (error) {
+      console.log(chalk.yellow(`⚠️ 生成提交信息失败: ${error.message}`));
+      return '自动提交';
+    }
+  }
+  /**
+ * 基于 diff 生成提交信息
+ * @param {Array<string>} files 文件列表
+ * @param {Object} diffs diff 信息
+ * @param {Object} gitOps Git 操作实例
+ * @returns {Promise<string>} 提交信息
+ */
+  async generateCommitMessageWithDiff(files, diffs, gitOps) {
+    if (!this.isAvailable()) {
+      return '自动提交';
+    }
+
+    try {
+      // 构建变更摘要
+      let diffSummary = '';
+      let totalAdd = 0;
+      let totalDelete = 0;
+
+      for (const file of files) {
+        const diff = diffs[file];
+        if (diff) {
+          diffSummary += `\n文件: ${file}\n`;
+          diffSummary += `变更: ${diff.summary}\n`;
+          totalAdd += diff.addCount || 0;
+          totalDelete += diff.deleteCount || 0;
+
+          // 添加具体的变更内容（最多5个关键变更）
+          let changeCount = 0;
+          for (const hunk of diff.changes) {
+            for (const change of hunk.changes) {
+              if (change.type === 'add' || change.type === 'delete') {
+                if (changeCount < 5) {
+                  const prefix = change.type === 'add' ? '+' : '-';
+                  // 只显示非空行和有意义的变更
+                  if (change.content.trim()) {
+                    diffSummary += `${prefix} ${change.content.trim().substring(0, 50)}\n`;
+                    changeCount++;
+                  }
+                }
+              }
+            }
+          }
+        }
+      }
+
+      // 如果有变更，添加到提示中
+      const changeSummary = totalAdd > 0 || totalDelete > 0
+        ? `\n总体变更: 新增 ${totalAdd} 行，删除 ${totalDelete} 行`
+        : '';
+
+      const response = await this.openai.chat.completions.create({
+        model: this.config.get('ai.model') || 'gpt-3.5-turbo',
+        messages: [
+          {
+            role: 'system',
+            content: '你是一个Git提交信息生成助手。根据代码变更生成符合常规提交格式的提交信息。'
+          },
+          {
+            role: 'user',
+            content: `根据以下代码变更生成一个简洁的Git提交信息：
+${changeSummary}
+${diffSummary}
+
+请返回一个符合常规提交格式的提交信息，例如：
+- feat: 添加用户登录功能
+- fix: 修复空指针异常
+- docs: 更新README文档
+- refactor: 重构用户验证逻辑
+- test: 添加单元测试
+- style: 格式化代码
+- chore: 更新依赖
+
+只需要返回提交信息本身，不要有其他内容。`
+          }
+        ],
+        temperature: this.config.get('ai.temperature') || 0.3,
+        max_tokens: this.config.get('ai.maxTokens') || 50
+      });
+
+      return response.choices[0].message.content.trim();
+    } catch (error) {
+      console.log(chalk.yellow(`⚠️ 生成提交信息失败: ${error.message}`));
+      return '自动提交';
+    }
+  }
+}
+
+module.exports = { AIReviewer };

package/lib/core/config.js

@@ -0,0 +1,120 @@
+const Conf = require('conf');
+const chalk = require('chalk');
+const { exec } = require('child_process');
+
+class ConfigManager {
+  constructor() {
+    this.config = new Conf({
+      projectName: 'git-ai',
+      defaults: this.getDefaultConfig()
+    });
+  }
+
+  getDefaultConfig() {
+    return {
+      ai: {
+        provider: 'openai',
+        apiKey: 'ms-114c4bf4-ab3e-442b-a30c-6b355db26263',
+        apiUrl: 'https://api-inference.modelscope.cn/v1',
+        model: 'Qwen/Qwen3-Coder-30B-A3B-Instruct',
+        maxTokens: 500,
+        temperature: 0.3
+      },
+      git: {
+        autoAdd: true,
+        autoPull: true,
+        autoPush: true,
+        remote: 'origin'
+      },
+      review: {
+        enabled: true,
+        skipPatterns: [
+          '*.lock',
+          '*.log',
+          '*.min.js',
+          '*.min.css',
+          '*.pyc',
+          '__pycache__',
+          'node_modules',
+          'dist',
+          'build'
+        ],
+        maxFileSize: 1024 * 100,
+        checkSecrets: true,
+        checkTodos: true,
+        checkDebug: true
+      },
+      ui: {
+        showProgress: true,
+        colorOutput: true
+      }
+    };
+  }
+
+  get(key) {
+    return this.config.get(key);
+  }
+
+  set(key, value) {
+    const keys = key.split('.');
+    if (keys.length > 1) {
+      const obj = {};
+      let current = obj;
+      for (let i = 0; i < keys.length - 1; i++) {
+        current[keys[i]] = {};
+        current = current[keys[i]];
+      }
+      current[keys[keys.length - 1]] = value;
+      this.config.set(obj);
+    } else {
+      this.config.set(key, value);
+    }
+  }
+
+  getAll() {
+    return this.config.store;
+  }
+
+  init() {
+    this.config.clear();
+    this.config.set(this.getDefaultConfig());
+
+    console.log(chalk.cyan('请配置你的AI服务提供商:'));
+
+    const readline = require('readline').createInterface({
+      input: process.stdin,
+      output: process.stdout
+    });
+
+    readline.question('OpenAI API密钥 (可选): ', (apiKey) => {
+      if (apiKey) {
+        this.set('ai.apiKey', apiKey);
+      }
+
+      readline.question('使用自定义API地址? (y/N): ', (customUrl) => {
+        if (customUrl.toLowerCase() === 'y') {
+          readline.question('API地址: ', (url) => {
+            this.set('ai.apiUrl', url);
+            readline.close();
+          });
+        } else {
+          readline.close();
+        }
+      });
+    });
+  }
+
+  openEditor() {
+    const configPath = this.config.path;
+    console.log(chalk.cyan(`配置文件路径: ${configPath}`));
+
+    const editor = process.env.EDITOR || 'notepad';
+    exec(`${editor} "${configPath}"`, (error) => {
+      if (error) {
+        console.error(chalk.red(`无法打开编辑器: ${error.message}`));
+      }
+    });
+  }
+}
+
+module.exports = { ConfigManager };