tokenforbes-cli 0.1.6 → 0.2.0

package/activity.js

@@ -0,0 +1,124 @@
+// 读本地 AI 活跃度:会话数/消息数/活跃时长/分时分布。
+// 只数「条数」和「时间戳」,绝不读取、不上传任何对话内容、代码和项目名。
+// 数据源:Claude Code(~/.claude/projects)和 Codex(~/.codex/sessions)的本地会话日志。
+// 注意:本地日志一般只保留最近约 30 天,所以活跃指标最多回填一个月;token 用量走 ccusage 老链路,不受影响。
+import fs from 'node:fs';
+import os from 'node:os';
+import path from 'node:path';
+import readline from 'node:readline';
+
+const ACTIVE_GAP = 5 * 60 * 1000;  // 活跃时长口径:相邻两条记录间隔 < 5 分钟,算连续在干活
+const TOTAL_GAP = 60 * 60 * 1000;  // 总时长口径:间隔 < 60 分钟,算同一段工作(中间想事、查资料也算)
+const MAX_DAYS = 400;              // 与用量上报同口径,只留最近 400 天
+
+// 递归找 .jsonl;目录不存在或没权限就当没有,活跃度是锦上添花,绝不因它报错。
+function jsonlFiles(dir, out = []) {
+  let entries;
+  try { entries = fs.readdirSync(dir, { withFileTypes: true }); } catch { return out; }
+  for (const e of entries) {
+    const p = path.join(dir, e.name);
+    if (e.isDirectory()) jsonlFiles(p, out);
+    else if (e.name.endsWith('.jsonl')) out.push(p);
+  }
+  return out;
+}
+
+// 本地时区的 YYYY-MM-DD,和 ccusage daily 的按天口径保持一致。
+function dayKey(ts) {
+  const d = new Date(ts);
+  return `${d.getFullYear()}-${String(d.getMonth() + 1).padStart(2, '0')}-${String(d.getDate()).padStart(2, '0')}`;
+}
+
+function dayOf(days, key) {
+  let day = days.get(key);
+  if (!day) { day = { sessions: new Set(), userMsgs: 0, aiMsgs: 0, stamps: [] }; days.set(key, day); }
+  return day;
+}
+
+// Claude Code 日志:一行一个 JSON,type=user/assistant 的行是对话消息。
+// 跳过子 Agent(isSidechain)和系统注入(isMeta);带 toolUseResult 的 user 行是工具结果回传,不算人发的消息。
+async function scanClaude(days) {
+  for (const f of jsonlFiles(path.join(os.homedir(), '.claude', 'projects'))) {
+    try {
+      const rl = readline.createInterface({ input: fs.createReadStream(f) });
+      for await (const line of rl) {
+        if (!line.includes('"timestamp"')) continue;
+        let o; try { o = JSON.parse(line); } catch { continue; }
+        if (o.type !== 'user' && o.type !== 'assistant') continue;
+        if (o.isSidechain || o.isMeta) continue;
+        const ts = Date.parse(o.timestamp);
+        if (!ts) continue;
+        const day = dayOf(days, dayKey(ts));
+        if (o.sessionId) day.sessions.add(o.sessionId);
+        if (o.type === 'assistant') day.aiMsgs++;
+        else if (!o.toolUseResult) day.userMsgs++;
+        day.stamps.push(ts);
+      }
+    } catch { /* 单个文件读坏了就跳过 */ }
+  }
+}
+
+// Codex 日志:每行开头都是 {"timestamp":"…"},一个 rollout 文件 = 一个会话。
+// 行可能巨大(工具输出全文),所以不整行 JSON.parse,用字符串特征匹配,只挑需要的行。
+const CODEX_TS = /^\{"timestamp":"([^"]+)"/;
+async function scanCodex(days) {
+  for (const f of jsonlFiles(path.join(os.homedir(), '.codex', 'sessions'))) {
+    const sessionId = path.basename(f, '.jsonl'); // 文件名含会话 UUID,够当会话标识
+    try {
+      const rl = readline.createInterface({ input: fs.createReadStream(f) });
+      for await (const line of rl) {
+        const m = CODEX_TS.exec(line);
+        if (!m) continue;
+        const ts = Date.parse(m[1]);
+        if (!ts) continue;
+        const day = dayOf(days, dayKey(ts));
+        day.sessions.add(sessionId);
+        day.stamps.push(ts);
+        // 消息只数 response_item 里 type=message 的行;环境上下文/预置指令是注入的,不算人发的。
+        if (!line.includes('"type":"response_item"') || !line.includes('"type":"message"')) continue;
+        if (line.includes('"role":"assistant"')) day.aiMsgs++;
+        else if (line.includes('"role":"user"')
+          && !line.includes('<environment_context>') && !line.includes('<user_instructions>')) day.userMsgs++;
+      }
+    } catch { /* 单个文件读坏了就跳过 */ }
+  }
+}
+
+// 把一天的时间戳折成两种时长(分钟)和 24 小时消息分布。
+function summarize(day) {
+  day.stamps.sort((a, b) => a - b);
+  let active = 0, total = 0;
+  const hours = Array(24).fill(0);
+  for (let i = 0; i < day.stamps.length; i++) {
+    hours[new Date(day.stamps[i]).getHours()]++;
+    if (i === 0) continue;
+    const gap = day.stamps[i] - day.stamps[i - 1];
+    if (gap < ACTIVE_GAP) active += gap;
+    if (gap < TOTAL_GAP) total += gap;
+  }
+  return {
+    sessions: day.sessions.size,
+    userMsgs: day.userMsgs,
+    aiMsgs: day.aiMsgs,
+    activeMin: Math.round(active / 60000),
+    totalMin: Math.round(total / 60000),
+    hours,
+  };
+}
+
+// 返回 [{date, sessions, userMsgs, aiMsgs, activeMin, totalMin, hours[24]}],按日期升序。
+// 任何一步出问题都返回 [](绝不让活跃度统计拖垮用量上报)。
+export async function readActivity() {
+  try {
+    const days = new Map();
+    await scanClaude(days);
+    await scanCodex(days);
+    return [...days]
+      .sort((a, b) => a[0].localeCompare(b[0]))
+      .slice(-MAX_DAYS)
+      .map(([date, day]) => ({ date, ...summarize(day) }))
+      .filter((d) => d.userMsgs + d.aiMsgs > 0);
+  } catch {
+    return [];
+  }
+}

package/bin.js

@@ -6,6 +6,7 @@ import os from 'node:os';
 import path from 'node:path';
 import { exec } from 'node:child_process';
 import { readUsage } from './ccusage.js';
+import { readActivity } from './activity.js';
 import { detectDevice } from './device.js';
 import { daemon, ensureDaemonInstalled } from './daemon.js';
 
@@ -215,10 +216,12 @@ async function uploadDays(days, { server, key, device }) {
     return;
   }
   stage = 'upload';
+  // 顺手统计活跃度(会话数/消息数/时长/分时),同样只有条数和时间戳;失败返回空数组,绝不影响用量上报。
+  const activity = await readActivity();
   const total = days.reduce((s, d) => s + d.input + d.output + (d.cacheCreation || 0) + (d.cacheRead || 0), 0);
   console.log(`\n设备：${device}`);
   console.log(`共 ${days.length} 天、约 ${Math.round(total / 1e4).toLocaleString()} 万 token，正在上传（只传数字，不传内容）…`);
-  const res = await postRaw(server + '/api/report', { key, device, days });
+  const res = await postRaw(server + '/api/report', { key, device, days, activity });
   if (!res.ok) die(sendFailText(res));
   stage = 'done';
   console.log(`✓ 上报成功！服务器已记下 ${res.json.days} 天的用量。\n`);

package/package.json

@@ -1,12 +1,12 @@
 {
   "name": "tokenforbes-cli",
-  "version": "0.1.6",
+  "version": "0.2.0",
   "description": "上报本地 AI 编程工具(Claude Code / Codex 等)的 token 用量到 Token 福布斯排行榜(ceorank.cn)。只读用量数字和设备型号，绝不碰你的代码内容。",
   "type": "module",
   "bin": {
     "tokenforbes": "./bin.js"
   },
-  "files": ["bin.js", "ccusage.js", "device.js", "daemon.js"],
+  "files": ["bin.js", "ccusage.js", "device.js", "daemon.js", "activity.js"],
   "engines": {
     "node": ">=18"
   },