@maintainer-pro/ai-bridge 0.1.35 → 0.1.36

package/package.json

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

package/src/daemon.mjs

@@ -81,6 +81,11 @@ const WS_RECONNECT_MAX_MS = 30_000;
 const WS_CONNECT_TIMEOUT_MS = 20_000;
 const WS_WATCHDOG_MS = 5_000;
 
+// Temporary: default debug in prod so we can catch leftover issues. Revisit later.
+if (!process.env.LOG_LEVEL?.trim() && !process.env.AI_LOG_LEVEL?.trim()) {
+  process.env.LOG_LEVEL = "debug";
+}
+
 /** @type {import("@maintainer-pro/ai-cli").Logger} */
 let logger = createLogger("ai-bridge");
 const SECRET_KEY_RE =
@@ -142,6 +147,8 @@ const intakeServers = new Map();
 
 /** @type {(payload: Record<string, unknown>) => boolean} */
 let bridgeSend = () => false;
+/** @type {() => Promise<void>} */
+let waitForBridgeDrain = async () => {};
 
 function activity(sandboxId, level, message) {
   const text = String(message || "").trim();
@@ -374,7 +381,7 @@ Flags:
   --help
 
 Logging (env):
-  LOG_LEVEL=info|debug|warn|error|silent   (default: info)
+  LOG_LEVEL=info|debug|warn|error|silent   (default: debug)
   LOG_PRETTY=0                             disable pretty TTY output
 `);
 }
@@ -2542,7 +2549,8 @@ const embeddedChat = new Map();
 /** @type {Map<string, Promise<object | null>>} */
 const embeddedChatStarting = new Map();
 
-const PROXY_CHUNK_BYTES = 256 * 1024;
+const PROXY_CHUNK_BYTES = 64 * 1024;
+const PROXY_SEND_HIGH_WATER = 512 * 1024;
 /** @type {Map<string, { req: import("node:http").ClientRequest, rewrite?: boolean, chunks?: Buffer[], headers?: Record<string, string>, portUrls?: Record<string, string> } | import("node:http").ClientRequest>} */
 const proxyHttpReqs = new Map();
 /** Reuse sockets to the local app — Next/Vite fetch many files per page. */
@@ -2632,6 +2640,10 @@ function replyProxyHttp(id, status, headers, body) {
     out[String(key)] = String(value);
   }
   bridgeSend({ type: "proxy.http.start", id, stream, status, headers: out });
+  void sendProxyHttpChunks(id, stream, buf);
+}
+
+async function sendProxyHttpChunks(id, stream, buf) {
   for (let offset = 0; offset < buf.length; offset += PROXY_CHUNK_BYTES) {
     const end = Math.min(offset + PROXY_CHUNK_BYTES, buf.length);
     bridgeSend({
@@ -2641,6 +2653,7 @@ function replyProxyHttp(id, status, headers, body) {
       data: buf.subarray(offset, end).toString("base64"),
       eof: false,
     });
+    await waitForBridgeDrain();
   }
   bridgeSend({ type: "proxy.http.chunk", id, stream, data: "", eof: true });
 }
@@ -2803,6 +2816,18 @@ async function handleAiProxyHttpFromAdmin(msg, ws) {
   );
 
   if (pathname === "/ai-ui.iife.js") {
+    if (method === "GET" || method === "HEAD") {
+      const cached = lookupShareResponse({
+        publicBase: ctx.publicBase,
+        path: reqPath,
+        ifNoneMatch: ctx.ifNoneMatch,
+        acceptEncoding: ctx.acceptEncoding,
+      });
+      if (cached) {
+        replyProxyHttp(id, cached.status, cached.headers, cached.body);
+        return;
+      }
+    }
     const file = findIife();
     if (!file || !fs.existsSync(file)) {
       sendProcessedProxyHttp(
@@ -3149,7 +3174,9 @@ function handleProxyHttpFromAdmin(msg) {
     return;
   }
   if (isAiServerAppId(appId, ws)) {
-    void handleAiProxyHttpFromAdmin(msg, ws);
+    setImmediate(() => {
+      void handleAiProxyHttpFromAdmin(msg, ws);
+    });
     return;
   }
   const existing = proxyHttpReqs.get(id);
@@ -3261,7 +3288,7 @@ function handleProxyHttpFromAdmin(msg) {
           }
           const status = res.statusCode || 502;
           applyShareCacheHeaders(outHeaders, path);
-          if (shouldProcessShareResponse(outHeaders)) {
+          if (shouldProcessShareResponse(outHeaders, path)) {
             /** @type {Buffer[]} */
             const chunks = [];
             res.on("data", (chunk) => {
@@ -3294,14 +3321,22 @@ function handleProxyHttpFromAdmin(msg) {
               offset += PROXY_CHUNK_BYTES
             ) {
               const end = Math.min(offset + PROXY_CHUNK_BYTES, buf.length);
-              bridgeSend({
+              const queued = bridgeSend({
                 type: "proxy.http.chunk",
                 id,
                 stream,
                 data: Buffer.from(buf.subarray(offset, end)).toString("base64"),
                 eof: false,
               });
+              if (queued === false) res.pause();
             }
+            void waitForBridgeDrain().then(() => {
+              try {
+                res.resume();
+              } catch {
+                /* ignore */
+              }
+            });
           });
           res.on("end", () => {
             proxyHttpReqs.delete(id);
@@ -5926,7 +5961,7 @@ async function main() {
   }
 
   logger.info(
-    { versions: PACKAGE_VERSIONS },
+    { versions: PACKAGE_VERSIONS, logLevel: logger.level },
     `bridge v${PACKAGE_VERSION} starting (${formatPackageVersions()})`
   );
   if (usingWsFallback) {
@@ -6029,6 +6064,7 @@ async function main() {
       return false;
     }
     sendChain = sendChain
+      .then(() => waitSocketDrain(current))
       .then(() => {
         if (!socket || socket !== current || socket.readyState !== 1) return;
         socket.send(text);
@@ -6036,7 +6072,25 @@ async function main() {
       .catch(() => {});
     return true;
   };
+  const waitSocketDrain = (current) =>
+    new Promise((resolve) => {
+      const tick = () => {
+        if (!current || current.readyState !== 1) {
+          resolve();
+          return;
+        }
+        const buffered =
+          typeof current.bufferedAmount === "number" ? current.bufferedAmount : 0;
+        if (buffered < PROXY_SEND_HIGH_WATER) {
+          setImmediate(resolve);
+          return;
+        }
+        setTimeout(tick, 25);
+      };
+      tick();
+    });
   bridgeSend = sendJson;
+  waitForBridgeDrain = () => waitSocketDrain(socket);
 
   const clearHeartbeatTimer = () => {
     if (heartbeatTimer) {

package/src/share-rewrite.mjs

@@ -140,7 +140,9 @@ export function shouldRewriteBody(headers) {
 }
 
 /** Redirects have no Content-Type, but Location still has to stay under the share prefix. */
-export function shouldProcessShareResponse(headers) {
+export function shouldProcessShareResponse(headers, path = "") {
+  // Widget bundle is self-contained — rewriting it as text blocks the bridge.
+  if (isLongCacheScriptPath(path)) return false;
   if (headerGet(headers, "location")) return true;
   if (shouldRewriteBody(headers)) return true;
   const ct = headerGet(headers, "content-type").toLowerCase();
@@ -1656,6 +1658,22 @@ export function processShareHttpResponse(opts) {
 
   rewriteHeaderUrls(headers, publicBase, port, portUrls);
 
+  if (isLongCacheScriptPath(path)) {
+    delete headers.etag;
+    delete headers["last-modified"];
+    const mode = applyShareCacheHeaders(headers, path);
+    const etag = etagFor(payload);
+    if (mode !== "no-store") headers.etag = etag;
+    const cachedKey = rewriteCacheKey(publicBase, path);
+    if (cachedKey) rememberRewrite(cachedKey, payload, status, headers, etag);
+    if (mode !== "no-store" && etagMatches(ifNoneMatch, etag)) {
+      return notModifiedResult(headers, etag);
+    }
+    payload = gzipIfRequested(payload, headers, acceptEncoding);
+    headers["content-length"] = String(payload.length);
+    return { status, headers, body: payload };
+  }
+
   const cachedKey = rewriteCacheKey(publicBase, path);
   const hit =
     cachedKey && Date.now() - (rewriteCache.get(cachedKey)?.at || 0) < REWRITE_CACHE_MS