tokenforbes-cli 0.1.2 → 0.1.3

package/bin.js

@@ -30,9 +30,8 @@ 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>');
 }
@@ -44,12 +43,12 @@ async function link() {
   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 + '（网页上核对一致再点确认）\n');
   openBrowser(url);
 
-  process.stdout.write('等待批准…');
+  process.stdout.write('已在浏览器等你点确认…');
   const until = Date.now() + s.expires_in * 1000;
   while (Date.now() < until) {
     await sleep(s.interval * 1000);
@@ -58,7 +57,7 @@ async function link() {
       c.key = p.key; c.server = server; saveConfig(c);
       console.log('\n✓ 已批准，密钥已存到 ' + CONFIG);
       console.log('\n顺手把你的本地用量传上去：\n');
-      await report();   // 登录即上报,一条命令闭环,别让用户卡在「登录完不知道下一步」
+      await report({ allowRelink: false });   // 登录即上报,一条命令闭环;刚拿到新密钥,别再触发二次登录
       console.log('\n想让它每半小时自动后台上报?  npm i -g tokenforbes-cli 后跑  tokenforbes daemon install');
       return;
     }
@@ -88,31 +87,53 @@ async function register(nickname) {
   console.log('现在跑  tokenforbes report  上报你的用量。');
 }
 
-async function report() {
+async function report({ allowRelink = true } = {}) {
   const c = loadConfig();
-  if (!c.key) die('还没登录。先跑  npx tokenforbes-cli  (浏览器一键登录),或 register / login');
+  // 没密钥:直接带去登录(登录成功会自动上报),不再让用户对着报错发愣。
+  if (!c.key) {
+    if (allowRelink) return link();
+    die('还没登录。先跑  npx tokenforbes-cli  (浏览器一键登录),或 register / login');
+  }
   const server = c.server || DEFAULT_SERVER;
-  console.log('读取本地用量(ccusage)…');
+  console.log('正在读取本地用量(ccusage)…');
   const days = readUsage();
-  if (!days.length) return console.log('没读到用量数据。');
+  if (!days.length) return console.log('没读到用量数据(这台电脑还没用过 Claude Code / Codex?)。');
   const device = detectDevice();
   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: 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));
 }
 
+// 会抛错就 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.3",
   "description": "上报本地 AI 编程工具(Claude Code / Codex 等)的 token 用量到 Token 福布斯排行榜(ceorank.cn)。只读用量数字和设备型号，绝不碰你的代码内容。",
   "type": "module",
   "bin": {