ai-git-tools 1.0.4 → 2.0.0

package/src/commands/pr.js

@@ -1,303 +1,187 @@
 /**
  * PR 命令
- * 
- * 生成 PR 標題、描述並創建 Pull Request
+ * 基于 scripts/ai-auto-pr.mjs 简化版
+ * 生成 PR 标题、描述并创建 Pull Request
  */
 
-import chalk from 'chalk';
-import inquirer from 'inquirer';
-import { loadConfig } from '../core/config-loader.js';
-import { AIClient } from '../core/ai-client.js';
+import { execSync } from 'child_process';
+import { loadPRConfig } from '../core/config-loader.js';
 import { GitOperations } from '../core/git-operations.js';
-import { GitHubAPI } from '../core/github-api.js';
+import { AIClient } from '../core/ai-client.js';
 import { Logger } from '../utils/logger.js';
-import { 
-  handleError, 
-  truncateText,
-  getProjectTypePrompt,
-} from '../utils/helpers.js';
-
-/**
- * 決定 base 分支
- */
-async function determineBaseBranch(config, logger) {
-  if (config.baseBranch && config.baseBranch !== 'auto') {
-    return config.baseBranch;
-  }
-
-  // 嘗試找最新的 release 分支
-  logger.startSpinner('尋找目標分支...');
-  const latestRelease = GitHubAPI.getLatestReleaseBranch();
-  
-  if (latestRelease) {
-    logger.succeedSpinner(`自動選擇目標分支: ${latestRelease}`);
-    return latestRelease;
-  }
-
-  // 預設使用 main 或 master
-  const defaultBranch = GitOperations.branchExists('main') ? 'main' : 'master';
-  logger.succeedSpinner(`使用預設分支: ${defaultBranch}`);
-  return defaultBranch;
-}
+import { handleError, getProjectTypePrompt } from '../utils/helpers.js';
 
 /**
- * 使用 AI 生成 PR 內容
+ * 生成 PR 内容
  */
-async function generatePRContent(baseBranch, headBranch, config, logger) {
-  logger.startSpinner('AI 正在分析變更並生成 PR 內容...');
-
-  // 獲取 diff 和 commits
-  const diff = GitOperations.getDiffBetweenBranches(baseBranch, headBranch);
-  const commits = GitOperations.getCommitsBetweenBranches(baseBranch, headBranch);
+async function generatePRContent(baseBranch, headBranch, config) {
+  const logger = new Logger();
+  logger.step('AI 正在分析变更并生成 PR 内容...');
 
-  const truncatedDiff = truncateText(diff, config.ai.maxDiffLength);
+  // 获取 diff 和 commits
+  const diff = GitOperations.getDiff(baseBranch, headBranch);
+  const commits = GitOperations.getCommits(baseBranch, headBranch);
 
-  const aiClient = new AIClient(config);
+  // 智能截断 diff
+  const truncatedDiff = GitOperations.truncateDiff(diff, config.ai.maxDiffLength);
 
   const prompt = `${getProjectTypePrompt()}
 
-請根據以下資訊生成一個 Pull Request 的標題和描述。
+请根据以下资讯生成一个 Pull Request 的标题和描述。
 
 **Commits 列表**：
-${commits.join('\n')}
+${commits}
 
 **Git Diff**：
 ${truncatedDiff}
 
-**輸出格式**（JSON）：
+**输出格式**（JSON）：
 {
-  "title": "PR 標題（簡短、繁體中文，50 字內）",
-  "description": "PR 描述（使用 Markdown 格式，繁體中文）"
+  "title": "PR 标题（简短、繁体中文，50 字内）",
+  "description": "PR 描述（使用 Markdown 格式，繁体中文）"
 }
 
-**PR 描述應包含**：
-1. ## 📝 變更摘要（簡述主要變更）
-2. ## ✨ 主要功能（列出新增功能或修正項目）
-3. ## 🔧 技術細節（選填，如有重要的技術變更）
-4. ## ✅ 測試（如何測試這些變更）
+**PR 描述应包含**：
+1. ## 📝 变更摘要（简述主要变更）
+2. ## ✨ 主要功能（列出新增功能或修正项目）
+3. ## 🔧 技术细节（选填，如有重要的技术变更）
+4. ## ✅ 测试（如何测试这些变更）
 
-請只輸出 JSON，不要其他文字。`;
+请只输出 JSON，不要其他文字。`;
 
-  const response = await aiClient.sendAndWait(prompt);
-  await aiClient.stop();
+  const response = await AIClient.sendAndWait(prompt, config.ai.model);
 
   try {
     const prContent = AIClient.parseJSON(response);
-    logger.succeedSpinner('PR 內容生成完成');
+    logger.success('PR 内容生成完成\n');
     return prContent;
   } catch (error) {
-    logger.failSpinner('生成 PR 內容失敗');
-    throw new Error(`無法解析 AI 回應: ${error.message}`);
+    throw new Error(`无法解析 AI 回应: ${error.message}`);
   }
 }
 
 /**
- * 選擇 Reviewers
+ * 显示 PR 预览
  */
-async function selectReviewers(changedFiles, config, logger) {
-  if (!config.reviewers.autoSelect) {
-    return [];
-  }
-
-  logger.startSpinner('分析潛在的 reviewers...');
-
-  const contributorsMap = new Map();
-
-  // 分析每個變更檔案的貢獻者
-  for (const file of changedFiles.slice(0, 10)) { // 限制分析前 10 個檔案
-    const contributors = GitOperations.getFileContributors(
-      file,
-      config.reviewers.gitHistoryDepth
-    );
-
-    for (const { email, name } of contributors) {
-      // 排除設定中的作者
-      if (config.reviewers.excludeAuthors.some(pattern => email.includes(pattern))) {
-        continue;
-      }
-
-      const count = contributorsMap.get(email) || 0;
-      contributorsMap.set(email, count + 1);
-    }
-  }
-
-  // 排序並取前 N 位
-  const suggested = Array.from(contributorsMap.entries())
-    .sort((a, b) => b[1] - a[1])
-    .slice(0, config.reviewers.maxSuggested)
-    .map(([email]) => email);
-
-  logger.succeedSpinner(`找到 ${suggested.length} 位潛在 reviewers`);
-
-  if (suggested.length === 0) {
-    return [];
-  }
-
-  // 互動式選擇
-  const { selectedReviewers } = await inquirer.prompt([
-    {
-      type: 'checkbox',
-      name: 'selectedReviewers',
-      message: '選擇 Reviewers:',
-      choices: suggested.map(email => ({
-        name: email,
-        value: email,
-      })),
-    },
-  ]);
-
-  return selectedReviewers;
+function displayPreview(prContent, stats) {
+  console.log('\n' + '═'.repeat(60));
+  console.log('📋 PR 预览');
+  console.log('═'.repeat(60));
+  console.log(`\n标题: ${prContent.title}\n`);
+  console.log(`统计: ${stats.stats}`);
+  console.log(`文件数: ${stats.filesChanged} 个文件\n`);
+  console.log('─'.repeat(60));
+  console.log('描述:\n');
+  console.log(prContent.description);
+  console.log('\n' + '═'.repeat(60) + '\n');
 }
 
 /**
- * 智能選擇 Labels
+ * PR 命令主函数
  */
-function suggestLabels(prContent, availableLabels) {
-  const suggestions = [];
-  const title = prContent.title.toLowerCase();
-  const description = prContent.description.toLowerCase();
-
-  // 根據關鍵字建議 labels
-  const labelKeywords = {
-    bug: ['fix', 'bug', '修正', '錯誤'],
-    enhancement: ['feat', 'feature', '新增', '功能'],
-    documentation: ['docs', '文件', 'readme'],
-    refactor: ['refactor', '重構'],
-    performance: ['perf', 'performance', '效能', '優化'],
-    test: ['test', '測試'],
-    ui: ['ui', 'style', '樣式', '介面'],
-    api: ['api', 'endpoint'],
-  };
-
-  for (const [label, keywords] of Object.entries(labelKeywords)) {
-    if (availableLabels.includes(label)) {
-      if (keywords.some(kw => title.includes(kw) || description.includes(kw))) {
-        suggestions.push(label);
-      }
-    }
-  }
-
-  return suggestions;
-}
-
-/**
- * PR 命令處理器
- */
-export async function prCommand(options) {
-  const logger = new Logger(options.verbose);
+export async function prCommand() {
+  const logger = new Logger();
 
   try {
     logger.header('AI Auto PR Generator');
 
-    // 檢查環境
-    if (!GitOperations.isGitRepository()) {
-      logger.error('當前目錄不是 Git 倉庫');
-      process.exit(1);
+    // 载入配置
+    const config = await loadPRConfig();
+
+    // 获取当前分支
+    const currentBranch = GitOperations.getCurrentBranch();
+    let headBranch = config.headBranch || currentBranch;
+    let baseBranch = config.baseBranch;
+
+    // 自动侦测 base branch
+    if (!baseBranch) {
+      logger.step('正在侦测 release 分支...\n');
+      const latestRelease = GitOperations.findLatestReleaseBranch();
+
+      if (latestRelease) {
+        baseBranch = latestRelease;
+        logger.success(`自动选择目标分支: ${baseBranch}\n`);
+      } else {
+        logger.error('未侦测到任何 release 分支');
+        console.log('💡 使用 --base 参数指定目标分支');
+        process.exit(1);
+      }
     }
 
-    if (!GitHubAPI.isAvailable()) {
-      logger.error('GitHub CLI 未安裝或未認證');
-      logger.info('請先安裝並設定 GitHub CLI:');
-      logger.info('  brew install gh');
-      logger.info('  gh auth login');
+    // 检查是否在 base branch
+    if (currentBranch === baseBranch) {
+      logger.error(`你目前在 ${baseBranch} 分支，请切换到 feature 分支`);
+      console.log('💡 切换到 feature 分支: git checkout <feature-branch>');
       process.exit(1);
     }
 
-    // 載入配置
-    const config = await loadConfig(options);
-
-    // 確定分支
-    const headBranch = config.headBranch || GitOperations.getCurrentBranch();
-    const baseBranch = await determineBaseBranch(config, logger);
-
-    console.log(chalk.cyan(`\n📌 分支資訊:`));
+    console.log('\n📌 分支资讯:');
     console.log(`   Base: ${baseBranch}`);
-    console.log(`   Head: ${headBranch}`);
-
-    // 檢查 PR 是否已存在
-    if (GitHubAPI.prExists(baseBranch, headBranch)) {
-      logger.warn('PR 已存在');
-      process.exit(0);
+    console.log(`   Head: ${headBranch}\n`);
+
+    // 推送到远端（预览模式跳过）
+    if (!config.preview) {
+      logger.step(`推送到远端分支: origin/${headBranch}`);
+      GitOperations.push(headBranch);
+      logger.success('推送成功\n');
+
+      // 等待 GitHub 同步
+      logger.info('等待 GitHub 同步...');
+      await new Promise(resolve => setTimeout(resolve, 2000));
+      GitOperations.fetch();
+      logger.success('同步完成\n');
     }
 
-    // 生成 PR 內容
-    const prContent = await generatePRContent(baseBranch, headBranch, config, logger);
+    // 获取变更统计
+    const stats = GitOperations.getChangeStats(baseBranch, headBranch);
+    console.log(`📈 变更统计: ${stats.stats}`);
+    console.log(`📁 影响文件: ${stats.filesChanged} 个\n`);
 
-    console.log(chalk.cyan('\n📝 PR 標題:'));
-    logger.code(prContent.title);
+    // 生成 PR 内容
+    const prContent = await generatePRContent(baseBranch, headBranch, config);
 
-    console.log(chalk.cyan('\n📄 PR 描述:'));
-    logger.code(prContent.description);
+    // 显示预览
+    displayPreview(prContent, stats);
 
-    // 如果是預覽模式，不創建 PR
+    // 预览模式：仅显示不创建
     if (config.preview) {
-      logger.info('預覽模式，不創建 PR');
-      process.exit(0);
+      logger.info('预览模式：未创建 PR');
+      return;
     }
 
-    // 確認是否創建 PR
-    if (!config.noConfirm) {
-      const { confirm } = await inquirer.prompt([
-        {
-          type: 'confirm',
-          name: 'confirm',
-          message: '是否創建 PR？',
-          default: true,
-        },
-      ]);
-
-      if (!confirm) {
-        logger.info('已取消');
-        process.exit(0);
-      }
-    }
+    // 创建 PR（使用 GitHub CLI）
+    logger.step('创建 Pull Request...');
 
-    // 選擇 Reviewers
-    let reviewers = [];
-    if (config.reviewers.autoSelect) {
-      const changes = GitOperations.getAllChanges();
-      const changedFiles = changes.map(c => c.filePath);
-      reviewers = await selectReviewers(changedFiles, config, logger);
+    // 准备 PR body
+    const prBody = prContent.description;
+
+    // 构建 gh pr create 命令
+    let ghCommand = `gh pr create --base ${baseBranch} --head ${headBranch} --title "${prContent.title.replace(/"/g, '\\"')}"`;
+
+    // 将 body 写入临时文件
+    const { writeFileSync, unlinkSync } = await import('fs');
+    const tmpFile = '/tmp/pr-body-temp.md';
+    writeFileSync(tmpFile, prBody, 'utf-8');
+    ghCommand += ` --body-file ${tmpFile}`;
+
+    if (config.draft) {
+      ghCommand += ' --draft';
     }
 
-    // 選擇 Labels
-    let labels = [];
-    if (config.github.autoLabels) {
-      const availableLabels = GitHubAPI.getLabels();
-      const suggestedLabels = suggestLabels(prContent, availableLabels);
-      
-      if (suggestedLabels.length > 0) {
-        const { selectedLabels } = await inquirer.prompt([
-          {
-            type: 'checkbox',
-            name: 'selectedLabels',
-            message: '選擇 Labels:',
-            choices: suggestedLabels.map(label => ({
-              name: label,
-              value: label,
-              checked: true,
-            })),
-          },
-        ]);
-        labels = selectedLabels;
+    try {
+      const result = execSync(ghCommand, { encoding: 'utf-8' });
+      unlinkSync(tmpFile);
+
+      logger.success('PR 创建成功！\n');
+      console.log(result);
+    } catch (error) {
+      try {
+        unlinkSync(tmpFile);
+      } catch (e) {
+        // 忽略
       }
+      throw new Error(`创建 PR 失败: ${error.message}`);
     }
-
-    // 創建 PR
-    logger.startSpinner('正在創建 PR...');
-    
-    GitHubAPI.createPR({
-      title: prContent.title,
-      body: prContent.description,
-      base: baseBranch,
-      head: headBranch,
-      draft: config.draft,
-      reviewers,
-      labels,
-    });
-
-    logger.succeedSpinner('PR 創建成功！');
-
   } catch (error) {
     handleError(error);
     process.exit(1);

package/src/core/ai-client.js

@@ -1,115 +1,50 @@
 /**
- * AI Client
- * 
- * 封裝 GitHub Copilot SDK 的 AI 客戶端
+ * AI 客户端
+ * 基于 @github/copilot-sdk
  */
 
 import { CopilotClient } from '@github/copilot-sdk';
 
 export class AIClient {
-  constructor(config = {}) {
-    this.config = config;
-    this.client = null;
-    this.session = null;
-  }
-
-  /**
-   * 初始化客戶端
-   */
-  async initialize() {
-    if (!this.client) {
-      this.client = new CopilotClient();
-    }
-    return this.client;
-  }
-
   /**
-   * 創建會話
+   * 发送 prompt 并等待回应（带重试机制）
    */
-  async createSession(options = {}) {
-    await this.initialize();
-    
-    this.session = await this.client.createSession({
-      model: options.model || this.config.ai?.model || 'gpt-4.1',
-      ...options,
-    });
-    
-    return this.session;
-  }
-
-  /**
-   * 發送請求並等待回應（帶重試機制）
-   */
-  async sendAndWait(prompt, options = {}) {
-    const maxRetries = options.maxRetries || this.config.ai?.maxRetries || 3;
+  static async sendAndWait(prompt, model = 'gpt-4.1', maxRetries = 3) {
+    const client = new CopilotClient();
     let lastError = null;
 
     for (let attempt = 1; attempt <= maxRetries; attempt++) {
       try {
-        if (!this.session) {
-          await this.createSession(options);
-        }
+        const session = await client.createSession({ model });
+        const response = await session.sendAndWait({ prompt });
+        await client.stop();
 
-        const response = await this.session.sendAndWait({ prompt });
-        return response?.data?.content || '';
+        const content = response?.data?.content || '';
+        return content.trim();
       } catch (error) {
         lastError = error;
-        
-        if (this.config.output?.verbose) {
-          console.log(`⚠️  嘗試 ${attempt}/${maxRetries} 失敗: ${error.message}`);
-        }
-        
-        // 如果還有重試機會，重新創建 session
         if (attempt < maxRetries) {
-          this.session = null;
+          console.log(`⚠️  AI 请求失败，重试第 ${attempt}/${maxRetries} 次...`);
+          await new Promise((resolve) => setTimeout(resolve, 1000 * attempt));
           continue;
         }
       }
     }
 
-    throw new Error(`AI 請求失敗（嘗試 ${maxRetries} 次）: ${lastError?.message || '未知錯誤'}`);
+    throw new Error(`AI 请求失败: ${lastError?.message || '未知错误'}`);
   }
 
   /**
-   * 停止客戶端
+   * 解析 JSON 响应
    */
-  async stop() {
-    if (this.client) {
-      await this.client.stop();
-      this.client = null;
-      this.session = null;
-    }
-  }
-
-  /**
-   * 清理回應內容（移除 markdown 程式碼區塊等）
-   */
-  static cleanResponse(response) {
-    if (!response) return '';
-
-    let cleaned = response.trim();
+  static parseJSON(content) {
+    // 移除可能的 markdown code block 标记
+    const jsonContent = content.replace(/```json\n?/g, '').replace(/```\n?/g, '').trim();
 
-    // 移除 markdown 程式碼區塊標記
-    cleaned = cleaned.replace(/^```[\w]*\n/gm, '');
-    cleaned = cleaned.replace(/\n```$/gm, '');
-    cleaned = cleaned.replace(/^```$/gm, '');
-
-    // 移除開頭和結尾的引號
-    cleaned = cleaned.replace(/^["']|["']$/g, '');
-
-    return cleaned.trim();
-  }
-
-  /**
-   * 解析 JSON 回應
-   */
-  static parseJSON(response) {
-    const cleaned = AIClient.cleanResponse(response);
-    
     try {
-      return JSON.parse(cleaned);
+      return JSON.parse(jsonContent);
     } catch (error) {
-      throw new Error(`無法解析 AI 回應為 JSON: ${error.message}`);
+      throw new Error(`无法解析 AI 回应为 JSON: ${error.message}`);
     }
   }
 }