tokenforbes-cli 0.1.3 → 0.1.5

package/bin.js

@@ -7,7 +7,7 @@ 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';
+import { daemon, ensureDaemonInstalled } from './daemon.js';
 
 const CONFIG = path.join(os.homedir(), '.tokenboard.json');
 const DEFAULT_SERVER = process.env.TB_SERVER || 'https://ceorank.cn';
@@ -36,35 +36,46 @@ if (cmd === 'login') {
   console.log('命令: link | register <昵称> | login <密钥> | report | daemon <install|status|uninstall>');
 }
 
-// 设备授权登录:发起 → 自动开浏览器批准 → 轮询拿密钥(免去手动复制粘贴密钥)。
-async function link() {
+// 设备授权登录:先弹浏览器让你批准 → 趁这工夫后台抓用量 → 批准一到立刻上传。
+// preDays: 若上层已抓好用量(比如 report 探到密钥失效),直接传进来,不再重抓。
+async function link(preDays) {
   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正在打开浏览器让你确认登录…没弹出来就手动复制下面这条链接:');
-  console.log('  → ' + url);
-  console.log('  验证码: ' + s.user_code + '（网页上核对一致再点确认）\n');
+  // 第一步就把浏览器弹出去,别让用户对着黑终端干等。
+  console.log('\n① 浏览器这就打开,填个昵称、点「注册并批准」就行。没弹出来就手动复制这条链接:');
+  console.log('   → ' + url);
+  console.log('   验证码: ' + s.user_code + '（跟网页上那串核对一致再批准）');
   openBrowser(url);
 
-  process.stdout.write('已在浏览器等你点确认…');
+  // 趁你在浏览器填昵称、点批准的工夫,后台把本地用量先抓好,批准一到就能立刻上传。
+  let days = preDays;
+  if (!days) {
+    process.stdout.write('\n② 正在后台读取本地用量(ccusage)…');
+    days = readUsage();
+    console.log(days.length ? ` 已抓好 ${days.length} 天,就等你在浏览器批准。` : ' 没读到用量。');
+  } else {
+    console.log('\n② 本地用量已经抓好,就等你在浏览器批准。');
+  }
+
+  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({ allowRelink: false });   // 登录即上报,一条命令闭环;刚拿到新密钥,别再触发二次登录
-      console.log('\n想让它每半小时自动后台上报?  npm i -g tokenforbes-cli 后跑  tokenforbes daemon install');
+      console.log('\n✓ 已批准,密钥已存到 ' + CONFIG);
+      await uploadDays(days, { server, key: p.key, device });
+      ensureDaemonInstalled();
       return;
     }
     if (p.status === 'expired') break;
     process.stdout.write('.');
   }
-  die('\n登录超时或验证码已过期，重新跑一次  npx tokenforbes-cli link');
+  die('\n登录超时或验证码已过期，重新跑一次  npx -y tokenforbes-cli');
 }
 
 function sleep(ms) { return new Promise((r) => setTimeout(r, ms)); }
@@ -84,37 +95,46 @@ async function register(nickname) {
   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  上报你的用量。');
+  console.log('正在读取本地用量(ccusage)…');
+  await uploadDays(readUsage(), { server, key: j.key, device }); // 注册完顺手上报,别让用户再跑一条
+  ensureDaemonInstalled();
 }
 
 async function report({ allowRelink = true } = {}) {
   const c = loadConfig();
-  // 没密钥:直接带去登录(登录成功会自动上报),不再让用户对着报错发愣。
+  // 没密钥:直接带去登录(link 会先弹浏览器,再后台抓用量),不让用户对着报错发愣。
   if (!c.key) {
     if (allowRelink) return link();
-    die('还没登录。先跑  npx tokenforbes-cli  (浏览器一键登录),或 register / login');
+    die('还没登录。先跑  npx -y tokenforbes-cli  (浏览器一键登录),或 register / login');
   }
   const server = c.server || DEFAULT_SERVER;
-  console.log('正在读取本地用量(ccusage)…');
-  const days = readUsage();
-  if (!days.length) return console.log('没读到用量数据(这台电脑还没用过 Claude Code / Codex?)。');
   const device = detectDevice();
+  // 先用一个很快的空请求验密钥,别等抓完几十天用量才发现要重登——那样浏览器弹得太迟。
+  const probe = await postRaw(server + '/api/report', { key: c.key, device, days: [] });
+  if (!probe.ok) {
+    const authFailed = probe.status === 401 || /密钥/.test(probe.json?.error || '');
+    if (authFailed && allowRelink) {
+      console.log('本地密钥已失效,这就打开浏览器带你重新登录…');
+      delete c.key; saveConfig(c);
+      return link(); // 密钥无效,还没抓用量;交给 link 先弹浏览器再后台抓
+    }
+    die('上报失败:' + (probe.json?.error || probe.netError || probe.status));
+  }
+  // 密钥有效:抓用量并上传。
+  console.log('正在读取本地用量(ccusage)…');
+  await uploadDays(readUsage(), { server, key: c.key, device });
+  ensureDaemonInstalled();
+}
+
+// 把抓好的用量传上去。抽出来给 report / link / register 共用,避免各写一遍、避免重复抓。
+async function uploadDays(days, { server, key, device }) {
+  if (!days || !days.length) return console.log('没读到用量数据(这台电脑还没用过 Claude Code / Codex?)。');
   const total = days.reduce((s, d) => s + d.input + d.output + (d.cacheCreation || 0) + (d.cacheRead || 0), 0);
   console.log(`设备:${device}`);
   console.log(`共 ${days.length} 天,合计约 ${Math.round(total / 1e4).toLocaleString()} 万 token,正在上传…`);
-  const res = await postRaw(server + '/api/report', { key: c.key, device, days });
-  if (res.ok) {
-    console.log(`上报成功 ✓ 已记录 ${res.json.days} 天。去 ${server} 看你的排名。`);
-    return;
-  }
-  // 密钥失效(最常见:服务器重置过用户库,旧密钥查无此人):清掉旧密钥,自动带去重新登录再上报。
-  const authFailed = res.status === 401 || /密钥/.test(res.json?.error || '');
-  if (authFailed && allowRelink) {
-    console.log('\n本地密钥已失效(服务器可能重置过数据),带你重新登录一次…\n');
-    delete c.key; saveConfig(c);
-    return link();
-  }
-  die('上报失败:' + (res.json?.error || res.netError || res.status));
+  const res = await postRaw(server + '/api/report', { key, device, days });
+  if (res.ok) console.log(`上报成功 ✓ 已记录 ${res.json.days} 天。去 ${server} 看你的排名。`);
+  else die('上报失败:' + (res.json?.error || res.netError || res.status));
 }
 
 // 会抛错就 die 的版本:给登录/注册这类「失败就该停」的调用用。

package/daemon.js

@@ -1,38 +1,72 @@
-// 后台自动同步:在 macOS 上装一个 launchd 定时任务,每 30 分钟自动跑一次 report。
-// ponytail: 只做 macOS(launchd)—— 负责人在 Mac,对标的竞品也是这套。
-//           Linux(systemd/cron)、Windows(计划任务)等真有人要再加,先不抽象。
+// 后台自动同步:macOS 用 launchd、Windows 用计划任务,每 30 分钟自动跑一次 report。
 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 VBS = path.join(os.homedir(), '.tokenboard.vbs'); // Windows 隐藏窗口启动器
 const INTERVAL = 30 * 60; // 秒
+// launchd 不读用户 shell 配置,PATH 得自己给全。nvm/fnm 装的 node 不在这些标准目录,
+// 所以安装时把「当前 node 的真实目录」排最前(realpath 解掉 fnm 每个终端的临时软链)。
+const FALLBACK_PATH = '/opt/homebrew/bin:/usr/local/bin:/usr/bin:/bin:/usr/sbin:/sbin';
+// 上线前旧采集脚本(opentoken)的 launchd 任务名,新版安装/卸载时顺手关掉,免得双份任务并存。
+export const LEGACY_LABELS = ['com.opentoken.daemon'];
 
-function ensureMac() {
-  if (process.platform !== 'darwin') {
-    console.error('后台自动同步目前只支持 macOS。其它系统可自己挂个定时任务跑  tokenforbes report');
+export function launchdPath() {
+  let nodeBin = '';
+  try { nodeBin = path.dirname(fs.realpathSync(process.execPath)); } catch {}
+  if (!nodeBin || FALLBACK_PATH.split(':').includes(nodeBin)) return FALLBACK_PATH;
+  return nodeBin + ':' + FALLBACK_PATH;
+}
+
+export function reportCommand() {
+  return ['npx', '-y', 'tokenforbes-cli@latest', 'report'];
+}
+
+// Windows 上直接让计划任务跑 cmd 会每半小时闪一个黑窗,所以包一层 VBS:
+// wscript 是无窗口宿主,由它再以隐藏方式(第二参 0)拉起 cmd。VBS 字符串里的双引号写成两个。
+export function buildVbs() {
+  const cmd = 'cmd /d /c ' + reportCommand().join(' ') + ' >> ""' + LOG + '"" 2>&1';
+  return 'CreateObject("WScript.Shell").Run "' + cmd + '", 0, False\r\n';
+}
+
+export function windowsTaskArgs() {
+  return ['/create', '/tn', LABEL, '/sc', 'minute', '/mo', '30', '/tr', 'wscript.exe //B //Nologo "' + VBS + '"', '/f'];
+}
+
+function supported() {
+  return process.platform === 'darwin' || process.platform === 'win32';
+}
+
+function ensureSupported() {
+  if (!supported()) {
+    console.error('后台自动同步目前支持 macOS / Windows。其它系统可自己挂定时任务跑  npx -y tokenforbes-cli@latest report');
     process.exit(1);
   }
 }
 
-// launchd 任务说明书(plist)。用绝对路径,日志重定向到文件,失败不静默。
+// launchd 任务说明书(plist)。日志重定向到文件,失败不静默。
+// ponytail: 命令必须用 /bin/sh -c 包一层 —— launchd 找程序不认 EnvironmentVariables 里的 PATH,
+//           裸写 npx 会以退出码 78(找不到程序)静默失败,真机实测过。PATH 是给 sh 找 npx 用的。
 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>EnvironmentVariables</key>
+  <dict>
+    <key>PATH</key><string>${launchdPath()}</string>
+  </dict>
   <key>ProgramArguments</key>
   <array>
-    <string>${process.execPath}</string>
-    <string>${BIN}</string>
-    <string>report</string>
+    <string>/bin/sh</string>
+    <string>-c</string>
+    <string>${reportCommand().join(' ')}</string>
   </array>
   <key>StartInterval</key><integer>${INTERVAL}</integer>
   <key>RunAtLoad</key><true/>
@@ -43,46 +77,134 @@ export function buildPlist() {
 `;
 }
 
-function install() {
-  ensureMac();
+function requireKey() {
   // 没登录就别装,否则只会每半小时报一次「未登录」错。
   let key;
   try { key = JSON.parse(fs.readFileSync(CONFIG, 'utf8')).key; } catch {}
-  if (!key) { console.error('还没登录,先跑  npx tokenforbes-cli  登录后再装。'); process.exit(1); }
+  return Boolean(key);
+}
 
-  // ponytail: plist 里写死了 BIN 的绝对路径。装 daemon 请用全局安装(npm i -g),
-  //           别用 npx —— npx 缓存目录会被清,清掉后 launchd 每半小时报一次「文件不存在」。
+function cleanupLegacyMac() {
+  for (const label of LEGACY_LABELS) {
+    const p = path.join(os.homedir(), 'Library', 'LaunchAgents', label + '.plist');
+    if (!fs.existsSync(p)) continue;
+    try { execFileSync('launchctl', ['unload', p], { stdio: 'ignore' }); } catch {}
+    try { fs.rmSync(p, { force: true }); } catch {}
+  }
+}
+
+function installMac() {
+  cleanupLegacyMac();
   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]);
+}
+
+function installWindows() {
+  fs.writeFileSync(VBS, buildVbs());
+  execFileSync('schtasks', windowsTaskArgs(), { stdio: 'ignore' });
+}
+
+function install() {
+  ensureSupported();
+  if (!requireKey()) {
+    console.error('还没登录,先跑  npx tokenforbes-cli  登录后再装。');
+    process.exit(1);
+  }
+  if (process.platform === 'darwin') installMac();
+  else installWindows();
   console.log('✓ 已开启后台自动同步,每 30 分钟自动上报一次。');
-  console.log('  任务文件: ' + PLIST);
-  console.log('  日志:     ' + LOG);
+  if (process.platform === 'darwin') {
+    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');
+function statusMac() {
+  if (!fs.existsSync(PLIST)) return false;
   let loaded = false;
   try { loaded = execFileSync('launchctl', ['list'], { encoding: 'utf8' }).includes(LABEL); } catch {}
   console.log(loaded ? '✓ 已安装并在运行,每 30 分钟自动上报。' : '已安装但未加载(重新 install 一下)。');
   console.log('  任务文件: ' + PLIST);
   console.log('  日志:     ' + LOG + '(看最近几次上报有没有出错)');
+  return true;
 }
 
-function uninstall() {
-  ensureMac();
+function statusWindows() {
+  try {
+    execFileSync('schtasks', ['/query', '/tn', LABEL], { stdio: 'ignore' });
+    console.log('✓ 已安装,Windows 会每 30 分钟自动上报。');
+    return true;
+  } catch {
+    return false;
+  }
+}
+
+function needsInstall() {
+  if (process.platform === 'darwin') {
+    if (!fs.existsSync(PLIST)) return true;
+    // 内容不含新版命令 = 旧格式;不含当前 PATH = 用户换过 node 安装位置,都重装自愈。
+    try {
+      const content = fs.readFileSync(PLIST, 'utf8');
+      return !content.includes('tokenforbes-cli@latest') || !content.includes(launchdPath());
+    } catch { return true; }
+  }
+  if (process.platform === 'win32') {
+    try { execFileSync('schtasks', ['/query', '/tn', LABEL], { stdio: 'ignore' }); return false; } catch { return true; }
+  }
+  return true;
+}
+
+function status() {
+  ensureSupported();
+  const ok = process.platform === 'darwin' ? statusMac() : statusWindows();
+  if (!ok) console.log('未安装。装它:  tokenforbes daemon install');
+}
+
+function uninstallMac() {
+  cleanupLegacyMac();
   if (!fs.existsSync(PLIST)) return console.log('本来就没装。');
   try { execFileSync('launchctl', ['unload', PLIST], { stdio: 'ignore' }); } catch {}
   fs.rmSync(PLIST, { force: true });
   console.log('✓ 已关闭后台自动同步,删除了任务文件。');
 }
 
+function uninstallWindows() {
+  try {
+    execFileSync('schtasks', ['/delete', '/tn', LABEL, '/f'], { stdio: 'ignore' });
+    console.log('✓ 已关闭后台自动同步。');
+  } catch {
+    console.log('本来就没装。');
+  }
+  fs.rmSync(VBS, { force: true });
+}
+
+function uninstall() {
+  ensureSupported();
+  if (process.platform === 'darwin') return uninstallMac();
+  return uninstallWindows();
+}
+
+export function ensureDaemonInstalled() {
+  if (!supported()) return;
+  // 老脚本清理独立于「要不要装」:就算新任务已装好,发现旧任务也顺手关掉。
+  if (process.platform === 'darwin') { try { cleanupLegacyMac(); } catch {} }
+  if (!needsInstall()) return;
+  try {
+    if (!requireKey()) return;
+    if (process.platform === 'darwin') installMac();
+    else installWindows();
+    console.log('后台自动同步已开启:之后每 30 分钟自动上报一次。');
+  } catch (e) {
+    console.log('后台自动同步没自动开启,可稍后手动跑: tokenforbes daemon install');
+  }
+}
+
 export function daemon(sub) {
   if (sub === 'install') return install();
   if (sub === 'status') return status();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "tokenforbes-cli",
-  "version": "0.1.3",
+  "version": "0.1.5",
   "description": "上报本地 AI 编程工具(Claude Code / Codex 等)的 token 用量到 Token 福布斯排行榜(ceorank.cn)。只读用量数字和设备型号，绝不碰你的代码内容。",
   "type": "module",
   "bin": {