ai-git-tools 2.0.28 → 2.0.30

package/bin/cli.js

@@ -34,7 +34,14 @@ program
 program
   .command('init')
   .description('初始化配置檔案 (.ai-git-config.mjs)')
-  .action(initCommand);
+  .action(async (options) => {
+    try {
+      await initCommand(options);
+      process.exit(0);
+    } catch (error) {
+      process.exit(1);
+    }
+  });
 
 // Commit 命令
 program
@@ -44,7 +51,14 @@ program
   .option('-v, --verbose', '顯示詳細輸出')
   .option('--max-diff <number>', '最大 diff 長度')
   .option('--max-retries <number>', '最大重試次數')
-  .action(commitCommand);
+  .action(async (options) => {
+    try {
+      await commitCommand(options);
+      process.exit(0);
+    } catch (error) {
+      process.exit(1);
+    }
+  });
 
 // Commit All 命令
 program
@@ -54,7 +68,14 @@ program
   .option('-v, --verbose', '顯示詳細輸出')
   .option('--max-diff <number>', '最大 diff 長度')
   .option('--max-retries <number>', '最大重試次數')
-  .action(commitAllCommand);
+  .action(async (options) => {
+    try {
+      await commitAllCommand(options);
+      process.exit(0);
+    } catch (error) {
+      process.exit(1);
+    }
+  });
 
 // PR 命令
 program
@@ -66,6 +87,13 @@ program
   .option('--no-confirm', '跳過確認直接創建')
   .option('--auto-labels', '自動添加 Labels (預設啟用)')
   .option('--include-impact', '在 PR 中包含影響範圍分析和注意事項 (預設關閉)')
-  .action(prCommand);
+  .action(async (options) => {
+    try {
+      await prCommand(options);
+      process.exit(0);
+    } catch (error) {
+      process.exit(1);
+    }
+  });
 
 program.parse();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-git-tools",
-  "version": "2.0.28",
+  "version": "2.0.30",
   "description": "AI-powered Git automation tools for commit messages and PR generation",
   "main": "src/index.js",
   "type": "module",

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

@@ -10,6 +10,70 @@ export class GitHubAPI {
   constructor() {
     // 自動從 git remote 偵測組織名稱
     this.orgName = this.detectOrgFromRemote();
+    // 檢測是否為 fork 環境
+    this.isFork = this.detectForkEnvironment();
+  }
+
+  /**
+   * 檢測是否為 fork 環境（有 upstream remote）
+   */
+  detectForkEnvironment() {
+    try {
+      execSync('git remote get-url upstream', {
+        encoding: 'utf-8',
+        stdio: ['pipe', 'pipe', 'pipe'],
+      });
+      log.info('偵測到 fork 環境（有 upstream remote）\n');
+      return true;
+    } catch (error) {
+      // 沒有 upstream，不是 fork
+      return false;
+    }
+  }
+
+  /**
+   * 獲取當前 gh CLI 登入的帳號
+   */
+  getGHAccount() {
+    try {
+      const account = execSync('gh api user --jq ".login"', {
+        encoding: 'utf-8',
+        stdio: ['pipe', 'pipe', 'pipe'],
+      }).trim();
+      return account;
+    } catch (error) {
+      return null;
+    }
+  }
+
+  /**
+   * 檢查 gh 帳號是否與倉庫擁有者匹配
+   */
+  checkAccountMatch() {
+    const ghAccount = this.getGHAccount();
+    if (!ghAccount) {
+      log.warning('⚠️  無法取得 gh CLI 登入帳號\n');
+      return true; // 無法確認就不阻止
+    }
+
+    if (!this.orgName) {
+      return true; // 無法確認倉庫擁有者
+    }
+
+    if (ghAccount.toLowerCase() !== this.orgName.toLowerCase()) {
+      log.error(`\n❌ 帳號不匹配！\n`);
+      console.log(`   倉庫擁有者: ${colors.cyan}${this.orgName}${colors.reset}`);
+      console.log(`   gh 登入帳號: ${colors.yellow}${ghAccount}${colors.reset}\n`);
+      console.log(`${colors.yellow}⚠️  GitHub 不允許非 collaborator 創建 PR${colors.reset}\n`);
+      console.log('解決方法：');
+      console.log(`  1. ${colors.cyan}切換 gh 帳號到倉庫擁有者${colors.reset}`);
+      console.log(`     執行: ${colors.green}gh auth switch${colors.reset}`);
+      console.log(`     或: ${colors.green}gh auth login${colors.reset}\n`);
+      console.log(`  2. 或者將 ${ghAccount} 加為倉庫 collaborator\n`);
+      return false;
+    }
+
+    return true;
   }
 
   /**
@@ -207,6 +271,12 @@ export class GitHubAPI {
     const bodyFile = '/tmp/pr-body.md';
     writeFileSync(bodyFile, body);
 
+    // 檢查帳號是否匹配
+    if (!this.checkAccountMatch()) {
+      this.safeUnlink(bodyFile);
+      throw new Error('GitHub 帳號與倉庫擁有者不匹配，無法創建 PR');
+    }
+
     try {
       // 檢查 PR 是否已存在
       let existingPRUrl = null;
@@ -242,13 +312,21 @@ export class GitHubAPI {
         // 創建新 PR
         log.step('正在創建 Pull Request...');
 
-        // 如果有 orgName 且是組織 repo，使用完整的分支格式
-        // 如果沒有 orgName，直接使用分支名稱（gh CLI 會自動處理）
+        // 判斷是否需要加 org: 前綴
+        // 1. 如果 headBranch 已經包含 ':'，直接使用
+        // 2. 如果是 fork 環境（有 upstream），需要加 'org:' 前綴
+        // 3. 如果不是 fork（在同一倉庫內操作），直接使用分支名稱
         let headRef = headBranch;
         if (headBranch.includes(':')) {
+          // 用戶已明確指定格式，直接使用
           headRef = headBranch;
-        } else if (this.orgName) {
+        } else if (this.isFork && this.orgName) {
+          // Fork 環境，需要加 org: 前綴以正確標識來源
           headRef = `${this.orgName}:${headBranch}`;
+          log.info(`Fork 環境：使用完整分支格式 ${headRef}\n`);
+        } else {
+          // 同一倉庫內操作，直接使用分支名稱
+          headRef = headBranch;
         }
         const createCmd = `gh pr create --base "${baseBranch}" --head "${headRef}" --title "${escapedTitle}" --body-file "${bodyFile}"`;