@maintainer-pro/ai-bridge 0.1.11 → 0.1.12

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@maintainer-pro/ai-bridge",
-  "version": "0.1.11",
+  "version": "0.1.12",
   "description": "Local bridge daemon that pairs a machine to Maintainer Pro and configures multiple client sandboxes.",
   "keywords": [
     "maintainer-pro",

package/src/daemon.mjs

@@ -2205,9 +2205,16 @@ const embeddedChat = new Map();
 /** @type {Map<string, Promise<object | null>>} */
 const embeddedChatStarting = new Map();
 
-const PROXY_CHUNK_BYTES = 32 * 1024;
+const PROXY_CHUNK_BYTES = 256 * 1024;
 /** @type {Map<string, import("node:http").ClientRequest>} */
 const proxyHttpReqs = new Map();
+/** Reuse sockets to the local app — Next/Vite fetch many files per page. */
+const proxyKeepAliveAgent = new http.Agent({
+  keepAlive: true,
+  maxSockets: 64,
+  maxFreeSockets: 16,
+  timeout: 30_000,
+});
 /** @type {Map<string, WebSocket>} */
 const proxyLocalSockets = new Map();
 /** @type {Record<string, unknown> | null} */
@@ -2463,7 +2470,7 @@ function handleProxyHttpFromAdmin(msg) {
         path,
         method,
         headers,
-        agent: false,
+        agent: proxyKeepAliveAgent,
       },
       (res) => {
         /** @type {Record<string, string>} */
@@ -2589,10 +2596,27 @@ function attachProxyLocalWs(id, socket) {
   });
 }
 
-function openProxyLocalWs(id, port, path) {
+function wsProtocolsFromMsg(msg) {
+  if (Array.isArray(msg?.protocols)) {
+    return msg.protocols.map((p) => String(p || "").trim()).filter(Boolean);
+  }
+  const headers = msg?.headers && typeof msg.headers === "object" ? msg.headers : {};
+  const raw = headers["sec-websocket-protocol"] || headers["Sec-WebSocket-Protocol"] || "";
+  return String(raw)
+    .split(",")
+    .map((part) => part.trim())
+    .filter(Boolean);
+}
+
+function openProxyLocalWs(id, port, path, protocols) {
   let socket;
   try {
-    socket = new WebSocket(`ws://127.0.0.1:${port}${path}`);
+    const url = `ws://127.0.0.1:${port}${path}`;
+    const proto = Array.isArray(protocols)
+      ? protocols.map((p) => String(p || "").trim()).filter(Boolean)
+      : [];
+    // Vite HMR only accepts upgrades with subprotocol `vite-hmr` / `vite-ping`.
+    socket = proto.length ? new WebSocket(url, proto) : new WebSocket(url);
   } catch (err) {
     bridgeSend({
       type: "proxy.ws.error",
@@ -2615,6 +2639,7 @@ function handleProxyWsOpenFromAdmin(msg) {
     return;
   }
   const path = safeProxyPath(msg.path);
+  const protocols = wsProtocolsFromMsg(msg);
   if (isAiServerAppId(appId, ws)) {
     void (async () => {
       const embedded = await ensureEmbeddedChat(ws);
@@ -2623,7 +2648,7 @@ function handleProxyWsOpenFromAdmin(msg) {
         bridgeSend({ type: "proxy.ws.error", id, error: "AI chat is not running" });
         return;
       }
-      openProxyLocalWs(id, port, path);
+      openProxyLocalWs(id, port, path, protocols);
     })();
     return;
   }
@@ -2632,7 +2657,7 @@ function handleProxyWsOpenFromAdmin(msg) {
     bridgeSend({ type: "proxy.ws.error", id, error: "No local port mapped" });
     return;
   }
-  openProxyLocalWs(id, port, path);
+  openProxyLocalWs(id, port, path, protocols);
 }
 
 function handleProxyWsFrameFromAdmin(msg) {