git-ai-shen 1.0.0

package/lib/core/git-operations.js

@@ -0,0 +1,285 @@
+const simpleGit = require('simple-git');
+const fs = require('fs').promises;
+const path = require('path');
+const { exec } = require('child_process');
+const util = require('util');
+const execAsync = util.promisify(exec);
+
+class GitOperations {
+  constructor(repoPath = process.cwd()) {
+    this.git = simpleGit(repoPath);
+    this.repoPath = repoPath;
+  }
+  /**
+   * 获取文件的详细变更信息（哪些行被修改）
+   * @param {string} filePath 文件路径
+   * @returns {Promise<Object>} 变更详情
+   */
+  async getFileDiff(filePath) {
+    try {
+      // 获取未暂存的变更
+      const diff = await this.git.diff([filePath]);
+
+      // 解析 diff 信息
+      return this.parseDiff(diff, filePath);
+    } catch (error) {
+      console.error(`获取 diff 失败: ${error.message}`);
+      return null;
+    }
+  }
+  /**
+ * 获取已暂存文件的详细变更信息
+ * @param {string} filePath 文件路径
+ * @returns {Promise<Object>} 变更详情
+ */
+  async getStagedDiff(filePath) {
+    try {
+      // 获取已暂存的变更
+      const diff = await this.git.diff(['--cached', filePath]);
+
+      // 解析 diff 信息
+      return this.parseDiff(diff, filePath);
+    } catch (error) {
+      console.error(`获取暂存 diff 失败: ${error.message}`);
+      return null;
+    }
+  }
+  /**
+ * 解析 git diff 输出
+ * @param {string} diff git diff 的输出
+ * @param {string} filePath 文件路径
+ * @returns {Object} 解析后的变更信息
+ */
+  parseDiff(diff, filePath) {
+    if (!diff) {
+      return {
+        file: filePath,
+        changes: [],
+        summary: '无变更'
+      };
+    }
+
+    const lines = diff.split('\n');
+    const changes = [];
+    let currentHunk = null;
+
+    // 解析 diff 的每一行
+    for (const line of lines) {
+      // 匹配 hunk 头信息，例如：@@ -1,4 +1,5 @@
+      const hunkMatch = line.match(/^@@ -(\d+),?(\d*) \+(\d+),?(\d*) @@/);
+      if (hunkMatch) {
+        currentHunk = {
+          oldStart: parseInt(hunkMatch[1]),
+          oldLines: parseInt(hunkMatch[2]) || 1,
+          newStart: parseInt(hunkMatch[3]),
+          newLines: parseInt(hunkMatch[4]) || 1,
+          changes: []
+        };
+        changes.push(currentHunk);
+        continue;
+      }
+
+      if (currentHunk) {
+        if (line.startsWith('+')) {
+          // 新增的行
+          currentHunk.changes.push({
+            type: 'add',
+            content: line.substring(1),
+            lineNumber: currentHunk.newStart + currentHunk.changes.filter(c => c.type !== 'delete').length
+          });
+        } else if (line.startsWith('-')) {
+          // 删除的行
+          currentHunk.changes.push({
+            type: 'delete',
+            content: line.substring(1),
+            lineNumber: currentHunk.oldStart + currentHunk.changes.filter(c => c.type !== 'add').length
+          });
+        } else if (!line.startsWith('\\')) {
+          // 上下文行
+          currentHunk.changes.push({
+            type: 'context',
+            content: line.substring(1)
+          });
+        }
+      }
+    }
+
+    // 生成变更摘要
+    const addCount = changes.flatMap(h => h.changes).filter(c => c.type === 'add').length;
+    const deleteCount = changes.flatMap(h => h.changes).filter(c => c.type === 'delete').length;
+
+    return {
+      file: filePath,
+      changes: changes,
+      summary: `新增 ${addCount} 行，删除 ${deleteCount} 行`,
+      addCount,
+      deleteCount
+    };
+  }
+  /**
+   * 获取所有变更文件的详细 diff
+   * @param {Array<string>} files 文件列表
+   * @param {boolean} staged 是否获取暂存区的变更
+   * @returns {Promise<Object>} 所有文件的 diff 信息
+   */
+  async getDiffs(files, staged = false) {
+    const diffs = {};
+
+    for (const file of files) {
+      if (staged) {
+        diffs[file] = await this.getStagedDiff(file);
+      } else {
+        diffs[file] = await this.getFileDiff(file);
+      }
+    }
+
+    return diffs;
+  }
+  async isGitRepo() {
+    try {
+      return await this.git.checkIsRepo();
+    } catch {
+      return false;
+    }
+  }
+
+  async getChangedFiles() {
+    try {
+      const status = await this.git.status();
+      return [
+        ...status.modified,
+        ...status.not_added,
+        ...status.created,
+        ...status.deleted
+      ];
+    } catch (error) {
+      return [];
+    }
+  }
+
+  async addFiles(files = ['.']) {
+    await this.git.add(files);
+    return true;
+  }
+
+  async commit(message) {
+    await this.git.commit(message);
+    return true;
+  }
+
+  async getCurrentBranch() {
+    const branch = await this.git.revparse(['--abbrev-ref', 'HEAD']);
+    return branch;
+  }
+
+  async pull(remote = 'origin', branch = null, sshPassword = null) {
+    try {
+      if (!branch) {
+        branch = await this.getCurrentBranch();
+      }
+
+      // 如果有 SSH 密码，使用 expect 脚本处理
+      if (sshPassword) {
+        return await this.pullWithPassword(remote, branch, sshPassword);
+      }
+
+      // 正常拉取
+      await this.git.pull(remote, branch);
+      return { success: true };
+      
+    } catch (error) {
+      // 检查是否是 SSH 密钥密码错误
+      if (error.message.includes('Permission denied') || 
+          error.message.includes('authenticity') ||
+          error.message.includes('password')) {
+        return { 
+          success: false, 
+          needPassword: true,
+          error: error.message 
+        };
+      }
+      return { success: false, error: error.message };
+    }
+  }
+
+  async pullWithPassword(remote, branch, password) {
+    try {
+      // 使用 expect 脚本自动输入 SSH 密码
+      const command = `
+        expect << 'EOF'
+        spawn git pull ${remote} ${branch}
+        expect {
+          "Enter passphrase for key" { send "${password}\r" }
+          "Password:" { send "${password}\r" }
+        }
+        expect eof
+        EOF
+      `;
+      
+      await execAsync(command);
+      return { success: true };
+    } catch (error) {
+      return { success: false, error: error.message };
+    }
+  }
+
+  async push(remote = 'origin', branch = null, sshPassword = null) {
+    try {
+      if (!branch) {
+        branch = await this.getCurrentBranch();
+      }
+
+      // 如果有 SSH 密码，使用 expect 脚本处理
+      if (sshPassword) {
+        return await this.pushWithPassword(remote, branch, sshPassword);
+      }
+
+      await this.git.push(remote, branch);
+      return { success: true };
+      
+    } catch (error) {
+      // 检查是否是 SSH 密钥密码错误
+      if (error.message.includes('Permission denied') || 
+          error.message.includes('authenticity') ||
+          error.message.includes('password')) {
+        return { 
+          success: false, 
+          needPassword: true,
+          error: error.message 
+        };
+      }
+      return { success: false, error: error.message };
+    }
+  }
+
+  async pushWithPassword(remote, branch, password) {
+    try {
+      const command = `
+        expect << 'EOF'
+        spawn git push ${remote} ${branch}
+        expect {
+          "Enter passphrase for key" { send "${password}\r" }
+          "Password:" { send "${password}\r" }
+        }
+        expect eof
+        EOF
+      `;
+      
+      await execAsync(command);
+      return { success: true };
+    } catch (error) {
+      return { success: false, error: error.message };
+    }
+  }
+
+  async getFileContent(filePath) {
+    try {
+      const fullPath = path.join(this.repoPath, filePath);
+      return await fs.readFile(fullPath, 'utf-8');
+    } catch {
+      return null;
+    }
+  }
+}
+
+module.exports = { GitOperations };

package/lib/core/simple-config.js

@@ -0,0 +1,171 @@
+
+const fs = require('fs');
+const path = require('path');
+const os = require('os');
+const chalk = require('chalk');
+const readline = require('readline');
+
+class SimpleConfigManager {
+  constructor() {
+    this.configDir = path.join(os.homedir(), '.git-ai');
+    this.configFile = path.join(this.configDir, 'config.json');
+    this.config = this.loadConfig();
+  }
+
+  getDefaultConfig() {
+    return {
+      ai: {
+        provider: 'openai',
+        apiKey: 'ms-114c4bf4-ab3e-442b-a30c-6b355db26263',
+        apiUrl: 'https://api-inference.modelscope.cn/v1',
+        model: 'deepseek-ai/DeepSeek-V3.2',
+        // maxTokens: 1000,
+        // temperature: 0.3
+      },
+      git: {
+        autoAdd: true,
+        autoPull: true,
+        autoPush: true,
+        remote: 'origin'
+      },
+      review: {
+        enabled: true,
+        skipPatterns: [
+          '*.lock',
+          '*.log',
+          '*.min.js',
+          'node_modules',
+          'dist',
+          'build',
+          '.git'
+        ],
+        maxFileSize: 1024 * 100,
+        checkSecrets: true,
+        checkTodos: true,
+        checkDebug: true
+      }
+    };
+  }
+
+  loadConfig() {
+    try {
+      if (!fs.existsSync(this.configDir)) {
+        fs.mkdirSync(this.configDir, { recursive: true });
+      }
+
+      if (fs.existsSync(this.configFile)) {
+        const data = fs.readFileSync(this.configFile, 'utf8');
+        return JSON.parse(data);
+      }
+    } catch (error) {
+      console.error(chalk.yellow(`⚠️ 读取配置文件失败: ${error.message}`));
+    }
+    return this.getDefaultConfig();
+  }
+
+  saveConfig() {
+    try {
+      if (!fs.existsSync(this.configDir)) {
+        fs.mkdirSync(this.configDir, { recursive: true });
+      }
+      fs.writeFileSync(this.configFile, JSON.stringify(this.config, null, 2), 'utf8');
+      return true;
+    } catch (error) {
+      console.error(chalk.red(`❌ 保存配置失败: ${error.message}`));
+      return false;
+    }
+  }
+
+  get(key) {
+    const keys = key.split('.');
+    let value = this.config;
+    
+    for (const k of keys) {
+      if (value && typeof value === 'object' && k in value) {
+        value = value[k];
+      } else {
+        return undefined;
+      }
+    }
+    return value;
+  }
+
+  set(key, value) {
+    const keys = key.split('.');
+    let current = this.config;
+    
+    for (let i = 0; i < keys.length - 1; i++) {
+      if (!(keys[i] in current)) {
+        current[keys[i]] = {};
+      }
+      current = current[keys[i]];
+    }
+    
+    current[keys[keys.length - 1]] = value;
+    this.saveConfig();
+  }
+
+  getAll() {
+    return this.config;
+  }
+
+  init() {
+    console.log('\n' + chalk.cyan('='.repeat(50)));
+    console.log(chalk.cyan('🔧 初始化 Git AI 配置'));
+    console.log(chalk.cyan('='.repeat(50)) + '\n');
+    
+    this.config = this.getDefaultConfig();
+    
+    const rl = readline.createInterface({
+      input: process.stdin,
+      output: process.stdout
+    });
+
+    console.log(chalk.yellow('(可选) 配置 OpenAI API 密钥，用于AI代码审查和生成提交信息'));
+    console.log(chalk.gray('如果不配置，将只使用基础检查功能\n'));
+
+    const askApiKey = () => {
+      rl.question(chalk.cyan('OpenAI API密钥 (直接回车跳过): '), (apiKey) => {
+        if (apiKey && apiKey.trim()) {
+          this.config.ai.apiKey = apiKey.trim();
+          console.log(chalk.green('✅ API密钥已保存'));
+        } else {
+          console.log(chalk.yellow('⚠️ 跳过API密钥配置，将使用基础功能'));
+        }
+        askCustomUrl();
+      });
+    };
+
+    const askCustomUrl = () => {
+      rl.question(chalk.cyan('\n使用自定义API地址? (y/N): '), (customUrl) => {
+        if (customUrl.toLowerCase() === 'y' || customUrl.toLowerCase() === 'yes') {
+          rl.question(chalk.cyan('API地址: '), (url) => {
+            if (url && url.trim()) {
+              this.config.ai.apiUrl = url.trim();
+              console.log(chalk.green('✅ API地址已保存'));
+            }
+            finishInit();
+          });
+        } else {
+          finishInit();
+        }
+      });
+    };
+
+    const finishInit = () => {
+      if (this.saveConfig()) {
+        console.log(chalk.cyan('\n📋 当前配置:'));
+        console.log(JSON.stringify(this.config, null, 2));
+        console.log(chalk.green('\n✅ 配置初始化完成！'));
+        console.log(chalk.yellow('\n现在你可以在项目中使用: gai commit'));
+      } else {
+        console.log(chalk.red('\n❌ 配置保存失败！'));
+      }
+      rl.close();
+    };
+
+    askApiKey();
+  }
+}
+
+module.exports = { SimpleConfigManager };

package/lib/index.js

@@ -0,0 +1,19 @@
+/**
+ * 主入口模块
+ */
+
+const { CommitCommand } = require('./commands/commit');
+const { ConfigManager } = require('./core/config');
+const { GitOperations } = require('./core/git-operations');
+const { AIReviewer } = require('./core/ai-reviewer');
+const { Logger } = require('./utils/logger');
+const { FileUtils } = require('./utils/file-utils');
+
+module.exports = {
+  CommitCommand,
+  ConfigManager,
+  GitOperations,
+  AIReviewer,
+  Logger,
+  FileUtils
+};

package/lib/utils/file-utils.js

@@ -0,0 +1,98 @@
+/**
+ * 文件工具模块
+ */
+
+const fs = require('fs').promises;
+const path = require('path');
+const glob = require('glob');
+
+class FileUtils {
+  static async readFile(filePath) {
+    try {
+      return await fs.readFile(filePath, 'utf-8');
+    } catch (error) {
+      return null;
+    }
+  }
+
+  static async writeFile(filePath, content) {
+    try {
+      await fs.writeFile(filePath, content, 'utf-8');
+      return true;
+    } catch (error) {
+      return false;
+    }
+  }
+
+  static async fileExists(filePath) {
+    try {
+      await fs.access(filePath);
+      return true;
+    } catch {
+      return false;
+    }
+  }
+
+  static async getFileSize(filePath) {
+    try {
+      const stats = await fs.stat(filePath);
+      return stats.size;
+    } catch {
+      return 0;
+    }
+  }
+
+  static getFileExtension(filePath) {
+    return path.extname(filePath).toLowerCase();
+  }
+
+  static getFileName(filePath) {
+    return path.basename(filePath);
+  }
+
+  static getFileDir(filePath) {
+    return path.dirname(filePath);
+  }
+
+  static async findFiles(pattern, options = {}) {
+    return new Promise((resolve, reject) => {
+      glob(pattern, options, (err, files) => {
+        if (err) reject(err);
+        else resolve(files);
+      });
+    });
+  }
+
+  static formatFileSize(bytes) {
+    const units = ['B', 'KB', 'MB', 'GB'];
+    let size = bytes;
+    let unitIndex = 0;
+
+    while (size >= 1024 && unitIndex < units.length - 1) {
+      size /= 1024;
+      unitIndex++;
+    }
+
+    return `${size.toFixed(2)} ${units[unitIndex]}`;
+  }
+
+  static async ensureDir(dirPath) {
+    try {
+      await fs.mkdir(dirPath, { recursive: true });
+      return true;
+    } catch {
+      return false;
+    }
+  }
+
+  static async deleteFile(filePath) {
+    try {
+      await fs.unlink(filePath);
+      return true;
+    } catch {
+      return false;
+    }
+  }
+}
+
+module.exports = { FileUtils };

package/lib/utils/logger.js

@@ -0,0 +1,71 @@
+/**
+ * 日志工具模块
+ */
+
+const chalk = require('chalk');
+const ora = require('ora');
+
+class Logger {
+  constructor(options = {}) {
+    this.spinner = null;
+    this.options = options;
+  }
+
+  info(message) {
+    console.log(chalk.blue('ℹ'), message);
+  }
+
+  success(message) {
+    console.log(chalk.green('✅'), message);
+  }
+
+  warning(message) {
+    console.log(chalk.yellow('⚠️'), message);
+  }
+
+  error(message) {
+    console.log(chalk.red('❌'), message);
+  }
+
+  startSpinner(text) {
+    if (this.options.showProgress) {
+      this.spinner = ora({
+        text,
+        color: 'cyan'
+      }).start();
+    }
+  }
+
+  stopSpinner(success = true) {
+    if (this.spinner) {
+      if (success) {
+        this.spinner.succeed();
+      } else {
+        this.spinner.fail();
+      }
+      this.spinner = null;
+    }
+  }
+
+  table(data) {
+    console.table(data);
+  }
+
+  divider() {
+    console.log(chalk.gray('─'.repeat(50)));
+  }
+
+  highlight(text) {
+    return chalk.cyan(text);
+  }
+
+  risk(text) {
+    return chalk.red(`[风险] ${text}`);
+  }
+
+  suggestion(text) {
+    return chalk.yellow(`[建议] ${text}`);
+  }
+}
+
+module.exports = { Logger };

package/package.json

@@ -0,0 +1,25 @@
+{
+  "name": "git-ai-shen",
+  "version": "1.0.0",
+  "description": "智能Git提交工具，支持行级代码审查和AI生成提交信息",
+  "main": "bin/git-ai.js",
+  "bin": {
+    "gai": "./bin/git-ai.js"
+  },
+  "scripts": {
+    "start": "node bin/git-ai.js"
+  },
+  "keywords": ["git", "ai", "code-review", "commit"],
+  "author": "shenzhichao",
+  "license": "MIT",
+  "dependencies": {
+    "simple-git": "^3.19.0",
+    "openai": "^4.0.0",
+    "chalk": "^4.1.2",
+    "inquirer": "^8.2.5",
+    "ignore": "^5.2.4"
+  },
+  "engines": {
+    "node": ">=14.0.0"
+  }
+}