tokenforbes-cli 0.1.2 → 0.1.4

package/bin.js

@@ -30,42 +30,52 @@ if (cmd === 'login') {
 } else if (cmd === 'report') {
   await report();
 } else if (!cmd) {
-  // 一条命令走完:没登录先引导登录(登录成功会自动上报一次),登录过就直接同步。对齐竞品 npx 体验。
-  const { key } = loadConfig();
-  if (key) await report(); else await link();
+  // 一条命令走完:report 内部会兜底——没密钥或密钥失效(服务器重置过)都自动带去登录再上报。
+  await report();
 } else {
   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→ 登录确认 ' + url);
-  console.log('  验证码: ' + s.user_code);
-  console.log('  浏览器会自动打开；没反应就手动复制上面的链接。\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();   // 登录即上报,一条命令闭环,别让用户卡在「登录完不知道下一步」
+      console.log('\n✓ 已批准,密钥已存到 ' + CONFIG);
+      await uploadDays(days, { server, key: p.key, device });
       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');
+  die('\n登录超时或验证码已过期，重新跑一次  npx -y tokenforbes-cli');
 }
 
 function sleep(ms) { return new Promise((r) => setTimeout(r, ms)); }
@@ -85,34 +95,63 @@ 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 }); // 注册完顺手上报,别让用户再跑一条
 }
 
-async function report() {
+async function report({ allowRelink = true } = {}) {
   const c = loadConfig();
-  if (!c.key) die('还没登录。先跑  npx tokenforbes-cli  (浏览器一键登录),或 register / login');
+  // 没密钥:直接带去登录(link 会先弹浏览器,再后台抓用量),不让用户对着报错发愣。
+  if (!c.key) {
+    if (allowRelink) return link();
+    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('没读到用量数据。');
   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 });
+}
+
+// 把抓好的用量传上去。抽出来给 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 j = await post(server + '/api/report', { key: c.key, device, days });
-  console.log(`上报成功 ✓ 已记录 ${j.days} 天。`);
+  console.log(`共 ${days.length} 天,合计约 ${Math.round(total / 1e4).toLocaleString()} 万 token,正在上传…`);
+  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 的版本:给登录/注册这类「失败就该停」的调用用。
 async function post(url, body) {
+  const res = await postRaw(url, body);
+  if (!res.ok) die('失败:' + (res.json?.error || res.netError || res.status));
+  return res.json;
+}
+
+// 不 die、把结果原样返回:给 report 这类「失败要自己判断怎么办」的调用用。
+async function postRaw(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}`);
+    return { ok: false, netError: `连不上服务器(${url}):${e.message}` };
   }
-  const j = await res.json().catch(() => ({}));
-  if (!res.ok) die('失败:' + (j.error || res.status));
-  return j;
+  const json = await res.json().catch(() => ({}));
+  return { ok: res.ok, status: res.status, json };
 }
 
 function die(msg) { console.error(msg); process.exit(1); }

package/package.json

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