@ricsam/isolate-daemon 0.1.14 → 0.1.15

package/dist/mjs/connection.mjs

@@ -7,12 +7,39 @@ import {
   ErrorCode,
   STREAM_CHUNK_SIZE,
   STREAM_DEFAULT_CREDIT,
-  marshalValue
+  marshalValue,
+  isPromiseRef,
+  isAsyncIteratorRef,
+  deserializeResponse,
+  IsolateEvents,
+  ClientEvents
 } from "@ricsam/isolate-protocol";
 import { createCallbackFileSystemHandler } from "./callback-fs-handler.mjs";
 import {
   createRuntime
 } from "@ricsam/isolate-runtime";
+var LINKER_CONFLICT_ERROR = "Module is currently being linked by another linker";
+function getErrorText(error) {
+  if (error instanceof Error) {
+    const cause = error.cause;
+    const causeText = cause instanceof Error ? `${cause.name}: ${cause.message}
+${cause.stack ?? ""}` : cause != null ? String(cause) : "";
+    return [error.name, error.message, error.stack, causeText].filter((part) => part != null && part !== "").join(`
+`);
+  }
+  if (typeof error === "string") {
+    return error;
+  }
+  try {
+    return JSON.stringify(error);
+  } catch {
+    return String(error ?? "");
+  }
+}
+function isLinkerConflictError(error) {
+  const text = getErrorText(error).toLowerCase();
+  return text.includes(LINKER_CONFLICT_ERROR.toLowerCase());
+}
 function handleConnection(socket, state) {
   const connection = {
     socket,
@@ -45,17 +72,17 @@ function handleConnection(socket, state) {
       const instance = state.isolates.get(isolateId);
       if (instance) {
         if (instance.namespaceId != null && !instance.isDisposed) {
-          softDeleteRuntime(instance, state);
+          if (instance.isPoisoned) {
+            hardDeleteRuntime(instance, state).catch(() => {});
+          } else {
+            softDeleteRuntime(instance, state);
+          }
         } else if (!instance.isDisposed) {
-          instance.runtime.dispose().catch(() => {});
-          state.isolates.delete(isolateId);
+          hardDeleteRuntime(instance, state).catch(() => {});
         }
       }
     }
     for (const [, pending] of connection.pendingCallbacks) {
-      if (pending.timeoutId) {
-        clearTimeout(pending.timeoutId);
-      }
       pending.reject(new Error("Connection closed"));
     }
     connection.pendingCallbacks.clear();
@@ -183,22 +210,32 @@ async function handleMessage(message, connection, state) {
     case MessageType.CALLBACK_STREAM_END:
       handleCallbackStreamEnd(message, connection);
       break;
-    case MessageType.CLIENT_WS_OPENED:
-      handleClientWsOpened(message, connection, state);
-      break;
-    case MessageType.CLIENT_WS_MESSAGE:
-      handleClientWsMessage(message, connection, state);
-      break;
-    case MessageType.CLIENT_WS_CLOSED:
-      handleClientWsClosed(message, connection, state);
-      break;
-    case MessageType.CLIENT_WS_ERROR:
-      handleClientWsError(message, connection, state);
+    case MessageType.CLIENT_EVENT:
+      handleClientEvent(message, connection, state);
       break;
     default:
       sendError(connection.socket, message.requestId ?? 0, ErrorCode.UNKNOWN_MESSAGE_TYPE, `Unknown message type: ${message.type}`);
   }
 }
+async function hardDeleteRuntime(instance, state) {
+  try {
+    await instance.runtime.dispose();
+  } finally {
+    state.isolates.delete(instance.isolateId);
+    if (instance.namespaceId != null) {
+      const indexed = state.namespacedRuntimes.get(instance.namespaceId);
+      if (indexed?.isolateId === instance.isolateId) {
+        state.namespacedRuntimes.delete(instance.namespaceId);
+      }
+    }
+    instance.isDisposed = true;
+    instance.disposedAt = undefined;
+    instance.ownerConnection = null;
+    if (instance.callbackContext) {
+      instance.callbackContext.connection = null;
+    }
+  }
+}
 function softDeleteRuntime(instance, state) {
   instance.isDisposed = true;
   instance.disposedAt = Date.now();
@@ -218,6 +255,7 @@ function softDeleteRuntime(instance, state) {
 function reuseNamespacedRuntime(instance, connection, message, state) {
   instance.ownerConnection = connection.socket;
   instance.isDisposed = false;
+  instance.isPoisoned = false;
   instance.disposedAt = undefined;
   instance.lastActivity = Date.now();
   connection.isolates.add(instance.isolateId);
@@ -332,18 +370,15 @@ async function evictOldestDisposedRuntime(state) {
   }
   if (oldest) {
     try {
-      await oldest.runtime.dispose();
+      await hardDeleteRuntime(oldest, state);
     } catch {}
-    state.isolates.delete(oldest.isolateId);
-    if (oldest.namespaceId != null) {
-      state.namespacedRuntimes.delete(oldest.namespaceId);
-    }
     return true;
   }
   return false;
 }
 async function handleCreateRuntime(message, connection, state) {
   const namespaceId = message.options.namespaceId;
+  let namespaceCreationLocked = false;
   if (namespaceId != null) {
     const existing = state.namespacedRuntimes.get(namespaceId);
     if (existing) {
@@ -365,14 +400,20 @@ async function handleCreateRuntime(message, connection, state) {
       });
       return;
     }
-  }
-  if (state.isolates.size >= state.options.maxIsolates) {
-    if (!await evictOldestDisposedRuntime(state)) {
-      sendError(connection.socket, message.requestId, ErrorCode.ISOLATE_MEMORY_LIMIT, `Maximum isolates (${state.options.maxIsolates}) reached`);
+    if (state.namespacedCreatesInFlight.has(namespaceId)) {
+      sendError(connection.socket, message.requestId, ErrorCode.SCRIPT_ERROR, `Namespace "${namespaceId}" creation already in progress`);
       return;
     }
+    state.namespacedCreatesInFlight.add(namespaceId);
+    namespaceCreationLocked = true;
   }
   try {
+    if (state.isolates.size >= state.options.maxIsolates) {
+      if (!await evictOldestDisposedRuntime(state)) {
+        sendError(connection.socket, message.requestId, ErrorCode.ISOLATE_MEMORY_LIMIT, `Maximum isolates (${state.options.maxIsolates}) reached`);
+        return;
+      }
+    }
     const isolateId = randomUUID();
     const consoleCallbacks = message.options.callbacks?.console;
     const fetchCallback = message.options.callbacks?.fetch;
@@ -415,6 +456,7 @@ async function handleCreateRuntime(message, connection, state) {
       nextLocalCallbackId: 1e6,
       namespaceId,
       isDisposed: false,
+      isPoisoned: false,
       callbackContext
     };
     let bridgedCustomFunctions;
@@ -437,8 +479,6 @@ async function handleCreateRuntime(message, connection, state) {
           return iteratorId;
         }
       });
-      const isPromiseRef = (value) => typeof value === "object" && value !== null && value.__type === "PromiseRef";
-      const isAsyncIteratorRef = (value) => typeof value === "object" && value !== null && value.__type === "AsyncIteratorRef";
       const addCallbackIdsToRefs = (value) => {
         if (value === null || typeof value !== "object")
           return value;
@@ -617,7 +657,7 @@ async function handleCreateRuntime(message, connection, state) {
             };
           }
           try {
-            const resultJson = await invokeClientCallback(conn, callbackId, [JSON.stringify(op)], 60000);
+            const resultJson = await invokeClientCallback(conn, callbackId, [JSON.stringify(op)]);
             return JSON.parse(resultJson);
           } catch (err) {
             const error = err;
@@ -668,7 +708,7 @@ async function handleCreateRuntime(message, connection, state) {
           headers: init.headers,
           body: init.rawBody
         };
-        const result = await invokeClientCallback(conn, callbackId, [serialized], 60000);
+        const result = await invokeClientCallback(conn, callbackId, [serialized]);
         if (result && typeof result === "object" && result.__streamingResponse) {
           const response = result.response;
           response.__isCallbackStream = true;
@@ -768,18 +808,19 @@ async function handleCreateRuntime(message, connection, state) {
       } else {
         data = cmd.data;
       }
-      const wsCommandMsg = {
-        type: MessageType.WS_COMMAND,
-        isolateId,
-        command: {
-          type: cmd.type,
-          connectionId: cmd.connectionId,
-          data,
-          code: cmd.code,
-          reason: cmd.reason
-        }
+      const payload = {
+        type: cmd.type,
+        connectionId: cmd.connectionId,
+        data,
+        code: cmd.code,
+        reason: cmd.reason
       };
-      sendMessage(targetConnection.socket, wsCommandMsg);
+      sendMessage(targetConnection.socket, {
+        type: MessageType.ISOLATE_EVENT,
+        isolateId,
+        event: IsolateEvents.WS_COMMAND,
+        payload
+      });
     });
     runtime.fetch.onClientWebSocketCommand((cmd) => {
       const targetConnection = callbackContext.connection;
@@ -793,40 +834,62 @@ async function handleCreateRuntime(message, connection, state) {
         data = cmd.data;
       }
       if (cmd.type === "connect") {
-        const msg = {
-          type: MessageType.CLIENT_WS_CONNECT,
-          requestId: 0,
-          isolateId,
+        const payload = {
           socketId: cmd.socketId,
           url: cmd.url,
           protocols: cmd.protocols
         };
-        sendMessage(targetConnection.socket, msg);
-      } else if (cmd.type === "send") {
-        const msg = {
-          type: MessageType.CLIENT_WS_SEND,
-          requestId: 0,
+        sendMessage(targetConnection.socket, {
+          type: MessageType.ISOLATE_EVENT,
           isolateId,
+          event: IsolateEvents.WS_CLIENT_CONNECT,
+          payload
+        });
+      } else if (cmd.type === "send") {
+        const payload = {
           socketId: cmd.socketId,
           data
         };
-        sendMessage(targetConnection.socket, msg);
-      } else if (cmd.type === "close") {
-        const msg = {
-          type: MessageType.CLIENT_WS_CLOSE,
-          requestId: 0,
+        sendMessage(targetConnection.socket, {
+          type: MessageType.ISOLATE_EVENT,
           isolateId,
+          event: IsolateEvents.WS_CLIENT_SEND,
+          payload
+        });
+      } else if (cmd.type === "close") {
+        const payload = {
           socketId: cmd.socketId,
           code: cmd.code,
           reason: cmd.reason
         };
-        sendMessage(targetConnection.socket, msg);
+        sendMessage(targetConnection.socket, {
+          type: MessageType.ISOLATE_EVENT,
+          isolateId,
+          event: IsolateEvents.WS_CLIENT_CLOSE,
+          payload
+        });
       }
     });
+    runtime.fetch.onEvent((eventName, payload) => {
+      const targetConnection = callbackContext.connection;
+      if (!targetConnection) {
+        return;
+      }
+      sendMessage(targetConnection.socket, {
+        type: MessageType.ISOLATE_EVENT,
+        isolateId,
+        event: eventName,
+        payload
+      });
+    });
     sendOk(connection.socket, message.requestId, { isolateId, reused: false });
   } catch (err) {
     const error = err;
     sendError(connection.socket, message.requestId, ErrorCode.SCRIPT_ERROR, error.message, { name: error.name, stack: error.stack });
+  } finally {
+    if (namespaceCreationLocked && namespaceId != null) {
+      state.namespacedCreatesInFlight.delete(namespaceId);
+    }
   }
 }
 async function handleDisposeRuntime(message, connection, state) {
@@ -842,14 +905,20 @@ async function handleDisposeRuntime(message, connection, state) {
   try {
     connection.isolates.delete(message.isolateId);
     if (instance.namespaceId != null) {
-      softDeleteRuntime(instance, state);
+      if (instance.isPoisoned) {
+        await hardDeleteRuntime(instance, state);
+      } else {
+        softDeleteRuntime(instance, state);
+      }
     } else {
-      await instance.runtime.dispose();
-      state.isolates.delete(message.isolateId);
+      await hardDeleteRuntime(instance, state);
     }
     sendOk(connection.socket, message.requestId);
   } catch (err) {
     const error = err;
+    if (instance.namespaceId != null && isLinkerConflictError(error)) {
+      instance.isPoisoned = true;
+    }
     sendError(connection.socket, message.requestId, ErrorCode.SCRIPT_ERROR, error.message, { name: error.name, stack: error.stack });
   }
 }
@@ -862,14 +931,15 @@ async function handleEval(message, connection, state) {
   instance.lastActivity = Date.now();
   try {
     await instance.runtime.eval(message.code, {
-      filename: message.filename,
-      maxExecutionMs: message.maxExecutionMs
+      filename: message.filename
     });
     sendOk(connection.socket, message.requestId, { value: undefined });
   } catch (err) {
     const error = err;
-    const isTimeoutError = error.message?.includes("Script execution timed out");
-    sendError(connection.socket, message.requestId, isTimeoutError ? ErrorCode.ISOLATE_TIMEOUT : ErrorCode.SCRIPT_ERROR, error.message, { name: error.name, stack: error.stack });
+    if (instance.namespaceId != null && isLinkerConflictError(error)) {
+      instance.isPoisoned = true;
+    }
+    sendError(connection.socket, message.requestId, ErrorCode.SCRIPT_ERROR, error.message, { name: error.name, stack: error.stack });
   }
 }
 async function handleDispatchRequest(message, connection, state) {
@@ -977,34 +1047,38 @@ async function handleWsClose(message, connection, state) {
     sendError(connection.socket, message.requestId, ErrorCode.SCRIPT_ERROR, error.message, { name: error.name, stack: error.stack });
   }
 }
-function handleClientWsOpened(message, connection, state) {
+function handleClientEvent(message, connection, state) {
   const instance = state.isolates.get(message.isolateId);
   if (!instance)
     return;
   instance.lastActivity = Date.now();
-  instance.runtime.fetch.dispatchClientWebSocketOpen(message.socketId, message.protocol, message.extensions);
-}
-function handleClientWsMessage(message, connection, state) {
-  const instance = state.isolates.get(message.isolateId);
-  if (!instance)
-    return;
-  instance.lastActivity = Date.now();
-  const data = message.data instanceof Uint8Array ? message.data.buffer.slice(message.data.byteOffset, message.data.byteOffset + message.data.byteLength) : message.data;
-  instance.runtime.fetch.dispatchClientWebSocketMessage(message.socketId, data);
-}
-function handleClientWsClosed(message, connection, state) {
-  const instance = state.isolates.get(message.isolateId);
-  if (!instance)
-    return;
-  instance.lastActivity = Date.now();
-  instance.runtime.fetch.dispatchClientWebSocketClose(message.socketId, message.code, message.reason, message.wasClean);
-}
-function handleClientWsError(message, connection, state) {
-  const instance = state.isolates.get(message.isolateId);
-  if (!instance)
-    return;
-  instance.lastActivity = Date.now();
-  instance.runtime.fetch.dispatchClientWebSocketError(message.socketId);
+  switch (message.event) {
+    case ClientEvents.WS_CLIENT_OPENED: {
+      const payload = message.payload;
+      instance.runtime.fetch.dispatchClientWebSocketOpen(payload.socketId, payload.protocol, payload.extensions);
+      break;
+    }
+    case ClientEvents.WS_CLIENT_MESSAGE: {
+      const payload = message.payload;
+      const data = payload.data instanceof Uint8Array ? payload.data.buffer.slice(payload.data.byteOffset, payload.data.byteOffset + payload.data.byteLength) : payload.data;
+      instance.runtime.fetch.dispatchClientWebSocketMessage(payload.socketId, data);
+      break;
+    }
+    case ClientEvents.WS_CLIENT_CLOSED: {
+      const payload = message.payload;
+      instance.runtime.fetch.dispatchClientWebSocketClose(payload.socketId, payload.code, payload.reason, payload.wasClean);
+      break;
+    }
+    case ClientEvents.WS_CLIENT_ERROR: {
+      const payload = message.payload;
+      instance.runtime.fetch.dispatchClientWebSocketError(payload.socketId);
+      break;
+    }
+    default: {
+      instance.runtime.fetch.dispatchEvent(message.event, message.payload);
+      break;
+    }
+  }
 }
 async function handleFetchGetUpgradeRequest(message, connection, state) {
   const instance = state.isolates.get(message.isolateId);
@@ -1148,9 +1222,6 @@ function handleCallbackResponse(message, connection) {
     return;
   }
   connection.pendingCallbacks.delete(message.requestId);
-  if (pending.timeoutId) {
-    clearTimeout(pending.timeoutId);
-  }
   if (message.error) {
     const error = new Error(message.error.message);
     error.name = message.error.name;
@@ -1162,17 +1233,12 @@ function handleCallbackResponse(message, connection) {
     pending.resolve(message.result);
   }
 }
-async function invokeClientCallback(connection, callbackId, args, timeout = 1e4) {
+async function invokeClientCallback(connection, callbackId, args) {
   const requestId = connection.nextCallbackId++;
   return new Promise((resolve, reject) => {
-    const timeoutId = setTimeout(() => {
-      connection.pendingCallbacks.delete(requestId);
-      reject(new Error("Callback timeout"));
-    }, timeout);
     const pending = {
       resolve,
-      reject,
-      timeoutId
+      reject
     };
     connection.pendingCallbacks.set(requestId, pending);
     const invoke = {
@@ -1184,13 +1250,6 @@ async function invokeClientCallback(connection, callbackId, args, timeout = 1e4)
     sendMessage(connection.socket, invoke);
   });
 }
-function deserializeResponse(data) {
-  return new Response(data.body, {
-    status: data.status,
-    statusText: data.statusText,
-    headers: data.headers
-  });
-}
 function handleStreamPush(message, connection) {
   const receiver = connection.streamReceivers.get(message.streamId);
   if (!receiver) {
@@ -1318,9 +1377,6 @@ function handleCallbackStreamStart(message, connection) {
   const pending = connection.pendingCallbacks.get(message.requestId);
   if (pending) {
     connection.pendingCallbacks.delete(message.requestId);
-    if (pending.timeoutId) {
-      clearTimeout(pending.timeoutId);
-    }
     const response = new Response(readableStream, {
       status: message.metadata.status,
       statusText: message.metadata.statusText,
@@ -1539,4 +1595,4 @@ export {
   handleConnection
 };
 
-//# debugId=C809F161C354C84D64756E2164756E21
+//# debugId=4DCD38BA729DF7E764756E2164756E21