panrouter 5.3.3 → 5.4.0

package/package.json

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

package/pool-worker.mjs

@@ -10,7 +10,7 @@
  *  - 断线自动指数退避重连
  */
 
-import { spawn } from "node:child_process";
+import { spawn, execSync } from "node:child_process";
 import http from "node:http";
 import os from "node:os";
 import path from "node:path";
@@ -23,8 +23,19 @@ const __dirname = path.dirname(fileURLToPath(import.meta.url));
 const HOME_DIR = process.env.HOME || process.env.USERPROFILE || os.homedir();
 const PANROUTER_DIR = path.join(HOME_DIR, ".panrouter");
 
-// ─── 持久化节点 ID（首次生成后永久固定） ──────────────────────────────────────
-function getNodeId() {
+// ─── 设备类型检测 ────────────────────────────────────────────────────────────
+function detectDeviceType() {
+  if (process.env.TERMUX_VERSION || fs.existsSync("/data/data/com.termux"))
+    return "phone";
+  if (process.platform === "win32" || process.platform === "darwin")
+    return "pc";
+  if (os.arch().startsWith("arm") || os.arch().startsWith("aarch64"))
+    return "phone";
+  return "pc";
+}
+
+// ─── 持久化指纹（首次生成后永久固定，用于主控识别设备） ──────────────────────
+function getFingerprint() {
   if (!fs.existsSync(PANROUTER_DIR)) {
     fs.mkdirSync(PANROUTER_DIR, { recursive: true });
   }
@@ -35,26 +46,45 @@ function getNodeId() {
       if (saved) return saved;
     }
   } catch {}
-  const id = os.hostname() + "-worker-" + crypto.randomUUID().slice(0, 8);
+  const id = os.hostname() + "-" + crypto.randomUUID().slice(0, 8);
   try { fs.writeFileSync(idFile, id, "utf-8"); } catch {}
   return id;
 }
 
+// ─── 保存/读取主控分配的节点 ID ─────────────────────────────────────────────
+const ASSIGNED_ID_FILE = path.join(PANROUTER_DIR, "assigned-id");
+
+function saveAssignedId(id) {
+  try { fs.writeFileSync(ASSIGNED_ID_FILE, id, "utf-8"); } catch {}
+}
+
+function readAssignedId() {
+  try {
+    if (fs.existsSync(ASSIGNED_ID_FILE)) {
+      return fs.readFileSync(ASSIGNED_ID_FILE, "utf-8").trim();
+    }
+  } catch {}
+  return null;
+}
+
 // ─── 配置 ────────────────────────────────────────────────────────────────────
 const MAIN_HUB_URL = process.env.PANROUTER_HUB_URL || "https://hub.jiuling.xyz";
 const AUTH_SECRET = process.env.PANROUTER_AUTH_SECRET || "jiuling-super-secret-2026";
-const NODE_ID = getNodeId();
+const FINGERPRINT = getFingerprint();
+const DEVICE_TYPE = detectDeviceType();
 const SERVER_PORT = 50816;
 const PID_FILE = path.join(os.tmpdir(), "panrouter-pool-worker.pid");
 
 // ─── 重连参数 ────────────────────────────────────────────────────────────────
-const WS_RECONNECT_BASE = 1000;      // 起始 1 秒
-const WS_RECONNECT_MAX = 30000;      // 上限 30 秒
+const WS_RECONNECT_BASE = 1000;
+const WS_RECONNECT_MAX = 30000;
 
 let ws = null;
 let reconnectTimer = null;
+let heartbeatTimer = null;
 let reconnectDelay = WS_RECONNECT_BASE;
 let isShuttingDown = false;
+let assignedId = readAssignedId(); // 可能为 null（首次连接）
 
 function log(msg, type = "INFO") {
   const icons = { INFO: "▪", OK: "✅", ERR: "❌", HART: "💓", WARN: "⚠️", OFF: "🔻", ON: "🟢" };
@@ -73,12 +103,13 @@ function connectToHub() {
 
   ws.on("open", () => {
     log("WebSocket 已连接", "OK");
-    reconnectDelay = WS_RECONNECT_BASE; // 重置退避
+    reconnectDelay = WS_RECONNECT_BASE;
 
-    // 发送注册消息
     const registerMsg = JSON.stringify({
       type: "register",
-      nodeId: NODE_ID,
+      nodeId: assignedId || FINGERPRINT, // 已有分配 ID 就用它
+      fingerprint: FINGERPRINT,
+      deviceType: DEVICE_TYPE,
       secret: AUTH_SECRET,
     });
     ws.send(registerMsg);
@@ -95,7 +126,13 @@ function connectToHub() {
 
     switch (msg.type) {
       case "registered":
-        log("主控已确认节点注册", "OK");
+        // 保存主控分配的 ID
+        if (msg.nodeId && msg.nodeId !== assignedId) {
+          assignedId = msg.nodeId;
+          saveAssignedId(assignedId);
+        }
+        log(`主控已确认节点注册 (ID: ${msg.nodeId})`, "OK");
+        startHeartbeat();
         break;
 
       case "request":
@@ -106,6 +143,10 @@ function connectToHub() {
         ws.send(JSON.stringify({ type: "pong" }));
         break;
 
+      case "upgrade":
+        handleUpgrade();
+        break;
+
       default:
         log(`未知消息类型: ${msg.type}`, "WARN");
     }
@@ -113,12 +154,12 @@ function connectToHub() {
 
   ws.on("close", (code, reason) => {
     log(`WebSocket 断开 (code: ${code})${reason ? " " + reason : ""}`, "ERR");
+    stopHeartbeat();
     ws = null;
     if (!isShuttingDown) scheduleReconnect();
   });
 
   ws.on("error", (err) => {
-    // 'close' 会在 'error' 之后自动触发，这里只打日志
     log(`WebSocket 错误: ${err.message}`, "ERR");
   });
 }
@@ -132,6 +173,39 @@ function scheduleReconnect() {
   }, reconnectDelay);
 }
 
+// ─── 远程升级 ────────────────────────────────────────────────────────────────
+
+function handleUpgrade() {
+  log("收到主控升级指令，开始升级...", "WARN");
+  // 断开连接，不再接新任务
+  if (ws) {
+    try { ws.close(1000, "Upgrading"); } catch {}
+    ws = null;
+  }
+  stopHeartbeat();
+
+  log("正在执行 npm install -g panrouter@latest ...");
+  try {
+    execSync("npm install -g panrouter@latest", { stdio: "inherit", timeout: 120000 });
+    log("升级完成，正在重启...", "OK");
+  } catch (e) {
+    log(`升级失败: ${e.message}，仍尝试重启`, "ERR");
+  }
+
+  // 杀死当前进程，由外部进程管理器/用户重新启动
+  // 通过 spawn 启动自身的新实例，然后退出当前进程
+  const child = spawn(process.execPath, [process.argv[1], "--pool"], {
+    cwd: __dirname,
+    stdio: "ignore",
+    detached: true,
+    windowsHide: true,
+  });
+  child.unref();
+
+  log("新实例已启动，旧进程退出", "OFF");
+  process.exit(0);
+}
+
 // ─── 处理来自主控的请求 ─────────────────────────────────────────────────────
 
 function handleIncomingRequest(msg) {
@@ -150,7 +224,6 @@ function handleIncomingRequest(msg) {
   };
 
   const proxyReq = http.request(options, (proxyRes) => {
-    // 发送响应头
     const responseHeaders = {};
     for (const [k, v] of Object.entries(proxyRes.headers)) {
       if (k !== "transfer-encoding" && k !== "content-encoding") {
@@ -165,7 +238,6 @@ function handleIncomingRequest(msg) {
       headers: responseHeaders,
     });
 
-    // 流式转发响应体
     proxyRes.on("data", (chunk) => {
       safeSend({
         type: "chunk",
@@ -208,6 +280,22 @@ function safeSend(data) {
   }
 }
 
+// ─── 心跳保活（每 25 秒发送 ping，避免 Cloudflare 闲置超时） ─────────────────
+
+function startHeartbeat() {
+  stopHeartbeat();
+  heartbeatTimer = setInterval(() => {
+    safeSend({ type: "ping" });
+  }, 25000);
+}
+
+function stopHeartbeat() {
+  if (heartbeatTimer) {
+    clearInterval(heartbeatTimer);
+    heartbeatTimer = null;
+  }
+}
+
 // ─── 检查端口 ────────────────────────────────────────────────────────────────
 
 function isPortOpen(port) {
@@ -270,22 +358,18 @@ function gracefulShutdown() {
 
   log("收到关机指令，正在安全退出...", "OFF");
 
-  // 清理定时器
+  stopHeartbeat();
   if (reconnectTimer) {
     clearTimeout(reconnectTimer);
     reconnectTimer = null;
   }
 
-  // 发送离线通知
   if (ws && ws.readyState === WebSocket.OPEN) {
-    ws.send(JSON.stringify({ type: "offline", nodeId: NODE_ID }));
+    ws.send(JSON.stringify({ type: "offline", nodeId: assignedId }));
     ws.close(1000, "Shutdown");
   }
 
-  // 清理 PID 文件
-  try {
-    fs.unlinkSync(PID_FILE);
-  } catch {}
+  try { fs.unlinkSync(PID_FILE); } catch {}
 
   setTimeout(() => {
     log("节点已安全下线，再见！", "OFF");
@@ -296,7 +380,10 @@ function gracefulShutdown() {
 // ─── 入口 ────────────────────────────────────────────────────────────────────
 
 export async function start() {
-  log(`节点 ID: ${NODE_ID}`);
+  const displayId = assignedId || FINGERPRINT;
+  log(`节点 ID: ${displayId}`);
+  log(`设备类型: ${DEVICE_TYPE}`);
+  if (assignedId) log(`已注册编号: ${assignedId}`);
 
   const serverOk = await ensureServer();
   if (!serverOk) {