magicrew-auth 0.3.26 → 0.3.29

package/src/hooks/loader.mjs

@@ -14,7 +14,45 @@
 // ⚠ 不要在这里读流、不要手拼 x-api-key ——
 //   流由 magpie 处理，头由 SDK 包自动加。见 README「最小交付面」。
 
-import { ensureKey, accountOf } from "../keys.mjs"
+import { ensureKey, accountOf, keyCandidates, adoptKey } from "../keys.mjs"
+
+/**
+ * 上游「这个账号不行」的两种形态（2026-10-29 实测）：
+ *
+ *   · HTTP 403
+ *   · **HTTP 200 + `{"code":403,"message":"token无效"}`**  ← 绝大多数是这种
+ *
+ * 所以光看状态码会漏。只解析 JSON（SSE 一律不碰，读流会破坏它）。
+ */
+async function isKeyRejected(res) {
+  if (!res) return false
+  if (res.status === 401 || res.status === 403) return true
+  const ct = res.headers.get("content-type") || ""
+  if (res.status !== 200 || !ct.includes("application/json")) return false
+  let text = ""
+  try {
+    text = await res.clone().text()
+  } catch {
+    return false
+  }
+  if (text.length > 8192) return false
+  let body
+  try {
+    body = JSON.parse(text)
+  } catch {
+    return false
+  }
+  // 只认**业务错误码**。正常 chat 完成的 JSON 是 {choices:[…]}，没有顶层 code；
+  // 用文本正则会把模型回复里恰好出现的 "unauthorized" 当成 key 拒绝 ——
+  // 那会白白重发请求（可能重复计费），甚至最后把正常响应丢掉换成 402。
+  const code = Number(body?.code ?? body?.error?.code)
+  if (code === 401 || code === 403) return true
+  const msg = String(body?.message ?? body?.error?.message ?? "")
+  if (!msg) return false
+  if (/token无效|token 无效|invalid token/i.test(msg)) return true
+  // 兜底：只有 message、没有 code 的少量形态；有 choices 的正常响应绝不判
+  return !Array.isArray(body?.choices) && /unauthorized/i.test(msg)
+}
 
 /**
  * 把 JSON Schema 里的 `additionalProperties` **非布尔值**原地换成 `true`。
@@ -166,6 +204,168 @@ export function loosenEmptyItems(node) {
   })
 }
 
+/**
+ * 修 **`items` 子树内部**的 schema：上游的 JSON Schema 校验器对数组元素的
+ * schema 比对根节点**更严**，下面这些形态在 `items` 子树里一律 400：
+ *
+ *   anyOf / oneOf / allOf / not / const   → 400
+ *   if / then / else                      → 400
+ *   {}（空 schema）                       → 400
+ *
+ * ⚠ 同样形态放在**顶层 parameters** 里上游**收**（200）—— 只有 `items` 子树
+ * 里才炸。所以限制必须精确落在 `items` 之下，不能全局摊平 union。
+ *
+ * 实测（2026-10-28~29，直打国内站 `i-magic-service.letsmagic.cn`，
+ * deepseek-v4.1-flash，同一把 key、同一份请求，只改这一个字段；
+ * `_probe/items-subtree-upstream.mjs` 可复跑）：
+ *
+ *   items 内部 {x:{anyOf:[{type:"string"},{type:"null"}]}} → 400
+ *   items 内部 {x:{oneOf:[…]}}                            → 400
+ *   items 内部 {x:{allOf:[…]}}                            → 400
+ *   items 内部 {x:{not:{type:"string"}}}                  → 400
+ *   items 内部 {x:{const:"a"}}                            → 400
+ *   items 内部 {x:{if:…,then:…}}                          → 400
+ *   items 内部 {x:{}}                                     → 400
+ *   items 内部 {x:{type:["object","string",…,"null"]}}    → 200   ← 「任意值」的合法写法
+ *   items 内部 {x:{type:"object"}}                        → 200
+ *   items 内部 {x:{type:"string",pattern:"^a$",enum:[…]}} → 200
+ *   items 内部 {x:{$ref:"#/$defs/s"}}, $defs 同层        → 200
+ *   items 是数组（tuple 形式）    国内站 → 200，**国际站 → 400**（两站不一致）
+ *   顶层   {x:{anyOf:…}} / {x:{const:…}} / {x:{not:…}} / {x:{}} → 200
+ *   （国内站 `i-magic-service.letsmagic.cn` 与国际站 `magic-service.magicrew.ai`
+ *     逐条对比过：除 tuple 形式外结论一致）
+ *
+ * 命中来源：**MCP 服务器**的工具 schema。REA MCP（mcp__rea__build_call_path 等
+ * 7 个工具）的 `functions` 数组参数，元素 schema 用了
+ * `anyOf: [{...}, {type:"null"}]`（可选参数的标准写法）、`oneOf`，还有
+ * `{}` 空 schema —— 一份 65 工具的请求里就有 7 个工具中招，整轮 400。
+ *
+ * 修法（全部是**降级放宽**，方向安全：宁可约束松一点，也不要整轮失败）：
+ *
+ *   anyOf / oneOf / allOf  → 取一个分支内联（**优先非纯 null 的分支**：
+ *                              `anyOf:[{…},{type:"null"}]` 取前者；反过来写时
+ *                              照抄第一个会把「可选」压成「只能 null」，那是收紧）
+ *   not / const / if+then+else → 删除（无等价物；删掉就是放宽）
+ *   {}                     → `{type:["object","string","number","boolean","array","null"]}`
+ *                             （六种 JSON 类型全覆盖 = 任意值，与 `{}` 严格等价；
+ *                              **不能**用 `{type:"object"}` —— 那会把「任意」收紧成「只能是对象」）
+ *
+ * ⚠ 两个容易漏的坑（都踩过）：
+ *   1. 删除 `not` / `const` 后节点可能变成 `{}` —— 必须在删除后再查一次空对象，
+ *      否则改完仍是 400（PR #10 初版就漏在这里，`not`/`const` 两例改写后仍 400）。
+ *   2. `items` 子树里 `{}` 和 `true` 都被拒，所以「任意值」只能落成显式 type 数组。
+ *
+ * ⚠ 只走 `items` 子树：顶层 parameters 的 union 是上游**收**的，动了它
+ * 会白白丢掉约束。
+ *
+ * @returns {boolean} 有没有改到
+ */
+export function loosenItemsSchema(node) {
+  let changed = false
+  // 用 walkSchema 找**每一个** schema 位置上的 items（它只进入真正的 schema
+  // 位置：properties 的值、union 的分支、items 的值……不碰 enum/default 的字面值）。
+  walkSchema(node, (n) => {
+    if (!Object.hasOwn(n, "items")) return false
+    const v = n.items
+    // tuple 形式（draft-04）：`items` 是 schema **数组**。
+    // 两站行为不一致 —— 国内站收、**国际站拒**（`[{"type":"string"}] is not of
+    // type 'object'`），实测见 `_probe/items-subtree-upstream.mjs`。
+    // 数组里的逐位约束在改写里表达不了（`anyOf` 在这层就是 400），
+    // 只能整体落成「任意值」—— 纯放宽，不会让原本合法的调用被拒。
+    if (Array.isArray(v)) {
+      n.items = { type: [...ANY_SCHEMA_TYPE] }
+      changed = true
+      return false
+    }
+    if (!v || typeof v !== "object") return false
+    if (fixItemsSubtree(v)) changed = true
+    return false
+  })
+  return changed
+}
+
+// 空 schema 在 JSON Schema 里等价于 `true` = 接受任意 JSON 值。但上游对
+// `items` 子树里的 `{}` 和 `true` 都拒，只能落成显式的「任意类型」schema。
+const ANY_SCHEMA_TYPE = ["object", "string", "number", "boolean", "array", "null"]
+
+/** 把一个空 schema 节点原地改成「任意值」。返回是否真的改到（只对空对象生效）。 */
+function toAnySchema(n) {
+  if (Object.keys(n).length > 0) return false
+  n.type = [...ANY_SCHEMA_TYPE]
+  return true
+}
+
+/** 分支是不是只声明了 null —— 取它内联会把「可选」压成「只能 null」（收紧） */
+function isNullOnly(branch) {
+  if (!branch || typeof branch !== "object" || Array.isArray(branch)) return false
+  if (branch.type === "null") return true
+  return Array.isArray(branch.type) && branch.type.length === 1 && branch.type[0] === "null"
+}
+
+/**
+ * 在 `items` 子树里就地修：union 摊平、const/not/if 删除、空 schema 换形状。
+ *
+ * 递归同样遵守 schema 位置纪律（见 {@link SCHEMA_MAPS} / {@link SCHEMA_LIST} /
+ * {@link SCHEMA_ONE}）：`enum` / `default` / `examples` 里的字面值不是 schema，
+ * 不能动。
+ *
+ * @returns {boolean} 有没有改到
+ */
+function fixItemsSubtree(n) {
+  if (!n || typeof n !== "object" || Array.isArray(n)) return false
+  let dirty = false
+
+  // 空 schema {} → 显式「任意类型」（与 {} 严格等价）
+  if (toAnySchema(n)) return true
+
+  for (const [k, v] of Object.entries(n)) {
+    if (k === "anyOf" || k === "oneOf" || k === "allOf") {
+      if (!Array.isArray(v) || v.length === 0) {
+        delete n[k]
+        dirty = true
+        continue
+      }
+      // 优先第一个非纯 null 的分支（顺序反过来时不至于把可选压成「只能 null」）
+      const branch =
+        v.find((b) => b && typeof b === "object" && !Array.isArray(b) && !isNullOnly(b)) ??
+        v.find((b) => b && typeof b === "object" && !Array.isArray(b))
+      delete n[k]
+      dirty = true
+      if (branch) {
+        // 分支本身也可能带 union / 空 schema ⇒ 先递归修，再内联
+        if (fixItemsSubtree(branch)) dirty = true
+        Object.assign(n, branch)
+      }
+      continue
+    }
+    if (k === "not" || k === "const" || k === "if" || k === "then" || k === "else") {
+      delete n[k]
+      dirty = true
+      continue
+    }
+    // 其余按 schema 位置递归
+    if (SCHEMA_MAPS.has(k) && !Array.isArray(v)) {
+      for (const child of Object.values(v)) if (fixItemsSubtree(child)) dirty = true
+      continue
+    }
+    if (SCHEMA_LIST.has(k) && Array.isArray(v)) {
+      for (const child of v) if (fixItemsSubtree(child)) dirty = true
+      continue
+    }
+    if (SCHEMA_ONE.has(k)) {
+      if (Array.isArray(v)) {
+        for (const child of v) if (fixItemsSubtree(child)) dirty = true
+      } else if (fixItemsSubtree(v)) {
+        dirty = true
+      }
+    }
+  }
+  // 删除 not/const/if… 之后节点可能变空 —— 空 schema 在 items 子树里仍然 400，
+  // 必须补成「任意值」。
+  if (toAnySchema(n)) dirty = true
+  return dirty
+}
+
 /**
  * 改写发往上游的请求体。**一次 parse，四处改写都在这**。
  *
@@ -221,6 +421,7 @@ export function rewriteRequestBody(raw) {
       if (!p || typeof p !== "object") continue
       if (loosenAdditionalProperties(p)) changed = true
       if (loosenEmptyItems(p)) changed = true
+      if (loosenItemsSchema(p)) changed = true
     }
   }
 
@@ -284,6 +485,8 @@ export async function rewriteCreditRefusal(res) {
 }
 
 export function makeLoader(c) {
+  /** 已经写回 store 的 key —— 避免每个请求都去读一次 store */
+  const adopted = new Set()
   return async (getAuth) => {
     const auth = await getAuth()
     if (!auth) return {}
@@ -326,12 +529,60 @@ export function makeLoader(c) {
         // 四处请求体改写，全在 rewriteRequestBody 里（那里有实测出处）
         body = rewriteRequestBody(body)
 
-        return rewriteCreditRefusal(await fetch(url, {
-          method: init?.method ?? "POST",
-          headers,
-          body,
-          signal: init?.signal ?? req?.signal ?? AbortSignal.timeout(300_000),
-        }))
+        const signal = init?.signal ?? req?.signal ?? AbortSignal.timeout(300_000)
+        const sendWith = (apiKey) =>
+          fetch(url, {
+            method: init?.method ?? "POST",
+            headers: { ...headers, authorization: `Bearer ${apiKey}` },
+            body,
+            signal,
+          })
+
+        // 同一账号可能握着两把 key（magpie 的 refresh 一条、插件 store 一条），
+        // 哪把还有效本地无法预判 —— 静态挑必然有一半踩空，只能逐把试。
+        // 详见 src/keys.mjs 的 keyCandidates 注释（2026-10-29 五个账号的实测表）。
+        // 传 fresh 而不是 getAuth：候选与下面 adoptKey 必须落在同一个账号上，
+        // 不能让两次 getAuth() 返回不同结果。
+        const cands = await keyCandidates(c, fresh)
+        if (!cands.length) throw new Error(`${c.SHORT} 未登录（拿不到可用的 key）`)
+
+        for (const cand of cands) {
+          const res = await sendWith(cand.apiKey)
+          if (!(await isKeyRejected(res))) {
+            // 这把能用 —— 写回 store，让 ensureKey / 额度那几路也用同一把
+            if (!adopted.has(cand.apiKey)) {
+              adopted.add(cand.apiKey)
+              adoptKey(c, fresh, cand.apiKey, cand.expiresAt)
+            }
+            return rewriteCreditRefusal(res)
+          }
+          // 这把被拒：响应不再使用，显式取消 body —— 否则逐把试时
+          // 每条死连接的响应体都挂着，等 GC 才释放。
+          try {
+            await res.body?.cancel()
+          } catch {}
+        }
+
+        // 这个账号的 key 全被拒。上游给的是 403「token无效」，magpie 会判成
+        // 「不可重试」直接抛给客户端（ZCode 显示 auth_failed / token无效）。
+        // 改写成 402：magpie 的 failure(402) 是 failCredit → 把这个账号 rest 掉、
+        // 去问下一个账号 / 组的下一个成员。与 0.3.26 的「积分不足」同一条通路。
+        return new Response(
+          JSON.stringify({
+            error: {
+              message:
+                `${c.SHORT}：本账号的 ${cands.length} 把密钥均已被上游拒绝，已交给 magpie 换下一个账号。` +
+                `（常见于登录 / 续期分叉；请重新登录这个账号）`,
+              type: "account_unavailable",
+              code: 402,
+            },
+          }),
+          {
+            status: 402,
+            statusText: "Payment Required",
+            headers: { "content-type": "application/json" },
+          },
+        )
       },
     }
   }