magicrew-auth 0.3.18 → 0.3.24

package/src/hooks/usage.mjs

@@ -1,12 +1,34 @@
 // ⑤ auth.usage(getAuth) —— 额度
 //
-// 画进度条，**同时是路由排序输入**：`used` 缺了会被当成 0% = 满额，
-// 空号反而排到队首。
+// 超级麦吉的积分是「预付钱包」：`/api/v1/quota/points/total` 只回当前余额
+// `total_points` 和一批会过期的明细（`expiring_quota_details`），没有总量、
+// 没有已消耗；积分流水接口恒 500。钱包没有分母，不能编造 used/limit 百分比。
 //
-// ⚠ 预充值积分制放 aside：只显示余额、不参与排序；余额 0 时 used 置 100，
-//   让它老老实实排到最后。
+// magpie 0.1.1790（issue #3950）起，窗口可以直接报余额：
+//
+//   windows: [{
+//     name: "积分余额",
+//     balance: 6291,   // 还剩多少，单位见 unit
+//     unit: "积分",
+//     // 可选：runsOutAt / runsOut / display / model / models / notModels
+//   }]
+//
+// 不写 used / limit / aside —— 宿主把余额当读数额度，不画百分比；同 provider、
+// 同单位的账号按余额从多到少排，余额 0 视为用完。`runsOutAt` 取最早过期的
+// 一批，让 smart 路由知道谁的积分先失效。
+//
+// 钱包只有一个总数，但过期批次有先后：今日 23:59 到期的每日积分、其他限时
+// 批次、永久积分要看得见。它们在余额窗口之后作为 `aside` 文本行显示（只展示，
+// 不进 Allowance.Left —— 否则最小的那一条会把总余额压低）。
+//
+// 不要再把「今日 / 限时 / 永久」做成带 used/limit 的假百分比窗口；旧写法是
+// 宿主支持余额之前的临时协议，见 CHANGELOG 0.3.20。
+
+import { fetchPoints, SITES } from "../api.mjs"
 
-import { fetchPoints } from "../api.mjs"
+// 每日登录积分：国内 +200/天、国际 +50/天（实测，见 docs/internals.md）。
+// 今日批次的分母，只用于展示「今日还剩 / 今日额度」。
+const DAILY_LOGIN_GIFT = { cn: 200, intl: 50 }
 
 export function makeUsage(c) {
   return async (getAuth) => {
@@ -14,27 +36,107 @@ export function makeUsage(c) {
     if (!auth) return { error: "未登录", windows: [] }
     try {
       const p = await fetchPoints(auth.jwt, auth.org, c.host)
-      const nextExpiry = p.expiring?.[0]?.expiresAt
+      const total = Number(p?.total ?? 0)
+      if (!Number.isFinite(total)) return { error: "积分接口没有返回可用余额", windows: [] }
+      const batches = Array.isArray(p?.expiring) ? p.expiring : []
+      const windows = [{
+        name: "积分余额",
+        balance: total,
+        unit: "积分",
+        display: total > 0 ? `${fmt(total)} 积分` : "已耗尽",
+        ...(total > 0 ? runsOutOf(batches) : {}),
+      }]
+
+      if (total > 0) {
+        const today = beijingDay(Date.now())
+        const todayBatches = batches.filter((b) => b.points > 0 && beijingDay(parseServerTime(b.expiresAt)) === today)
+        const todayLeft = sum(todayBatches)
+        const timedLeft = sum(batches)
+        const timedRest = Math.max(0, timedLeft - todayLeft)
+        const permanent = Math.max(0, total - timedLeft)
+        const dailyCap = DAILY_LOGIN_GIFT[c.host === SITES.cn.host ? "cn" : "intl"]
+
+        // 只展示的批次行：没有 used/limit/balance，宿主把它们当 aside 文本，
+        // 不参与 Allowance.Left；路由仍只看上面的总余额。
+        if (todayLeft > 0) {
+          windows.push({
+            name: "今日积分",
+            display: `${fmt(todayLeft)} / ${fmt(dailyCap)} 积分 · 今日 23:59 到期`,
+            unit: "积分",
+            aside: true,
+          })
+        }
+        if (timedRest > 0) {
+          const next = pickEarliest(batches.filter((b) => !todayBatches.includes(b)))
+          const when = next ? ` · 最早 ${beijingDate(next.ms)} 到期` : ""
+          windows.push({
+            name: "限时积分",
+            display: `${fmt(timedRest)} 积分${when}`,
+            unit: "积分",
+            aside: true,
+          })
+        }
+        if (permanent > 0) {
+          windows.push({
+            name: "永久积分",
+            display: `${fmt(permanent)} 积分`,
+            unit: "积分",
+            aside: true,
+          })
+        }
+      }
+
       return {
-        plan: p.total > 0 ? "积分制" : "已耗尽",
-        balance: `${p.total.toLocaleString("zh-CN")} 积分`,
-        windows: [
-          p.total > 0
-            ? {
-                name: "积分余额",
-                used: 0,
-                display: `${p.total.toLocaleString("zh-CN")} 积分`,
-                amount: 0,
-                limit: p.total,
-                unit: "积分",
-                resetsAt: nextExpiry,
-                aside: true,
-              }
-            : { name: "积分余额", used: 100, display: "已耗尽" },
-        ],
+        plan: total > 0 ? "积分制" : "已耗尽",
+        balance: `${fmt(total)} 积分`,
+        windows,
       }
     } catch (e) {
       return { error: String(e?.message ?? e), windows: [] }
     }
   }
 }
+
+/** 最早失效的那批 → magpie 的 `runsOutAt` / `runsOut`。 */
+function runsOutOf(expiring) {
+  const next = pickEarliest(expiring)
+  if (!next) return {}
+  return {
+    runsOutAt: new Date(next.ms).toISOString(),
+    runsOut: next.points,
+  }
+}
+
+function pickEarliest(expiring) {
+  const list = Array.isArray(expiring) ? expiring : []
+  let best = null
+  for (const e of list) {
+    const pts = Number(e?.points ?? 0)
+    const ms = parseServerTime(e?.expiresAt)
+    if (!(pts > 0) || ms === null) continue
+    if (!best || ms < best.ms) best = { points: pts, ms }
+  }
+  return best
+}
+
+/** `"2026-10-28 23:59:59"` → 毫秒；解析不了返回 null */
+function parseServerTime(s) {
+  const m = /^(\d{4})-(\d{2})-(\d{2})[ T](\d{2}):(\d{2}):(\d{2})/.exec(String(s ?? ""))
+  if (!m) return null
+  const [, y, mo, d, h, mi, se] = m
+  const ms = new Date(Number(y), Number(mo) - 1, Number(d), Number(h), Number(mi), Number(se)).getTime()
+  return Number.isFinite(ms) ? ms : null
+}
+
+/** 北京时间的日期键 `YYYY-MM-DD`（服务端的时间都是 UTC+8） */
+function beijingDay(ms) {
+  if (ms === null) return ""
+  return new Date(ms + 8 * 3600e3).toISOString().slice(0, 10)
+}
+
+function beijingDate(ms) {
+  return beijingDay(ms)
+}
+
+const sum = (list) => list.reduce((n, b) => n + Number(b?.points ?? 0), 0)
+const fmt = (n) => Number(n).toLocaleString("zh-CN")

package/src/keys.mjs

@@ -28,6 +28,7 @@ import {
   otherSlot,
   currentKey,
   safeAccount,
+  listAccounts,
 } from "./key-store.mjs"
 
 /**
@@ -37,6 +38,52 @@ export function accountOf(auth) {
   return safeAccount(auth?.org || auth?.accountId || auth?.uid || "default")
 }
 
+/** 这个账号的 store 里到底有没有东西（登录过 / 接管过 / 续过期都算） */
+function storeHasIdentity(siteId, account) {
+  const st = readStore(siteId, account)
+  return !!(
+    st?.refreshToken ||
+    st?.keys?.a?.apiKey ||
+    st?.keys?.b?.apiKey
+  )
+}
+
+/**
+ * 这条请求属于哪个账号的 key 槽 —— 比 `accountOf` 多一层兜底。
+ *
+ * 为什么要多这一层（2026-10-28 实测的链路）：
+ *
+ *   网关路径下（ZCode 等 agent 从 magpie 的 `/v1/chat/completions` 进来），
+ *   `getAuth()` 返回的 auth **可能不带 org / accountId / uid** ——
+ *   于是 `accountOf()` 落到 `"default"`，`readStore(c.id, "default")` 读不到
+ *   真实账号的槽，`ensureKey` 只能退到「从 magpie 存的凭据接管一次」用
+ *   `auth.access`；而它过了 2 小时就 403，表现成
+ *   `502 {"message":"超级麦吉 did not stream: token无效"}`。
+ *
+ * 兜底只做**一半**，这是关键：
+ *
+ *   · 该站点**恰好一个账号**（绝大多数用户）→ 那就是它，安全
+ *   · 0 个 → 交给原来的「首次接管」
+ *   · **≥2 个 → 不猜**，退回 `"default"`
+ *
+ * 为什么不猜：key 与账号一一对应，用错账号的 key 就是拿别人的积分、别人的
+ * 组织出话，而且本地看不出来（不是报错，是错账）。多账号场景宁可让这次请求
+ * 失败，也不要借身份 —— 判据见 docs/internals.md 第十五节。
+ *
+ * ⚠ 多账号 + 网关路径的组合目前仍然无解：需要 magpie 把 provider 实例 id
+ *   传进来（`plugin-auth.json` 里已经有 `magicrew#p4nwa7` 这种带后缀的 key），
+ *   插件侧拿不到。在那之前这里只能保守。
+ */
+export function accountFor(c, auth) {
+  const direct = accountOf(auth)
+  if (direct !== "default") return direct
+  try {
+    const live = listAccounts(c.id).filter((acc) => storeHasIdentity(c.id, acc))
+    if (live.length === 1) return live[0]
+  } catch {}
+  return direct
+}
+
 /**
  * 拿到这个账号一把能用的 key。**永不抛** —— 续不上就交出手里那把（哪怕不新鲜），
  * 让请求自己去撞上游的错误，比在这里抛更能说明问题。
@@ -50,7 +97,7 @@ export async function ensureKey(c, getAuth, { leadMs } = {}) {
   try {
     auth = typeof getAuth === "function" ? await getAuth() : getAuth
   } catch {}
-  const account = accountOf(auth)
+  const account = accountFor(c, auth)
 
   const st = readStore(c.id, account)
   const cur = currentKey(st)