@ricsam/isolate-daemon 0.1.11 → 0.1.12

package/dist/cjs/package.json

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

package/dist/mjs/connection.mjs

@@ -26,6 +26,12 @@ import {
 import {
   createInternalRuntime
 } from "@ricsam/isolate-runtime";
+import {
+  transformEntryCode,
+  transformModuleCode,
+  contentHash,
+  mapErrorStack
+} from "@ricsam/isolate-transform";
 function handleConnection(socket, state) {
   const connection = {
     socket,
@@ -587,7 +593,14 @@ async function handleEval(message, connection, state) {
   instance.lastActivity = Date.now();
   try {
     const filename = normalizeEntryFilename(message.filename);
-    const mod = await instance.runtime.isolate.compileModule(message.code, {
+    const transformed = await 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) {
@@ -599,13 +612,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 = mapErrorStack(error.stack, instance.sourceMaps);
+    }
     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 });
   }
@@ -1370,7 +1395,19 @@ function createModuleResolver(instance, connection) {
     const importerResolveDir = path.posix.dirname(importerPath);
     const result = await invokeClientCallback(connection, instance.moduleLoaderCallbackId, [specifier, { path: importerPath, resolveDir: importerResolveDir }]);
     const { code, resolveDir } = result;
-    const mod = await instance.runtime.isolate.compileModule(code, {
+    const hash = contentHash(code);
+    const cacheKey = `${specifier}:${hash}`;
+    const hashCached = instance.moduleCache?.get(cacheKey);
+    if (hashCached)
+      return hashCached;
+    const transformed = await 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 = path.posix.join(resolveDir, path.posix.basename(specifier));
@@ -1378,6 +1415,7 @@ function createModuleResolver(instance, connection) {
     const resolver = createModuleResolver(instance, connection);
     await mod.instantiate(instance.runtime.context, resolver);
     instance.moduleCache?.set(specifier, mod);
+    instance.moduleCache?.set(cacheKey, mod);
     return mod;
   };
 }
@@ -1520,6 +1558,10 @@ function handleCallbackStreamStart(message, connection) {
         resolver();
       }
       connection.callbackStreamReceivers.delete(message.streamId);
+      sendMessage(connection.socket, {
+        type: MessageType.CALLBACK_STREAM_CANCEL,
+        streamId: message.streamId
+      });
       return Promise.resolve();
     }
   });
@@ -1795,4 +1837,4 @@ export {
   handleConnection
 };
 
-//# debugId=E68FD11623EE904064756E2164756E21
+//# debugId=514032B8427B94A564756E2164756E21