magicrew-auth 0.3.25 → 0.3.28

package/src/hooks/loader.mjs

@@ -1,14 +1,18 @@
 // ④ auth.loader(getAuth) —— 请求接管
 //
 // 返回 baseURL / apiKey / headers / fetch(a,b)，magpie 拿它发真实请求。
-// 这是唯一能改请求体的地方。本项目用它干两件事：
+// 这是唯一能改请求体和上游错误状态的地方。本项目用它干三件事：
 //
 //   · 国际站网关强制要 organization-code（国内站不要），不带就是
 //     HTTP 500 "Internal Server Error."（22 字节纯文本，不是 JSON）
 //   · 推理模型低于 512 的 max_tokens 会被 reasoning 吃光，content 返回空串
+//   · 上游用 HTTP 400 报「可用积分已不足」：这个 400 在 magpie 的
+//     retryable() 里不被识别为可重试（400 分支只查 quotaWords，而这条文案
+//     命中的是 creditWords 的「充值」），于是整轮直接失败、账号也不 rest。
+//     插件把它改写为语义正确的 402 Payment Required，让 magpie 换下一个账号。
 //
-// ⚠ 不要在这里读流、不要重试换号、不要手拼 x-api-key ——
-//   前两件 magpie 做了，第三件看 SDK 包自动加。见 README「最小交付面」。
+// ⚠ 不要在这里读流、不要手拼 x-api-key ——
+//   流由 magpie 处理，头由 SDK 包自动加。见 README「最小交付面」。
 
 import { ensureKey, accountOf } from "../keys.mjs"
 
@@ -162,6 +166,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，四处改写都在这**。
  *
@@ -217,12 +383,69 @@ 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
     }
   }
 
   return changed ? JSON.stringify(j) : raw
 }
 
+// ═══ 上游「积分/余额不足」的 400 → 402 ═══════════════════════════════════
+//
+// 实测（2026-10-28 23:38，magpie 0.1.1899，DSH 走 group）：
+//
+//   try #1  magicrew@181****710  200  fail=loop   （模型自己绕圈，不是账号）
+//   try #2  magicrew@181****710  400  fail=credit
+//           body: 超级麦吉: 非常抱歉，当前对话需要调用大模型 / 付费工具，
+//                 您的可用积分已不足。…通过「充值」通道补充积分…
+//   route   done=true status=400   ← 第三个账号（余额 6,467）一次都没被问
+//
+// magpie 的 failure() 已经把这条归成 failCredit（creditWords 命中「充值」），
+// 但 retryable() 的 400/422 分支只查 quotaWords（quota|余额|额度|欠费|…），
+// 不含「积分」「充值」⇒ hw.failed() 为 false ⇒ 走不到 `!last && hw.failed()`
+// 的换账号分支，连 rest 都不记。账号于是永远留在轮换里，下一轮继续撞。
+//
+// 修法只改插件这一层：400/422 + 这条文案 → 402 Payment Required。
+// 402 在 magpie 的 retryable() 里无条件 true，failure(402) 也是 failCredit，
+// 会 rest 该账号 30 分钟并继续问同 provider 的下一个账号（或组的下一个成员）。
+// body 原样保留，最终错误文案不变。
+//
+// 注意：只转换「上游拒绝」这一个语义；400 的 schema 错误、参数错误绝不碰。
+const CREDIT_EXHAUSTED = /(?:积分|余额)(?:已|额度)?不足|可用积分不足|insufficient\s+(?:credit|credits|balance|fund|funds|point|points)|(?:credit|credits|point|points|balance)\s+(?:is\s+|are\s+)?(?:insufficient|exhausted)|out\s+of\s+(?:credit|credits|points?|balance)/i
+
+/** 这条错误文案是不是「账户没钱了」（而不是请求本身有问题） */
+export function isCreditExhausted(text) {
+  return CREDIT_EXHAUSTED.test(String(text ?? ""))
+}
+
+/**
+ * 把上游的「积分/余额不足」400/422 改写成 402，让 magpie 的 `retryable()`
+ * 愿意换下一个账号；其余响应原样返回（同一个 Response 对象）。
+ *
+ * @param {Response} res
+ * @returns {Promise<Response>}
+ */
+export async function rewriteCreditRefusal(res) {
+  if (!res || (res.status !== 400 && res.status !== 422)) return res
+  let text
+  try {
+    // 只克隆错误响应：成功响应（尤其是 SSE）不能在这里读，否则会破坏流
+    text = await res.clone().text()
+  } catch {
+    return res
+  }
+  if (!isCreditExhausted(text)) return res
+  const headers = new Headers(res.headers)
+  // body 已由 undici 解压，原 content-encoding/length 不再成立
+  headers.delete("content-length")
+  headers.delete("content-encoding")
+  return new Response(text, {
+    status: 402,
+    statusText: "Payment Required",
+    headers,
+  })
+}
+
 export function makeLoader(c) {
   return async (getAuth) => {
     const auth = await getAuth()
@@ -266,12 +489,12 @@ export function makeLoader(c) {
         // 四处请求体改写，全在 rewriteRequestBody 里（那里有实测出处）
         body = rewriteRequestBody(body)
 
-        return fetch(url, {
+        return rewriteCreditRefusal(await fetch(url, {
           method: init?.method ?? "POST",
           headers,
           body,
           signal: init?.signal ?? req?.signal ?? AbortSignal.timeout(300_000),
-        })
+        }))
       },
     }
   }