@grabbit-labs/dynafetch 0.2.2 → 0.2.3

package/dist/index.js

@@ -814,6 +814,34 @@ function shouldSkipDynamicScriptUrl(url, pageUrl, policy) {
 }
 
 // ../../src/phantom/execute.ts
+var esbuildModulePromise = null;
+var esbuildRefCount = 0;
+async function acquireEsbuildModule() {
+  if (!esbuildModulePromise) {
+    esbuildModulePromise = import("esbuild");
+  }
+  esbuildRefCount++;
+  try {
+    return await esbuildModulePromise;
+  } catch (error2) {
+    esbuildRefCount = Math.max(0, esbuildRefCount - 1);
+    if (esbuildRefCount === 0) esbuildModulePromise = null;
+    throw error2;
+  }
+}
+function releaseEsbuildModule(esbuildMod) {
+  if (!esbuildMod) return;
+  esbuildRefCount = Math.max(0, esbuildRefCount - 1);
+  if (esbuildRefCount > 0) return;
+  const stopFn = esbuildMod?.stop || esbuildMod?.default?.stop;
+  if (typeof stopFn === "function") {
+    try {
+      stopFn.call(esbuildMod?.default ?? esbuildMod);
+    } catch {
+    }
+  }
+  esbuildModulePromise = null;
+}
 var Executor = class {
   constructor(harvestData, options = null) {
     this.logs = [];
@@ -845,6 +873,9 @@ var Executor = class {
     this.moduleInFlight = /* @__PURE__ */ new Map();
     // entryUrl -> promise
     this.windowClosed = false;
+    this.esbuildModule = null;
+    this.originalGlobalMessageChannel = void 0;
+    this.originalGlobalMessagePort = void 0;
     // Simple telemetry counters (useful for debugging).
     this.telemetry_stubbed = 0;
     this.telemetry_proxy = 0;
@@ -936,6 +967,16 @@ var Executor = class {
     if (!warning) return;
     this.warnings.add(warning);
   }
+  unrefNewMessagePorts(initialHandles) {
+    for (const handle of process._getActiveHandles()) {
+      if (initialHandles.has(handle)) continue;
+      if (handle?.constructor?.name !== "MessagePort") continue;
+      try {
+        handle.unref?.();
+      } catch {
+      }
+    }
+  }
   applyDefaults(quiescence, moduleWaitMsOverride) {
     const hardMaxCap = this.clampMs(Number(process.env.PHANTOM_QUIESCENCE_MAX_CAP_MS ?? 8e3), 500, 6e4);
     const minWaitMs = this.clampMs(quiescence?.minWaitMs ?? 75, 0, 1e4);
@@ -1213,7 +1254,8 @@ var Executor = class {
           if (!this.windowClosed) window.eval(cached);
           return;
         }
-        const esbuildMod = await import("esbuild");
+        const esbuildMod = this.esbuildModule ?? await acquireEsbuildModule();
+        this.esbuildModule = esbuildMod;
         const buildFn = esbuildMod?.build || esbuildMod?.default?.build;
         if (typeof buildFn !== "function") {
           throw new Error("esbuild.build not available (esbuild import failed)");
@@ -1439,8 +1481,27 @@ var Executor = class {
     }
     {
       const _g = globalThis;
-      if (!window.MessageChannel && _g.MessageChannel) window.MessageChannel = _g.MessageChannel;
-      if (!window.MessagePort && _g.MessagePort) window.MessagePort = _g.MessagePort;
+      if (_g.MessageChannel) {
+        if (this.originalGlobalMessageChannel === void 0) {
+          this.originalGlobalMessageChannel = _g.MessageChannel;
+        }
+        const NativeMessageChannel = _g.MessageChannel;
+        const UnrefMessageChannel = class MessageChannel extends NativeMessageChannel {
+          constructor() {
+            super();
+            this.port1?.unref?.();
+            this.port2?.unref?.();
+          }
+        };
+        window.MessageChannel = UnrefMessageChannel;
+        _g.MessageChannel = UnrefMessageChannel;
+      }
+      if (_g.MessagePort) {
+        if (this.originalGlobalMessagePort === void 0) {
+          this.originalGlobalMessagePort = _g.MessagePort;
+        }
+        window.MessagePort = _g.MessagePort;
+      }
     }
     if (!window.requestIdleCallback) {
       window.requestIdleCallback = (cb) => window.setTimeout(() => cb({
@@ -1664,6 +1725,7 @@ var Executor = class {
   async execute() {
     const onNodeUncaught = (err) => this.recordExecutionError(err, "uncaughtException");
     const onNodeUnhandled = (reason) => this.recordExecutionError(reason, "unhandledRejection");
+    const initialActiveHandles = new Set(process._getActiveHandles());
     process.on("uncaughtException", onNodeUncaught);
     process.on("unhandledRejection", onNodeUnhandled);
     try {
@@ -2091,8 +2153,20 @@ var Executor = class {
       };
       const shutdownGraceMs = this.clampMs(Number(process.env.PHANTOM_SHUTDOWN_GRACE_MS ?? 50), 10, 5e3);
       await new Promise((r) => setTimeout(r, shutdownGraceMs));
+      this.unrefNewMessagePorts(initialActiveHandles);
       return result;
     } finally {
+      const g = globalThis;
+      if (this.originalGlobalMessageChannel !== void 0) {
+        g.MessageChannel = this.originalGlobalMessageChannel;
+        this.originalGlobalMessageChannel = void 0;
+      }
+      if (this.originalGlobalMessagePort !== void 0) {
+        g.MessagePort = this.originalGlobalMessagePort;
+        this.originalGlobalMessagePort = void 0;
+      }
+      releaseEsbuildModule(this.esbuildModule);
+      this.esbuildModule = null;
       process.off("uncaughtException", onNodeUncaught);
       process.off("unhandledRejection", onNodeUnhandled);
     }