tokenforbes-cli 0.1.0

package/bin.js

@@ -0,0 +1,118 @@
+#!/usr/bin/env node
+// TokenBoard 上报命令行。只上报数字和设备型号,不读、不传你的任何项目内容。
+// 命令:register <昵称> | login <密钥> | report(默认)
+import fs from 'node:fs';
+import os from 'node:os';
+import path from 'node:path';
+import { exec } from 'node:child_process';
+import { readUsage } from './ccusage.js';
+import { detectDevice } from './device.js';
+import { daemon } from './daemon.js';
+
+const CONFIG = path.join(os.homedir(), '.tokenboard.json');
+const DEFAULT_SERVER = process.env.TB_SERVER || 'https://ceorank.cn';
+
+const loadConfig = () => { try { return JSON.parse(fs.readFileSync(CONFIG, 'utf8')); } catch { return {}; } };
+const saveConfig = (c) => fs.writeFileSync(CONFIG, JSON.stringify(c, null, 2));
+
+const [cmd, arg] = process.argv.slice(2);
+
+if (cmd === 'login') {
+  if (!arg) die('用法: tokenforbes login <密钥>');
+  const c = loadConfig(); c.key = arg; c.server = c.server || DEFAULT_SERVER; saveConfig(c);
+  console.log('密钥已保存 →', CONFIG);
+} else if (cmd === 'link') {
+  await link();
+} else if (cmd === 'daemon') {
+  daemon(arg);
+} else if (cmd === 'register') {
+  await register(arg);
+} else if (cmd === 'report') {
+  await report();
+} else if (!cmd) {
+  // 一条命令走完:没登录先引导登录(登录成功会自动上报一次),登录过就直接同步。对齐竞品 npx 体验。
+  const { key } = loadConfig();
+  if (key) await report(); else await link();
+} else {
+  console.log('命令: link | register <昵称> | login <密钥> | report | daemon <install|status|uninstall>');
+}
+
+// 设备授权登录:发起 → 自动开浏览器批准 → 轮询拿密钥(免去手动复制粘贴密钥)。
+async function link() {
+  const c = loadConfig();
+  const server = c.server || DEFAULT_SERVER;
+  const device = detectDevice();
+  const s = await post(server + '/api/device/start', { device });
+  const url = `${server}/link.html?code=${s.user_code}`;
+  console.log('\n→ 登录确认 ' + url);
+  console.log('  验证码: ' + s.user_code);
+  console.log('  浏览器会自动打开；没反应就手动复制上面的链接。\n');
+  openBrowser(url);
+
+  process.stdout.write('等待批准…');
+  const until = Date.now() + s.expires_in * 1000;
+  while (Date.now() < until) {
+    await sleep(s.interval * 1000);
+    const p = await post(server + '/api/device/poll', { device_code: s.device_code });
+    if (p.status === 'approved') {
+      c.key = p.key; c.server = server; saveConfig(c);
+      console.log('\n✓ 已批准，密钥已存到 ' + CONFIG);
+      console.log('\n顺手把你的本地用量传上去：\n');
+      await report();   // 登录即上报,一条命令闭环,别让用户卡在「登录完不知道下一步」
+      console.log('\n想让它每半小时自动后台上报?  npm i -g tokenforbes-cli 后跑  tokenforbes daemon install');
+      return;
+    }
+    if (p.status === 'expired') break;
+    process.stdout.write('.');
+  }
+  die('\n登录超时或验证码已过期，重新跑一次  npx tokenforbes-cli link');
+}
+
+const sleep = (ms) => new Promise((r) => setTimeout(r, ms));
+
+function openBrowser(url) {
+  const opener = process.platform === 'darwin' ? 'open'
+    : process.platform === 'win32' ? 'start ""'
+    : 'xdg-open';
+  exec(`${opener} "${url}"`, () => {}); // 打不开就算了,上面已打印链接供手动打开
+}
+
+async function register(nickname) {
+  if (!nickname) die('用法: tokenforbes register <昵称>');
+  const c = loadConfig();
+  const server = c.server || DEFAULT_SERVER;
+  const device = detectDevice();
+  const j = await post(server + '/api/register', { nickname, device });
+  c.key = j.key; c.server = server; saveConfig(c);
+  console.log(`注册成功,昵称「${j.nickname}」。密钥已存到 ${CONFIG}`);
+  console.log('现在跑  tokenforbes report  上报你的用量。');
+}
+
+async function report() {
+  const c = loadConfig();
+  if (!c.key) die('还没登录。先跑  npx tokenforbes-cli  (浏览器一键登录),或 register / login');
+  const server = c.server || DEFAULT_SERVER;
+  console.log('读取本地用量(ccusage)…');
+  const days = readUsage();
+  if (!days.length) return console.log('没读到用量数据。');
+  const device = detectDevice();
+  const total = days.reduce((s, d) => s + d.input + d.output, 0);
+  console.log(`设备:${device}`);
+  console.log(`共 ${days.length} 天,合计约 ${Math.round(total / 1e4).toLocaleString()} 万 token,上报中…`);
+  const j = await post(server + '/api/report', { key: c.key, device, days });
+  console.log(`上报成功 ✓ 已记录 ${j.days} 天。`);
+}
+
+async function post(url, body) {
+  let res;
+  try {
+    res = await fetch(url, { method: 'POST', headers: { 'Content-Type': 'application/json' }, body: JSON.stringify(body) });
+  } catch (e) {
+    die(`连不上服务器(${url}):${e.message}`);
+  }
+  const j = await res.json().catch(() => ({}));
+  if (!res.ok) die('失败:' + (j.error || res.status));
+  return j;
+}
+
+function die(msg) { console.error(msg); process.exit(1); }

package/ccusage.js

@@ -0,0 +1,44 @@
+// 读本地 AI 编码用量。只用 ccusage 的统计数字,绝不碰你的项目内容。
+// 榜单口径:每天 input + output(不含缓存)。
+import { execSync } from 'node:child_process';
+
+const num = (x) => Math.max(0, Math.round(Number(x) || 0));
+
+export function readUsage() {
+  let out;
+  try {
+    out = execSync('npx ccusage@latest daily --breakdown --json', {
+      encoding: 'utf8', timeout: 120000, maxBuffer: 64 * 1024 * 1024,
+    });
+  } catch {
+    throw new Error('读取用量失败:ccusage 跑不起来。先确认能联网、能跑 npx。');
+  }
+  let data;
+  try {
+    data = JSON.parse(out);
+  } catch {
+    throw new Error('读取用量失败:ccusage 输出看不懂(可能版本变了)。');
+  }
+  if (!data || !Array.isArray(data.daily)) {
+    throw new Error('读取用量失败:没找到每日数据(ccusage 输出结构变了)。');
+  }
+  return data.daily
+    .map((d) => {
+      const input = num(d.inputTokens), output = num(d.outputTokens);
+      return {
+        date: d.period,
+        input,
+        output,
+        models: d.modelsUsed || [],
+        tools: d.metadata?.agents || [], // 用了哪些工具:claude / codex …
+        breakdown: Array.isArray(d.modelBreakdowns)
+          ? d.modelBreakdowns.map((m) => ({
+            model: m.modelName,
+            input: num(m.inputTokens),
+            output: num(m.outputTokens),
+          }))
+          : [],
+      };
+    })
+    .filter((d) => d.date && d.input + d.output > 0);
+}

package/daemon.js

@@ -0,0 +1,91 @@
+// 后台自动同步:在 macOS 上装一个 launchd 定时任务,每 30 分钟自动跑一次 report。
+// ponytail: 只做 macOS(launchd)—— 负责人在 Mac,对标的竞品也是这套。
+//           Linux(systemd/cron)、Windows(计划任务)等真有人要再加,先不抽象。
+import fs from 'node:fs';
+import os from 'node:os';
+import path from 'node:path';
+import { execFileSync } from 'node:child_process';
+import { fileURLToPath } from 'node:url';
+
+const LABEL = 'ai.tokenboard.report';
+const PLIST = path.join(os.homedir(), 'Library', 'LaunchAgents', LABEL + '.plist');
+const LOG = path.join(os.homedir(), '.tokenboard.log');
+const CONFIG = path.join(os.homedir(), '.tokenboard.json');
+const BIN = fileURLToPath(new URL('./bin.js', import.meta.url)); // 这个 CLI 的绝对路径
+const INTERVAL = 30 * 60; // 秒
+
+function ensureMac() {
+  if (process.platform !== 'darwin') {
+    console.error('后台自动同步目前只支持 macOS。其它系统可自己挂个定时任务跑  tokenforbes report');
+    process.exit(1);
+  }
+}
+
+// launchd 任务说明书(plist)。用绝对路径,日志重定向到文件,失败不静默。
+export function buildPlist() {
+  return `<?xml version="1.0" encoding="UTF-8"?>
+<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
+<plist version="1.0">
+<dict>
+  <key>Label</key><string>${LABEL}</string>
+  <key>ProgramArguments</key>
+  <array>
+    <string>${process.execPath}</string>
+    <string>${BIN}</string>
+    <string>report</string>
+  </array>
+  <key>StartInterval</key><integer>${INTERVAL}</integer>
+  <key>RunAtLoad</key><true/>
+  <key>StandardOutPath</key><string>${LOG}</string>
+  <key>StandardErrorPath</key><string>${LOG}</string>
+</dict>
+</plist>
+`;
+}
+
+function install() {
+  ensureMac();
+  // 没登录就别装,否则只会每半小时报一次「未登录」错。
+  let key;
+  try { key = JSON.parse(fs.readFileSync(CONFIG, 'utf8')).key; } catch {}
+  if (!key) { console.error('还没登录,先跑  npx tokenforbes-cli  登录后再装。'); process.exit(1); }
+
+  // ponytail: plist 里写死了 BIN 的绝对路径。装 daemon 请用全局安装(npm i -g),
+  //           别用 npx —— npx 缓存目录会被清,清掉后 launchd 每半小时报一次「文件不存在」。
+  fs.mkdirSync(path.dirname(PLIST), { recursive: true });
+  fs.writeFileSync(PLIST, buildPlist());
+  // ponytail: 用 load/unload(老接口,现 macOS 仍可用);若哪天失效,换 bootstrap/bootout gui/$UID。
+  // 先 unload 再 load = 幂等,重复装不报错。
+  try { execFileSync('launchctl', ['unload', PLIST], { stdio: 'ignore' }); } catch {}
+  execFileSync('launchctl', ['load', PLIST]);
+  console.log('✓ 已开启后台自动同步,每 30 分钟自动上报一次。');
+  console.log('  任务文件: ' + PLIST);
+  console.log('  日志:     ' + LOG);
+  console.log('  查看状态: tokenforbes daemon status');
+  console.log('  关掉它:   tokenforbes daemon uninstall');
+}
+
+function status() {
+  ensureMac();
+  if (!fs.existsSync(PLIST)) return console.log('未安装。装它:  tokenforbes daemon install');
+  let loaded = false;
+  try { loaded = execFileSync('launchctl', ['list'], { encoding: 'utf8' }).includes(LABEL); } catch {}
+  console.log(loaded ? '✓ 已安装并在运行,每 30 分钟自动上报。' : '已安装但未加载(重新 install 一下)。');
+  console.log('  任务文件: ' + PLIST);
+  console.log('  日志:     ' + LOG + '(看最近几次上报有没有出错)');
+}
+
+function uninstall() {
+  ensureMac();
+  if (!fs.existsSync(PLIST)) return console.log('本来就没装。');
+  try { execFileSync('launchctl', ['unload', PLIST], { stdio: 'ignore' }); } catch {}
+  fs.rmSync(PLIST, { force: true });
+  console.log('✓ 已关闭后台自动同步,删除了任务文件。');
+}
+
+export function daemon(sub) {
+  if (sub === 'install') return install();
+  if (sub === 'status') return status();
+  if (sub === 'uninstall' || sub === 'remove') return uninstall();
+  console.log('用法: tokenforbes daemon install | status | uninstall');
+}

package/device.js

@@ -0,0 +1,26 @@
+// 识别设备型号。Mac 走 system_profiler,其他系统用 Node 自带信息降级。
+import { execSync } from 'node:child_process';
+import os from 'node:os';
+
+export function detectDevice() {
+  try {
+    if (process.platform === 'darwin') return mac();
+  } catch {}
+  return generic();
+}
+
+function mac() {
+  const out = execSync('system_profiler SPHardwareDataType -json', {
+    encoding: 'utf8', timeout: 8000,
+  });
+  const h = JSON.parse(out).SPHardwareDataType[0] || {};
+  return [h.machine_name, h.chip_type || h.cpu_type, h.physical_memory]
+    .filter(Boolean).join(' · ');
+}
+
+function generic() {
+  const plat = { win32: 'Windows', linux: 'Linux' }[process.platform] || process.platform;
+  const cpu = (os.cpus()[0]?.model || '').trim();
+  const mem = Math.round(os.totalmem() / 1e9) + ' GB';
+  return [plat, cpu, mem].filter(Boolean).join(' · ');
+}

package/package.json

@@ -0,0 +1,16 @@
+{
+  "name": "tokenforbes-cli",
+  "version": "0.1.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"],
+  "engines": {
+    "node": ">=18"
+  },
+  "keywords": ["tokenforbes", "ccusage", "claude-code", "codex", "token", "leaderboard", "ai-coding"],
+  "homepage": "https://ceorank.cn",
+  "license": "MIT"
+}