tokenforbes-cli 0.1.3 → 0.1.4

package/bin.js

@@ -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✓ 已批准,密钥已存到 ' + 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)); }
@@ -84,37 +95,44 @@ 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({ 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 });
+}
+
+// 把抓好的用量传上去。抽出来给 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/package.json

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