panrouter 5.1.0 → 5.1.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "panrouter",
-  "version": "5.1.0",
+  "version": "5.1.1",
   "description": "让 Claude Code 免费使用 DeepSeek 等模型，无需 API Key",
   "type": "module",
   "bin": {

package/pool-worker.mjs

@@ -44,11 +44,45 @@ function notifyHub(status, url = "") {
       "x-secret-token": AUTH_SECRET,
     },
   });
-  req.on("error", () => {}); // 不等待确认
+  req.on("error", () => {});
   req.write(payload);
   req.end();
 }
 
+// 可等待的 notify — 用于首次上线确认
+function notifyHubWait(status, url = "") {
+  return new Promise((resolve) => {
+    const payload = JSON.stringify({ nodeId: NODE_ID, status, url });
+    const req = https.request(MAIN_HUB_URL, {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json",
+        "Content-Length": Buffer.byteLength(payload),
+        "x-secret-token": AUTH_SECRET,
+      },
+    }, (res) => resolve(res.statusCode === 200));
+    req.on("error", () => resolve(false));
+    req.setTimeout(5000, () => { req.destroy(); resolve(false); });
+    req.write(payload);
+    req.end();
+  });
+}
+
+// 首次上线确认 — 失败则重试，最多 5 次
+async function confirmOnline(url, retries = 5) {
+  const ok = await notifyHubWait("online", url);
+  if (ok) {
+    log("主控已确认节点注册", "OK");
+    return;
+  }
+  if (retries > 0) {
+    log(`注册尚未确认，3 秒后重试 (剩余 ${retries} 次)`, "INFO");
+    await new Promise((r) => setTimeout(r, 3000));
+    return confirmOnline(url, retries - 1);
+  }
+  log("注册确认已达最大重试次数", "WARN");
+}
+
 // ─── 检查端口 ────────────────────────────────────────────────────────────────
 function isPortOpen(port) {
   return new Promise((resolve) => {
@@ -138,7 +172,7 @@ function startTunnel() {
     if (match && match[0] !== currentUrl) {
       currentUrl = match[0];
       log(`成功获取公网入口: ${currentUrl}`, "OK");
-      notifyHub("online", currentUrl);
+      confirmOnline(currentUrl); // 确保主控注册成功（异步重试）
     }
 
     // 网络波动提示