feat: 添加HF Space卡死自愈功能，增强代理健康检查逻辑，更新配置文件

cf-worker/.wrangler/tmp/deploy-PAi0l6/index.js

@@ -0,0 +1,215 @@
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+
+// src/index.js
+var LOGIN_PATH = "/admin/login";
+var CHECK_ALL_PATH = "/admin/proxies/check-all";
+var DEFAULT_HF_SPACE_REPO = "a3216/ds2api";
+var RESTART_COOLDOWN_MS = 5 * 60 * 1e3;
+var LAST_RESTART_KV_KEY = "last_restart";
+function vercelHeaders(env) {
+  const headers = {};
+  const bypass = env.VERCEL_BYPASS_TOKEN;
+  if (bypass) {
+    headers["x-vercel-protection-bypass"] = bypass;
+  }
+  return headers;
+}
+__name(vercelHeaders, "vercelHeaders");
+async function login(baseURL, adminKey, extraHeaders) {
+  const res = await fetch(`${baseURL}${LOGIN_PATH}`, {
+    method: "POST",
+    headers: { "Content-Type": "application/json", ...extraHeaders },
+    body: JSON.stringify({ admin_key: adminKey, expire_hours: 1 })
+  });
+  if (!res.ok) {
+    const text = await res.text();
+    throw new Error(`login failed (${res.status}): ${text}`);
+  }
+  const data = await res.json();
+  return data.token;
+}
+__name(login, "login");
+async function checkAll(baseURL, token, extraHeaders) {
+  const res = await fetch(`${baseURL}${CHECK_ALL_PATH}`, {
+    method: "POST",
+    headers: {
+      "Content-Type": "application/json",
+      Authorization: `Bearer ${token}`,
+      ...extraHeaders
+    }
+  });
+  if (!res.ok) {
+    const text = await res.text();
+    throw new Error(`check-all failed (${res.status}): ${text}`);
+  }
+  return res.json();
+}
+__name(checkAll, "checkAll");
+async function runCheck(env) {
+  const baseURL = (env.DS2API_BASE_URL || "").replace(/\/+$/, "");
+  const adminKey = env.ADMIN_KEY || "";
+  if (!baseURL || !adminKey) {
+    throw new Error("DS2API_BASE_URL and ADMIN_KEY secrets must be set");
+  }
+  const extraHeaders = vercelHeaders(env);
+  const token = await login(baseURL, adminKey, extraHeaders);
+  const result = await checkAll(baseURL, token, extraHeaders);
+  return result;
+}
+__name(runCheck, "runCheck");
+async function fetchSpaceRuntime(repo, token) {
+  const url = `https://huggingface.co/api/spaces/${repo}/runtime`;
+  const res = await fetch(url, {
+    headers: { Authorization: `Bearer ${token}` }
+  });
+  if (!res.ok) {
+    const text = await res.text();
+    throw new Error(`runtime query failed (${res.status}): ${text}`);
+  }
+  const data = await res.json();
+  return {
+    stage: data.stage,
+    hardwareCurrent: data.hardware ? data.hardware.current : null
+  };
+}
+__name(fetchSpaceRuntime, "fetchSpaceRuntime");
+function isSpaceStuck(runtime) {
+  if (!runtime) return true;
+  if (runtime.stage !== "RUNNING") return true;
+  if (!runtime.hardwareCurrent) return true;
+  return false;
+}
+__name(isSpaceStuck, "isSpaceStuck");
+async function restartSpace(repo, token) {
+  const url = `https://huggingface.co/api/spaces/${repo}/restart`;
+  const res = await fetch(url, {
+    method: "POST",
+    headers: { Authorization: `Bearer ${token}` }
+  });
+  if (!res.ok) {
+    const text = await res.text();
+    throw new Error(`restart failed (${res.status}): ${text}`);
+  }
+  return true;
+}
+__name(restartSpace, "restartSpace");
+async function getLastRestart(kv) {
+  if (!kv) return 0;
+  try {
+    const v = await kv.get(LAST_RESTART_KV_KEY);
+    const n = parseInt(v, 10);
+    return Number.isFinite(n) ? n : 0;
+  } catch {
+    return 0;
+  }
+}
+__name(getLastRestart, "getLastRestart");
+async function setLastRestart(kv) {
+  if (!kv) return;
+  try {
+    await kv.put(LAST_RESTART_KV_KEY, String(Date.now()));
+  } catch {
+  }
+}
+__name(setLastRestart, "setLastRestart");
+async function runSpaceCheck(env) {
+  const token = env.HF_TOKEN || "";
+  if (!token) {
+    throw new Error("HF_TOKEN secret must be set for space self-healing");
+  }
+  const repo = (env.HF_SPACE_REPO || DEFAULT_HF_SPACE_REPO).trim();
+  const runtime = await fetchSpaceRuntime(repo, token);
+  if (!isSpaceStuck(runtime)) {
+    return {
+      ok: true,
+      action: "none",
+      repo,
+      stage: runtime.stage,
+      hardwareCurrent: runtime.hardwareCurrent,
+      message: "space healthy, no action"
+    };
+  }
+  const last = await getLastRestart(env.SPACE_STATE);
+  const elapsed = Date.now() - last;
+  if (elapsed < RESTART_COOLDOWN_MS) {
+    const waitMs = RESTART_COOLDOWN_MS - elapsed;
+    return {
+      ok: true,
+      action: "cooldown",
+      repo,
+      stage: runtime.stage,
+      hardwareCurrent: runtime.hardwareCurrent,
+      message: `space stuck but in restart cooldown, wait ${Math.ceil(waitMs / 1e3)}s`
+    };
+  }
+  await restartSpace(repo, token);
+  await setLastRestart(env.SPACE_STATE);
+  return {
+    ok: true,
+    action: "restarted",
+    repo,
+    stage: runtime.stage,
+    hardwareCurrent: runtime.hardwareCurrent,
+    message: "space was stuck, restart triggered"
+  };
+}
+__name(runSpaceCheck, "runSpaceCheck");
+function isSpaceCheckCron(cron) {
+  return cron === "*/10 * * * *";
+}
+__name(isSpaceCheckCron, "isSpaceCheckCron");
+var index_default = {
+  async scheduled(event, env, ctx) {
+    if (isSpaceCheckCron(event.cron)) {
+      ctx.waitUntil(
+        runSpaceCheck(env).then((r) => {
+          if (r.action === "restarted") {
+            console.warn(`[space] ${r.message} repo=${r.repo} stage=${r.stage} hardware=${r.hardwareCurrent}`);
+          } else {
+            console.log(`[space] ${r.message} repo=${r.repo} stage=${r.stage} hardware=${r.hardwareCurrent}`);
+          }
+        }).catch((err) => {
+          console.error(`[space] check failed: ${err.message}`);
+        })
+      );
+      return;
+    }
+    ctx.waitUntil(
+      runCheck(env).then((result) => {
+        const items = result.items || [];
+        const healthy = items.filter((i) => i.healthy && !i.disabled).length;
+        const banned = items.filter((i) => i.disabled).length;
+        console.log(
+          `proxy health check done: ${healthy}/${items.length} healthy, ${banned} banned`
+        );
+      }).catch((err) => {
+        console.error(`proxy health check failed: ${err.message}`);
+      })
+    );
+  },
+  async fetch(request, env) {
+    const url = new URL(request.url);
+    if (url.pathname === "/check") {
+      try {
+        const result = await runCheck(env);
+        return Response.json(result);
+      } catch (err) {
+        return Response.json({ error: err.message }, { status: 500 });
+      }
+    }
+    if (url.pathname === "/check-space") {
+      try {
+        const result = await runSpaceCheck(env);
+        return Response.json(result);
+      } catch (err) {
+        return Response.json({ error: err.message }, { status: 500 });
+      }
+    }
+    return new Response("not found", { status: 404 });
+  }
+};
+export {
+  index_default as default
+};
+//# sourceMappingURL=index.js.map

cf-worker/.wrangler/tmp/deploy-PAi0l6/index.js.map

@@ -0,0 +1,8 @@
+{
+  "version": 3,
+  "sources": ["../../../src/index.js"],
+  "sourceRoot": "F:\\deepseek\\deepseek2api-main_2\\ds2api\\cf-worker\\.wrangler\\tmp\\deploy-PAi0l6",
+  "sourcesContent": ["/**\n * Cloudflare Worker \u2014 ds2api \u4EE3\u7406\u5065\u5EB7\u68C0\u67E5 + HF Space \u5361\u6B7B\u81EA\u6108\n *\n * \u90E8\u7F72\u6B65\u9AA4:\n *   1. cd cf-worker && npm install\n *   2. \u8BBE\u7F6E secrets:\n *        wrangler secret put DS2API_BASE_URL\n *        wrangler secret put ADMIN_KEY\n *        wrangler secret put VERCEL_BYPASS_TOKEN   (\u5982\u679C\u5F00\u4E86 Vercel Deployment Protection)\n *        wrangler secret put HF_TOKEN              (HF API token\uFF0C\u7528\u4E8E Space \u5361\u6B7B\u81EA\u6108)\n *   3. wrangler deploy\n *\n * \u4E24\u4E2A cron\uFF08\u89C1 wrangler.toml\uFF09\uFF1A\n *   - 5 \u5C0F\u65F6\u4E00\u6B21\uFF08\"0 0,5,10,15,20 * * *\" \u7B49\u4EF7\u5199\u6CD5\uFF09\uFF1A\u4EE3\u7406\u5065\u5EB7\u68C0\u67E5\uFF08runCheck\uFF09\n *   - 10 \u5206\u949F\u4E00\u6B21\uFF1AHF Space \u5361\u6B7B\u68C0\u6D4B+\u81EA\u52A8 restart\uFF08runSpaceCheck\uFF09\n *\n * \u624B\u52A8\u89E6\u53D1\uFF1A\n *   GET/POST https://<worker>/check          \u2192 \u4EE3\u7406\u5065\u5EB7\u68C0\u67E5\n *   GET/POST https://<worker>/check-space    \u2192 Space \u5361\u6B7B\u68C0\u6D4B+\u81EA\u6108\n */\n\nconst LOGIN_PATH = \"/admin/login\"\nconst CHECK_ALL_PATH = \"/admin/proxies/check-all\"\n\n// Space \u5361\u6B7B\u81EA\u6108\u76F8\u5173\u5E38\u91CF\u3002namespace/name \u901A\u8FC7 env.HF_SPACE_REPO\n// \uFF08\u683C\u5F0F \"a3216/ds2api\"\uFF09\u914D\u7F6E\uFF1B\u4E0D\u914D\u5219\u7528\u9ED8\u8BA4\u503C\u3002\nconst DEFAULT_HF_SPACE_REPO = \"a3216/ds2api\"\n// \u91CD\u542F\u51B7\u5374\uFF1A\u907F\u514D\u8FDE\u7EED restart \u628A Space \u6253\u7206\u3002\u91CD\u542F\u540E\u81F3\u5C11\u7B49 5 \u5206\u949F\u624D\u5141\u8BB8\u518D\u6B21\u91CD\u542F\u3002\nconst RESTART_COOLDOWN_MS = 5 * 60 * 1000\n// KV \u4E2D\u5B58\u4E0A\u6B21\u91CD\u542F\u65F6\u95F4\u7684 key\u3002\nconst LAST_RESTART_KV_KEY = \"last_restart\"\n\nfunction vercelHeaders(env) {\n  const headers = {}\n  const bypass = env.VERCEL_BYPASS_TOKEN\n  if (bypass) {\n    headers[\"x-vercel-protection-bypass\"] = bypass\n  }\n  return headers\n}\n\nasync function login(baseURL, adminKey, extraHeaders) {\n  const res = await fetch(`${baseURL}${LOGIN_PATH}`, {\n    method: \"POST\",\n    headers: { \"Content-Type\": \"application/json\", ...extraHeaders },\n    body: JSON.stringify({ admin_key: adminKey, expire_hours: 1 }),\n  })\n  if (!res.ok) {\n    const text = await res.text()\n    throw new Error(`login failed (${res.status}): ${text}`)\n  }\n  const data = await res.json()\n  return data.token\n}\n\nasync function checkAll(baseURL, token, extraHeaders) {\n  const res = await fetch(`${baseURL}${CHECK_ALL_PATH}`, {\n    method: \"POST\",\n    headers: {\n      \"Content-Type\": \"application/json\",\n      Authorization: `Bearer ${token}`,\n      ...extraHeaders,\n    },\n  })\n  if (!res.ok) {\n    const text = await res.text()\n    throw new Error(`check-all failed (${res.status}): ${text}`)\n  }\n  return res.json()\n}\n\nasync function runCheck(env) {\n  const baseURL = (env.DS2API_BASE_URL || \"\").replace(/\\/+$/, \"\")\n  const adminKey = env.ADMIN_KEY || \"\"\n\n  if (!baseURL || !adminKey) {\n    throw new Error(\"DS2API_BASE_URL and ADMIN_KEY secrets must be set\")\n  }\n\n  const extraHeaders = vercelHeaders(env)\n  const token = await login(baseURL, adminKey, extraHeaders)\n  const result = await checkAll(baseURL, token, extraHeaders)\n  return result\n}\n\n// === HF Space \u5361\u6B7B\u81EA\u6108 ===\n\n// \u67E5\u8BE2 Space runtime \u72B6\u6001\u3002\u8FD4\u56DE { stage, hardwareCurrent }\u3002\nasync function fetchSpaceRuntime(repo, token) {\n  const url = `https://huggingface.co/api/spaces/${repo}/runtime`\n  const res = await fetch(url, {\n    headers: { Authorization: `Bearer ${token}` },\n  })\n  if (!res.ok) {\n    const text = await res.text()\n    throw new Error(`runtime query failed (${res.status}): ${text}`)\n  }\n  const data = await res.json()\n  return {\n    stage: data.stage,\n    hardwareCurrent: data.hardware ? data.hardware.current : null,\n  }\n}\n\n// \u5224\u5B9A Space \u662F\u5426\u5361\u6B7B\uFF1A\u975E RUNNING \u6216\u672A\u5206\u914D\u786C\u4EF6\u3002\nfunction isSpaceStuck(runtime) {\n  if (!runtime) return true\n  if (runtime.stage !== \"RUNNING\") return true\n  if (!runtime.hardwareCurrent) return true\n  return false\n}\n\n// \u89E6\u53D1 HF Space restart\u3002\u8FD4\u56DE true \u8868\u793A\u8BF7\u6C42\u5DF2\u88AB\u63A5\u53D7\u3002\nasync function restartSpace(repo, token) {\n  const url = `https://huggingface.co/api/spaces/${repo}/restart`\n  const res = await fetch(url, {\n    method: \"POST\",\n    headers: { Authorization: `Bearer ${token}` },\n  })\n  if (!res.ok) {\n    const text = await res.text()\n    throw new Error(`restart failed (${res.status}): ${text}`)\n  }\n  return true\n}\n\n// \u4ECE KV \u8BFB\u4E0A\u6B21\u91CD\u542F\u65F6\u95F4\u6233\uFF08ms\uFF09\u3002KV \u672A\u7ED1\u5B9A\u6216\u65E0\u503C\u8FD4\u56DE 0\u3002\nasync function getLastRestart(kv) {\n  if (!kv) return 0\n  try {\n    const v = await kv.get(LAST_RESTART_KV_KEY)\n    const n = parseInt(v, 10)\n    return Number.isFinite(n) ? n : 0\n  } catch {\n    return 0\n  }\n}\n\nasync function setLastRestart(kv) {\n  if (!kv) return\n  try {\n    await kv.put(LAST_RESTART_KV_KEY, String(Date.now()))\n  } catch {\n    // KV \u5199\u5931\u8D25\u4E0D\u963B\u65AD\u4E3B\u6D41\u7A0B\u3002\n  }\n}\n\n// Space \u5361\u6B7B\u68C0\u6D4B+\u81EA\u6108\u4E3B\u903B\u8F91\u3002\nasync function runSpaceCheck(env) {\n  const token = env.HF_TOKEN || \"\"\n  if (!token) {\n    throw new Error(\"HF_TOKEN secret must be set for space self-healing\")\n  }\n  const repo = (env.HF_SPACE_REPO || DEFAULT_HF_SPACE_REPO).trim()\n\n  const runtime = await fetchSpaceRuntime(repo, token)\n  if (!isSpaceStuck(runtime)) {\n    return {\n      ok: true,\n      action: \"none\",\n      repo,\n      stage: runtime.stage,\n      hardwareCurrent: runtime.hardwareCurrent,\n      message: \"space healthy, no action\",\n    }\n  }\n\n  // \u5361\u6B7B\uFF0C\u68C0\u67E5\u51B7\u5374\u3002\n  const last = await getLastRestart(env.SPACE_STATE)\n  const elapsed = Date.now() - last\n  if (elapsed < RESTART_COOLDOWN_MS) {\n    const waitMs = RESTART_COOLDOWN_MS - elapsed\n    return {\n      ok: true,\n      action: \"cooldown\",\n      repo,\n      stage: runtime.stage,\n      hardwareCurrent: runtime.hardwareCurrent,\n      message: `space stuck but in restart cooldown, wait ${Math.ceil(waitMs / 1000)}s`,\n    }\n  }\n\n  // \u89E6\u53D1\u91CD\u542F\u3002\n  await restartSpace(repo, token)\n  await setLastRestart(env.SPACE_STATE)\n  return {\n    ok: true,\n    action: \"restarted\",\n    repo,\n    stage: runtime.stage,\n    hardwareCurrent: runtime.hardwareCurrent,\n    message: \"space was stuck, restart triggered\",\n  }\n}\n\n// \u533A\u5206\u4E24\u79CD cron\uFF1A5 \u5C0F\u65F6\u4E00\u6B21\u7684\u4EE3\u7406\u68C0\u67E5 vs 10 \u5206\u949F\u4E00\u6B21\u7684 Space \u68C0\u67E5\u3002\n// scheduled.cron \u5373\u89E6\u53D1\u7684 cron \u8868\u8FBE\u5F0F\uFF08\u5B57\u7B26\u4E32\uFF09\u3002\nfunction isSpaceCheckCron(cron) {\n  return cron === \"*/10 * * * *\"\n}\n\nexport default {\n  async scheduled(event, env, ctx) {\n    if (isSpaceCheckCron(event.cron)) {\n      ctx.waitUntil(\n        runSpaceCheck(env)\n          .then((r) => {\n            if (r.action === \"restarted\") {\n              console.warn(`[space] ${r.message} repo=${r.repo} stage=${r.stage} hardware=${r.hardwareCurrent}`)\n            } else {\n              console.log(`[space] ${r.message} repo=${r.repo} stage=${r.stage} hardware=${r.hardwareCurrent}`)\n            }\n          })\n          .catch((err) => {\n            console.error(`[space] check failed: ${err.message}`)\n          })\n      )\n      return\n    }\n\n    // \u9ED8\u8BA4\uFF1A\u4EE3\u7406\u5065\u5EB7\u68C0\u67E5 cron\u3002\n    ctx.waitUntil(\n      runCheck(env)\n        .then((result) => {\n          const items = result.items || []\n          const healthy = items.filter((i) => i.healthy && !i.disabled).length\n          const banned = items.filter((i) => i.disabled).length\n          console.log(\n            `proxy health check done: ${healthy}/${items.length} healthy, ${banned} banned`\n          )\n        })\n        .catch((err) => {\n          console.error(`proxy health check failed: ${err.message}`)\n        })\n    )\n  },\n\n  async fetch(request, env) {\n    const url = new URL(request.url)\n\n    if (url.pathname === \"/check\") {\n      try {\n        const result = await runCheck(env)\n        return Response.json(result)\n      } catch (err) {\n        return Response.json({ error: err.message }, { status: 500 })\n      }\n    }\n\n    if (url.pathname === \"/check-space\") {\n      try {\n        const result = await runSpaceCheck(env)\n        return Response.json(result)\n      } catch (err) {\n        return Response.json({ error: err.message }, { status: 500 })\n      }\n    }\n\n    return new Response(\"not found\", { status: 404 })\n  },\n}\n"],
+  "mappings": ";;;;AAqBA,IAAM,aAAa;AACnB,IAAM,iBAAiB;AAIvB,IAAM,wBAAwB;AAE9B,IAAM,sBAAsB,IAAI,KAAK;AAErC,IAAM,sBAAsB;AAE5B,SAAS,cAAc,KAAK;AAC1B,QAAM,UAAU,CAAC;AACjB,QAAM,SAAS,IAAI;AACnB,MAAI,QAAQ;AACV,YAAQ,4BAA4B,IAAI;AAAA,EAC1C;AACA,SAAO;AACT;AAPS;AAST,eAAe,MAAM,SAAS,UAAU,cAAc;AACpD,QAAM,MAAM,MAAM,MAAM,GAAG,OAAO,GAAG,UAAU,IAAI;AAAA,IACjD,QAAQ;AAAA,IACR,SAAS,EAAE,gBAAgB,oBAAoB,GAAG,aAAa;AAAA,IAC/D,MAAM,KAAK,UAAU,EAAE,WAAW,UAAU,cAAc,EAAE,CAAC;AAAA,EAC/D,CAAC;AACD,MAAI,CAAC,IAAI,IAAI;AACX,UAAM,OAAO,MAAM,IAAI,KAAK;AAC5B,UAAM,IAAI,MAAM,iBAAiB,IAAI,MAAM,MAAM,IAAI,EAAE;AAAA,EACzD;AACA,QAAM,OAAO,MAAM,IAAI,KAAK;AAC5B,SAAO,KAAK;AACd;AAZe;AAcf,eAAe,SAAS,SAAS,OAAO,cAAc;AACpD,QAAM,MAAM,MAAM,MAAM,GAAG,OAAO,GAAG,cAAc,IAAI;AAAA,IACrD,QAAQ;AAAA,IACR,SAAS;AAAA,MACP,gBAAgB;AAAA,MAChB,eAAe,UAAU,KAAK;AAAA,MAC9B,GAAG;AAAA,IACL;AAAA,EACF,CAAC;AACD,MAAI,CAAC,IAAI,IAAI;AACX,UAAM,OAAO,MAAM,IAAI,KAAK;AAC5B,UAAM,IAAI,MAAM,qBAAqB,IAAI,MAAM,MAAM,IAAI,EAAE;AAAA,EAC7D;AACA,SAAO,IAAI,KAAK;AAClB;AAde;AAgBf,eAAe,SAAS,KAAK;AAC3B,QAAM,WAAW,IAAI,mBAAmB,IAAI,QAAQ,QAAQ,EAAE;AAC9D,QAAM,WAAW,IAAI,aAAa;AAElC,MAAI,CAAC,WAAW,CAAC,UAAU;AACzB,UAAM,IAAI,MAAM,mDAAmD;AAAA,EACrE;AAEA,QAAM,eAAe,cAAc,GAAG;AACtC,QAAM,QAAQ,MAAM,MAAM,SAAS,UAAU,YAAY;AACzD,QAAM,SAAS,MAAM,SAAS,SAAS,OAAO,YAAY;AAC1D,SAAO;AACT;AAZe;AAiBf,eAAe,kBAAkB,MAAM,OAAO;AAC5C,QAAM,MAAM,qCAAqC,IAAI;AACrD,QAAM,MAAM,MAAM,MAAM,KAAK;AAAA,IAC3B,SAAS,EAAE,eAAe,UAAU,KAAK,GAAG;AAAA,EAC9C,CAAC;AACD,MAAI,CAAC,IAAI,IAAI;AACX,UAAM,OAAO,MAAM,IAAI,KAAK;AAC5B,UAAM,IAAI,MAAM,yBAAyB,IAAI,MAAM,MAAM,IAAI,EAAE;AAAA,EACjE;AACA,QAAM,OAAO,MAAM,IAAI,KAAK;AAC5B,SAAO;AAAA,IACL,OAAO,KAAK;AAAA,IACZ,iBAAiB,KAAK,WAAW,KAAK,SAAS,UAAU;AAAA,EAC3D;AACF;AAde;AAiBf,SAAS,aAAa,SAAS;AAC7B,MAAI,CAAC,QAAS,QAAO;AACrB,MAAI,QAAQ,UAAU,UAAW,QAAO;AACxC,MAAI,CAAC,QAAQ,gBAAiB,QAAO;AACrC,SAAO;AACT;AALS;AAQT,eAAe,aAAa,MAAM,OAAO;AACvC,QAAM,MAAM,qCAAqC,IAAI;AACrD,QAAM,MAAM,MAAM,MAAM,KAAK;AAAA,IAC3B,QAAQ;AAAA,IACR,SAAS,EAAE,eAAe,UAAU,KAAK,GAAG;AAAA,EAC9C,CAAC;AACD,MAAI,CAAC,IAAI,IAAI;AACX,UAAM,OAAO,MAAM,IAAI,KAAK;AAC5B,UAAM,IAAI,MAAM,mBAAmB,IAAI,MAAM,MAAM,IAAI,EAAE;AAAA,EAC3D;AACA,SAAO;AACT;AAXe;AAcf,eAAe,eAAe,IAAI;AAChC,MAAI,CAAC,GAAI,QAAO;AAChB,MAAI;AACF,UAAM,IAAI,MAAM,GAAG,IAAI,mBAAmB;AAC1C,UAAM,IAAI,SAAS,GAAG,EAAE;AACxB,WAAO,OAAO,SAAS,CAAC,IAAI,IAAI;AAAA,EAClC,QAAQ;AACN,WAAO;AAAA,EACT;AACF;AATe;AAWf,eAAe,eAAe,IAAI;AAChC,MAAI,CAAC,GAAI;AACT,MAAI;AACF,UAAM,GAAG,IAAI,qBAAqB,OAAO,KAAK,IAAI,CAAC,CAAC;AAAA,EACtD,QAAQ;AAAA,EAER;AACF;AAPe;AAUf,eAAe,cAAc,KAAK;AAChC,QAAM,QAAQ,IAAI,YAAY;AAC9B,MAAI,CAAC,OAAO;AACV,UAAM,IAAI,MAAM,oDAAoD;AAAA,EACtE;AACA,QAAM,QAAQ,IAAI,iBAAiB,uBAAuB,KAAK;AAE/D,QAAM,UAAU,MAAM,kBAAkB,MAAM,KAAK;AACnD,MAAI,CAAC,aAAa,OAAO,GAAG;AAC1B,WAAO;AAAA,MACL,IAAI;AAAA,MACJ,QAAQ;AAAA,MACR;AAAA,MACA,OAAO,QAAQ;AAAA,MACf,iBAAiB,QAAQ;AAAA,MACzB,SAAS;AAAA,IACX;AAAA,EACF;AAGA,QAAM,OAAO,MAAM,eAAe,IAAI,WAAW;AACjD,QAAM,UAAU,KAAK,IAAI,IAAI;AAC7B,MAAI,UAAU,qBAAqB;AACjC,UAAM,SAAS,sBAAsB;AACrC,WAAO;AAAA,MACL,IAAI;AAAA,MACJ,QAAQ;AAAA,MACR;AAAA,MACA,OAAO,QAAQ;AAAA,MACf,iBAAiB,QAAQ;AAAA,MACzB,SAAS,6CAA6C,KAAK,KAAK,SAAS,GAAI,CAAC;AAAA,IAChF;AAAA,EACF;AAGA,QAAM,aAAa,MAAM,KAAK;AAC9B,QAAM,eAAe,IAAI,WAAW;AACpC,SAAO;AAAA,IACL,IAAI;AAAA,IACJ,QAAQ;AAAA,IACR;AAAA,IACA,OAAO,QAAQ;AAAA,IACf,iBAAiB,QAAQ;AAAA,IACzB,SAAS;AAAA,EACX;AACF;AA7Ce;AAiDf,SAAS,iBAAiB,MAAM;AAC9B,SAAO,SAAS;AAClB;AAFS;AAIT,IAAO,gBAAQ;AAAA,EACb,MAAM,UAAU,OAAO,KAAK,KAAK;AAC/B,QAAI,iBAAiB,MAAM,IAAI,GAAG;AAChC,UAAI;AAAA,QACF,cAAc,GAAG,EACd,KAAK,CAAC,MAAM;AACX,cAAI,EAAE,WAAW,aAAa;AAC5B,oBAAQ,KAAK,WAAW,EAAE,OAAO,SAAS,EAAE,IAAI,UAAU,EAAE,KAAK,aAAa,EAAE,eAAe,EAAE;AAAA,UACnG,OAAO;AACL,oBAAQ,IAAI,WAAW,EAAE,OAAO,SAAS,EAAE,IAAI,UAAU,EAAE,KAAK,aAAa,EAAE,eAAe,EAAE;AAAA,UAClG;AAAA,QACF,CAAC,EACA,MAAM,CAAC,QAAQ;AACd,kBAAQ,MAAM,yBAAyB,IAAI,OAAO,EAAE;AAAA,QACtD,CAAC;AAAA,MACL;AACA;AAAA,IACF;AAGA,QAAI;AAAA,MACF,SAAS,GAAG,EACT,KAAK,CAAC,WAAW;AAChB,cAAM,QAAQ,OAAO,SAAS,CAAC;AAC/B,cAAM,UAAU,MAAM,OAAO,CAAC,MAAM,EAAE,WAAW,CAAC,EAAE,QAAQ,EAAE;AAC9D,cAAM,SAAS,MAAM,OAAO,CAAC,MAAM,EAAE,QAAQ,EAAE;AAC/C,gBAAQ;AAAA,UACN,4BAA4B,OAAO,IAAI,MAAM,MAAM,aAAa,MAAM;AAAA,QACxE;AAAA,MACF,CAAC,EACA,MAAM,CAAC,QAAQ;AACd,gBAAQ,MAAM,8BAA8B,IAAI,OAAO,EAAE;AAAA,MAC3D,CAAC;AAAA,IACL;AAAA,EACF;AAAA,EAEA,MAAM,MAAM,SAAS,KAAK;AACxB,UAAM,MAAM,IAAI,IAAI,QAAQ,GAAG;AAE/B,QAAI,IAAI,aAAa,UAAU;AAC7B,UAAI;AACF,cAAM,SAAS,MAAM,SAAS,GAAG;AACjC,eAAO,SAAS,KAAK,MAAM;AAAA,MAC7B,SAAS,KAAK;AACZ,eAAO,SAAS,KAAK,EAAE,OAAO,IAAI,QAAQ,GAAG,EAAE,QAAQ,IAAI,CAAC;AAAA,MAC9D;AAAA,IACF;AAEA,QAAI,IAAI,aAAa,gBAAgB;AACnC,UAAI;AACF,cAAM,SAAS,MAAM,cAAc,GAAG;AACtC,eAAO,SAAS,KAAK,MAAM;AAAA,MAC7B,SAAS,KAAK;AACZ,eAAO,SAAS,KAAK,EAAE,OAAO,IAAI,QAAQ,GAAG,EAAE,QAAQ,IAAI,CAAC;AAAA,MAC9D;AAAA,IACF;AAEA,WAAO,IAAI,SAAS,aAAa,EAAE,QAAQ,IAAI,CAAC;AAAA,EAClD;AACF;",
+  "names": []
+}

cf-worker/src/index.js

@@ -1,5 +1,5 @@
 /**
- * Cloudflare Worker — 定时刷新 ds2api 代理健康状态
+ * Cloudflare Worker — ds2api 代理健康检查 + HF Space 卡死自愈
  *
  * 部署步骤:
  *   1. cd cf-worker && npm install
@@ -7,15 +7,29 @@
  *        wrangler secret put DS2API_BASE_URL
  *        wrangler secret put ADMIN_KEY
  *        wrangler secret put VERCEL_BYPASS_TOKEN   (如果开了 Vercel Deployment Protection)
+ *        wrangler secret put HF_TOKEN              (HF API token，用于 Space 卡死自愈)
  *   3. wrangler deploy
  *
- * Cron 每 5 小时触发一次 (与 wrangler.toml 中的 crons 对应)。
- * 也可以手动 GET/POST https://<worker>/check 触发。
+ * 两个 cron（见 wrangler.toml）：
+ *   - 5 小时一次（"0 0,5,10,15,20 * * *" 等价写法）：代理健康检查（runCheck）
+ *   - 10 分钟一次：HF Space 卡死检测+自动 restart（runSpaceCheck）
+ *
+ * 手动触发：
+ *   GET/POST https://<worker>/check          → 代理健康检查
+ *   GET/POST https://<worker>/check-space    → Space 卡死检测+自愈
  */
 
 const LOGIN_PATH = "/admin/login"
 const CHECK_ALL_PATH = "/admin/proxies/check-all"
 
+// Space 卡死自愈相关常量。namespace/name 通过 env.HF_SPACE_REPO
+// （格式 "a3216/ds2api"）配置；不配则用默认值。
+const DEFAULT_HF_SPACE_REPO = "a3216/ds2api"
+// 重启冷却：避免连续 restart 把 Space 打爆。重启后至少等 5 分钟才允许再次重启。
+const RESTART_COOLDOWN_MS = 5 * 60 * 1000
+// KV 中存上次重启时间的 key。
+const LAST_RESTART_KV_KEY = "last_restart"
+
 function vercelHeaders(env) {
   const headers = {}
   const bypass = env.VERCEL_BYPASS_TOKEN
@@ -69,8 +83,142 @@ async function runCheck(env) {
   return result
 }
 
+// === HF Space 卡死自愈 ===
+
+// 查询 Space runtime 状态。返回 { stage, hardwareCurrent }。
+async function fetchSpaceRuntime(repo, token) {
+  const url = `https://huggingface.co/api/spaces/${repo}/runtime`
+  const res = await fetch(url, {
+    headers: { Authorization: `Bearer ${token}` },
+  })
+  if (!res.ok) {
+    const text = await res.text()
+    throw new Error(`runtime query failed (${res.status}): ${text}`)
+  }
+  const data = await res.json()
+  return {
+    stage: data.stage,
+    hardwareCurrent: data.hardware ? data.hardware.current : null,
+  }
+}
+
+// 判定 Space 是否卡死：非 RUNNING 或未分配硬件。
+function isSpaceStuck(runtime) {
+  if (!runtime) return true
+  if (runtime.stage !== "RUNNING") return true
+  if (!runtime.hardwareCurrent) return true
+  return false
+}
+
+// 触发 HF Space restart。返回 true 表示请求已被接受。
+async function restartSpace(repo, token) {
+  const url = `https://huggingface.co/api/spaces/${repo}/restart`
+  const res = await fetch(url, {
+    method: "POST",
+    headers: { Authorization: `Bearer ${token}` },
+  })
+  if (!res.ok) {
+    const text = await res.text()
+    throw new Error(`restart failed (${res.status}): ${text}`)
+  }
+  return true
+}
+
+// 从 KV 读上次重启时间戳（ms）。KV 未绑定或无值返回 0。
+async function getLastRestart(kv) {
+  if (!kv) return 0
+  try {
+    const v = await kv.get(LAST_RESTART_KV_KEY)
+    const n = parseInt(v, 10)
+    return Number.isFinite(n) ? n : 0
+  } catch {
+    return 0
+  }
+}
+
+async function setLastRestart(kv) {
+  if (!kv) return
+  try {
+    await kv.put(LAST_RESTART_KV_KEY, String(Date.now()))
+  } catch {
+    // KV 写失败不阻断主流程。
+  }
+}
+
+// Space 卡死检测+自愈主逻辑。
+async function runSpaceCheck(env) {
+  const token = env.HF_TOKEN || ""
+  if (!token) {
+    throw new Error("HF_TOKEN secret must be set for space self-healing")
+  }
+  const repo = (env.HF_SPACE_REPO || DEFAULT_HF_SPACE_REPO).trim()
+
+  const runtime = await fetchSpaceRuntime(repo, token)
+  if (!isSpaceStuck(runtime)) {
+    return {
+      ok: true,
+      action: "none",
+      repo,
+      stage: runtime.stage,
+      hardwareCurrent: runtime.hardwareCurrent,
+      message: "space healthy, no action",
+    }
+  }
+
+  // 卡死，检查冷却。
+  const last = await getLastRestart(env.SPACE_STATE)
+  const elapsed = Date.now() - last
+  if (elapsed < RESTART_COOLDOWN_MS) {
+    const waitMs = RESTART_COOLDOWN_MS - elapsed
+    return {
+      ok: true,
+      action: "cooldown",
+      repo,
+      stage: runtime.stage,
+      hardwareCurrent: runtime.hardwareCurrent,
+      message: `space stuck but in restart cooldown, wait ${Math.ceil(waitMs / 1000)}s`,
+    }
+  }
+
+  // 触发重启。
+  await restartSpace(repo, token)
+  await setLastRestart(env.SPACE_STATE)
+  return {
+    ok: true,
+    action: "restarted",
+    repo,
+    stage: runtime.stage,
+    hardwareCurrent: runtime.hardwareCurrent,
+    message: "space was stuck, restart triggered",
+  }
+}
+
+// 区分两种 cron：5 小时一次的代理检查 vs 10 分钟一次的 Space 检查。
+// scheduled.cron 即触发的 cron 表达式（字符串）。
+function isSpaceCheckCron(cron) {
+  return cron === "*/10 * * * *"
+}
+
 export default {
   async scheduled(event, env, ctx) {
+    if (isSpaceCheckCron(event.cron)) {
+      ctx.waitUntil(
+        runSpaceCheck(env)
+          .then((r) => {
+            if (r.action === "restarted") {
+              console.warn(`[space] ${r.message} repo=${r.repo} stage=${r.stage} hardware=${r.hardwareCurrent}`)
+            } else {
+              console.log(`[space] ${r.message} repo=${r.repo} stage=${r.stage} hardware=${r.hardwareCurrent}`)
+            }
+          })
+          .catch((err) => {
+            console.error(`[space] check failed: ${err.message}`)
+          })
+      )
+      return
+    }
+
+    // 默认：代理健康检查 cron。
     ctx.waitUntil(
       runCheck(env)
         .then((result) => {
@@ -89,15 +237,25 @@ export default {
 
   async fetch(request, env) {
     const url = new URL(request.url)
-    if (url.pathname !== "/check") {
-      return new Response("not found", { status: 404 })
+
+    if (url.pathname === "/check") {
+      try {
+        const result = await runCheck(env)
+        return Response.json(result)
+      } catch (err) {
+        return Response.json({ error: err.message }, { status: 500 })
+      }
     }
 
-    try {
-      const result = await runCheck(env)
-      return Response.json(result)
-    } catch (err) {
-      return Response.json({ error: err.message }, { status: 500 })
+    if (url.pathname === "/check-space") {
+      try {
+        const result = await runSpaceCheck(env)
+        return Response.json(result)
+      } catch (err) {
+        return Response.json({ error: err.message }, { status: 500 })
+      }
     }
+
+    return new Response("not found", { status: 404 })
   },
 }

cf-worker/wrangler.toml

@@ -2,8 +2,20 @@ name = "ds2api-health-check"
 main = "src/index.js"
 compatibility_date = "2024-12-01"
 
+# 两个 cron：
+#   - "0 */5 * * *"   每 5 小时：代理健康检查（runCheck）
+#   - "*/10 * * * *"  每 10 分钟：HF Space 卡死检测+自动 restart（runSpaceCheck）
 [triggers]
-crons = ["0 */5 * * *"]
+crons = ["0 */5 * * *", "*/10 * * * *"]
 
 [vars]
 # DS2API_BASE_URL = "https://your-app.vercel.app"
+# HF_SPACE_REPO = "a3216/ds2api"   # 可选，覆盖默认监控的 Space
+
+# KV 用于存上次 Space restart 时间，做重启冷却（避免连续 restart 打爆 Space）。
+# 部署前需先创建 KV namespace：
+#   wrangler kv namespace create SPACE_STATE
+# 然后把下面的 id 替换为返回的 id。
+[[kv_namespaces]]
+binding = "SPACE_STATE"
+id = "da57e26ab14248c79649397afc12bc73"