claude-coder 1.6.2 → 1.7.1

package/src/runner.js

@@ -1,394 +1,396 @@
-'use strict';
-
-const fs = require('fs');
-const path = require('path');
-const readline = require('readline');
-const { execSync } = require('child_process');
-const { paths, log, COLOR, loadConfig, ensureLoopDir, getProjectRoot } = require('./config');
-const { loadTasks, getFeatures, getStats, findNextTask, forceStatus } = require('./tasks');
-const { validate } = require('./validator');
-const { scan } = require('./scanner');
-const { loadSDK, runCodingSession, runAddSession } = require('./session');
-
-const MAX_RETRY = 3;
-
-function sleep(ms) {
-  return new Promise(resolve => setTimeout(resolve, ms));
-}
-
-function getHead() {
-  try {
-    return execSync('git rev-parse HEAD', { cwd: getProjectRoot(), encoding: 'utf8' }).trim();
-  } catch {
-    return 'none';
-  }
-}
-
-function allTasksDone() {
-  const data = loadTasks();
-  if (!data) return false;
-  const features = getFeatures(data);
-  if (features.length === 0) return true;
-  return features.every(f => f.status === 'done');
-}
-
-function killServicesByProfile() {
-  const p = paths();
-  if (!fs.existsSync(p.profile)) return;
-  try {
-    const profile = JSON.parse(fs.readFileSync(p.profile, 'utf8'));
-    const services = profile.services || [];
-    const ports = services.map(s => s.port).filter(Boolean);
-    if (ports.length === 0) return;
-
-    const isWin = process.platform === 'win32';
-    for (const port of ports) {
-      try {
-        if (isWin) {
-          const out = execSync(`netstat -ano | findstr :${port} | findstr LISTENING`, { encoding: 'utf8', stdio: 'pipe' }).trim();
-          const pids = [...new Set(out.split('\n').map(l => l.trim().split(/\s+/).pop()).filter(Boolean))];
-          for (const pid of pids) { try { execSync(`taskkill /F /PID ${pid}`, { stdio: 'pipe' }); } catch { /* ignore */ } }
-        } else {
-          execSync(`lsof -ti :${port} | xargs kill -9 2>/dev/null`, { stdio: 'pipe' });
-        }
-      } catch { /* no process on port */ }
-    }
-    log('info', `已停止端口 ${ports.join(', ')} 上的服务`);
-  } catch { /* ignore profile read errors */ }
-}
-
-async function rollback(headBefore, reason) {
-  if (!headBefore || headBefore === 'none') return;
-
-  killServicesByProfile();
-
-  if (process.platform === 'win32') await sleep(1500);
-
-  const cwd = getProjectRoot();
-  const gitEnv = { ...process.env, GIT_TERMINAL_PROMPT: '0' };
-
-  log('warn', `回滚到 ${headBefore} ...`);
-
-  let success = false;
-  for (let attempt = 1; attempt <= 2; attempt++) {
-    try {
-      execSync(`git reset --hard ${headBefore}`, { cwd, stdio: 'pipe', env: gitEnv });
-      log('ok', '回滚完成');
-      success = true;
-      break;
-    } catch (err) {
-      if (attempt === 1) {
-        log('warn', `回滚首次失败，等待后重试: ${err.message}`);
-        await sleep(2000);
-      } else {
-        log('error', `回滚失败: ${err.message}`);
-      }
-    }
-  }
-
-  appendProgress({
-    type: 'rollback',
-    timestamp: new Date().toISOString(),
-    reason: reason || 'harness 校验失败',
-    rollbackTo: headBefore,
-    success,
-  });
-}
-
-function markTaskFailed() {
-  const data = loadTasks();
-  if (!data) return;
-  const result = forceStatus(data, 'failed');
-  if (result) {
-    log('warn', `已将任务 ${result.id} 强制标记为 failed`);
-  }
-}
-
-function tryPush() {
-  try {
-    const remotes = execSync('git remote', { cwd: getProjectRoot(), encoding: 'utf8' }).trim();
-    if (!remotes) return;
-    log('info', '正在推送代码...');
-    execSync('git push', { cwd: getProjectRoot(), stdio: 'inherit' });
-    log('ok', '推送成功');
-  } catch {
-    log('warn', '推送失败 (请检查网络或权限)，继续执行...');
-  }
-}
-
-function appendProgress(entry) {
-  const p = paths();
-  let progress = { sessions: [] };
-  if (fs.existsSync(p.progressFile)) {
-    try {
-      const text = fs.readFileSync(p.progressFile, 'utf8');
-      progress = JSON.parse(text);
-    } catch { /* reset */ }
-  }
-  if (!Array.isArray(progress.sessions)) progress.sessions = [];
-  progress.sessions.push(entry);
-  fs.writeFileSync(p.progressFile, JSON.stringify(progress, null, 2) + '\n', 'utf8');
-}
-
-function printStats() {
-  const data = loadTasks();
-  if (!data) return;
-  const stats = getStats(data);
-  log('info', `进度: ${stats.done}/${stats.total} done, ${stats.in_progress} in_progress, ${stats.testing} testing, ${stats.failed} failed, ${stats.pending} pending`);
-}
-
-async function promptContinue() {
-  if (!process.stdin.isTTY) return true;
-  const rl = readline.createInterface({ input: process.stdin, output: process.stdout });
-  return new Promise(resolve => {
-    rl.question('是否继续？(y/n) ', answer => {
-      rl.close();
-      resolve(/^[Yy]/.test(answer.trim()));
-    });
-  });
-}
-
-async function run(requirement, opts = {}) {
-  const p = paths();
-  const projectRoot = getProjectRoot();
-  ensureLoopDir();
-
-  const maxSessions = opts.max || 50;
-  const pauseEvery = opts.pause ?? 0;
-  const dryRun = opts.dryRun || false;
-
-  console.log('');
-  console.log('============================================');
-  console.log(`  Claude Coder${dryRun ? ' (预览模式)' : ''}`);
-  console.log('============================================');
-  console.log('');
-
-  const config = loadConfig();
-  if (config.provider !== 'claude' && config.baseUrl) {
-    log('ok', `模型配置已加载: ${config.provider}${config.model ? ` (${config.model})` : ''}`);
-  }
-
-  const reqFile = path.join(projectRoot, 'requirements.md');
-  if (fs.existsSync(reqFile) && !requirement) {
-    requirement = fs.readFileSync(reqFile, 'utf8');
-    log('ok', '已读取需求文件: requirements.md');
-  }
-
-  try {
-    execSync('git rev-parse --is-inside-work-tree', { cwd: projectRoot, stdio: 'ignore' });
-  } catch {
-    log('info', '初始化 git 仓库...');
-    execSync('git init', { cwd: projectRoot, stdio: 'inherit' });
-    execSync('git add -A && git commit -m "init: 项目初始化" --allow-empty', {
-      cwd: projectRoot,
-      stdio: 'inherit',
-    });
-  }
-
-  if (!fs.existsSync(p.profile) || !fs.existsSync(p.tasksFile)) {
-    if (!requirement) {
-      log('error', '首次运行需要提供需求描述');
-      console.log('');
-      console.log('用法（二选一）:');
-      console.log('  方式 1: 在项目根目录创建 requirements.md');
-      console.log('          claude-coder run');
-      console.log('');
-      console.log('  方式 2: 直接传入一句话需求');
-      console.log('          claude-coder run "你的需求描述"');
-      process.exit(1);
-    }
-
-    if (dryRun) {
-      log('info', '[DRY-RUN] 将执行初始化扫描（跳过）');
-      const reqPreview = (requirement || '').slice(0, 100);
-      log('info', `[DRY-RUN] 需求: ${reqPreview}${reqPreview.length >= 100 ? '...' : ''}`);
-      return;
-    }
-
-    await loadSDK();
-    const scanResult = await scan(requirement, { projectRoot });
-    if (!scanResult.success) {
-      console.log('');
-      console.log(`${COLOR.yellow}═══════════════════════════════════════════════${COLOR.reset}`);
-      console.log(`${COLOR.yellow}  若出现 "Credit balance is too low"，请运行:${COLOR.reset}`);
-      console.log(`  ${COLOR.green}claude-coder setup${COLOR.reset}`);
-      console.log(`${COLOR.yellow}═══════════════════════════════════════════════${COLOR.reset}`);
-      process.exit(1);
-    }
-  } else {
-    log('ok', '检测到已有 project_profile.json + tasks.json，跳过初始化');
-    printStats();
-  }
-
-  if (!dryRun) await loadSDK();
-  log('info', `开始编码循环 (最多 ${maxSessions} 个会话) ...`);
-  console.log('');
-
-  let consecutiveFailures = 0;
-
-  for (let session = 1; session <= maxSessions; session++) {
-    console.log('');
-    console.log('--------------------------------------------');
-    log('info', `Session ${session} / ${maxSessions}`);
-    console.log('--------------------------------------------');
-
-    const taskData = loadTasks();
-    if (!taskData) {
-      log('error', 'tasks.json 无法读取，终止循环');
-      break;
-    }
-
-    if (allTasksDone()) {
-      console.log('');
-      log('ok', '所有任务已完成！');
-      printStats();
-      break;
-    }
-
-    printStats();
-
-    if (dryRun) {
-      const next = findNextTask(loadTasks());
-      log('info', `[DRY-RUN] 下一个任务: ${next ? `${next.id} - ${next.description}` : '无'}`);
-      if (!next) break;
-      continue;
-    }
-
-    const headBefore = getHead();
-    const nextTask = findNextTask(taskData);
-    const taskId = nextTask?.id || 'unknown';
-
-    const sessionResult = await runCodingSession(session, {
-      projectRoot,
-      taskId,
-      consecutiveFailures,
-      maxSessions,
-      lastValidateLog: consecutiveFailures > 0 ? '上次校验失败' : '',
-    });
-
-    if (sessionResult.stalled) {
-      log('warn', `Session ${session} 因停顿超时中断，跳过校验直接重试`);
-      consecutiveFailures++;
-      await rollback(headBefore, '停顿超时');
-      if (consecutiveFailures >= MAX_RETRY) {
-        log('error', `连续失败 ${MAX_RETRY} 次，跳过当前任务`);
-        markTaskFailed();
-        consecutiveFailures = 0;
-      }
-      appendProgress({
-        session,
-        timestamp: new Date().toISOString(),
-        result: 'stalled',
-        cost: sessionResult.cost,
-        taskId,
-      });
-      continue;
-    }
-
-    log('info', '开始 harness 校验 ...');
-    const validateResult = await validate(headBefore, taskId);
-
-    if (!validateResult.fatal) {
-      if (validateResult.hasWarnings) {
-        log('warn', `Session ${session} 校验通过 (有自动修复或警告)`);
-      } else {
-        log('ok', `Session ${session} 校验通过`);
-      }
-      tryPush();
-      consecutiveFailures = 0;
-
-      appendProgress({
-        session,
-        timestamp: new Date().toISOString(),
-        result: 'success',
-        cost: sessionResult.cost,
-        taskId,
-        statusAfter: validateResult.sessionData?.status_after || null,
-        notes: validateResult.sessionData?.notes || null,
-      });
-
-    } else {
-      consecutiveFailures++;
-      log('error', `Session ${session} 校验失败 (连续失败: ${consecutiveFailures}/${MAX_RETRY})`);
-
-      appendProgress({
-        session,
-        timestamp: new Date().toISOString(),
-        result: 'fatal',
-        cost: sessionResult.cost,
-        taskId,
-        reason: validateResult.sessionData?.reason || '校验失败',
-      });
-
-      await rollback(headBefore, '校验失败');
-
-      if (consecutiveFailures >= MAX_RETRY) {
-        log('error', `连续失败 ${MAX_RETRY} 次，跳过当前任务`);
-        markTaskFailed();
-        consecutiveFailures = 0;
-        log('warn', '已将任务标记为 failed，继续下一个任务');
-      }
-    }
-
-    if (pauseEvery > 0 && session % pauseEvery === 0) {
-      console.log('');
-      printStats();
-      const shouldContinue = await promptContinue();
-      if (!shouldContinue) {
-        log('info', '手动停止');
-        break;
-      }
-    }
-  }
-
-  killServicesByProfile();
-
-  console.log('');
-  console.log('============================================');
-  console.log('  运行结束');
-  console.log('============================================');
-  console.log('');
-  printStats();
-}
-
-async function add(instruction, opts = {}) {
-  await loadSDK();
-  const p = paths();
-  const projectRoot = getProjectRoot();
-  ensureLoopDir();
-
-  const config = loadConfig();
-
-  if (!opts.model) {
-    if (config.defaultOpus) {
-      opts.model = config.defaultOpus;
-    } else if (config.model) {
-      opts.model = config.model;
-    }
-  }
-
-  const displayModel = opts.model || config.model || '(default)';
-  log('ok', `模型配置已加载: ${config.provider || 'claude'} (add 使用: ${displayModel})`);
-
-  if (!fs.existsSync(p.profile) || !fs.existsSync(p.tasksFile)) {
-    log('error', 'add 需要先完成初始化（至少运行一次 claude-coder run）');
-    process.exit(1);
-  }
-
-  deployTestRule(p);
-
-  await runAddSession(instruction, { projectRoot, ...opts });
-  printStats();
-}
-
-function deployTestRule(p) {
-  const dest = path.join(p.loopDir, 'test_rule.md');
-  if (fs.existsSync(dest)) return;
-  if (!fs.existsSync(p.testRuleTemplate)) return;
-  try {
-    fs.copyFileSync(p.testRuleTemplate, dest);
-    log('ok', '已部署测试指导规则 → .claude-coder/test_rule.md');
-  } catch { /* ignore */ }
-}
-
-module.exports = { run, add };
+'use strict';
+
+const fs = require('fs');
+const path = require('path');
+const readline = require('readline');
+const { execSync } = require('child_process');
+const { paths, log, COLOR, loadConfig, ensureLoopDir, getProjectRoot } = require('./config');
+const { loadTasks, getFeatures, getStats, findNextTask, forceStatus } = require('./tasks');
+const { validate } = require('./validator');
+const { scan } = require('./scanner');
+const { loadSDK, runCodingSession, runAddSession } = require('./session');
+
+const MAX_RETRY = 3;
+
+function sleep(ms) {
+  return new Promise(resolve => setTimeout(resolve, ms));
+}
+
+function getHead() {
+  try {
+    return execSync('git rev-parse HEAD', { cwd: getProjectRoot(), encoding: 'utf8' }).trim();
+  } catch {
+    return 'none';
+  }
+}
+
+function allTasksDone() {
+  const data = loadTasks();
+  if (!data) return false;
+  const features = getFeatures(data);
+  if (features.length === 0) return true;
+  return features.every(f => f.status === 'done');
+}
+
+function killServicesByProfile() {
+  const p = paths();
+  if (!fs.existsSync(p.profile)) return;
+  try {
+    const profile = JSON.parse(fs.readFileSync(p.profile, 'utf8'));
+    const services = profile.services || [];
+    const ports = services.map(s => s.port).filter(Boolean);
+    if (ports.length === 0) return;
+
+    const isWin = process.platform === 'win32';
+    for (const port of ports) {
+      try {
+        if (isWin) {
+          const out = execSync(`netstat -ano | findstr :${port} | findstr LISTENING`, { encoding: 'utf8', stdio: 'pipe' }).trim();
+          const pids = [...new Set(out.split('\n').map(l => l.trim().split(/\s+/).pop()).filter(Boolean))];
+          for (const pid of pids) { try { execSync(`taskkill /F /PID ${pid}`, { stdio: 'pipe' }); } catch { /* ignore */ } }
+        } else {
+          execSync(`lsof -ti :${port} | xargs kill -9 2>/dev/null`, { stdio: 'pipe' });
+        }
+      } catch { /* no process on port */ }
+    }
+    log('info', `已停止端口 ${ports.join(', ')} 上的服务`);
+  } catch { /* ignore profile read errors */ }
+}
+
+async function rollback(headBefore, reason) {
+  if (!headBefore || headBefore === 'none') return;
+
+  killServicesByProfile();
+
+  if (process.platform === 'win32') await sleep(1500);
+
+  const cwd = getProjectRoot();
+  const gitEnv = { ...process.env, GIT_TERMINAL_PROMPT: '0' };
+
+  log('warn', `回滚到 ${headBefore} ...`);
+
+  let success = false;
+  for (let attempt = 1; attempt <= 2; attempt++) {
+    try {
+      execSync(`git reset --hard ${headBefore}`, { cwd, stdio: 'pipe', env: gitEnv });
+      log('ok', '回滚完成');
+      success = true;
+      break;
+    } catch (err) {
+      if (attempt === 1) {
+        log('warn', `回滚首次失败，等待后重试: ${err.message}`);
+        await sleep(2000);
+      } else {
+        log('error', `回滚失败: ${err.message}`);
+      }
+    }
+  }
+
+  appendProgress({
+    type: 'rollback',
+    timestamp: new Date().toISOString(),
+    reason: reason || 'harness 校验失败',
+    rollbackTo: headBefore,
+    success,
+  });
+}
+
+function markTaskFailed() {
+  const data = loadTasks();
+  if (!data) return;
+  const result = forceStatus(data, 'failed');
+  if (result) {
+    log('warn', `已将任务 ${result.id} 强制标记为 failed`);
+  }
+}
+
+function tryPush() {
+  try {
+    const remotes = execSync('git remote', { cwd: getProjectRoot(), encoding: 'utf8' }).trim();
+    if (!remotes) return;
+    log('info', '正在推送代码...');
+    execSync('git push', { cwd: getProjectRoot(), stdio: 'inherit' });
+    log('ok', '推送成功');
+  } catch {
+    log('warn', '推送失败 (请检查网络或权限)，继续执行...');
+  }
+}
+
+function appendProgress(entry) {
+  const p = paths();
+  let progress = { sessions: [] };
+  if (fs.existsSync(p.progressFile)) {
+    try {
+      const text = fs.readFileSync(p.progressFile, 'utf8');
+      progress = JSON.parse(text);
+    } catch { /* reset */ }
+  }
+  if (!Array.isArray(progress.sessions)) progress.sessions = [];
+  progress.sessions.push(entry);
+  fs.writeFileSync(p.progressFile, JSON.stringify(progress, null, 2) + '\n', 'utf8');
+}
+
+function printStats() {
+  const data = loadTasks();
+  if (!data) return;
+  const stats = getStats(data);
+  log('info', `进度: ${stats.done}/${stats.total} done, ${stats.in_progress} in_progress, ${stats.testing} testing, ${stats.failed} failed, ${stats.pending} pending`);
+}
+
+async function promptContinue() {
+  if (!process.stdin.isTTY) return true;
+  const rl = readline.createInterface({ input: process.stdin, output: process.stdout });
+  return new Promise(resolve => {
+    rl.question('是否继续？(y/n) ', answer => {
+      rl.close();
+      resolve(/^[Yy]/.test(answer.trim()));
+    });
+  });
+}
+
+async function run(opts = {}) {
+  const p = paths();
+  const projectRoot = getProjectRoot();
+  ensureLoopDir();
+
+  const maxSessions = opts.max || 50;
+  const pauseEvery = opts.pause ?? 0;
+  const dryRun = opts.dryRun || false;
+
+  console.log('');
+  console.log('============================================');
+  console.log(`  Claude Coder${dryRun ? ' (预览模式)' : ''}`);
+  console.log('============================================');
+  console.log('');
+
+  const config = loadConfig();
+  if (config.provider !== 'claude' && config.baseUrl) {
+    log('ok', `模型配置已加载: ${config.provider}${config.model ? ` (${config.model})` : ''}`);
+  }
+
+  try {
+    execSync('git rev-parse --is-inside-work-tree', { cwd: projectRoot, stdio: 'ignore' });
+  } catch {
+    log('info', '初始化 git 仓库...');
+    execSync('git init', { cwd: projectRoot, stdio: 'inherit' });
+    execSync('git add -A && git commit -m "init: 项目初始化" --allow-empty', {
+      cwd: projectRoot,
+      stdio: 'inherit',
+    });
+  }
+
+  // 检查前置条件
+  if (!fs.existsSync(p.profile)) {
+    log('error', 'profile.json 不存在，请先运行 claude-coder add 添加任务');
+    process.exit(1);
+  }
+
+  if (!fs.existsSync(p.tasksFile)) {
+    log('error', 'tasks.json 不存在，请先运行 claude-coder add 添加任务');
+    process.exit(1);
+  }
+
+  printStats();
+
+  if (!dryRun) await loadSDK();
+  log('info', `开始编码循环 (最多 ${maxSessions} 个会话) ...`);
+  console.log('');
+
+  let consecutiveFailures = 0;
+
+  for (let session = 1; session <= maxSessions; session++) {
+    console.log('');
+    console.log('--------------------------------------------');
+    log('info', `Session ${session} / ${maxSessions}`);
+    console.log('--------------------------------------------');
+
+    const taskData = loadTasks();
+    if (!taskData) {
+      log('error', 'tasks.json 无法读取，终止循环');
+      break;
+    }
+
+    if (allTasksDone()) {
+      console.log('');
+      log('ok', '所有任务已完成！');
+      printStats();
+      break;
+    }
+
+    printStats();
+
+    if (dryRun) {
+      const next = findNextTask(loadTasks());
+      log('info', `[DRY-RUN] 下一个任务: ${next ? `${next.id} - ${next.description}` : '无'}`);
+      if (!next) break;
+      continue;
+    }
+
+    const headBefore = getHead();
+    const nextTask = findNextTask(taskData);
+    const taskId = nextTask?.id || 'unknown';
+
+    const sessionResult = await runCodingSession(session, {
+      projectRoot,
+      taskId,
+      consecutiveFailures,
+      maxSessions,
+      lastValidateLog: consecutiveFailures > 0 ? '上次校验失败' : '',
+    });
+
+    if (sessionResult.stalled) {
+      log('warn', `Session ${session} 因停顿超时中断，跳过校验直接重试`);
+      consecutiveFailures++;
+      await rollback(headBefore, '停顿超时');
+      if (consecutiveFailures >= MAX_RETRY) {
+        log('error', `连续失败 ${MAX_RETRY} 次，跳过当前任务`);
+        markTaskFailed();
+        consecutiveFailures = 0;
+      }
+      appendProgress({
+        session,
+        timestamp: new Date().toISOString(),
+        result: 'stalled',
+        cost: sessionResult.cost,
+        taskId,
+      });
+      continue;
+    }
+
+    log('info', '开始 harness 校验 ...');
+    const validateResult = await validate(headBefore, taskId);
+
+    if (!validateResult.fatal) {
+      if (validateResult.hasWarnings) {
+        log('warn', `Session ${session} 校验通过 (有自动修复或警告)`);
+      } else {
+        log('ok', `Session ${session} 校验通过`);
+      }
+      tryPush();
+      consecutiveFailures = 0;
+
+      appendProgress({
+        session,
+        timestamp: new Date().toISOString(),
+        result: 'success',
+        cost: sessionResult.cost,
+        taskId,
+        statusAfter: validateResult.sessionData?.status_after || null,
+        notes: validateResult.sessionData?.notes || null,
+      });
+
+    } else {
+      consecutiveFailures++;
+      log('error', `Session ${session} 校验失败 (连续失败: ${consecutiveFailures}/${MAX_RETRY})`);
+
+      appendProgress({
+        session,
+        timestamp: new Date().toISOString(),
+        result: 'fatal',
+        cost: sessionResult.cost,
+        taskId,
+        reason: validateResult.sessionData?.reason || '校验失败',
+      });
+
+      await rollback(headBefore, '校验失败');
+
+      if (consecutiveFailures >= MAX_RETRY) {
+        log('error', `连续失败 ${MAX_RETRY} 次，跳过当前任务`);
+        markTaskFailed();
+        consecutiveFailures = 0;
+        log('warn', '已将任务标记为 failed，继续下一个任务');
+      }
+    }
+
+    if (pauseEvery > 0 && session % pauseEvery === 0) {
+      console.log('');
+      printStats();
+      const shouldContinue = await promptContinue();
+      if (!shouldContinue) {
+        log('info', '手动停止');
+        break;
+      }
+    }
+  }
+
+  killServicesByProfile();
+
+  console.log('');
+  console.log('============================================');
+  console.log('  运行结束');
+  console.log('============================================');
+  console.log('');
+  printStats();
+}
+
+async function promptAutoRun() {
+  if (!process.stdin.isTTY) return false;
+  const rl = readline.createInterface({ input: process.stdin, output: process.stdout });
+  return new Promise(resolve => {
+    rl.question('任务分解完成后是否自动开始执行？(y/n) ', answer => {
+      rl.close();
+      resolve(/^[Yy]/.test(answer.trim()));
+    });
+  });
+}
+
+async function add(instruction, opts = {}) {
+  await loadSDK();
+  const p = paths();
+  const projectRoot = getProjectRoot();
+  ensureLoopDir();
+
+  const config = loadConfig();
+
+  if (!opts.model) {
+    if (config.defaultOpus) {
+      opts.model = config.defaultOpus;
+    } else if (config.model) {
+      opts.model = config.model;
+    }
+  }
+
+  const displayModel = opts.model || config.model || '(default)';
+  log('ok', `模型配置已加载: ${config.provider || 'claude'} (add 使用: ${displayModel})`);
+
+  // 如果 profile 不存在，先执行项目扫描
+  if (!fs.existsSync(p.profile)) {
+    log('info', '首次使用，正在执行项目扫描...');
+    const scanResult = await scan(instruction, { projectRoot });
+    if (!scanResult.success) {
+      console.log('');
+      console.log(`${COLOR.yellow}═══════════════════════════════════════════════${COLOR.reset}`);
+      console.log(`${COLOR.yellow}  若出现 "Credit balance is too low"，请运行:${COLOR.reset}`);
+      console.log(`  ${COLOR.green}claude-coder setup${COLOR.reset}`);
+      console.log(`${COLOR.yellow}═══════════════════════════════════════════════${COLOR.reset}`);
+      process.exit(1);
+    }
+  }
+
+  // 询问用户是否在完成后自动运行
+  const shouldAutoRun = await promptAutoRun();
+
+  deployTestRule(p);
+
+  await runAddSession(instruction, { projectRoot, ...opts });
+  printStats();
+
+  // 如果用户选择自动运行，则调用 run()
+  if (shouldAutoRun) {
+    console.log('');
+    log('info', '开始自动执行任务...');
+    await run(opts);
+  }
+}
+
+function deployTestRule(p) {
+  const dest = path.join(p.loopDir, 'test_rule.md');
+  if (fs.existsSync(dest)) return;
+  if (!fs.existsSync(p.testRuleTemplate)) return;
+  try {
+    fs.copyFileSync(p.testRuleTemplate, dest);
+    log('ok', '已部署测试指导规则 → .claude-coder/test_rule.md');
+  } catch { /* ignore */ }
+}
+
+module.exports = { run, add };