poe-code 14.0.57 → 14.0.58

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

@@ -22,7 +22,7 @@ import {
   resolveFsConfig,
   splitFrontmatter,
   withBrokenPipeGuard
-} from "./chunks/chunk-H4O5OPM2.js";
+} from "./chunks/chunk-3VTLNK4E.js";
 import {
   Budget,
   SandboxError,
@@ -34,7 +34,7 @@ import {
   replaceErrorStack,
   restore,
   run
-} from "./chunks/chunk-N3CN2ACZ.js";
+} from "./chunks/chunk-DHD453E5.js";
 import "./chunks/chunk-UXMKEUH3.js";
 import "./chunks/chunk-5DYEOCT5.js";
 import "./chunks/chunk-4YIDX44G.js";

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

@@ -5,7 +5,7 @@ import {
   defineExtension,
   lint,
   run
-} from "./chunks/chunk-N3CN2ACZ.js";
+} from "./chunks/chunk-DHD453E5.js";
 import "./chunks/chunk-4YIDX44G.js";
 export {
   Budget,

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

@@ -26,7 +26,7 @@ import {
   runWithSpawnUsageAccumulator,
   splitFrontmatter,
   supportsSpawnMode
-} from "./chunks/chunk-H4O5OPM2.js";
+} from "./chunks/chunk-3VTLNK4E.js";
 import {
   Budget,
   FileSnapshotBackend,
@@ -50,7 +50,7 @@ import {
   registerPendingHostCallPolicy,
   restore,
   run
-} from "./chunks/chunk-N3CN2ACZ.js";
+} from "./chunks/chunk-DHD453E5.js";
 import "./chunks/chunk-UXMKEUH3.js";
 import "./chunks/chunk-5DYEOCT5.js";
 import "./chunks/chunk-4YIDX44G.js";

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

@@ -4,6 +4,7 @@ import { toPropertyKey } from "./property-key.js";
 import { assertPromiseExecutionAllowed } from "./promise-tracker.js";
 import { SandboxJobQueue, runAsyncPrefix, suspendJob } from "./jobs.js";
 import { withCancellationSignal } from "./cancel.js";
+import { runResources } from "./resources.js";
 import { attachErrorSpan, formatErrorStack, replaceErrorStack } from "../error/shape.js";
 import { evaluateArrowFunctionExpression, evaluateFunctionExpression, evaluateAwaitExpression, emitResumeBreakpoint, createInterpretedClosure, normalizeClosureResult } from "./async.js";
 import { HostCallResumabilityError } from "./host-call.js";
@@ -519,66 +520,61 @@ async function evaluateMemberAssignmentExpression(node, context) {
     if (node.left.type !== "MemberExpression") {
         throw new TypeError("Expected member assignment target.");
     }
-    const member = await evaluateMemberAccess(node.left, context);
-    if (member.kind === "error") {
-        return member;
-    }
-    if (member.kind === "completion") {
-        return member.result;
-    }
-    if (member.kind === "nullish") {
-        throw new TypeError("Cannot assign properties of null or undefined.");
-    }
-    let property;
-    let current = undefined;
-    if (node.operator !== "=") {
+    return evaluateMemberAccess(node.left, context, async (member) => {
+        if (member.kind === "nullish") {
+            throw new TypeError("Cannot assign properties of null or undefined.");
+        }
+        let property;
+        let current = undefined;
+        if (node.operator !== "=") {
+            if (member.object === null || member.object === undefined) {
+                throw new TypeError("Cannot assign properties of null or undefined.");
+            }
+            property = await toPropertyKey(member.property, context.budget, createCoercionContext(context));
+            current = getPropertyValue(member.object, property, context);
+        }
+        if (node.operator === "&&=" && !isTruthy(current)) {
+            return {
+                kind: "normal",
+                hasValue: true,
+                value: current
+            };
+        }
+        if (node.operator === "||=" && isTruthy(current)) {
+            return {
+                kind: "normal",
+                hasValue: true,
+                value: current
+            };
+        }
+        if (node.operator === "??=" && current !== null && current !== undefined) {
+            return {
+                kind: "normal",
+                hasValue: true,
+                value: current
+            };
+        }
+        const right = await evaluateNode(node.right, context);
+        if (right.kind !== "normal") {
+            return right;
+        }
+        const value = node.operator === "=" ||
+            node.operator === "&&=" ||
+            node.operator === "||=" ||
+            node.operator === "??="
+            ? right.value
+            : await applyCompoundAssignmentOperator(node.operator, current, right.value, context);
         if (member.object === null || member.object === undefined) {
             throw new TypeError("Cannot assign properties of null or undefined.");
         }
-        property = await toPropertyKey(member.property, context.budget, createCoercionContext(context));
-        current = getPropertyValue(member.object, property, context);
-    }
-    if (node.operator === "&&=" && !isTruthy(current)) {
-        return {
-            kind: "normal",
-            hasValue: true,
-            value: current
-        };
-    }
-    if (node.operator === "||=" && isTruthy(current)) {
+        property ??= await toPropertyKey(member.property, context.budget, createCoercionContext(context));
+        setSandboxProperty(member.object, property, value, context.budget);
         return {
             kind: "normal",
             hasValue: true,
-            value: current
-        };
-    }
-    if (node.operator === "??=" && current !== null && current !== undefined) {
-        return {
-            kind: "normal",
-            hasValue: true,
-            value: current
+            value
         };
-    }
-    const right = await evaluateNode(node.right, context);
-    if (right.kind !== "normal") {
-        return right;
-    }
-    const value = node.operator === "=" ||
-        node.operator === "&&=" ||
-        node.operator === "||=" ||
-        node.operator === "??="
-        ? right.value
-        : await applyCompoundAssignmentOperator(node.operator, current, right.value, context);
-    if (member.object === null || member.object === undefined) {
-        throw new TypeError("Cannot assign properties of null or undefined.");
-    }
-    property ??= await toPropertyKey(member.property, context.budget, createCoercionContext(context));
-    setSandboxProperty(member.object, property, value, context.budget);
-    return {
-        kind: "normal",
-        hasValue: true,
-        value
-    };
+    });
 }
 async function evaluateLogicalExpression(node, context) {
     const left = await evaluateNode(node.left, context);
@@ -1572,33 +1568,28 @@ async function evaluateDeleteExpression(node, context) {
     if (node.argument.type !== "MemberExpression") {
         throw createError("UNSUPPORTED_NODE", node, "Unary operator 'delete' requires a member target.");
     }
-    const member = await evaluateMemberAccess(node.argument, context);
-    if (member.kind === "error") {
-        return member;
-    }
-    if (member.kind === "completion") {
-        return member.result;
-    }
-    if (member.kind === "nullish" || member.object === null || member.object === undefined) {
-        if (member.kind === "nullish" && node.argument.optional) {
-            return {
-                kind: "normal",
-                hasValue: true,
-                value: true
-            };
+    return evaluateMemberAccess(node.argument, context, async (member) => {
+        if (member.kind === "nullish" || member.object === null || member.object === undefined) {
+            if (member.kind === "nullish") {
+                return {
+                    kind: "normal",
+                    hasValue: true,
+                    value: true
+                };
+            }
+            throw new TypeError("Cannot delete properties of null or undefined.");
         }
-        throw new TypeError("Cannot delete properties of null or undefined.");
-    }
-    if (!isIndexableSandboxValue(member.object)) {
-        throw new TypeError("Unary operator 'delete' requires a sandbox object property.");
-    }
-    const property = await toPropertyKey(member.property, context.budget, createCoercionContext(context));
-    const deleted = deleteSandboxProperty(member.object, property);
-    return {
-        kind: "normal",
-        hasValue: true,
-        value: deleted
-    };
+        if (!isIndexableSandboxValue(member.object)) {
+            throw new TypeError("Unary operator 'delete' requires a sandbox object property.");
+        }
+        const property = await toPropertyKey(member.property, context.budget, createCoercionContext(context));
+        const deleted = deleteSandboxProperty(member.object, property);
+        return {
+            kind: "normal",
+            hasValue: true,
+            value: deleted
+        };
+    });
 }
 async function evaluateUpdateExpression(node, context) {
     if (node.argument.type === "Identifier") {
@@ -1630,42 +1621,34 @@ async function evaluateMemberUpdateExpression(node, context) {
     if (node.argument.type !== "MemberExpression") {
         throw new TypeError("Expected member update target.");
     }
-    const member = await evaluateMemberAccess(node.argument, context);
-    if (member.kind === "error") {
-        return member;
-    }
-    if (member.kind === "completion") {
-        return member.result;
-    }
-    if (member.kind === "nullish" || member.object === null || member.object === undefined) {
-        throw new TypeError("Cannot update properties of null or undefined.");
-    }
-    const property = await toPropertyKey(member.property, context.budget, createCoercionContext(context));
-    const current = toNumber(await toNumericPrimitive(getPropertyValue(member.object, property, context), context));
-    const next = node.operator === "++" ? current + 1 : current - 1;
-    setSandboxProperty(member.object, property, next, context.budget);
-    return {
-        kind: "normal",
-        hasValue: true,
-        value: node.prefix ? next : current
-    };
+    return evaluateMemberAccess(node.argument, context, async (member) => {
+        if (member.kind === "nullish" || member.object === null || member.object === undefined) {
+            throw new TypeError("Cannot update properties of null or undefined.");
+        }
+        const property = await toPropertyKey(member.property, context.budget, createCoercionContext(context));
+        const current = toNumber(await toNumericPrimitive(getPropertyValue(member.object, property, context), context));
+        const next = node.operator === "++" ? current + 1 : current - 1;
+        setSandboxProperty(member.object, property, next, context.budget);
+        return {
+            kind: "normal",
+            hasValue: true,
+            value: node.prefix ? next : current
+        };
+    });
 }
 async function evaluateMemberExpression(node, context) {
-    const member = await evaluateMemberAccess(node, context);
-    if (member.kind === "error")
-        return member;
-    if (member.kind === "completion")
-        return member.result;
-    if (member.kind === "nullish" || member.object === null || member.object === undefined) {
-        if (member.kind === "nullish" && node.optional)
-            return { kind: "normal", hasValue: true, value: undefined };
-        throw new TypeError("Cannot read properties of null or undefined.");
-    }
-    return {
-        kind: "normal",
-        hasValue: true,
-        value: getPropertyValue(member.object, await toPropertyKey(member.property, context.budget, createCoercionContext(context)), context)
-    };
+    return evaluateMemberAccess(node, context, async (member) => {
+        if (member.kind === "nullish" || member.object === null || member.object === undefined) {
+            if (member.kind === "nullish")
+                return { kind: "normal", hasValue: true, value: undefined };
+            throw new TypeError("Cannot read properties of null or undefined.");
+        }
+        return {
+            kind: "normal",
+            hasValue: true,
+            value: getPropertyValue(member.object, await toPropertyKey(member.property, context.budget, createCoercionContext(context)), context)
+        };
+    });
 }
 function getPropertyValue(target, property, context) {
     if (isGuestHostObject(target))
@@ -1778,29 +1761,32 @@ function attachFatalSandboxErrorContext(error, node, stackFrames) {
     attachErrorSpan(error, node.span);
     replaceErrorStack(error, stackFrames);
 }
-async function evaluateMemberAccess(node, context) {
+async function evaluateMemberAccess(node, context, consume) {
     const object = await evaluateNode(node.object, context);
-    if (object.kind !== "normal") {
-        return {
-            kind: "completion",
-            result: object
-        };
-    }
+    if (object.kind !== "normal")
+        return object;
     if ((object.value === null || object.value === undefined) && node.optional) {
-        return {
-            kind: "nullish"
-        };
+        return consume({ kind: "nullish" });
+    }
+    const retained = {};
+    let property;
+    context.budget.setRetainedValues(retained, () => [object.value, property]);
+    const release = () => context.budget.setRetainedValues(retained, undefined);
+    const pending = runResources.getStore()?.referenceReleases;
+    pending?.add(release);
+    try {
+        const result = node.computed
+            ? await evaluateNode(node.property, context)
+            : { kind: "normal", value: getStaticPropertyName(node.property) };
+        if (result.kind !== "normal")
+            return result;
+        property = result.value;
+        return await consume({ kind: "resolved", object: object.value, property });
+    }
+    finally {
+        release();
+        pending?.delete(release);
     }
-    const property = node.computed
-        ? await evaluateNode(node.property, context)
-        : { kind: "normal", value: getStaticPropertyName(node.property) };
-    if (property.kind !== "normal")
-        return { kind: "completion", result: property };
-    return {
-        kind: "resolved",
-        object: object.value,
-        property: property.value
-    };
 }
 async function evaluateMemberProperty(node, context) {
     const property = await evaluateNode(node, context);
@@ -1837,89 +1823,84 @@ async function evaluateMemberCallExpression(node, context) {
     if (node.callee.type !== "MemberExpression") {
         throw new TypeError("Expected member call expression.");
     }
-    const reference = await evaluateMemberAccess(node.callee, context);
-    if (reference.kind === "error") {
-        return reference;
-    }
-    if (reference.kind === "completion") {
-        return reference.result;
-    }
-    if (reference.kind === "nullish" || reference.object === null || reference.object === undefined) {
-        if (reference.kind === "nullish") {
-            return {
-                kind: "normal",
-                hasValue: true,
-                value: undefined
-            };
+    return evaluateMemberAccess(node.callee, context, async (reference) => {
+        if (reference.kind === "nullish" || reference.object === null || reference.object === undefined) {
+            if (reference.kind === "nullish") {
+                return {
+                    kind: "normal",
+                    hasValue: true,
+                    value: undefined
+                };
+            }
+            throw new TypeError("Cannot read properties of null or undefined.");
         }
-        throw new TypeError("Cannot read properties of null or undefined.");
-    }
-    const member = {
-        ...reference,
-        property: await toPropertyKey(reference.property, context.budget, createCoercionContext(context))
-    };
-    if (typeof member.object === "string" && isStringMethodName(member.property)) {
-        return evaluateStringMethodCall(node, member.object, member.property, context);
-    }
-    if (typeof member.object === "number" && isNumberMethodName(member.property)) {
-        return evaluateNumberMethodCall(node, member.object, member.property, context);
-    }
-    if (Array.isArray(member.object) &&
-        isArrayMethodName(member.property) &&
-        !Object.hasOwn(member.object, member.property)) {
-        return evaluateArrayMethodCall(node, member.object, member.property, context);
-    }
-    if (isSandboxMap(member.object) && isMapMethodName(member.property)) {
-        return evaluateMapMethodCall(node, member.object, member.property, context);
-    }
-    if (isSandboxSet(member.object) && isSetMethodName(member.property)) {
-        return evaluateSetMethodCall(node, member.object, member.property, context);
-    }
-    if (typeof member.object === "string") {
-        return evaluatePrimitiveMemberCall(node, "String", member.property, getStringMember(member.object, member.property, context.budget), context);
-    }
-    if (typeof member.object === "number") {
-        return evaluatePrimitiveMemberCall(node, "Number", member.property, getNumberMember(member.property, context.budget), context);
-    }
-    if (Array.isArray(member.object) && !Object.hasOwn(member.object, member.property)) {
-        return evaluatePrimitiveMemberCall(node, "Array", member.property, getArrayMemberValue(member.object, member.property, context), context);
-    }
-    if (isSandboxMap(member.object)) {
-        return evaluatePrimitiveMemberCall(node, "Map", member.property, getMapMember(member.object, member.property, createMapMethodOptions(context)), context);
-    }
-    if (isSandboxDate(member.object)) {
-        return evaluateResolvedCallExpression(node, getDateMember(member.property, context.budget, context.compilation?.owner), context, member.object);
-    }
-    if (isFloat32Array(member.object)) {
-        return evaluateResolvedCallExpression(node, getFloat32Member(member.object, member.property, context.budget), context, member.object);
-    }
-    if (isSandboxSet(member.object)) {
-        return evaluatePrimitiveMemberCall(node, "Set", member.property, getSetMember(member.object, member.property, createSetMethodOptions(context)), context);
-    }
-    if (isSandboxGenerator(member.object)) {
-        const memberValue = getGeneratorMember(member.object, member.property, context.budget);
-        if (memberValue === undefined) {
-            throw new TypeError(`Generator#${String(member.property)} is not a supported method.`);
+        const member = {
+            ...reference,
+            property: await toPropertyKey(reference.property, context.budget, createCoercionContext(context))
+        };
+        if (typeof member.object === "string" && isStringMethodName(member.property)) {
+            return evaluateStringMethodCall(node, member.object, member.property, context);
         }
-        return evaluateResolvedCallExpression(node, memberValue, context, member.object);
-    }
-    if (isSandboxClosure(member.object)) {
-        const memberValue = getClosureMemberValue(member.object, member.property, context);
-        if (memberValue === undefined) {
-            throw new TypeError(`Function#${String(member.property)} is not a supported method.`);
+        if (typeof member.object === "number" && isNumberMethodName(member.property)) {
+            return evaluateNumberMethodCall(node, member.object, member.property, context);
         }
-        return evaluateResolvedCallExpression(node, memberValue, context, member.object);
-    }
-    if (isSandboxPromise(member.object)) {
-        return evaluateResolvedCallExpression(node, getPromiseMember(member.property, context.budget), context, member.object);
-    }
-    if (isSandboxRegex(member.object) && isRegexMethodName(member.property)) {
-        return evaluateResolvedCallExpression(node, getRegexMember(member.object, member.property, context.budget), context, member.object);
-    }
-    if (!isIndexableSandboxValue(member.object)) {
-        throw new TypeError("Attempted to read a property from a non-object value.");
-    }
-    return evaluateResolvedCallExpression(node, getMemberValue(member.object, member.property, context), context, member.object);
+        if (Array.isArray(member.object) &&
+            isArrayMethodName(member.property) &&
+            !Object.hasOwn(member.object, member.property)) {
+            return evaluateArrayMethodCall(node, member.object, member.property, context);
+        }
+        if (isSandboxMap(member.object) && isMapMethodName(member.property)) {
+            return evaluateMapMethodCall(node, member.object, member.property, context);
+        }
+        if (isSandboxSet(member.object) && isSetMethodName(member.property)) {
+            return evaluateSetMethodCall(node, member.object, member.property, context);
+        }
+        if (typeof member.object === "string") {
+            return evaluatePrimitiveMemberCall(node, "String", member.property, getStringMember(member.object, member.property, context.budget), context);
+        }
+        if (typeof member.object === "number") {
+            return evaluatePrimitiveMemberCall(node, "Number", member.property, getNumberMember(member.property, context.budget), context);
+        }
+        if (Array.isArray(member.object) && !Object.hasOwn(member.object, member.property)) {
+            return evaluatePrimitiveMemberCall(node, "Array", member.property, getArrayMemberValue(member.object, member.property, context), context);
+        }
+        if (isSandboxMap(member.object)) {
+            return evaluatePrimitiveMemberCall(node, "Map", member.property, getMapMember(member.object, member.property, createMapMethodOptions(context)), context);
+        }
+        if (isSandboxDate(member.object)) {
+            return evaluateResolvedCallExpression(node, getDateMember(member.property, context.budget, context.compilation?.owner), context, member.object);
+        }
+        if (isFloat32Array(member.object)) {
+            return evaluateResolvedCallExpression(node, getFloat32Member(member.object, member.property, context.budget), context, member.object);
+        }
+        if (isSandboxSet(member.object)) {
+            return evaluatePrimitiveMemberCall(node, "Set", member.property, getSetMember(member.object, member.property, createSetMethodOptions(context)), context);
+        }
+        if (isSandboxGenerator(member.object)) {
+            const memberValue = getGeneratorMember(member.object, member.property, context.budget);
+            if (memberValue === undefined) {
+                throw new TypeError(`Generator#${String(member.property)} is not a supported method.`);
+            }
+            return evaluateResolvedCallExpression(node, memberValue, context, member.object);
+        }
+        if (isSandboxClosure(member.object)) {
+            const memberValue = getClosureMemberValue(member.object, member.property, context);
+            if (memberValue === undefined) {
+                throw new TypeError(`Function#${String(member.property)} is not a supported method.`);
+            }
+            return evaluateResolvedCallExpression(node, memberValue, context, member.object);
+        }
+        if (isSandboxPromise(member.object)) {
+            return evaluateResolvedCallExpression(node, getPromiseMember(member.property, context.budget), context, member.object);
+        }
+        if (isSandboxRegex(member.object) && isRegexMethodName(member.property)) {
+            return evaluateResolvedCallExpression(node, getRegexMember(member.object, member.property, context.budget), context, member.object);
+        }
+        if (!isIndexableSandboxValue(member.object)) {
+            throw new TypeError("Attempted to read a property from a non-object value.");
+        }
+        return evaluateResolvedCallExpression(node, getMemberValue(member.object, member.property, context), context, member.object);
+    });
 }
 function evaluatePrimitiveMemberCall(node, receiverType, property, value, context) {
     if (value === undefined) {

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 () => {