coding-tool-x 3.2.0

package/src/commands/search.js

@@ -0,0 +1,190 @@
+// 搜索会话命令
+const chalk = require('chalk');
+const ora = require('ora');
+const inquirer = require('inquirer');
+const { promptSelectSession, promptSearchKeyword, promptForkConfirm } = require('../ui/prompts');
+const { resumeSession } = require('./resume');
+const { getProjects, searchSessions: searchSessionsInProject, parseRealProjectPath } = require('../server/services/sessions');
+const { loadAliases } = require('../server/services/alias');
+
+/**
+ * 跨所有项目搜索会话内容
+ */
+async function searchSessionsAcrossProjects(config, keyword) {
+  const spinner = ora(`🔍 正在搜索 "${keyword}"...`).start();
+
+  const projects = await getProjects(config);
+  const aliases = loadAliases();
+  const allResults = [];
+
+  // 跨所有项目搜索
+  for (const projectName of projects) {
+    try {
+      const { projectName: displayName } = parseRealProjectPath(projectName);
+      spinner.text = `🔍 正在搜索项目: ${displayName}...`;
+      const results = searchSessionsInProject(config, projectName, keyword, 15);
+
+      if (results.length > 0) {
+        results.forEach(result => {
+          allResults.push({
+            ...result,
+            projectName: projectName,
+            projectDisplayName: displayName,
+            alias: aliases[result.sessionId] || null
+          });
+        });
+      }
+    } catch (e) {
+      // 忽略单个项目的错误
+    }
+  }
+
+  spinner.stop();
+  spinner.clear();
+
+  if (allResults.length === 0) {
+    console.clear();
+    console.log(chalk.red(`\n❌ 未找到包含 "${keyword}" 的对话\n`));
+    return [];
+  }
+
+  // 按匹配数量排序
+  allResults.sort((a, b) => b.matchCount - a.matchCount);
+
+  // 统计总匹配数
+  const totalMatches = allResults.reduce((sum, r) => sum + r.matchCount, 0);
+
+  console.clear();
+  console.log(chalk.green(`\n✨ 找到 ${allResults.length} 个对话，共 ${totalMatches} 处匹配\n`));
+
+  const choices = [];
+
+  allResults.forEach((result, index) => {
+    // 构建显示名称
+    let displayName = '';
+
+    // 序号
+    displayName += chalk.bold.white(`${index + 1}. `);
+
+    // 项目名（洋红色高亮）
+    displayName += chalk.magenta.bold(`[${result.projectDisplayName}] `);
+
+    // 会话别名或 ID
+    if (result.alias) {
+      displayName += chalk.yellow.bold(`[${result.alias}] `);
+    } else {
+      displayName += chalk.gray(`[${result.sessionId.substring(0, 8)}] `);
+    }
+
+    // 匹配数量
+    displayName += chalk.cyan(`(${result.matchCount} 处匹配)`);
+
+    choices.push({
+      name: displayName,
+      value: { sessionId: result.sessionId, projectName: result.projectName },
+      short: result.alias || result.sessionId.substring(0, 8)
+    });
+
+    // 显示前 3 个匹配的上下文
+    const matchesToShow = result.matches.slice(0, 3);
+    matchesToShow.forEach((match, idx) => {
+      const roleColor = match.role === 'user' ? chalk.blue : chalk.green;
+      const roleLabel = match.role === 'user' ? '用户' : '助手';
+
+      choices.push({
+        name: `    ${roleColor(`[${roleLabel}]`)} ${chalk.gray(match.context)}`,
+        value: null,
+        disabled: true
+      });
+    });
+
+    // 如果还有更多匹配，显示提示
+    if (result.matches.length > 3) {
+      choices.push({
+        name: chalk.gray(`    ... 还有 ${result.matches.length - 3} 处匹配`),
+        value: null,
+        disabled: true
+      });
+    }
+
+    // 添加分隔线（不是最后一个）
+    if (index < allResults.length - 1) {
+      choices.push(new inquirer.Separator(chalk.gray('─'.repeat(10))));
+    }
+  });
+
+  return choices;
+}
+
+/**
+ * 处理搜索会话
+ */
+async function handleSearch(config, switchProjectCallback) {
+  while (true) {
+    const keyword = await promptSearchKeyword();
+    const choices = await searchSessionsAcrossProjects(config, keyword);
+
+    if (choices.length === 0) {
+      const { action } = await inquirer.prompt([
+        {
+          type: 'list',
+          name: 'action',
+          message: '未找到匹配的对话',
+          choices: [
+            { name: chalk.blue('↩️  返回主菜单'), value: 'back' },
+            { name: chalk.cyan('🔎  重新搜索'), value: 'retry' },
+          ],
+        },
+      ]);
+
+      if (action === 'back') return;
+      if (action === 'retry') continue;
+    }
+
+    // 添加操作选项
+    choices.push(new inquirer.Separator(chalk.gray('═'.repeat(80))));
+    choices.push({ name: chalk.blue('↩️  返回主菜单'), value: 'back' });
+    choices.push({ name: chalk.cyan('🔎  重新搜索'), value: 'retry' });
+
+    // 使用自定义 pageSize 以便显示更多结果
+    const { selected } = await inquirer.prompt([
+      {
+        type: 'list',
+        name: 'selected',
+        message: '选择对话:',
+        pageSize: 20,
+        choices: choices,
+      },
+    ]);
+
+    if (selected === 'back') {
+      return;
+    }
+
+    if (selected === 'retry') {
+      continue;
+    }
+
+    // selected 是 { sessionId, projectName }
+    const sessionId = selected.sessionId;
+    const projectName = selected.projectName;
+
+    // 切换到该项目
+    config.currentProject = projectName;
+
+    // 询问是否 fork
+    const action = await promptForkConfirm();
+
+    if (action === 'back') {
+      continue;
+    }
+
+    const fork = action === 'fork';
+    await resumeSession(config, sessionId, fork);
+  }
+}
+
+module.exports = {
+  searchSessionsAcrossProjects,
+  handleSearch,
+};

package/src/commands/security.js

@@ -0,0 +1,37 @@
+const chalk = require('chalk');
+const fs = require('fs');
+const path = require('path');
+const { PATHS, ensureStorageDirMigrated } = require('../config/paths');
+
+const SECURITY_FILE = path.join(PATHS.base, 'security.json');
+
+function showSecurityHelp() {
+  console.log(chalk.yellow('\n🔐 安全设置命令:'));
+  console.log('  ctx security reset       关闭访问密码（删除安全配置文件）');
+  console.log('');
+}
+
+async function handleSecurityReset() {
+  console.log(chalk.cyan('\n🔐 安全设置 - 关闭访问密码\n'));
+  ensureStorageDirMigrated();
+
+  if (!fs.existsSync(SECURITY_FILE)) {
+    console.log(chalk.yellow('⚠️  未检测到安全配置文件'));
+    console.log(chalk.gray(`路径: ${SECURITY_FILE}\n`));
+    return;
+  }
+
+  try {
+    fs.unlinkSync(SECURITY_FILE);
+    console.log(chalk.green('✅ 访问密码已关闭'));
+    console.log(chalk.gray(`已删除: ${SECURITY_FILE}\n`));
+  } catch (error) {
+    console.error(chalk.red('❌ 关闭密码失败:'), error.message);
+    console.log(chalk.gray(`路径: ${SECURITY_FILE}\n`));
+  }
+}
+
+module.exports = {
+  showSecurityHelp,
+  handleSecurityReset
+};

package/src/commands/stats.js

@@ -0,0 +1,398 @@
+const chalk = require('chalk');
+const http = require('http');
+const { loadConfig } = require('../config/loader');
+
+const TOOL_TYPES = ['claude', 'codex', 'gemini', 'opencode'];
+const TOOL_ENDPOINTS = {
+  claude: '/api/statistics',
+  codex: '/api/codex/statistics',
+  gemini: '/api/gemini/statistics',
+  opencode: '/api/opencode/statistics'
+};
+
+/**
+ * HTTP 请求辅助函数
+ */
+function httpRequest(method, path, data = null) {
+  const config = loadConfig();
+  const port = config.ports?.webUI || 19999;
+
+  return new Promise((resolve, reject) => {
+    const postData = data ? JSON.stringify(data) : null;
+    const options = {
+      hostname: 'localhost',
+      port: port,
+      path: path,
+      method: method,
+      headers: {
+        'Content-Type': 'application/json',
+        ...(postData && { 'Content-Length': Buffer.byteLength(postData) })
+      },
+      timeout: 5000
+    };
+
+    const req = http.request(options, (res) => {
+      let responseData = '';
+
+      res.on('data', (chunk) => {
+        responseData += chunk;
+      });
+
+      res.on('end', () => {
+        try {
+          const json = JSON.parse(responseData);
+          resolve({ data: json, status: res.statusCode });
+        } catch (err) {
+          reject(new Error('Invalid JSON response'));
+        }
+      });
+    });
+
+    req.on('error', (err) => {
+      reject(err);
+    });
+
+    req.on('timeout', () => {
+      req.destroy();
+      reject(new Error('Request timeout'));
+    });
+
+    if (postData) {
+      req.write(postData);
+    }
+
+    req.end();
+  });
+}
+
+/**
+ * 检查 UI 服务是否运行
+ */
+async function checkUIService() {
+  try {
+    await httpRequest('GET', '/api/proxy/status');
+    return true;
+  } catch (err) {
+    return false;
+  }
+}
+
+function validateToolType(type) {
+  if (!type) return true;
+  return TOOL_TYPES.includes(type);
+}
+
+function getDateString(offsetDays = 0) {
+  const date = new Date();
+  date.setHours(0, 0, 0, 0);
+  date.setDate(date.getDate() - offsetDays);
+  const y = date.getFullYear();
+  const m = String(date.getMonth() + 1).padStart(2, '0');
+  const d = String(date.getDate()).padStart(2, '0');
+  return `${y}-${m}-${d}`;
+}
+
+function emptySummary() {
+  return {
+    requests: 0,
+    tokens: 0,
+    cost: 0,
+    inputTokens: 0,
+    outputTokens: 0,
+    cacheCreation: 0,
+    cacheRead: 0,
+    reasoningTokens: 0,
+    cachedTokens: 0
+  };
+}
+
+function normalizeNumber(value) {
+  if (typeof value === 'number' && Number.isFinite(value)) return value;
+  if (typeof value === 'string') {
+    const parsed = Number(value);
+    return Number.isFinite(parsed) ? parsed : 0;
+  }
+  return 0;
+}
+
+function extractSummary(stats) {
+  const summary = emptySummary();
+  const sourceSummary = stats?.summary || {};
+  const sourceGlobal = stats?.global || {};
+
+  summary.requests = normalizeNumber(
+    sourceSummary.totalRequests !== undefined ? sourceSummary.totalRequests : sourceSummary.requests
+  ) || normalizeNumber(sourceGlobal.totalRequests);
+
+  summary.tokens = normalizeNumber(
+    sourceSummary.totalTokens !== undefined ? sourceSummary.totalTokens : sourceSummary.tokens
+  ) || normalizeNumber(sourceGlobal.totalTokens);
+
+  summary.cost = normalizeNumber(
+    sourceSummary.totalCost !== undefined ? sourceSummary.totalCost : sourceSummary.cost
+  ) || normalizeNumber(sourceGlobal.totalCost);
+
+  summary.inputTokens = normalizeNumber(sourceSummary.inputTokens ?? sourceSummary.input);
+  summary.outputTokens = normalizeNumber(sourceSummary.outputTokens ?? sourceSummary.output);
+  summary.cacheCreation = normalizeNumber(sourceSummary.cacheCreation ?? sourceSummary.cache_creation);
+  summary.cacheRead = normalizeNumber(sourceSummary.cacheRead ?? sourceSummary.cache_read);
+  summary.reasoningTokens = normalizeNumber(sourceSummary.reasoningTokens ?? sourceSummary.reasoning);
+  summary.cachedTokens = normalizeNumber(sourceSummary.cachedTokens ?? sourceSummary.cached);
+
+  const detailedTotal =
+    summary.inputTokens +
+    summary.outputTokens +
+    summary.cacheCreation +
+    summary.cacheRead +
+    summary.reasoningTokens +
+    summary.cachedTokens;
+
+  if (!summary.tokens && detailedTotal > 0) {
+    summary.tokens = detailedTotal;
+  }
+
+  return summary;
+}
+
+function mergeSummaries(target, source) {
+  target.requests += normalizeNumber(source.requests);
+  target.tokens += normalizeNumber(source.tokens);
+  target.cost += normalizeNumber(source.cost);
+  target.inputTokens += normalizeNumber(source.inputTokens);
+  target.outputTokens += normalizeNumber(source.outputTokens);
+  target.cacheCreation += normalizeNumber(source.cacheCreation);
+  target.cacheRead += normalizeNumber(source.cacheRead);
+  target.reasoningTokens += normalizeNumber(source.reasoningTokens);
+  target.cachedTokens += normalizeNumber(source.cachedTokens);
+  return target;
+}
+
+function getRangeDays(timeRange) {
+  if (timeRange === 'week') return 7;
+  if (timeRange === 'month') return 30;
+  return 0;
+}
+
+async function fetchToolStats(toolType, timeRange) {
+  const endpointBase = TOOL_ENDPOINTS[toolType];
+  if (!endpointBase) {
+    throw new Error(`不支持的渠道类型: ${toolType}`);
+  }
+
+  if (timeRange === 'today') {
+    const response = await httpRequest('GET', `${endpointBase}/today`);
+    return extractSummary(response.data);
+  }
+
+  if (timeRange === 'all') {
+    const response = await httpRequest('GET', `${endpointBase}/summary`);
+    return extractSummary(response.data);
+  }
+
+  const days = getRangeDays(timeRange);
+  const merged = emptySummary();
+  for (let i = 0; i < days; i++) {
+    const date = getDateString(i);
+    const response = await httpRequest('GET', `${endpointBase}/daily/${date}`);
+    const dailySummary = extractSummary(response.data);
+    mergeSummaries(merged, dailySummary);
+  }
+  return merged;
+}
+
+async function fetchOverallStats(timeRange) {
+  const byToolType = {};
+  const summary = emptySummary();
+
+  for (const toolType of TOOL_TYPES) {
+    const toolSummary = await fetchToolStats(toolType, timeRange);
+    byToolType[toolType] = toolSummary;
+    mergeSummaries(summary, toolSummary);
+  }
+
+  return { summary, byToolType };
+}
+
+function buildDisplayPayload(type, timeRange, data) {
+  if (type) {
+    return {
+      type,
+      timeRange,
+      summary: data,
+      byToolType: null
+    };
+  }
+
+  return {
+    type: null,
+    timeRange,
+    summary: data.summary,
+    byToolType: data.byToolType
+  };
+}
+
+/**
+ * 查看统计信息
+ */
+async function handleStats(type = null, options = {}) {
+  // 检查 UI 服务
+  const uiRunning = await checkUIService();
+  if (!uiRunning) {
+    console.error(chalk.red('\n❌ UI 服务未运行\n'));
+    console.log(chalk.yellow('💡 请先启动 UI 服务: ') + chalk.cyan('ctx start\n'));
+    process.exit(1);
+  }
+
+  const timeRange = options.today ? 'today' : options.week ? 'week' : options.month ? 'month' : 'all';
+
+  try {
+    if (!validateToolType(type)) {
+      console.error(chalk.red(`\n❌ 无效的渠道类型: ${type}\n`));
+      console.log(chalk.gray('支持的类型: claude, codex, gemini, opencode\n'));
+      process.exit(1);
+    }
+
+    let payload;
+    if (type) {
+      const summary = await fetchToolStats(type, timeRange);
+      payload = buildDisplayPayload(type, timeRange, summary);
+    } else {
+      const overall = await fetchOverallStats(timeRange);
+      payload = buildDisplayPayload(null, timeRange, overall);
+    }
+
+    displayStats(payload);
+  } catch (error) {
+    console.error(chalk.red(`\n❌ 获取统计失败: ${error.message}\n`));
+    process.exit(1);
+  }
+}
+
+/**
+ * 显示统计信息
+ */
+function displayStats(stats) {
+  const type = stats.type;
+  const timeRange = stats.timeRange;
+  const title = type ? `${type.toUpperCase()} 统计信息` : '总体统计信息';
+  const rangeText = {
+    today: '今日',
+    week: '本周',
+    month: '本月',
+    all: '全部'
+  }[timeRange];
+
+  console.log(chalk.bold.cyan(`\n╔══════════════════════════════════════╗`));
+  console.log(chalk.bold.cyan(`║        ${title} (${rangeText})        ║`));
+  console.log(chalk.bold.cyan(`╚══════════════════════════════════════╝\n`));
+
+  if (!stats || !stats.summary) {
+    console.log(chalk.gray('  暂无统计数据\n'));
+    return;
+  }
+
+  const summary = stats.summary;
+
+  // 请求统计
+  console.log(chalk.bold('📊 请求统计:'));
+  console.log(chalk.gray('  总请求数: ') + chalk.cyan(formatNumber(summary.requests)));
+
+  // Token 使用
+  if (summary.tokens !== undefined) {
+    console.log(chalk.bold('\n🎯 Token 使用:'));
+    console.log(chalk.gray('  输入 Tokens: ') + chalk.cyan(formatNumber(summary.inputTokens || 0)));
+    console.log(chalk.gray('  输出 Tokens: ') + chalk.cyan(formatNumber(summary.outputTokens || 0)));
+    console.log(chalk.gray('  缓存创建: ') + chalk.cyan(formatNumber(summary.cacheCreation || 0)));
+    console.log(chalk.gray('  缓存读取: ') + chalk.cyan(formatNumber(summary.cacheRead || 0)));
+    console.log(chalk.gray('  推理 Tokens: ') + chalk.cyan(formatNumber(summary.reasoningTokens || 0)));
+    console.log(chalk.gray('  缓存 Tokens: ') + chalk.cyan(formatNumber(summary.cachedTokens || 0)));
+    console.log(chalk.gray('  总计: ') + chalk.bold.cyan(formatNumber(summary.tokens || 0)));
+  }
+
+  // 成本统计
+  if (summary.cost !== undefined) {
+    console.log(chalk.bold('\n💰 成本统计:'));
+    console.log(chalk.gray('  总成本: ') + chalk.yellow(`$${normalizeNumber(summary.cost).toFixed(4)}`));
+  }
+
+  if (!type && stats.byToolType) {
+    const iconMap = { claude: '🟢', codex: '🔵', gemini: '🟣', opencode: '🟠' };
+    console.log(chalk.bold('\n📡 分渠道汇总:'));
+    TOOL_TYPES.forEach((toolType) => {
+      const item = stats.byToolType[toolType] || emptySummary();
+      console.log(chalk.gray(`  ${iconMap[toolType]} ${toolType.toUpperCase()}:`));
+      console.log(
+        chalk.gray(
+          `     请求: ${formatNumber(item.requests)}  |  Tokens: ${formatNumber(item.tokens)}  |  成本: $${normalizeNumber(item.cost).toFixed(4)}`
+        )
+      );
+    });
+  }
+
+  console.log(chalk.gray('\n💡 提示:'));
+  console.log(chalk.gray('  • 使用 ') + chalk.cyan('ctx stats --today') + chalk.gray(' 查看今日统计'));
+  console.log(chalk.gray('  • 使用 ') + chalk.cyan('ctx stats claude') + chalk.gray(' 查看特定渠道'));
+  console.log(chalk.gray('  • 使用 ') + chalk.cyan('ctx stats opencode') + chalk.gray(' 查看 OpenCode 统计'));
+  console.log(chalk.gray('  • 使用 ') + chalk.cyan('ctx stats export') + chalk.gray(' 导出统计数据\n'));
+}
+
+/**
+ * 格式化数字
+ */
+function formatNumber(num) {
+  const normalized = normalizeNumber(num);
+  return normalized.toString().replace(/\B(?=(\d{3})+(?!\d))/g, ',');
+}
+
+/**
+ * 导出统计数据
+ */
+async function handleStatsExport(type = null, format = 'json') {
+  console.log(chalk.cyan('\n📤 导出统计数据...\n'));
+
+  const uiRunning = await checkUIService();
+  if (!uiRunning) {
+    console.error(chalk.red('❌ UI 服务未运行\n'));
+    process.exit(1);
+  }
+
+  try {
+    if (!validateToolType(type)) {
+      console.error(chalk.red(`\n❌ 无效的渠道类型: ${type}\n`));
+      console.log(chalk.gray('支持的类型: claude, codex, gemini, opencode\n'));
+      process.exit(1);
+    }
+
+    const exportFormat = format || 'json';
+    if (exportFormat !== 'json') {
+      console.log(chalk.yellow(`⚠️ 暂不支持 ${exportFormat} 格式，已回退为 json`));
+    }
+
+    let payload;
+    if (type) {
+      const summary = await fetchToolStats(type, 'all');
+      payload = buildDisplayPayload(type, 'all', summary);
+    } else {
+      const overall = await fetchOverallStats('all');
+      payload = buildDisplayPayload(null, 'all', overall);
+    }
+
+    const fs = require('fs');
+    const path = require('path');
+    const filename = `cc-tool-stats-${type || 'all'}-${Date.now()}.json`;
+    const filepath = path.join(process.cwd(), filename);
+
+    fs.writeFileSync(filepath, JSON.stringify(payload, null, 2));
+
+    console.log(chalk.green(`✅ 统计数据已导出\n`));
+    console.log(chalk.gray(`文件路径: ${filepath}\n`));
+  } catch (error) {
+    console.error(chalk.red(`\n❌ 导出失败: ${error.message}\n`));
+    process.exit(1);
+  }
+}
+
+module.exports = {
+  handleStats,
+  handleStatsExport
+};

package/src/commands/switch.js

@@ -0,0 +1,48 @@
+// 切换项目命令
+const chalk = require('chalk');
+const os = require('os');
+const { getAvailableProjects } = require('../utils/session');
+const { promptSelectProject } = require('../ui/prompts');
+const { saveConfig } = require('../config/loader');
+
+/**
+ * 切换项目
+ */
+async function switchProject(config) {
+  const projects = await getAvailableProjects(config);
+
+  if (projects.length === 0) {
+    console.log(chalk.yellow('没有找到项目'));
+    return false;
+  }
+
+  const selectedProject = await promptSelectProject(projects);
+
+  // 用户取消切换
+  if (!selectedProject) {
+    console.log(chalk.gray('\n取消切换\n'));
+    return false;
+  }
+
+  // 更新配置
+  config.currentProject = selectedProject;
+  config.defaultProject = selectedProject;
+
+  // 保存到配置文件（保留其余字段）
+  saveConfig({
+    ...config,
+    projectsDir: config.projectsDir.replace(os.homedir(), '~')
+  });
+
+  // 使用解析后的名称显示
+  const { parseRealProjectPath } = require('../server/services/sessions');
+  const { displayName, fullPath } = parseRealProjectPath(selectedProject);
+
+  console.log(chalk.green(`\n✅ 已切换到: ${displayName}\n`));
+  console.log(chalk.gray(`   路径: ${fullPath}\n`));
+  return true;
+}
+
+module.exports = {
+  switchProject,
+};