tokenforbes-cli 0.1.4 → 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';
@@ -69,7 +69,7 @@ async function link(preDays) {
       c.key = p.key; c.server = server; saveConfig(c);
       console.log('\n✓ 已批准,密钥已存到 ' + CONFIG);
       await uploadDays(days, { server, key: p.key, device });
-      console.log('\n想让它每半小时自动后台上报?  npm i -g tokenforbes-cli 后跑  tokenforbes daemon install');
+      ensureDaemonInstalled();
       return;
     }
     if (p.status === 'expired') break;
@@ -97,6 +97,7 @@ async function register(nickname) {
   console.log(`注册成功,昵称「${j.nickname}」。密钥已存到 ${CONFIG}`);
   console.log('正在读取本地用量(ccusage)…');
   await uploadDays(readUsage(), { server, key: j.key, device }); // 注册完顺手上报,别让用户再跑一条
+  ensureDaemonInstalled();
 }
 
 async function report({ allowRelink = true } = {}) {
@@ -122,6 +123,7 @@ async function report({ allowRelink = true } = {}) {
   // 密钥有效:抓用量并上传。
   console.log('正在读取本地用量(ccusage)…');
   await uploadDays(readUsage(), { server, key: c.key, device });
+  ensureDaemonInstalled();
 }
 
 // 把抓好的用量传上去。抽出来给 report / link / register 共用,避免各写一遍、避免重复抓。

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