@ricsam/isolate-daemon 0.1.17 → 0.1.18

package/dist/cjs/index.cjs

@@ -2,27 +2,37 @@ var __defProp = Object.defineProperty;
 var __getOwnPropNames = Object.getOwnPropertyNames;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
-var __moduleCache = /* @__PURE__ */ new WeakMap;
+function __accessProp(key) {
+  return this[key];
+}
 var __toCommonJS = (from) => {
-  var entry = __moduleCache.get(from), desc;
+  var entry = (__moduleCache ??= new WeakMap).get(from), desc;
   if (entry)
     return entry;
   entry = __defProp({}, "__esModule", { value: true });
-  if (from && typeof from === "object" || typeof from === "function")
-    __getOwnPropNames(from).map((key) => !__hasOwnProp.call(entry, key) && __defProp(entry, key, {
-      get: () => from[key],
-      enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable
-    }));
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (var key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(entry, key))
+        __defProp(entry, key, {
+          get: __accessProp.bind(from, key),
+          enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable
+        });
+  }
   __moduleCache.set(from, entry);
   return entry;
 };
+var __moduleCache;
+var __returnValue = (v) => v;
+function __exportSetter(name, newValue) {
+  this[name] = __returnValue.bind(null, newValue);
+}
 var __export = (target, all) => {
   for (var name in all)
     __defProp(target, name, {
       get: all[name],
       enumerable: true,
       configurable: true,
-      set: (newValue) => all[name] = () => newValue
+      set: __exportSetter.bind(all, name)
     });
 };
 
@@ -130,4 +140,4 @@ function updateStats(state) {
   state.stats.activeConnections = state.connections.size;
 }
 
-//# debugId=4E8854D8C09B9EE864756E2164756E21
+//# debugId=08DBEA612F7F726864756E2164756E21

package/dist/cjs/index.cjs.map

@@ -4,7 +4,7 @@
   "sourcesContent": [
     "/**\n * @ricsam/isolate-daemon\n *\n * Node.js daemon for running isolated-vm runtimes accessible via IPC.\n */\n\nimport { createServer, type Server } from \"node:net\";\nimport { unlink } from \"node:fs/promises\";\nimport { handleConnection } from \"./connection.cjs\";\nimport type {\n  DaemonOptions,\n  DaemonHandle,\n  DaemonState,\n  DaemonStats,\n} from \"./types.cjs\";\n\nexport type { DaemonOptions, DaemonHandle, DaemonStats };\n\nconst DEFAULT_OPTIONS: Required<DaemonOptions> = {\n  socketPath: \"/tmp/isolate-daemon.sock\",\n  host: \"127.0.0.1\",\n  port: 47891,\n  maxIsolates: 100,\n  defaultMemoryLimitMB: 128,\n};\n\n/**\n * Start the isolate daemon.\n *\n * @param options - Daemon configuration options\n * @returns Handle to control the daemon\n */\nexport async function startDaemon(\n  options: DaemonOptions = {}\n): Promise<DaemonHandle> {\n  const resolvedOptions: Required<DaemonOptions> = {\n    ...DEFAULT_OPTIONS,\n    ...options,\n  };\n\n  const state: DaemonState = {\n    isolates: new Map(),\n    connections: new Map(),\n    stats: {\n      activeIsolates: 0,\n      activeConnections: 0,\n      totalIsolatesCreated: 0,\n      totalRequestsProcessed: 0,\n    },\n    options: resolvedOptions,\n    namespacedRuntimes: new Map(),\n    namespacedCreatesInFlight: new Set(),\n  };\n\n  const server = createServer((socket) => {\n    handleConnection(socket, state);\n    updateStats(state);\n  });\n\n  // Try to remove existing socket file\n  if (resolvedOptions.socketPath) {\n    try {\n      await unlink(resolvedOptions.socketPath);\n    } catch {\n      // Ignore if doesn't exist\n    }\n  }\n\n  // Start listening\n  await new Promise<void>((resolve, reject) => {\n    server.on(\"error\", reject);\n\n    if (resolvedOptions.socketPath) {\n      server.listen(resolvedOptions.socketPath, () => {\n        server.removeListener(\"error\", reject);\n        resolve();\n      });\n    } else {\n      server.listen(resolvedOptions.port, resolvedOptions.host, () => {\n        server.removeListener(\"error\", reject);\n        resolve();\n      });\n    }\n  });\n\n  const address = resolvedOptions.socketPath\n    ? resolvedOptions.socketPath\n    : `${resolvedOptions.host}:${resolvedOptions.port}`;\n\n  console.log(`Isolate daemon listening on ${address}`);\n\n  return {\n    address,\n    getStats: () => ({\n      ...state.stats,\n      activeIsolates: state.isolates.size,\n      activeConnections: state.connections.size,\n    }),\n    close: async () => {\n      // Close all connections\n      for (const [socket] of state.connections) {\n        socket.destroy();\n      }\n\n      // Dispose all isolates\n      for (const [, instance] of state.isolates) {\n        try {\n          instance.runtime.dispose();\n        } catch {\n          // Ignore\n        }\n      }\n\n      state.isolates.clear();\n      state.connections.clear();\n      state.namespacedRuntimes.clear();\n\n      // Close server\n      await closeServer(server);\n\n      // Remove socket file\n      if (resolvedOptions.socketPath) {\n        try {\n          await unlink(resolvedOptions.socketPath);\n        } catch {\n          // Ignore\n        }\n      }\n\n      console.log(\"Isolate daemon stopped\");\n    },\n  };\n}\n\nfunction closeServer(server: Server): Promise<void> {\n  return new Promise((resolve, reject) => {\n    server.close((err) => {\n      if (err) {\n        reject(err);\n      } else {\n        resolve();\n      }\n    });\n  });\n}\n\nfunction updateStats(state: DaemonState): void {\n  state.stats.activeIsolates = state.isolates.size;\n  state.stats.activeConnections = state.connections.size;\n}\n"
   ],
-  "mappings": ";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAM0C,IAA1C;AACuB,IAAvB;AACiC,IAAjC;AAUA,IAAM,kBAA2C;AAAA,EAC/C,YAAY;AAAA,EACZ,MAAM;AAAA,EACN,MAAM;AAAA,EACN,aAAa;AAAA,EACb,sBAAsB;AACxB;AAQA,eAAsB,WAAW,CAC/B,UAAyB,CAAC,GACH;AAAA,EACvB,MAAM,kBAA2C;AAAA,OAC5C;AAAA,OACA;AAAA,EACL;AAAA,EAEA,MAAM,QAAqB;AAAA,IACzB,UAAU,IAAI;AAAA,IACd,aAAa,IAAI;AAAA,IACjB,OAAO;AAAA,MACL,gBAAgB;AAAA,MAChB,mBAAmB;AAAA,MACnB,sBAAsB;AAAA,MACtB,wBAAwB;AAAA,IAC1B;AAAA,IACA,SAAS;AAAA,IACT,oBAAoB,IAAI;AAAA,IACxB,2BAA2B,IAAI;AAAA,EACjC;AAAA,EAEA,MAAM,SAAS,6BAAa,CAAC,WAAW;AAAA,IACtC,mCAAiB,QAAQ,KAAK;AAAA,IAC9B,YAAY,KAAK;AAAA,GAClB;AAAA,EAGD,IAAI,gBAAgB,YAAY;AAAA,IAC9B,IAAI;AAAA,MACF,MAAM,uBAAO,gBAAgB,UAAU;AAAA,MACvC,MAAM;AAAA,EAGV;AAAA,EAGA,MAAM,IAAI,QAAc,CAAC,SAAS,WAAW;AAAA,IAC3C,OAAO,GAAG,SAAS,MAAM;AAAA,IAEzB,IAAI,gBAAgB,YAAY;AAAA,MAC9B,OAAO,OAAO,gBAAgB,YAAY,MAAM;AAAA,QAC9C,OAAO,eAAe,SAAS,MAAM;AAAA,QACrC,QAAQ;AAAA,OACT;AAAA,IACH,EAAO;AAAA,MACL,OAAO,OAAO,gBAAgB,MAAM,gBAAgB,MAAM,MAAM;AAAA,QAC9D,OAAO,eAAe,SAAS,MAAM;AAAA,QACrC,QAAQ;AAAA,OACT;AAAA;AAAA,GAEJ;AAAA,EAED,MAAM,UAAU,gBAAgB,aAC5B,gBAAgB,aAChB,GAAG,gBAAgB,QAAQ,gBAAgB;AAAA,EAE/C,QAAQ,IAAI,+BAA+B,SAAS;AAAA,EAEpD,OAAO;AAAA,IACL;AAAA,IACA,UAAU,OAAO;AAAA,SACZ,MAAM;AAAA,MACT,gBAAgB,MAAM,SAAS;AAAA,MAC/B,mBAAmB,MAAM,YAAY;AAAA,IACvC;AAAA,IACA,OAAO,YAAY;AAAA,MAEjB,YAAY,WAAW,MAAM,aAAa;AAAA,QACxC,OAAO,QAAQ;AAAA,MACjB;AAAA,MAGA,cAAc,aAAa,MAAM,UAAU;AAAA,QACzC,IAAI;AAAA,UACF,SAAS,QAAQ,QAAQ;AAAA,UACzB,MAAM;AAAA,MAGV;AAAA,MAEA,MAAM,SAAS,MAAM;AAAA,MACrB,MAAM,YAAY,MAAM;AAAA,MACxB,MAAM,mBAAmB,MAAM;AAAA,MAG/B,MAAM,YAAY,MAAM;AAAA,MAGxB,IAAI,gBAAgB,YAAY;AAAA,QAC9B,IAAI;AAAA,UACF,MAAM,uBAAO,gBAAgB,UAAU;AAAA,UACvC,MAAM;AAAA,MAGV;AAAA,MAEA,QAAQ,IAAI,wBAAwB;AAAA;AAAA,EAExC;AAAA;AAGF,SAAS,WAAW,CAAC,QAA+B;AAAA,EAClD,OAAO,IAAI,QAAQ,CAAC,SAAS,WAAW;AAAA,IACtC,OAAO,MAAM,CAAC,QAAQ;AAAA,MACpB,IAAI,KAAK;AAAA,QACP,OAAO,GAAG;AAAA,MACZ,EAAO;AAAA,QACL,QAAQ;AAAA;AAAA,KAEX;AAAA,GACF;AAAA;AAGH,SAAS,WAAW,CAAC,OAA0B;AAAA,EAC7C,MAAM,MAAM,iBAAiB,MAAM,SAAS;AAAA,EAC5C,MAAM,MAAM,oBAAoB,MAAM,YAAY;AAAA;",
-  "debugId": "4E8854D8C09B9EE864756E2164756E21",
+  "mappings": ";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAM0C,IAA1C;AACuB,IAAvB;AACiC,IAAjC;AAUA,IAAM,kBAA2C;AAAA,EAC/C,YAAY;AAAA,EACZ,MAAM;AAAA,EACN,MAAM;AAAA,EACN,aAAa;AAAA,EACb,sBAAsB;AACxB;AAQA,eAAsB,WAAW,CAC/B,UAAyB,CAAC,GACH;AAAA,EACvB,MAAM,kBAA2C;AAAA,OAC5C;AAAA,OACA;AAAA,EACL;AAAA,EAEA,MAAM,QAAqB;AAAA,IACzB,UAAU,IAAI;AAAA,IACd,aAAa,IAAI;AAAA,IACjB,OAAO;AAAA,MACL,gBAAgB;AAAA,MAChB,mBAAmB;AAAA,MACnB,sBAAsB;AAAA,MACtB,wBAAwB;AAAA,IAC1B;AAAA,IACA,SAAS;AAAA,IACT,oBAAoB,IAAI;AAAA,IACxB,2BAA2B,IAAI;AAAA,EACjC;AAAA,EAEA,MAAM,SAAS,6BAAa,CAAC,WAAW;AAAA,IACtC,mCAAiB,QAAQ,KAAK;AAAA,IAC9B,YAAY,KAAK;AAAA,GAClB;AAAA,EAGD,IAAI,gBAAgB,YAAY;AAAA,IAC9B,IAAI;AAAA,MACF,MAAM,uBAAO,gBAAgB,UAAU;AAAA,MACvC,MAAM;AAAA,EAGV;AAAA,EAGA,MAAM,IAAI,QAAc,CAAC,SAAS,WAAW;AAAA,IAC3C,OAAO,GAAG,SAAS,MAAM;AAAA,IAEzB,IAAI,gBAAgB,YAAY;AAAA,MAC9B,OAAO,OAAO,gBAAgB,YAAY,MAAM;AAAA,QAC9C,OAAO,eAAe,SAAS,MAAM;AAAA,QACrC,QAAQ;AAAA,OACT;AAAA,IACH,EAAO;AAAA,MACL,OAAO,OAAO,gBAAgB,MAAM,gBAAgB,MAAM,MAAM;AAAA,QAC9D,OAAO,eAAe,SAAS,MAAM;AAAA,QACrC,QAAQ;AAAA,OACT;AAAA;AAAA,GAEJ;AAAA,EAED,MAAM,UAAU,gBAAgB,aAC5B,gBAAgB,aAChB,GAAG,gBAAgB,QAAQ,gBAAgB;AAAA,EAE/C,QAAQ,IAAI,+BAA+B,SAAS;AAAA,EAEpD,OAAO;AAAA,IACL;AAAA,IACA,UAAU,OAAO;AAAA,SACZ,MAAM;AAAA,MACT,gBAAgB,MAAM,SAAS;AAAA,MAC/B,mBAAmB,MAAM,YAAY;AAAA,IACvC;AAAA,IACA,OAAO,YAAY;AAAA,MAEjB,YAAY,WAAW,MAAM,aAAa;AAAA,QACxC,OAAO,QAAQ;AAAA,MACjB;AAAA,MAGA,cAAc,aAAa,MAAM,UAAU;AAAA,QACzC,IAAI;AAAA,UACF,SAAS,QAAQ,QAAQ;AAAA,UACzB,MAAM;AAAA,MAGV;AAAA,MAEA,MAAM,SAAS,MAAM;AAAA,MACrB,MAAM,YAAY,MAAM;AAAA,MACxB,MAAM,mBAAmB,MAAM;AAAA,MAG/B,MAAM,YAAY,MAAM;AAAA,MAGxB,IAAI,gBAAgB,YAAY;AAAA,QAC9B,IAAI;AAAA,UACF,MAAM,uBAAO,gBAAgB,UAAU;AAAA,UACvC,MAAM;AAAA,MAGV;AAAA,MAEA,QAAQ,IAAI,wBAAwB;AAAA;AAAA,EAExC;AAAA;AAGF,SAAS,WAAW,CAAC,QAA+B;AAAA,EAClD,OAAO,IAAI,QAAQ,CAAC,SAAS,WAAW;AAAA,IACtC,OAAO,MAAM,CAAC,QAAQ;AAAA,MACpB,IAAI,KAAK;AAAA,QACP,OAAO,GAAG;AAAA,MACZ,EAAO;AAAA,QACL,QAAQ;AAAA;AAAA,KAEX;AAAA,GACF;AAAA;AAGH,SAAS,WAAW,CAAC,OAA0B;AAAA,EAC7C,MAAM,MAAM,iBAAiB,MAAM,SAAS;AAAA,EAC5C,MAAM,MAAM,oBAAoB,MAAM,YAAY;AAAA;",
+  "debugId": "08DBEA612F7F726864756E2164756E21",
   "names": []
 }

package/dist/cjs/package.json

@@ -1,5 +1,5 @@
 {
   "name": "@ricsam/isolate-daemon",
-  "version": "0.1.17",
+  "version": "0.1.18",
   "type": "commonjs"
 }

package/dist/cjs/types.cjs

@@ -2,23 +2,29 @@ var __defProp = Object.defineProperty;
 var __getOwnPropNames = Object.getOwnPropertyNames;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
-var __moduleCache = /* @__PURE__ */ new WeakMap;
+function __accessProp(key) {
+  return this[key];
+}
 var __toCommonJS = (from) => {
-  var entry = __moduleCache.get(from), desc;
+  var entry = (__moduleCache ??= new WeakMap).get(from), desc;
   if (entry)
     return entry;
   entry = __defProp({}, "__esModule", { value: true });
-  if (from && typeof from === "object" || typeof from === "function")
-    __getOwnPropNames(from).map((key) => !__hasOwnProp.call(entry, key) && __defProp(entry, key, {
-      get: () => from[key],
-      enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable
-    }));
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (var key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(entry, key))
+        __defProp(entry, key, {
+          get: __accessProp.bind(from, key),
+          enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable
+        });
+  }
   __moduleCache.set(from, entry);
   return entry;
 };
+var __moduleCache;
 
 // packages/isolate-daemon/src/types.ts
 var exports_types = {};
 module.exports = __toCommonJS(exports_types);
 
-//# debugId=A80F84CE523F92D564756E2164756E21
+//# debugId=5C509FE4BE77B71164756E2164756E21

package/dist/cjs/types.cjs.map

@@ -4,6 +4,6 @@
   "sourcesContent": [
   ],
   "mappings": "",
-  "debugId": "A80F84CE523F92D564756E2164756E21",
+  "debugId": "5C509FE4BE77B71164756E2164756E21",
   "names": []
 }

package/dist/mjs/connection.mjs

@@ -70,6 +70,9 @@ function handleConnection(socket, state) {
     }
   });
   socket.on("close", () => {
+    for (const streamId of Array.from(connection.activeStreams.keys())) {
+      closeActiveDownloadSession(connection, streamId);
+    }
     for (const [, controller] of connection.dispatchAbortControllers) {
       controller.abort();
     }
@@ -103,6 +106,22 @@ function sendMessage(socket, message) {
   const frame = buildFrame(message);
   socket.write(frame);
 }
+function closeActiveDownloadSession(connection, streamId) {
+  const session = connection.activeStreams.get(streamId);
+  if (!session) {
+    return;
+  }
+  session.state = "closed";
+  if (session.creditResolver) {
+    session.creditResolver();
+    session.creditResolver = undefined;
+  }
+  if (session.cancelReader) {
+    session.cancelReader();
+    session.cancelReader = undefined;
+  }
+  connection.activeStreams.delete(streamId);
+}
 function sendError(socket, requestId, code, message, details) {
   const response = {
     type: MessageType.RESPONSE_ERROR,
@@ -1362,8 +1381,7 @@ function handleStreamError(message, connection) {
   }
   const session = connection.activeStreams.get(message.streamId);
   if (session) {
-    session.state = "closed";
-    connection.activeStreams.delete(message.streamId);
+    closeActiveDownloadSession(connection, message.streamId);
   }
   const callbackReceiver = connection.callbackStreamReceivers.get(message.streamId);
   if (callbackReceiver && callbackReceiver.state === "active") {
@@ -1482,6 +1500,7 @@ function waitForCredit(session) {
   });
 }
 async function sendStreamedResponse(connection, requestId, response) {
+  const SEND_LOOP_YIELD_BYTES = 64 * 1024;
   const streamId = connection.nextStreamId++;
   const headers = [];
   response.headers.forEach((value, key) => {
@@ -1517,38 +1536,61 @@ async function sendStreamedResponse(connection, requestId, response) {
   };
   connection.activeStreams.set(streamId, session);
   const reader = response.body.getReader();
+  session.cancelReader = () => {
+    reader.cancel("Stream cancelled by client").catch(() => {});
+  };
+  let cancelledByClient = false;
+  let bytesSinceYield = 0;
+  let chunksSinceYield = 0;
   try {
-    while (true) {
-      while (session.credit < STREAM_CHUNK_SIZE && session.state === "active") {
-        await waitForCredit(session);
-      }
-      if (session.state !== "active") {
-        throw new Error("Stream cancelled");
-      }
-      const { done, value } = await reader.read();
-      if (done) {
-        const endMsg = {
-          type: MessageType.RESPONSE_STREAM_END,
-          requestId,
-          streamId
-        };
-        sendMessage(connection.socket, endMsg);
-        break;
-      }
-      for (let offset = 0;offset < value.length; offset += STREAM_CHUNK_SIZE) {
-        const chunk = value.slice(offset, offset + STREAM_CHUNK_SIZE);
-        const chunkMsg = {
-          type: MessageType.RESPONSE_STREAM_CHUNK,
-          requestId,
-          streamId,
-          chunk
-        };
-        sendMessage(connection.socket, chunkMsg);
-        session.credit -= chunk.length;
-        session.bytesTransferred += chunk.length;
+    streamLoop:
+      while (true) {
+        while (session.credit < STREAM_CHUNK_SIZE && session.state === "active") {
+          await waitForCredit(session);
+        }
+        if (session.state !== "active") {
+          cancelledByClient = true;
+          break;
+        }
+        const { done, value } = await reader.read();
+        if (done) {
+          const endMsg = {
+            type: MessageType.RESPONSE_STREAM_END,
+            requestId,
+            streamId
+          };
+          sendMessage(connection.socket, endMsg);
+          break;
+        }
+        for (let offset = 0;offset < value.length; offset += STREAM_CHUNK_SIZE) {
+          const chunk = value.slice(offset, offset + STREAM_CHUNK_SIZE);
+          const chunkMsg = {
+            type: MessageType.RESPONSE_STREAM_CHUNK,
+            requestId,
+            streamId,
+            chunk
+          };
+          sendMessage(connection.socket, chunkMsg);
+          const creditCost = Math.max(chunk.length, STREAM_CHUNK_SIZE);
+          session.credit -= creditCost;
+          session.bytesTransferred += chunk.length;
+          bytesSinceYield += chunk.length;
+          chunksSinceYield += 1;
+          if (chunk.length < 1024 || bytesSinceYield >= SEND_LOOP_YIELD_BYTES || chunksSinceYield >= 256) {
+            bytesSinceYield = 0;
+            chunksSinceYield = 0;
+            await new Promise((resolve) => setImmediate(resolve));
+            if (session.state !== "active") {
+              cancelledByClient = true;
+              break streamLoop;
+            }
+          }
+        }
       }
-    }
   } catch (err) {
+    if (cancelledByClient || session.state !== "active") {
+      return;
+    }
     const errorMsg = {
       type: MessageType.STREAM_ERROR,
       streamId,
@@ -1556,8 +1598,15 @@ async function sendStreamedResponse(connection, requestId, response) {
     };
     sendMessage(connection.socket, errorMsg);
   } finally {
+    if (cancelledByClient) {
+      await Promise.race([
+        reader.cancel("Stream cancelled by client").catch(() => {}),
+        new Promise((resolve) => setTimeout(resolve, 50))
+      ]);
+    }
+    session.cancelReader = undefined;
     reader.releaseLock();
-    connection.activeStreams.delete(streamId);
+    closeActiveDownloadSession(connection, streamId);
   }
 }
 async function handleRunTests(message, connection, state) {
@@ -1677,4 +1726,4 @@ export {
   handleConnection
 };
 
-//# debugId=23FC39B31009F9AA64756E2164756E21
+//# debugId=1FC256F121C44C4364756E2164756E21