cc-team-viewer 1.4.16

package/server/stats-worker.js

@@ -0,0 +1,377 @@
+// Stats Worker — 后台线程，扫描 JSONL 日志生成项目级统计 JSON
+import { parentPort } from 'node:worker_threads';
+import { readFileSync, writeFileSync, existsSync, readdirSync, statSync } from 'node:fs';
+import { join, basename } from 'node:path';
+
+// 统计 schema 版本号，新增统计字段时递增，强制旧缓存失效重新解析
+const STATS_VERSION = 3;
+
+/**
+ * 解析单个 JSONL 文件，提取模型使用次数和 token 统计
+ * @param {string} filePath JSONL 文件绝对路径
+ * @returns {{ models: Object, summary: Object, members: Object }}
+ */
+function parseJsonlFile(filePath) {
+  const models = {};
+  let requestCount = 0;
+  let sessionCount = 0;
+  let totalInput = 0;
+  let totalOutput = 0;
+  let totalCacheRead = 0;
+  let totalCacheCreation = 0;
+
+  // 成员统计
+  const members = {};
+
+  try {
+    const content = readFileSync(filePath, 'utf-8');
+    if (!content.trim()) return { models, summary: { requestCount: 0, sessionCount: 0, input_tokens: 0, output_tokens: 0, cache_read_input_tokens: 0, cache_creation_input_tokens: 0 }, members: {} };
+
+    const entries = content.split('\n---\n').filter(p => p.trim());
+    for (const raw of entries) {
+      try {
+        const entry = JSON.parse(raw);
+        requestCount++;
+
+        // 会话轮次：MainAgent 且 messages.length === 1 表示一次新会话开始
+        if (entry.mainAgent && Array.isArray(entry.body?.messages) && entry.body.messages.length === 1) {
+          sessionCount++;
+        }
+
+        // 提取模型名：优先 body.model，其次 response.body.model
+        const model = entry.body?.model || entry.response?.body?.model;
+        if (!model) continue;
+
+        if (!models[model]) {
+          models[model] = { count: 0, input_tokens: 0, output_tokens: 0, cache_read_input_tokens: 0, cache_creation_input_tokens: 0 };
+        }
+        models[model].count++;
+
+        // 提取 usage — 可能在 response.body.usage
+        const usage = entry.response?.body?.usage;
+        if (usage) {
+          const inp = usage.input_tokens || 0;
+          const out = usage.output_tokens || 0;
+          const cacheRead = usage.cache_read_input_tokens || usage.cache_creation_input_tokens ? (usage.cache_read_input_tokens || 0) : 0;
+          const cacheCreate = usage.cache_creation_input_tokens || 0;
+
+          models[model].input_tokens += inp;
+          models[model].output_tokens += out;
+          models[model].cache_read_input_tokens += cacheRead;
+          models[model].cache_creation_input_tokens += cacheCreate;
+
+          totalInput += inp;
+          totalOutput += out;
+          totalCacheRead += cacheRead;
+          totalCacheCreation += cacheCreate;
+        }
+
+        // 成员统计：提取 teamMember
+        const teamMember = entry.teamMember;
+        if (teamMember) {
+          if (!members[teamMember]) {
+            members[teamMember] = {
+              requestCount: 0,
+              successCount: 0,
+              errorCount: 0,
+              inputTokens: 0,
+              outputTokens: 0,
+              toolCalls: {},
+              skillCalls: {},
+              lastActive: null,
+              systemPrompt: entry.body?.system || '',
+            };
+          }
+          members[teamMember].requestCount++;
+          members[teamMember].inputTokens += usage?.input_tokens || 0;
+          members[teamMember].outputTokens += usage?.output_tokens || 0;
+          members[teamMember].lastActive = entry.timestamp;
+
+          // 统计成功/失败
+          const error = entry.response?.body?.error;
+          if (error) {
+            members[teamMember].errorCount++;
+          } else {
+            members[teamMember].successCount++;
+          }
+
+          // 统计 Tool 调用
+          const content = entry.response?.body?.content;
+          if (Array.isArray(content)) {
+            for (const block of content) {
+              if (block.type === 'tool_use' && block.name) {
+                members[teamMember].toolCalls[block.name] = (members[teamMember].toolCalls[block.name] || 0) + 1;
+              }
+            }
+          }
+
+          // 统计 Skill 调用（从请求体中检测 # skill-name）
+          const messages = entry.body?.messages;
+          if (Array.isArray(messages)) {
+            for (const msg of messages) {
+              const msgContent = msg?.content;
+              if (typeof msgContent === 'string') {
+                const skillMatch = msgContent.match(/#\s+(\S+)/g);
+                if (skillMatch) {
+                  for (const skill of skillMatch) {
+                    const skillName = skill.replace(/^#\s+/, '').trim();
+                    if (skillName) {
+                      members[teamMember].skillCalls[skillName] = (members[teamMember].skillCalls[skillName] || 0) + 1;
+                    }
+                  }
+                }
+              } else if (Array.isArray(msgContent)) {
+                for (const block of msgContent) {
+                  if (block.type === 'text' && typeof block.text === 'string') {
+                    const skillMatch = block.text.match(/#\s+(\S+)/g);
+                    if (skillMatch) {
+                      for (const skill of skillMatch) {
+                        const skillName = skill.replace(/^#\s+/, '').trim();
+                        if (skillName) {
+                          members[teamMember].skillCalls[skillName] = (members[teamMember].skillCalls[skillName] || 0) + 1;
+                        }
+                      }
+                    }
+                  }
+                }
+              }
+            }
+          }
+        }
+      } catch {
+        // 跳过无法解析的条目
+      }
+    }
+  } catch {
+    // 文件读取失败
+  }
+
+  return {
+    models,
+    summary: {
+      requestCount,
+      sessionCount,
+      input_tokens: totalInput,
+      output_tokens: totalOutput,
+      cache_read_input_tokens: totalCacheRead,
+      cache_creation_input_tokens: totalCacheCreation,
+    },
+    members,
+  };
+}
+
+/**
+ * 为单个项目生成或增量更新统计 JSON
+ * @param {string} projectDir 项目日志目录
+ * @param {string} projectName 项目名
+ * @param {string|null} onlyFile 仅更新此文件（增量），null 表示智能增量
+ */
+function generateProjectStats(projectDir, projectName, onlyFile) {
+  const statsFile = join(projectDir, `${projectName}.json`);
+
+  // 读取已有统计（用于增量更新）
+  let existing = null;
+  try {
+    if (existsSync(statsFile)) {
+      existing = JSON.parse(readFileSync(statsFile, 'utf-8'));
+    }
+  } catch {
+    existing = null;
+  }
+
+  // 列出所有 JSONL 文件（排除 _temp.jsonl）
+  let jsonlFiles;
+  try {
+    jsonlFiles = readdirSync(projectDir)
+      .filter(f => f.endsWith('.jsonl') && !f.endsWith('_temp.jsonl'))
+      .sort();
+  } catch {
+    return;
+  }
+
+  if (jsonlFiles.length === 0) return;
+
+  const filesStats = {};
+  const topModels = {};
+
+  for (const f of jsonlFiles) {
+    const filePath = join(projectDir, f);
+    let stat;
+    try {
+      stat = statSync(filePath);
+    } catch {
+      continue;
+    }
+
+    const size = stat.size;
+    const lastModified = stat.mtime.toISOString();
+
+    // 增量优化：如果有已有统计且文件未变化且 schema 版本一致，直接复用
+    if (existing?._v === STATS_VERSION
+        && existing?.files?.[f] && existing.files[f].size === size && existing.files[f].lastModified === lastModified) {
+      // 如果指定了 onlyFile 且不是此文件，跳过重新解析
+      if (!onlyFile || onlyFile !== f) {
+        filesStats[f] = existing.files[f];
+        // 汇总模型
+        if (filesStats[f].models) {
+          for (const [model, data] of Object.entries(filesStats[f].models)) {
+            if (!topModels[model]) topModels[model] = 0;
+            topModels[model] += data.count;
+          }
+        }
+        continue;
+      }
+    }
+
+    // 需要重新解析
+    const parsed = parseJsonlFile(filePath);
+    filesStats[f] = {
+      models: parsed.models,
+      summary: parsed.summary,
+      members: parsed.members,
+      size,
+      lastModified,
+    };
+
+    // 汇总模型使用次数
+    for (const [model, data] of Object.entries(parsed.models)) {
+      if (!topModels[model]) topModels[model] = 0;
+      topModels[model] += data.count;
+    }
+  }
+
+  // 计算全局汇总
+  let totalRequests = 0;
+  let totalSessions = 0;
+  let totalInput = 0;
+  let totalOutput = 0;
+  let totalCacheRead = 0;
+  let totalCacheCreation = 0;
+
+  // 汇总成员统计
+  const allMembers = {};
+
+  for (const f of Object.values(filesStats)) {
+    totalRequests += f.summary.requestCount;
+    totalSessions += f.summary.sessionCount || 0;
+    totalInput += f.summary.input_tokens;
+    totalOutput += f.summary.output_tokens;
+    totalCacheRead += f.summary.cache_read_input_tokens;
+    totalCacheCreation += f.summary.cache_creation_input_tokens;
+
+    // 合并成员统计
+    if (f.members) {
+      for (const [memberName, memberData] of Object.entries(f.members)) {
+        if (!allMembers[memberName]) {
+          allMembers[memberName] = {
+            requestCount: 0,
+            successCount: 0,
+            errorCount: 0,
+            inputTokens: 0,
+            outputTokens: 0,
+            toolCalls: {},
+            skillCalls: {},
+            lastActive: null,
+            systemPrompt: memberData.systemPrompt || '',
+          };
+        }
+        allMembers[memberName].requestCount += memberData.requestCount || 0;
+        allMembers[memberName].successCount += memberData.successCount || 0;
+        allMembers[memberName].errorCount += memberData.errorCount || 0;
+        allMembers[memberName].inputTokens += memberData.inputTokens || 0;
+        allMembers[memberName].outputTokens += memberData.outputTokens || 0;
+
+        // 合并 toolCalls
+        for (const [toolName, count] of Object.entries(memberData.toolCalls || {})) {
+          allMembers[memberName].toolCalls[toolName] = (allMembers[memberName].toolCalls[toolName] || 0) + count;
+        }
+
+        // 合并 skillCalls
+        for (const [skillName, count] of Object.entries(memberData.skillCalls || {})) {
+          allMembers[memberName].skillCalls[skillName] = (allMembers[memberName].skillCalls[skillName] || 0) + count;
+        }
+
+        // 更新最近活跃时间
+        if (memberData.lastActive && (!allMembers[memberName].lastActive || memberData.lastActive > allMembers[memberName].lastActive)) {
+          allMembers[memberName].lastActive = memberData.lastActive;
+        }
+
+        // 保存 systemPrompt（取最新的）
+        if (memberData.systemPrompt) {
+          allMembers[memberName].systemPrompt = memberData.systemPrompt;
+        }
+      }
+    }
+  }
+
+  const stats = {
+    _v: STATS_VERSION,
+    project: projectName,
+    updatedAt: new Date().toISOString(),
+    models: topModels,
+    files: filesStats,
+    members: allMembers,
+    summary: {
+      requestCount: totalRequests,
+      sessionCount: totalSessions,
+      fileCount: jsonlFiles.length,
+      input_tokens: totalInput,
+      output_tokens: totalOutput,
+      cache_read_input_tokens: totalCacheRead,
+      cache_creation_input_tokens: totalCacheCreation,
+    },
+  };
+
+  try {
+    writeFileSync(statsFile, JSON.stringify(stats, null, 2));
+  } catch (err) {
+    parentPort?.postMessage({ type: 'error', message: `Failed to write stats: ${err.message}` });
+  }
+}
+
+/**
+ * 扫描 logDir 下所有项目目录，逐个生成统计
+ */
+function scanAllProjects(logDir) {
+  try {
+    const entries = readdirSync(logDir, { withFileTypes: true });
+    for (const entry of entries) {
+      if (!entry.isDirectory()) continue;
+      const projectDir = join(logDir, entry.name);
+      generateProjectStats(projectDir, entry.name, null);
+    }
+    parentPort?.postMessage({ type: 'scan-all-done' });
+  } catch (err) {
+    parentPort?.postMessage({ type: 'error', message: `scan-all failed: ${err.message}` });
+  }
+}
+
+// Worker 消息处理
+parentPort?.on('message', (msg) => {
+  switch (msg.type) {
+    case 'init': {
+      const { logDir, projectName } = msg;
+      const projectDir = join(logDir, projectName);
+      if (existsSync(projectDir)) {
+        generateProjectStats(projectDir, projectName, null);
+        parentPort?.postMessage({ type: 'init-done', projectName });
+      }
+      break;
+    }
+    case 'update': {
+      const { logDir, projectName, logFile } = msg;
+      const projectDir = join(logDir, projectName);
+      const fileName = basename(logFile);
+      if (existsSync(projectDir)) {
+        generateProjectStats(projectDir, projectName, fileName);
+        parentPort?.postMessage({ type: 'update-done', projectName, logFile: fileName });
+      }
+      break;
+    }
+    case 'scan-all': {
+      const { logDir } = msg;
+      scanAllProjects(logDir);
+      break;
+    }
+  }
+});

package/src/vite.config.js

@@ -0,0 +1,89 @@
+import { defineConfig } from 'vite';
+import react from '@vitejs/plugin-react';
+import { execSync } from 'node:child_process';
+import { resolve, dirname } from 'node:path';
+import { fileURLToPath } from 'node:url';
+
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = dirname(__filename);
+
+function findCurrentPort() {
+  const currentProject = process.cwd().split('/').pop();
+  console.error('[Vite] Current project:', currentProject);
+
+  // 1. 开发模式：使用环境变量 CC_VIEWER_PORT 指定开发环境服务器端口
+  if (process.env.CC_VIEWER_PORT) {
+    console.error('[Vite] Using port from CC_VIEWER_PORT (dev mode):', process.env.CC_VIEWER_PORT);
+    return parseInt(process.env.CC_VIEWER_PORT, 10);
+  }
+
+  // 2. 自动查找：遍历端口，找到与当前项目匹配且有最新活动的端口
+  let matchedPort = 7008;
+  let latestTimestamp = '';
+
+  for (let port = 7008; port <= 7008; port++) {
+    try {
+      const result = execSync(`lsof -i :${port} -sTCP:LISTEN -t`, { encoding: 'utf-8' });
+      if (result.trim()) {
+        // 先验证这是 cc-team-viewer server（通过检查 /api/teams 端点）
+        let isCCViewerServer = false;
+        try {
+          const teamsResponse = execSync(`curl -s "http://127.0.0.1:${port}/api/teams" --max-time 1`, { encoding: 'utf-8' });
+          // cc-team-viewer server 返回 JSON，ccv proxy 返回 HTML
+          if (teamsResponse.trim().startsWith('{') || teamsResponse.trim().startsWith('[')) {
+            isCCViewerServer = true;
+          }
+        } catch (e) {
+          // /api/teams 可能不存在，跳过这个端口
+          continue;
+        }
+
+        if (!isCCViewerServer) {
+          console.error(`[Vite] Port ${port}: not a cc-team-viewer server, skipping`);
+          continue;
+        }
+
+        // 获取这个端口的最新请求
+        try {
+          const response = execSync(`curl -s "http://127.0.0.1:${port}/api/requests?limit=1" --max-time 2`, { encoding: 'utf-8' });
+          const requests = JSON.parse(response);
+          if (requests && requests.length > 0) {
+            const lastRequest = requests[0];
+            const lastProject = lastRequest.project;
+            const lastTimestamp = lastRequest.timestamp;
+
+            console.error(`[Vite] Port ${port}: project="${lastProject}", timestamp="${lastTimestamp}"`);
+
+            // 如果项目名匹配，且时间戳更新，则选择这个端口
+            if (lastProject === currentProject && lastTimestamp > latestTimestamp) {
+              latestTimestamp = lastTimestamp;
+              matchedPort = port;
+            }
+          }
+        } catch (e) {
+          // Ignore curl errors
+        }
+      }
+    } catch { }
+  }
+
+  console.error(`[Vite] Selected port: ${matchedPort} (latest timestamp: ${latestTimestamp})`);
+  return matchedPort;
+}
+
+export default defineConfig(() => {
+  const port = findCurrentPort();
+  console.error('[Vite] CC-team-Viewer backend port:', port);
+  return {
+    plugins: [react()],
+    build: {
+      outDir: resolve(__dirname, '..', 'dist'),
+    },
+    server: {
+      proxy: {
+        '/events': `http://127.0.0.1:${port}`,
+        '/api': `http://127.0.0.1:${port}`,
+      },
+    },
+  };
+});