ai-git-tools 2.0.8 → 2.0.9

package/src/pr-modules/core/git-operations.js

@@ -0,0 +1,211 @@
+import { execSync } from 'child_process';
+import { CONSTANTS } from '../utils/constants.js';
+import { PRError } from '../utils/helpers.js';
+
+/**
+ * Git 操作封裝
+ */
+export class GitOperations {
+  /**
+   * 偵測可用的 release 分支
+   */
+  detectReleaseBranches() {
+    try {
+      execSync('git fetch origin', { stdio: 'ignore' });
+      const branches = execSync('git branch -r')
+        .toString()
+        .split('\n')
+        .map((b) => b.trim())
+        .filter((b) => b.startsWith('origin/release-'))
+        .map((b) => b.replace('origin/', ''));
+      return branches;
+    } catch (error) {
+      return [];
+    }
+  }
+
+  /**
+   * 找到最新的 release 分支
+   */
+  findLatestReleaseBranch() {
+    const branches = this.detectReleaseBranches();
+    if (branches.length === 0) return null;
+
+    const monthlyBranches = branches.filter((b) => b.includes('-m'));
+    const weeklyBranches = branches.filter((b) => b.includes('-w'));
+    const priorityBranches = monthlyBranches.length > 0 ? monthlyBranches : weeklyBranches;
+
+    priorityBranches.sort().reverse();
+    return priorityBranches[0];
+  }
+
+  /**
+   * 獲取當前分支
+   */
+  getCurrentBranch() {
+    return execSync('git rev-parse --abbrev-ref HEAD').toString().trim();
+  }
+
+  /**
+   * 獲取變更統計
+   */
+  getChangeStats(baseBranch, headBranch) {
+    try {
+      const stats = execSync(`git diff --shortstat origin/${baseBranch}...${headBranch}`, {
+        encoding: 'utf-8',
+      }).trim();
+
+      const filesChanged = execSync(
+        `git diff --name-only origin/${baseBranch}...${headBranch} | wc -l`,
+        { encoding: 'utf-8' }
+      ).trim();
+
+      return { stats, filesChanged: parseInt(filesChanged, 10) };
+    } catch (error) {
+      return { stats: '無法獲取統計', filesChanged: 0 };
+    }
+  }
+
+  /**
+   * 獲取變更的檔案列表
+   */
+  getChangedFiles(baseBranch, headBranch) {
+    try {
+      const files = execSync(`git diff --name-only origin/${baseBranch}...${headBranch}`, {
+        encoding: 'utf-8',
+      })
+        .split('\n')
+        .filter(Boolean);
+      return files;
+    } catch (error) {
+      return [];
+    }
+  }
+
+  /**
+   * 獲取 commit 列表
+   */
+  getCommits(baseBranch, headBranch, options = {}) {
+    const { oneline = true, noDecorate = true } = options;
+    try {
+      let cmd = `git log origin/${baseBranch}..origin/${headBranch}`;
+      if (oneline) cmd += ' --oneline';
+      if (noDecorate) cmd += ' --no-decorate';
+
+      return execSync(cmd, { encoding: 'utf-8' });
+    } catch (error) {
+      throw new PRError(
+        '無法比較分支差異',
+        'GIT_COMPARE_FAILED',
+        ['檢查遠端分支是否存在: git branch -r', '執行診斷: npm run diagnose:pr'],
+        `git log origin/${baseBranch}..origin/${headBranch}`
+      );
+    }
+  }
+
+  /**
+   * 獲取 diff
+   */
+  getDiff(baseBranch, headBranch, maxBuffer = CONSTANTS.MAX_BUFFER_SIZE) {
+    try {
+      return execSync(`git diff origin/${baseBranch}...${headBranch}`, {
+        encoding: 'utf-8',
+        maxBuffer,
+      });
+    } catch (error) {
+      // 嘗試替代方案
+      return execSync(`git diff origin/${baseBranch}..${headBranch}`, {
+        encoding: 'utf-8',
+        maxBuffer,
+      });
+    }
+  }
+
+  /**
+   * 推送到遠端
+   */
+  async push(branch) {
+    try {
+      execSync(`git push -u origin ${branch}`, { stdio: 'inherit' });
+      return true;
+    } catch (error) {
+      throw new PRError(
+        '推送失敗',
+        'GIT_PUSH_FAILED',
+        ['檢查是否有推送權限', '檢查遠端分支是否有衝突', '檢查網路連接'],
+        `git push -u origin ${branch}`
+      );
+    }
+  }
+
+  /**
+   * 同步遠端資訊
+   */
+  async fetch() {
+    try {
+      execSync('git fetch origin', { stdio: 'ignore' });
+      return true;
+    } catch (error) {
+      return false;
+    }
+  }
+
+  /**
+   * 智能截斷 diff
+   */
+  truncateDiff(diff, maxLength = CONSTANTS.MAX_DIFF_LENGTH) {
+    if (diff.length <= maxLength) return diff;
+
+    const lines = diff.split('\n');
+    const header = lines.slice(0, CONSTANTS.DIFF_CONTEXT_LINES).join('\n');
+    const footer = lines.slice(-CONSTANTS.DIFF_CONTEXT_LINES).join('\n');
+
+    const middle = `\n\n... [已省略 ${lines.length - 100} 行變更] ...\n\n`;
+
+    return header + middle + footer;
+  }
+
+  /**
+   * 獲取當前用戶資訊
+   */
+  getCurrentUser() {
+    try {
+      const email = execSync('git config user.email', { encoding: 'utf-8' }).trim();
+      const name = execSync('git config user.name', { encoding: 'utf-8' }).trim();
+      let githubUser = null;
+
+      try {
+        githubUser = execSync('gh api user --jq .login', {
+          encoding: 'utf-8',
+          stdio: ['pipe', 'pipe', 'pipe'],
+        }).trim();
+      } catch {
+        // GitHub CLI 未認證或未安裝
+      }
+
+      return { email, name, githubUser };
+    } catch (error) {
+      return null;
+    }
+  }
+
+  /**
+   * 獲取 repository owner
+   */
+  getRepoOwner() {
+    try {
+      const remoteUrl = execSync('git config --get remote.origin.url', {
+        encoding: 'utf-8',
+      }).trim();
+
+      // 解析 GitHub URL
+      const httpsMatch = remoteUrl.match(/github\.com[/:]([^/]+)\//);
+      const sshMatch = remoteUrl.match(/github\.com:([^/]+)\//);
+
+      const owner = httpsMatch?.[1] || sshMatch?.[1];
+      return owner || null;
+    } catch (error) {
+      return null;
+    }
+  }
+}

package/src/pr-modules/core/github-api.js

@@ -0,0 +1,376 @@
+import { execSync } from 'child_process';
+import { writeFileSync, unlinkSync, existsSync } from 'fs';
+import { log } from '../utils/helpers.js';
+import { colors } from '../utils/constants.js';
+
+/**
+ * GitHub API 操作封裝
+ */
+export class GitHubAPI {
+  constructor(config = {}) {
+    this.orgName = config.orgName || 'kingsinfo-project';
+  }
+
+  /**
+   * 安全刪除檔案（如果存在）
+   */
+  safeUnlink(filePath) {
+    try {
+      if (existsSync(filePath)) {
+        unlinkSync(filePath);
+      }
+    } catch (error) {
+      // 忽略刪除錯誤
+      log.warning(`⚠️  無法刪除臨時檔案: ${filePath}`);
+    }
+  }
+
+  /**
+   * 檢查 GitHub CLI 認證狀態
+   */
+  checkAuth() {
+    try {
+      const authStatus = execSync('gh auth status 2>&1', {
+        encoding: 'utf-8',
+        stdio: ['pipe', 'pipe', 'pipe'],
+      });
+
+      return {
+        authenticated: authStatus.includes('Logged in'),
+        details: authStatus,
+      };
+    } catch (error) {
+      return {
+        authenticated: false,
+        details: error.message,
+      };
+    }
+  }
+
+  /**
+   * 從 GitHub 抓取組織的成員列表
+   */
+  async fetchOrgMembers(orgName = this.orgName) {
+    try {
+      log.step(`正在嘗試抓取 ${orgName} 組織的成員列表...`);
+
+      const membersJson = execSync(`gh api orgs/${orgName}/members --jq '.'`, {
+        encoding: 'utf-8',
+        stdio: ['pipe', 'pipe', 'pipe'],
+      });
+
+      const membersData = JSON.parse(membersJson);
+      const members = membersData.map((member) => ({
+        login: member.login,
+        name: member.name || member.login,
+      }));
+
+      if (members.length > 0) {
+        log.success(`成功抓取 ${members.length} 位組織成員\n`);
+        return members;
+      }
+
+      return [];
+    } catch (error) {
+      return [];
+    }
+  }
+
+  /**
+   * 從 GitHub 抓取組織的團隊列表
+   */
+  async fetchTeams(orgName = this.orgName) {
+    try {
+      // 先檢查認證狀態
+      const authStatus = this.checkAuth();
+      if (!authStatus.authenticated) {
+        log.warning('GitHub CLI 未認證');
+        log.info('請執行: gh auth login\n');
+        return { teams: {}, members: [] };
+      }
+
+      log.step(`正在從 GitHub 抓取 ${orgName} 的團隊資訊...`);
+
+      let teamsJson;
+      try {
+        teamsJson = execSync(`gh api orgs/${orgName}/teams --jq '.'`, {
+          encoding: 'utf-8',
+          stdio: ['pipe', 'pipe', 'pipe'],
+        });
+      } catch (error) {
+        // 無法存取團隊 API，使用替代方案
+        log.warning('無法存取組織團隊資訊（可能是權限問題）');
+        log.info('嘗試使用替代方案：直接抓取組織成員...\n');
+
+        const members = await this.fetchOrgMembers(orgName);
+        if (members.length > 0) {
+          return { teams: {}, members };
+        }
+
+        throw error;
+      }
+
+      const teams = JSON.parse(teamsJson);
+
+      if (!teams || teams.length === 0) {
+        log.warning('未找到任何團隊，嘗試直接抓取組織成員...');
+        const members = await this.fetchOrgMembers(orgName);
+        return { teams: {}, members };
+      }
+
+      const teamData = {};
+
+      for (const team of teams) {
+        try {
+          const membersJson = execSync(
+            `gh api orgs/${orgName}/teams/${team.slug}/members --jq '.'`,
+            {
+              encoding: 'utf-8',
+              stdio: ['pipe', 'pipe', 'pipe'],
+            }
+          );
+
+          const members = JSON.parse(membersJson);
+          teamData[team.slug] = {
+            name: team.name,
+            slug: team.slug,
+            description: team.description || '',
+            members: members.map((m) => ({
+              login: m.login,
+              name: m.name || m.login,
+            })),
+          };
+        } catch (error) {
+          log.warning(`無法抓取團隊 ${team.slug} 的成員: ${error.message}`);
+        }
+      }
+
+      // 同時取得所有成員作為備選
+      const allMembers = await this.fetchOrgMembers(orgName);
+
+      log.success(
+        `成功抓取 ${Object.keys(teamData).length} 個團隊和 ${allMembers.length} 位成員\n`
+      );
+      return { teams: teamData, members: allMembers };
+    } catch (error) {
+      log.warning('無法從 GitHub 抓取資訊');
+      log.info('請確認：');
+      console.log('  1. 已安裝 GitHub CLI: brew install gh');
+      console.log('  2. 已執行認證: gh auth login');
+      console.log('  3. 選擇正確的認證範圍（需要 read:org 權限）');
+      console.log(`  4. 有權限存取 ${orgName} 組織`);
+      console.log('  5. 組織名稱正確（可用 --org 參數指定）\n');
+
+      log.info('提示: 你仍可以手動輸入 reviewer 的 GitHub username\n');
+
+      return { teams: {}, members: [] };
+    }
+  }
+
+  /**
+   * 創建或更新 PR
+   */
+  async createOrUpdatePR(params) {
+    const { title, body, baseBranch, headBranch, reviewers, config } = params;
+    const bodyFile = '/tmp/pr-body.md';
+    writeFileSync(bodyFile, body);
+
+    try {
+      // 檢查 PR 是否已存在
+      let existingPRUrl = null;
+      try {
+        existingPRUrl = execSync(
+          `gh pr list --head "${headBranch}" --base "${baseBranch}" --json url --jq '.[0].url'`,
+          { encoding: 'utf-8', stdio: ['pipe', 'pipe', 'pipe'] }
+        ).trim();
+      } catch (error) {
+        // PR 不存在
+      }
+
+      const draftFlag = config.draft ? '--draft' : '';
+      const escapedTitle = title.replace(/"/g, '\\"');
+
+      if (existingPRUrl) {
+        // 更新現有 PR
+        log.step('偵測到現有 PR，正在更新內容...');
+
+        const updateCmd = `gh pr edit "${existingPRUrl}" --title "${escapedTitle}" --body-file "${bodyFile}"`;
+        execSync(updateCmd, { stdio: 'inherit' });
+
+        // 更新 reviewers
+        if (reviewers) {
+          await this.addReviewers(existingPRUrl, reviewers);
+        }
+
+        log.success('Pull Request 已更新！');
+        console.log(`🔗 ${existingPRUrl}\n`);
+
+        this.safeUnlink(bodyFile);
+        return existingPRUrl;
+      } else {
+        // 創建新 PR
+        log.step('正在創建 Pull Request...');
+
+        // 使用完整的分支格式（包含組織名稱）以避免 GitHub API 錯誤
+        const fullHeadBranch = headBranch.includes(':')
+          ? headBranch
+          : `${this.orgName}:${headBranch}`;
+        let createCmd = `gh pr create --base "${baseBranch}" --head "${fullHeadBranch}" --title "${escapedTitle}" --body-file "${bodyFile}"`;
+
+        if (draftFlag) {
+          createCmd += ` ${draftFlag}`;
+        }
+
+        try {
+          const result = execSync(createCmd, {
+            encoding: 'utf-8',
+            stdio: ['pipe', 'pipe', 'pipe'],
+          });
+          log.success(`Pull Request 創建成功！${config.draft ? ' (草稿模式)' : ''}`);
+
+          // 提取 PR URL
+          const prUrl = result
+            .trim()
+            .split('\n')
+            .find((line) => line.includes('https://'));
+
+          if (prUrl) {
+            console.log(`🔗 ${prUrl}`);
+
+            // 添加 reviewers
+            if (reviewers) {
+              const prNumber = prUrl.split('/').pop();
+              await this.addReviewersByAPI(prNumber, reviewers);
+            }
+          }
+
+          this.safeUnlink(bodyFile);
+          return prUrl;
+        } catch (error) {
+          this.safeUnlink(bodyFile);
+          throw error;
+        }
+      }
+    } catch (error) {
+      this.safeUnlink(bodyFile);
+      throw error;
+    }
+  }
+
+  /**
+   * 添加 Reviewers (給已存在的 PR)
+   */
+  async addReviewers(prUrl, reviewers) {
+    if (!reviewers) return;
+
+    const prNumber = prUrl.split('/').pop();
+
+    if (reviewers.individuals && reviewers.individuals.length > 0) {
+      await this.addReviewersByAPI(prNumber, { individuals: reviewers.individuals });
+    }
+
+    if (reviewers.teams && reviewers.teams.length > 0) {
+      await this.addTeamReviewers(prNumber, reviewers.teams);
+    }
+  }
+
+  /**
+   * 使用 API 添加個人 Reviewers
+   */
+  async addReviewersByAPI(prNumber, reviewers) {
+    if (!reviewers.individuals || reviewers.individuals.length === 0) return;
+
+    log.info(`正在添加個人 Reviewers (共 ${reviewers.individuals.length} 位)...`);
+    console.log(
+      `${colors.blue}準備添加: ${reviewers.individuals.map((r) => `@${r}`).join(', ')}${
+        colors.reset
+      }\n`
+    );
+
+    const successfulReviewers = [];
+    const failedReviewers = [];
+
+    for (const username of reviewers.individuals) {
+      try {
+        console.log(`${colors.cyan}正在添加 @${username}...${colors.reset}`);
+
+        const requestBody = JSON.stringify({ reviewers: [username] });
+
+        execSync(
+          `printf '%s' '${requestBody.replace(
+            /'/g,
+            "'\\''"
+          )}' | gh api repos/:owner/:repo/pulls/${prNumber}/requested_reviewers --input - -X POST`,
+          { encoding: 'utf-8', stdio: ['pipe', 'pipe', 'pipe'] }
+        );
+
+        // 等待 GitHub 處理
+        await new Promise((resolve) => setTimeout(resolve, 500));
+
+        // 驗證是否成功
+        const verifyResponse = execSync(
+          `gh api repos/:owner/:repo/pulls/${prNumber} --jq '.requested_reviewers[].login'`,
+          { encoding: 'utf-8', stdio: ['pipe', 'pipe', 'pipe'] }
+        ).trim();
+
+        const currentReviewers = verifyResponse.split('\n').filter(Boolean);
+
+        if (currentReviewers.includes(username)) {
+          console.log(`  ${colors.green}✓ 成功添加 @${username}${colors.reset}\n`);
+          successfulReviewers.push(username);
+        } else {
+          console.log(`  ${colors.yellow}⚠ 無法添加 @${username}${colors.reset}\n`);
+          failedReviewers.push({ username, reason: 'API 調用成功但未添加' });
+        }
+      } catch (error) {
+        console.log(`  ${colors.red}✗ 失敗: ${error.message}${colors.reset}\n`);
+        failedReviewers.push({ username, reason: error.message });
+      }
+    }
+
+    // 顯示結果
+    if (successfulReviewers.length > 0) {
+      log.success(`成功添加 ${successfulReviewers.length} 位 Reviewers`);
+    }
+
+    if (failedReviewers.length > 0) {
+      log.warning(`失敗 ${failedReviewers.length} 位，請手動添加`);
+    }
+  }
+
+  /**
+   * 添加團隊 Reviewers
+   */
+  async addTeamReviewers(prNumber, teams) {
+    if (!teams || teams.length === 0) return;
+
+    try {
+      log.info('正在添加團隊 Reviewers...');
+      const requestBody = JSON.stringify({ team_reviewers: teams });
+
+      execSync(
+        `printf '%s' '${requestBody.replace(
+          /'/g,
+          "'\\''"
+        )}' | gh api repos/:owner/:repo/pulls/${prNumber}/requested_reviewers --input - -X POST`,
+        { encoding: 'utf-8', stdio: ['pipe', 'pipe', 'pipe'] }
+      );
+
+      await new Promise((resolve) => setTimeout(resolve, 1000));
+
+      // 驗證
+      const verifyResponse = execSync(
+        `gh api repos/:owner/:repo/pulls/${prNumber} --jq '.requested_teams[].slug'`,
+        { encoding: 'utf-8', stdio: ['pipe', 'pipe', 'pipe'] }
+      ).trim();
+
+      const actualTeams = verifyResponse.split('\n').filter(Boolean);
+
+      if (actualTeams.length > 0) {
+        log.success(`成功添加團隊: ${actualTeams.join(', ')}`);
+      }
+    } catch (error) {
+      log.error('無法添加團隊 reviewers，請手動操作');
+    }
+  }
+}