poe-code 14.0.57 → 14.0.59

package/packages/safe-js/dist/interp/resources.d.ts

@@ -1,6 +1,7 @@
 import { AsyncLocalStorage } from "node:async_hooks";
 export type RunResources = {
     signal: AbortSignal;
+    referenceReleases: Set<() => void>;
     add(close: () => Promise<void>): void;
 };
 export declare const runResources: AsyncLocalStorage<RunResources>;

package/packages/safe-js/dist/interp/resources.js

@@ -6,6 +6,7 @@ export async function withRunResources(signal, execute) {
     const cleanups = new Set();
     const resources = {
         signal: controller.signal,
+        referenceReleases: new Set(),
         add(close) {
             cleanups.add(close);
         }
@@ -25,6 +26,9 @@ export async function withRunResources(signal, execute) {
     finally {
         signal?.removeEventListener("abort", cancel);
         controller.abort(new Error("SafeJS run finished."));
+        for (const release of resources.referenceReleases)
+            release();
+        resources.referenceReleases.clear();
         const outcomes = await Promise.allSettled([...cleanups].map((close) => Promise.resolve().then(close)));
         errors = outcomes.flatMap((outcome) => (outcome.status === "rejected" ? [outcome.reason] : []));
     }

package/packages/safe-js/dist/realm.js

@@ -34,6 +34,7 @@ class RealmState {
     extensions;
     consoleExtension;
     cleanups = [];
+    referenceReleases = new Set();
     callbacks = new Map();
     pendingCallbacks = new Set();
     callbackCache = new WeakMap();
@@ -458,7 +459,7 @@ class RealmState {
         };
         try {
             const active = this.active !== undefined;
-            const pending = withSandboxPromiseRejectionTracker(this.tracker, () => runResources.run({ signal: this.controller.signal, add: this.onCleanup }, () => withCancellationSignal(this.controller.signal, () => active && this.phase.getStore()?.active ? runAsyncPrefix(invoke) : this.queue.run(invoke))));
+            const pending = withSandboxPromiseRejectionTracker(this.tracker, () => runResources.run({ signal: this.controller.signal, referenceReleases: this.referenceReleases, add: this.onCleanup }, () => withCancellationSignal(this.controller.signal, () => active && this.phase.getStore()?.active ? runAsyncPrefix(invoke) : this.queue.run(invoke))));
             record.promise = active ? pending : this.perform(() => pending);
             return await record.promise;
         }
@@ -645,7 +646,7 @@ class RealmState {
         this.assertOpen();
         if (this.active !== undefined)
             throw new SandboxError("reentry");
-        const pending = Promise.resolve().then(() => withSandboxPromiseRejectionTracker(this.tracker, () => runResources.run({ signal: this.controller.signal, add: this.onCleanup }, () => withCancellationSignal(this.controller.signal, task))));
+        const pending = Promise.resolve().then(() => withSandboxPromiseRejectionTracker(this.tracker, () => runResources.run({ signal: this.controller.signal, referenceReleases: this.referenceReleases, add: this.onCleanup }, () => withCancellationSignal(this.controller.signal, task))));
         this.active = pending;
         try {
             const result = await pending;
@@ -701,6 +702,9 @@ class RealmState {
         for (const reference of this.guestReferences.keys())
             revokeGuestReference(reference, this);
         this.guestReferences.clear();
+        for (const release of this.referenceReleases)
+            release();
+        this.referenceReleases.clear();
         this.budget.setRetainedValues(this, undefined);
         releaseObjectPrototype(this.budget);
         this.disposal = (async () => {