@ricsam/isolate-daemon 0.1.10 → 0.1.12

package/dist/cjs/connection.cjs

@@ -46,12 +46,14 @@ __export(exports_connection, {
 });
 module.exports = __toCommonJS(exports_connection);
 var import_node_crypto = require("node:crypto");
+var import_node_path = __toESM(require("node:path"));
 var import_isolated_vm = __toESM(require("isolated-vm"));
 var import_isolate_protocol = require("@ricsam/isolate-protocol");
 var import_callback_fs_handler = require("./callback-fs-handler.cjs");
 var import_isolate_test_environment = require("@ricsam/isolate-test-environment");
 var import_isolate_playwright = require("@ricsam/isolate-playwright");
 var import_isolate_runtime = require("@ricsam/isolate-runtime");
+var import_isolate_transform = require("@ricsam/isolate-transform");
 function handleConnection(socket, state) {
   const connection = {
     socket,
@@ -434,7 +436,7 @@ async function handleCreateRuntime(message, connection, state) {
         }
       },
       fs: {
-        getDirectory: async (path) => {
+        getDirectory: async (path2) => {
           const conn = callbackContext.connection;
           if (!conn) {
             throw new Error("FS callbacks not available");
@@ -443,7 +445,7 @@ async function handleCreateRuntime(message, connection, state) {
             connection: conn,
             callbackContext,
             invokeClientCallback,
-            basePath: path
+            basePath: path2
           });
         }
       }
@@ -467,6 +469,7 @@ async function handleCreateRuntime(message, connection, state) {
     if (moduleLoaderCallback) {
       instance.moduleLoaderCallbackId = moduleLoaderCallback.callbackId;
       instance.moduleCache = new Map;
+      instance.moduleToFilename = new Map;
     }
     if (customCallbacks) {
       await setupCustomFunctions(runtime.context, customCallbacks, connection, instance);
@@ -611,10 +614,19 @@ async function handleEval(message, connection, state) {
   }
   instance.lastActivity = Date.now();
   try {
-    const mod = await instance.runtime.isolate.compileModule(message.code, {
-      filename: message.filename ?? "<eval>"
+    const filename = import_isolate_protocol.normalizeEntryFilename(message.filename);
+    const transformed = await import_isolate_transform.transformEntryCode(message.code, filename);
+    if (transformed.sourceMap) {
+      if (!instance.sourceMaps) {
+        instance.sourceMaps = new Map;
+      }
+      instance.sourceMaps.set(filename, transformed.sourceMap);
+    }
+    const mod = await instance.runtime.isolate.compileModule(transformed.code, {
+      filename
     });
     if (instance.moduleLoaderCallbackId) {
+      instance.moduleToFilename?.set(mod, filename);
       const resolver = createModuleResolver(instance, connection);
       await mod.instantiate(instance.runtime.context, resolver);
     } else {
@@ -622,13 +634,25 @@ async function handleEval(message, connection, state) {
         throw new Error(`No module loader registered. Cannot import: ${specifier}`);
       });
     }
-    const timeout = message.maxExecutionMs;
-    await mod.evaluate(timeout ? { timeout } : undefined);
+    await mod.evaluate();
+    const ns = mod.namespace;
+    const runRef = await ns.get("default", { reference: true });
+    try {
+      await runRef.apply(undefined, [], {
+        result: { promise: true },
+        ...message.maxExecutionMs ? { timeout: message.maxExecutionMs } : {}
+      });
+    } finally {
+      runRef.release();
+    }
     await Promise.all(instance.pendingCallbacks);
     instance.pendingCallbacks.length = 0;
     sendOk(connection.socket, message.requestId, { value: undefined });
   } catch (err) {
     const error = err;
+    if (error.stack && instance.sourceMaps?.size) {
+      error.stack = import_isolate_transform.mapErrorStack(error.stack, instance.sourceMaps);
+    }
     const isTimeoutError = error.message?.includes("Script execution timed out");
     sendError(connection.socket, message.requestId, isTimeoutError ? import_isolate_protocol.ErrorCode.ISOLATE_TIMEOUT : import_isolate_protocol.ErrorCode.SCRIPT_ERROR, error.message, { name: error.name, stack: error.stack });
   }
@@ -1382,20 +1406,38 @@ async function setupCustomFunctions(context, customCallbacks, connection, instan
   }
 }
 function createModuleResolver(instance, connection) {
-  return async (specifier, _referrer) => {
+  return async (specifier, referrer) => {
     const cached = instance.moduleCache?.get(specifier);
     if (cached)
       return cached;
     if (!instance.moduleLoaderCallbackId) {
       throw new Error(`Module not found: ${specifier}`);
     }
-    const code = await invokeClientCallback(connection, instance.moduleLoaderCallbackId, [specifier]);
-    const mod = await instance.runtime.isolate.compileModule(code, {
+    const importerPath = instance.moduleToFilename?.get(referrer) ?? "<unknown>";
+    const importerResolveDir = import_node_path.default.posix.dirname(importerPath);
+    const result = await invokeClientCallback(connection, instance.moduleLoaderCallbackId, [specifier, { path: importerPath, resolveDir: importerResolveDir }]);
+    const { code, resolveDir } = result;
+    const hash = import_isolate_transform.contentHash(code);
+    const cacheKey = `${specifier}:${hash}`;
+    const hashCached = instance.moduleCache?.get(cacheKey);
+    if (hashCached)
+      return hashCached;
+    const transformed = await import_isolate_transform.transformModuleCode(code, specifier);
+    if (transformed.sourceMap) {
+      if (!instance.sourceMaps) {
+        instance.sourceMaps = new Map;
+      }
+      instance.sourceMaps.set(specifier, transformed.sourceMap);
+    }
+    const mod = await instance.runtime.isolate.compileModule(transformed.code, {
       filename: specifier
     });
+    const resolvedPath = import_node_path.default.posix.join(resolveDir, import_node_path.default.posix.basename(specifier));
+    instance.moduleToFilename?.set(mod, resolvedPath);
     const resolver = createModuleResolver(instance, connection);
     await mod.instantiate(instance.runtime.context, resolver);
     instance.moduleCache?.set(specifier, mod);
+    instance.moduleCache?.set(cacheKey, mod);
     return mod;
   };
 }
@@ -1538,6 +1580,10 @@ function handleCallbackStreamStart(message, connection) {
         resolver();
       }
       connection.callbackStreamReceivers.delete(message.streamId);
+      sendMessage(connection.socket, {
+        type: import_isolate_protocol.MessageType.CALLBACK_STREAM_CANCEL,
+        streamId: message.streamId
+      });
       return Promise.resolve();
     }
   });
@@ -1810,4 +1856,4 @@ async function handleClearCollectedData(message, connection, state) {
   }
 }
 
-//# debugId=DE5E35C70F14FD3964756E2164756E21
+//# debugId=32017BAA615D918C64756E2164756E21