ai-git-tools 2.0.59 → 2.0.61

package/bin/cli.js

@@ -142,8 +142,9 @@ program
   .command('write-and-test')
   .description('AI 為指定檔案生成 Jest 測試，並自動執行與修復（最多 2 次）')
   .requiredOption('--file <path>', '原始碼路徑')
-  .option('--test-type <type>', '測試類型：unit 或 component（預設 unit）', 'unit')
+  .option('--test-type <type>', '測試類型：auto、unit 或 component（預設 auto）', 'auto')
   .option('--max-fixes <number>', '最大自動修復次數（預設 2）', '2')
+  .option('--no-confirm', '跳過確認直接生成並執行測試')
   .option('--model <model>', '指定 AI 模型')
   .action(async (options) => {
     try {
@@ -159,10 +160,11 @@ program
   .command('auto-dev')
   .description('一鍵自動化：從 GitHub Issue 到代碼生成、測試與提交')
   .requiredOption('--issue <number>', 'GitHub Issue 編號')
-  .option('--file <path>', '目標檔案路徑（若無則提示）')
+  .option('--file <path>', '目標檔案路徑（若無則自動推斷）')
   .option('--context <description>', '額外說明或補充需求')
-  .option('--test-type <type>', '測試類型：unit 或 component（預設 unit）', 'unit')
+  .option('--test-type <type>', '測試類型：auto、unit 或 component（預設 auto）', 'auto')
   .option('--max-lines <number>', '最大行數限制（預設 500）', '500')
+  .option('--max-fixes <number>', '最大自動修復次數（預設 2）', '2')
   .option('--no-confirm', '全自動模式，跳過所有確認')
   .option('--model <model>', '指定 AI 模型')
   .action(async (options) => {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-git-tools",
-  "version": "2.0.59",
+  "version": "2.0.61",
   "description": "AI-powered Git automation tools for commit messages and PR generation",
   "main": "src/index.js",
   "type": "module",
@@ -63,9 +63,6 @@
   "jest": {
     "testEnvironment": "node",
     "transform": {},
-    "extensionsToTreatAsEsm": [
-      ".js"
-    ],
     "moduleNameMapper": {
       "^(\\.{1,2}/.+)\\.js$": "$1"
     }

package/src/commands/auto-dev.js

@@ -4,14 +4,16 @@
  * 支援全自動模式（--no-confirm）和分步互動模式
  */
 
-import { resolve, dirname, basename, join } from 'path';
+import { resolve } from 'path';
+import { existsSync, readFileSync } from 'fs';
+import { execSync } from 'child_process';
 import inquirer from 'inquirer';
 import { IssueReader } from '../core/issue-reader.js';
 import { CodeGenerator } from '../core/code-generator.js';
-import { TestGenerator } from '../core/test-generator.js';
-import { TestRunner } from '../core/test-runner.js';
 import { AIClient } from '../core/ai-client.js';
 import { GitOperations } from '../core/git-operations.js';
+import { TestGenerator } from '../core/test-generator.js';
+import { writeAndTestCommand, TEST_TYPE_CHOICES, normalizeTestType } from './write-and-test.js';
 import { Logger } from '../utils/logger.js';
 
 export async function autoDevCommand(options = {}) {
@@ -71,13 +73,25 @@ export async function autoDevCommand(options = {}) {
   }
 
   // ============================================================
-  // 第四步：詢問目標檔案路徑（若未指定）
+  // 第四步：決定目標檔案路徑（優先自動推斷）
   // ============================================================
   if (!filePath) {
+    filePath = await inferTargetFilePath(issue, plan, options.model);
+    if (filePath) {
+      logger.info(`已自動推斷目標檔案：${filePath}`);
+    }
+  }
+
+  if (!filePath) {
+    if (noConfirm) {
+      logger.error('無法自動推斷目標檔案路徑，請改用 --file 明確指定。');
+      throw new Error('缺少可推斷的目標檔案路徑');
+    }
+
     const { file } = await inquirer.prompt([{
       type: 'input',
       name: 'file',
-      message: '請輸入要生成的檔案路徑（例如：src/core/new-feature.js）：',
+      message: '無法自動推斷，請輸入要生成的檔案路徑（例如：src/core/new-feature.js）：',
       validate: (input) => input.trim().length > 0 ? true : '路徑不能為空',
     }]);
     filePath = file;
@@ -94,7 +108,7 @@ export async function autoDevCommand(options = {}) {
   try {
     codeResult = await CodeGenerator.generateFile(issue, absoluteFilePath, {
       maxLines: parseInt(options.maxLines || '500', 10),
-      language: filePath.endsWith('.ts') ? 'ts' : 'js',
+      language: getLanguageFromFilePath(filePath),
       extraContext: options.context || '',
       model: options.model,
     });
@@ -106,80 +120,42 @@ export async function autoDevCommand(options = {}) {
 
   console.log('');
 
-  // ============================================================
-  // 第六步：詢問是否執行測試（若非全自動）
-  // ============================================================
-  if (!noConfirm) {
-    const { runTest } = await inquirer.prompt([{
-      type: 'confirm',
-      name: 'runTest',
-      message: '代碼已生成，是否為該檔案生成並執行測試？',
-      default: true,
-    }]);
+  const selectedTestType = await resolveTestTypeForAutoDev({
+    noConfirm,
+    requestedTestType: options.testType,
+  });
 
-    if (!runTest) {
-      logger.success('完成！代碼已生成。可稍後手動運行：');
-      console.log(`  ai-git-tools write-and-test --file ${filePath}`);
-      return;
-    }
-  }
+  const resolvedRuntimeTestType = resolveRuntimeTestType(absoluteFilePath, selectedTestType);
 
-  // ============================================================
-  // 第七步：生成與執行測試（自動修復最多 2 次）
-  // ============================================================
-  const testFilePath = deriveTestFilePath(absoluteFilePath);
-  const maxFixes = 2;
-
-  logger.step('正在生成測試...');
-  await TestGenerator.generateTests(
-    absoluteFilePath,
-    testFilePath,
-    options.testType || 'unit',
-    options.model
-  );
-  logger.success(`測試已生成：${testFilePath}`);
+  logger.step('正在驗證專案可用的測試腳本...');
+  ensureJestAvailable(resolvedRuntimeTestType);
+  logger.success('已確認可執行 Jest 測試。');
   console.log('');
 
-  let lastErrors = [];
-  for (let attempt = 0; attempt <= maxFixes; attempt++) {
-    if (attempt > 0) {
-      logger.step(`進行第 ${attempt}/${maxFixes} 次自動修復...`);
-      await TestGenerator.generateFix(absoluteFilePath, lastErrors, attempt, options.model);
-      logger.success('修復代碼已更新。');
-    }
-
-    logger.step(`執行測試${attempt > 0 ? `（第 ${attempt} 次修復後）` : ''}...`);
-    const result = await TestRunner.runTests(testFilePath);
-
-    if (result.success) {
-      logger.success('所有測試通過！ 🎉');
-      console.log('');
-      break;
-    }
+  // ============================================================
+  // 第六步：強制執行測試流程
+  // ============================================================
+  const testResult = await writeAndTestCommand({
+    file: filePath,
+    testType: selectedTestType,
+    maxFixes: options.maxFixes || '2',
+    model: options.model,
+    noConfirm: true,
+    skipCommit: true,
+  });
 
-    lastErrors = result.errors;
-    if (attempt < maxFixes) {
-      logger.warning(`測試失敗（${result.errors.length} 個錯誤），嘗試自動修復...`);
-    } else {
-      logger.warning(`測試在第 ${maxFixes} 次修復後仍未通過`);
-      console.log('');
-      lastErrors.slice(0, 2).forEach((e, i) => {
-        console.log(`\n[錯誤 ${i + 1}]`);
-        console.log(e.split('\n').slice(0, 8).join('\n'));
-      });
-      console.log('');
-      logger.error('自動修復失敗，需要手動介入。');
-      return;
-    }
+  if (!testResult?.success) {
+    logger.error('測試流程未完成，自動化工作流中止。');
+    return;
   }
 
   // ============================================================
-  // 第八步：自動 Commit
+  // 第七步：自動 Commit
   // ============================================================
   logger.step('正在自動提交...');
   try {
-    GitOperations.exec(`git add "${absoluteFilePath}" "${testFilePath}"`, { silent: true });
-    const commitMsg = `feat(auto-dev): Issue #${issue.number} - ${issue.title}\n\n生成檔案：\n- ${filePath}\n- ${testFilePath.replace(process.cwd(), '.')}`;
+    GitOperations.exec(`git add "${absoluteFilePath}" "${testResult.testFilePath}"`, { silent: true });
+    const commitMsg = `feat(auto-dev): Issue #${issue.number} - ${issue.title}\n\n生成檔案：\n- ${filePath}\n- ${testResult.testFilePath.replace(process.cwd(), '.')}`;
     GitOperations.exec(`git commit -m "${commitMsg.split('\n')[0]}"`, { silent: true });
     logger.success('已自動提交');
   } catch {
@@ -190,7 +166,8 @@ export async function autoDevCommand(options = {}) {
   logger.section('✅ 自動化工作流完成');
   console.log(`Issue #${issue.number}：${issue.title}`);
   console.log(`生成檔案：${filePath}`);
-  console.log(`測試檔案：${testFilePath.replace(process.cwd(), '.')}`);
+  console.log(`測試檔案：${testResult.testFilePath.replace(process.cwd(), '.')}`);
+  console.log(`測試類型：${testResult.testType}`);
   console.log(`合計行數：${codeResult.linesCount + ' (程式碼)'}`);
 }
 
@@ -220,11 +197,144 @@ ${issue.labels.length ? `標籤：${issue.labels.join(', ')}` : ''}
 }
 
 /**
- * 依據原始檔案路徑推導測試檔案路徑
+ * 優先以規則與計畫內容推斷最可能的目標檔案
  */
-function deriveTestFilePath(sourceFilePath) {
-  const dir = dirname(sourceFilePath);
-  const name = basename(sourceFilePath).replace(/\.(js|ts|mjs|cjs)$/, '');
-  const ext = sourceFilePath.endsWith('.ts') ? '.ts' : '.js';
-  return join(dir, '__tests__', `${name}.test${ext}`);
+async function inferTargetFilePath(issue, plan, model) {
+  const planCandidates = extractFileCandidates(plan);
+  if (planCandidates.length === 1) {
+    return planCandidates[0];
+  }
+
+  if (planCandidates.length > 1) {
+    const preferredCandidate = planCandidates.find((candidate) => candidate.startsWith('src/'));
+    if (preferredCandidate) {
+      return preferredCandidate;
+    }
+    return planCandidates[0];
+  }
+
+  const prompt = `請根據以下 GitHub Issue 與實作計畫，推斷「最主要的實作檔案路徑」。
+
+## 規則
+- 只回傳單一檔案路徑
+- 優先回傳 src/ 底下的實作檔，不要回傳測試檔、文件檔、設定檔
+- 副檔名限制為 .js、.ts、.jsx、.tsx、.mjs、.cjs
+- 不要包含程式碼區塊、不要加說明
+
+## Issue #${issue.number}
+標題：${issue.title}
+描述：
+${issue.body || '（無描述）'}
+
+## 實作計畫
+${plan}`.trim();
+
+  const inferred = (await AIClient.sendAndWait(prompt, model)).trim().split(/\s+/)[0];
+  return isValidTargetPath(inferred) ? inferred : null;
+}
+
+function extractFileCandidates(plan) {
+  const matches = plan.match(/([A-Za-z0-9_./-]+\.(?:js|ts|jsx|tsx|mjs|cjs))/g) || [];
+  const uniqueMatches = [...new Set(matches.filter(isValidTargetPath))];
+
+  return uniqueMatches.filter((candidate) => {
+    const normalized = candidate.toLowerCase();
+    return !normalized.includes('__tests__/')
+      && !normalized.includes('.test.')
+      && !normalized.includes('.spec.')
+      && !normalized.endsWith('.config.js');
+  });
+}
+
+function isValidTargetPath(value) {
+  return typeof value === 'string'
+    && /^[A-Za-z0-9_./-]+\.(js|ts|jsx|tsx|mjs|cjs)$/.test(value)
+    && !value.startsWith('/');
+}
+
+function getLanguageFromFilePath(filePath) {
+  return /\.(ts|tsx)$/.test(filePath) ? 'ts' : 'js';
+}
+
+async function resolveTestTypeForAutoDev({ noConfirm, requestedTestType }) {
+  if (noConfirm) {
+    return normalizeTestType(requestedTestType || 'auto');
+  }
+
+  const { testType } = await inquirer.prompt([{
+    type: 'list',
+    name: 'testType',
+    message: '請選擇這次要執行的測試方式：',
+    choices: TEST_TYPE_CHOICES,
+    default: normalizeTestType(requestedTestType || 'auto'),
+  }]);
+
+  return normalizeTestType(testType);
+}
+
+function resolveRuntimeTestType(sourceFilePath, requestedTestType) {
+  if (requestedTestType !== 'auto') {
+    return requestedTestType;
+  }
+
+  const sourceCode = readFileSync(sourceFilePath, 'utf-8');
+  return TestGenerator.resolveTestType(sourceFilePath, sourceCode, 'auto');
+}
+
+function ensureJestAvailable(runtimeTestType = 'unit') {
+  const packageJson = readProjectPackageJson();
+
+  try {
+    execSync('npx jest --version', {
+      encoding: 'utf-8',
+      stdio: 'pipe',
+      cwd: process.cwd(),
+    });
+    ensureComponentTestDependencies(packageJson, runtimeTestType);
+    return;
+  } catch {
+    const hasJestDependency = Boolean(
+      packageJson.dependencies?.jest || packageJson.devDependencies?.jest
+    );
+
+    if (!hasJestDependency) {
+      throw new Error('目前專案未安裝 Jest，請先安裝後再執行 auto-dev。');
+    }
+
+    ensureComponentTestDependencies(packageJson, runtimeTestType);
+
+    throw new Error('偵測到 Jest 依賴，但無法執行 npx jest，請檢查安裝或 lockfile 狀態。');
+  }
+}
+
+function readProjectPackageJson() {
+  const packageJsonPath = resolve(process.cwd(), 'package.json');
+  if (!existsSync(packageJsonPath)) {
+    throw new Error('目前目錄沒有 package.json，無法執行 Jest 測試。');
+  }
+
+  try {
+    return JSON.parse(readFileSync(packageJsonPath, 'utf-8'));
+  } catch {
+    throw new Error('無法讀取目前專案的 package.json，請確認 Jest 已安裝。');
+  }
+}
+
+function ensureComponentTestDependencies(packageJson, runtimeTestType) {
+  if (runtimeTestType !== 'component') {
+    return;
+  }
+
+  const hasJsdomDependency = Boolean(
+    packageJson.devDependencies?.['jest-environment-jsdom']
+    || packageJson.dependencies?.['jest-environment-jsdom']
+  );
+  const hasTestingLibraryDependency = Boolean(
+    packageJson.devDependencies?.['@testing-library/react']
+    || packageJson.dependencies?.['@testing-library/react']
+  );
+
+  if (!hasJsdomDependency || !hasTestingLibraryDependency) {
+    throw new Error('元件測試需要 jest-environment-jsdom 與 @testing-library/react，請先安裝後再執行 auto-dev。');
+  }
 }

package/src/commands/generate-code.js

@@ -8,6 +8,7 @@ import { resolve } from 'path';
 import inquirer from 'inquirer';
 import { IssueReader } from '../core/issue-reader.js';
 import { CodeGenerator } from '../core/code-generator.js';
+import { writeAndTestCommand, TEST_TYPE_CHOICES } from './write-and-test.js';
 import { Logger } from '../utils/logger.js';
 
 export async function generateCodeCommand(options = {}) {
@@ -61,7 +62,7 @@ export async function generateCodeCommand(options = {}) {
 
   const result = await CodeGenerator.generateFile(issue, absolutePath, {
     maxLines,
-    language: absolutePath.endsWith('.ts') ? 'ts' : 'js',
+    language: /\.(ts|tsx)$/.test(absolutePath) ? 'ts' : 'js',
     extraContext,
     model: options.model,
   });
@@ -84,5 +85,31 @@ export async function generateCodeCommand(options = {}) {
   console.log('\n💡 下一步：執行 write-and-test 為此檔案生成並執行測試');
   console.log(`   ai-git-tools write-and-test --file ${filePath}`);
 
+  if (!options.noConfirm) {
+    const { nextAction } = await inquirer.prompt([{
+      type: 'list',
+      name: 'nextAction',
+      message: '代碼已生成，接下來要如何處理測試？',
+      choices: [
+        ...TEST_TYPE_CHOICES,
+        {
+          name: '先不執行測試',
+          value: 'skip',
+        },
+      ],
+      default: 'auto',
+    }]);
+
+    if (nextAction !== 'skip') {
+      await writeAndTestCommand({
+        file: filePath,
+        testType: nextAction,
+        maxFixes: options.maxFixes || '2',
+        model: options.model,
+        noConfirm: true,
+      });
+    }
+  }
+
   return result;
 }

package/src/commands/write-and-test.js

@@ -11,11 +11,44 @@ import { TestGenerator } from '../core/test-generator.js';
 import { TestRunner } from '../core/test-runner.js';
 import { Logger } from '../utils/logger.js';
 
+export const TEST_TYPE_CHOICES = [
+  {
+    name: '寫測試（自動判斷 Jest 單元測試 / 元件測試）',
+    value: 'auto',
+  },
+  {
+    name: '寫 Jest 單元測試',
+    value: 'unit',
+  },
+  {
+    name: '寫元件測試',
+    value: 'component',
+  },
+];
+
+export function normalizeTestType(testType = 'auto') {
+  const normalized = String(testType).trim().toLowerCase();
+
+  if (normalized === 'auto' || normalized === 'unit' || normalized === 'component') {
+    return normalized;
+  }
+
+  if (normalized === 'jest' || normalized === 'jest-unit' || normalized === 'unit-test') {
+    return 'unit';
+  }
+
+  if (normalized === 'component-test' || normalized === 'react-component') {
+    return 'component';
+  }
+
+  throw new Error(`不支援的測試類型：${testType}`);
+}
+
 export async function writeAndTestCommand(options = {}) {
   const logger = new Logger();
 
   const filePath = options.file;
-  const testType = options.testType || 'unit';
+  const requestedTestType = normalizeTestType(options.testType || 'auto');
   const maxFixes = parseInt(options.maxFixes || '2', 10);
 
   if (!filePath) {
@@ -35,27 +68,35 @@ export async function writeAndTestCommand(options = {}) {
   logger.header('write-and-test');
   console.log(`原始碼  ：${absoluteSourcePath}`);
   console.log(`測試檔案：${testFilePath}`);
-  console.log(`測試類型：${testType}`);
+  console.log(`測試類型：${requestedTestType}`);
   console.log(`最大修復：${maxFixes} 次`);
   console.log('');
 
   // 1. 詢問是否繼續
-  const { confirmed } = await inquirer.prompt([{
-    type: 'confirm',
-    name: 'confirmed',
-    message: `確認為 ${filePath} 生成 ${testType} 測試並執行？`,
-    default: true,
-  }]);
-
-  if (!confirmed) {
-    logger.info('已取消。');
-    return;
+  if (!options.noConfirm) {
+    const { confirmed } = await inquirer.prompt([{
+      type: 'confirm',
+      name: 'confirmed',
+      message: `確認為 ${filePath} 生成 ${requestedTestType} 測試並執行？`,
+      default: true,
+    }]);
+
+    if (!confirmed) {
+      logger.info('已取消。');
+      return;
+    }
   }
 
   // 2. 生成測試
   logger.step('正在生成測試代碼...');
-  await TestGenerator.generateTests(absoluteSourcePath, testFilePath, testType, options.model);
+  const generatedTest = await TestGenerator.generateTests(
+    absoluteSourcePath,
+    testFilePath,
+    requestedTestType,
+    options.model
+  );
   logger.success(`測試已寫入：${testFilePath}`);
+  logger.info(`實際測試類型：${generatedTest.testType}`);
 
   // 3. 執行測試 + 自動修復循環
   let lastErrors = [];
@@ -73,8 +114,10 @@ export async function writeAndTestCommand(options = {}) {
       logger.success('所有測試通過！ 🎉');
 
       // 4. 自動 commit 測試檔案
-      await commitTestFile(testFilePath, absoluteSourcePath, logger);
-      return { success: true, testFilePath };
+      if (!options.skipCommit) {
+        await commitGeneratedFiles(testFilePath, absoluteSourcePath, logger);
+      }
+      return { success: true, testFilePath, testType: generatedTest.testType };
     }
 
     lastErrors = result.errors;
@@ -97,7 +140,12 @@ export async function writeAndTestCommand(options = {}) {
       console.log('');
       console.log(`原始碼路徑：${absoluteSourcePath}`);
       console.log(`測試路徑  ：${testFilePath}`);
-      return { success: false, testFilePath, errors: lastErrors };
+      return {
+        success: false,
+        testFilePath,
+        errors: lastErrors,
+        testType: generatedTest.testType,
+      };
     }
   }
 }
@@ -106,23 +154,23 @@ export async function writeAndTestCommand(options = {}) {
  * 依據原始檔案路徑推導測試檔案路徑
  * e.g. src/core/foo.js → src/core/__tests__/foo.test.js
  */
-function deriveTestFilePath(sourceFilePath) {
+export function deriveTestFilePath(sourceFilePath) {
   const dir = dirname(sourceFilePath);
-  const name = basename(sourceFilePath).replace(/\.(js|ts|mjs|cjs)$/, '');
-  const ext = sourceFilePath.endsWith('.ts') ? '.ts' : '.js';
+  const name = basename(sourceFilePath).replace(/\.(js|jsx|ts|tsx|mjs|cjs)$/, '');
+  const ext = /\.(ts|tsx)$/.test(sourceFilePath) ? '.ts' : '.js';
   return join(dir, '__tests__', `${name}.test${ext}`);
 }
 
 /**
- * 將測試檔案 commit 到 git
+ * 將原始碼與測試檔案一併 commit 到 git
  */
-async function commitTestFile(testFilePath, sourceFilePath, logger) {
+async function commitGeneratedFiles(testFilePath, sourceFilePath, logger) {
   try {
-    execSync(`git add "${testFilePath}"`, { stdio: 'pipe' });
+    execSync(`git add "${sourceFilePath}" "${testFilePath}"`, { stdio: 'pipe' });
     const name = basename(sourceFilePath);
     execSync(`git commit -m "test: 為 ${name} 新增自動生成的測試"`, { stdio: 'pipe' });
-    logger.success('測試檔案已自動 commit。');
-  } catch (error) {
-    logger.warning('無法自動 commit 測試檔案（可能沒有 staged 變更或不在 git 倉庫中）。');
+    logger.success('原始碼與測試檔案已自動 commit。');
+  } catch {
+    logger.warning('無法自動 commit 產生的檔案（可能沒有 staged 變更或不在 git 倉庫中）。');
   }
 }

package/src/core/test-generator.js

@@ -12,11 +12,11 @@ export class TestGenerator {
    * 為原始檔案生成測試並寫入測試檔案
    * @param {string} sourceFilePath - 原始碼路徑（絕對路徑）
    * @param {string} testFilePath - 測試檔案輸出路徑（絕對路徑）
-   * @param {'unit'|'component'} testType - 測試類型（預設 'unit'）
+   * @param {'auto'|'unit'|'component'} testType - 測試類型（預設 'auto'）
    * @param {string} [model] - AI 模型（可選）
-   * @returns {Promise<{ testFilePath, linesCount }>}
+   * @returns {Promise<{ testFilePath, linesCount, testType: 'unit'|'component' }>}
    */
-  static async generateTests(sourceFilePath, testFilePath, testType = 'unit', model) {
+  static async generateTests(sourceFilePath, testFilePath, testType = 'auto', model) {
     let sourceCode;
     try {
       sourceCode = readFileSync(sourceFilePath, 'utf-8');
@@ -24,15 +24,47 @@ export class TestGenerator {
       throw new Error(`無法讀取原始碼：${sourceFilePath}`);
     }
 
-    const prompt = TestGenerator._buildPrompt(sourceCode, sourceFilePath, testType);
+    const resolvedTestType = TestGenerator.resolveTestType(sourceFilePath, sourceCode, testType);
+    const prompt = TestGenerator._buildPrompt(sourceCode, sourceFilePath, resolvedTestType);
     const rawTests = await AIClient.sendAndWait(prompt, model);
-    const testCode = TestGenerator._cleanCode(rawTests);
+    const testCode = TestGenerator._finalizeTestCode(
+      TestGenerator._cleanCode(rawTests),
+      resolvedTestType
+    );
 
     // 確保目錄存在
     mkdirSync(dirname(testFilePath), { recursive: true });
     writeFileSync(testFilePath, testCode, 'utf-8');
 
-    return { testFilePath, linesCount: testCode.split('\n').length };
+    return {
+      testFilePath,
+      linesCount: testCode.split('\n').length,
+      testType: resolvedTestType,
+    };
+  }
+
+  /**
+   * 解析最終測試類型
+   * @param {string} sourceFilePath
+   * @param {string} sourceCode
+   * @param {'auto'|'unit'|'component'} requestedTestType
+   * @returns {'unit'|'component'}
+   */
+  static resolveTestType(sourceFilePath, sourceCode, requestedTestType = 'auto') {
+    if (requestedTestType === 'unit' || requestedTestType === 'component') {
+      return requestedTestType;
+    }
+
+    const normalizedPath = sourceFilePath.toLowerCase();
+    const looksLikeComponentPath = /\.(jsx|tsx)$/.test(normalizedPath)
+      || /component|page|view|screen/.test(normalizedPath);
+    const hasReactImport = /from\s+['"]react['"]|from\s+['"]react-dom['"]/.test(sourceCode);
+    const hasTestingLibraryHint = /use(State|Effect|Memo|Callback|Reducer|Ref|Context)|React\./.test(sourceCode);
+    const hasJsx = /<([A-Z][\w]*|[a-z][\w-]*)(\s[^>]*)?>/.test(sourceCode);
+
+    return looksLikeComponentPath || hasReactImport || hasTestingLibraryHint || hasJsx
+      ? 'component'
+      : 'unit';
   }
 
   /**
@@ -75,11 +107,14 @@ ${sourceCode}
 3. Mock 所有外部依賴（檔案系統、網路、子程序等）
 4. 使用 beforeEach / afterEach 管理測試狀態
 5. 測試名稱使用繁體中文描述
+${testType === 'component'
+      ? '6. 使用 @testing-library/react 撰寫互動與渲染測試\n7. 檔案最前面必須加上 /** @jest-environment jsdom */ 註解'
+      : '6. 以純邏輯與副作用隔離為主，避免使用 DOM API'}
 
 ## 輸出規則
 - 只輸出可直接執行的測試程式碼
 - 不要 markdown 區塊（\`\`\`）、不要任何說明文字
-- 第一行必須是 import 或 require 陳述式`.trim();
+- 第一行必須是 /** @jest-environment jsdom */ 或 import / require 陳述式`.trim();
   }
 
   /**
@@ -112,4 +147,19 @@ ${testErrors.join('\n')}
       .replace(/\n?```$/gm, '')
       .trim();
   }
+
+  /**
+   * 針對元件測試補上必要的 jsdom 設定
+   */
+  static _finalizeTestCode(testCode, testType) {
+    if (testType !== 'component') {
+      return testCode;
+    }
+
+    if (testCode.startsWith('/** @jest-environment jsdom */')) {
+      return testCode;
+    }
+
+    return `/** @jest-environment jsdom */\n${testCode}`;
+  }
 }

package/src/core/test-runner.js

@@ -5,6 +5,7 @@
 
 import { execSync } from 'child_process';
 import { existsSync } from 'fs';
+import { resolve } from 'path';
 
 export class TestRunner {
   /**
@@ -18,10 +19,12 @@ export class TestRunner {
     }
 
     try {
-      execSync(`npx jest "${testFilePath}" --no-coverage --forceExit`, {
+      const absoluteTestFilePath = resolve(testFilePath);
+      execSync(`npx jest --runInBand --runTestsByPath "${absoluteTestFilePath}" --no-coverage`, {
         encoding: 'utf-8',
         stdio: 'pipe',
         timeout: 120000, // 2 分鐘超時
+        cwd: process.cwd(),
       });
       return { success: true, errors: [] };
     } catch (error) {