poe-code 11.0.22 → 11.0.24

package/packages/safejs/dist/chunks/{chunk-K3DCXHEE.js → chunk-I5FH3SKL.js}

@@ -27,7 +27,7 @@ import {
   validateMigrationSemantics,
   validateSnapshotData,
   validateSnapshotMigration
-} from "./chunk-IB5DSFBQ.js";
+} from "./chunk-CGS4H4L3.js";
 
 // packages/safejs/src/migrate.ts
 import { createHash } from "node:crypto";
@@ -8055,4 +8055,4 @@ export {
   parseMcpConfig,
   makeMcpModule
 };
-//# sourceMappingURL=chunk-K3DCXHEE.js.map
+//# sourceMappingURL=chunk-I5FH3SKL.js.map

package/packages/safejs/dist/cli.js

@@ -20,7 +20,7 @@ import {
   parseMcpConfig,
   splitFrontmatter,
   withBrokenPipeGuard
-} from "./chunks/chunk-K3DCXHEE.js";
+} from "./chunks/chunk-I5FH3SKL.js";
 import {
   Budget,
   SandboxError,
@@ -32,7 +32,7 @@ import {
   replaceErrorStack,
   restore,
   run
-} from "./chunks/chunk-IB5DSFBQ.js";
+} from "./chunks/chunk-CGS4H4L3.js";
 import "./chunks/chunk-4YIDX44G.js";
 
 // packages/safejs/src/cli.ts

package/packages/safejs/dist/core.js

@@ -3,7 +3,7 @@ import {
   createReplayableRandom,
   lint,
   run
-} from "./chunks/chunk-IB5DSFBQ.js";
+} from "./chunks/chunk-CGS4H4L3.js";
 import "./chunks/chunk-4YIDX44G.js";
 export {
   Budget,

package/packages/safejs/dist/index.js

@@ -27,7 +27,7 @@ import {
   runWithSpawnUsageAccumulator,
   splitFrontmatter,
   supportsSpawnMode
-} from "./chunks/chunk-K3DCXHEE.js";
+} from "./chunks/chunk-I5FH3SKL.js";
 import {
   Budget,
   FileSnapshotBackend,
@@ -49,7 +49,7 @@ import {
   registerPendingHostCallPolicy,
   restore,
   run
-} from "./chunks/chunk-IB5DSFBQ.js";
+} from "./chunks/chunk-CGS4H4L3.js";
 import "./chunks/chunk-4YIDX44G.js";
 
 // packages/safejs/src/runner/run-harness.ts

package/packages/safejs/dist/interp/globals/object-array.js

@@ -227,7 +227,7 @@ function isAssignableSandboxTarget(value) {
         !isSandboxRegex(value));
 }
 async function arrayFromSandboxValues(args, budget) {
-    const [items, mapFn] = args;
+    const [items, mapFn, thisValue] = args;
     const iterator = getSandboxIterator(items);
     const values = iterator === undefined
         ? Reflect.apply(Array.from, Array, [items])
@@ -240,7 +240,7 @@ async function arrayFromSandboxValues(args, budget) {
     }
     const mappedValues = [];
     for (const [index, value] of values.entries()) {
-        const result = await mapFn.call([value, index]);
+        const result = await mapFn.call([value, index], { stack: [], thisValue });
         if (isSandboxPromise(result) && result.synchronousPrefix !== undefined) {
             await result.synchronousPrefix;
         }

package/packages/safejs/dist/interp/host-bridge.js

@@ -150,7 +150,7 @@ function invokeHostCallback(invoke, options) {
     }
     lifecycle.hostCallbackDepth += 1;
     try {
-        const result = invoke();
+        const result = lifecycle.hostCallbackContext.run(true, invoke);
         if (isPromiseLike(result)) {
             return Promise.resolve(result).finally(() => {
                 lifecycle.hostCallbackDepth -= 1;

package/packages/safejs/dist/interp/interpreter.js

@@ -2286,19 +2286,19 @@ async function evaluateNumberMethodCall(node, target, methodName, context) {
 function createArrayMethodOptions(context) {
     return {
         budget: context.budget,
-        callClosure: (closure, args, stack) => invokeSandboxClosure(closure, args, context, stack)
+        callClosure: (closure, args, stack, thisValue) => invokeSandboxClosure(closure, args, context, stack, undefined, thisValue)
     };
 }
 function createMapMethodOptions(context) {
     return {
         budget: context.budget,
-        callClosure: (closure, args, stack) => invokeSandboxClosure(closure, args, context, stack)
+        callClosure: (closure, args, stack, thisValue) => invokeSandboxClosure(closure, args, context, stack, undefined, thisValue)
     };
 }
 function createSetMethodOptions(context) {
     return {
         budget: context.budget,
-        callClosure: (closure, args, stack) => invokeSandboxClosure(closure, args, context, stack)
+        callClosure: (closure, args, stack, thisValue) => invokeSandboxClosure(closure, args, context, stack, undefined, thisValue)
     };
 }
 function createFunctionMethodOptions(context) {

package/packages/safejs/dist/interp/methods/array.d.ts

@@ -3,7 +3,7 @@ import { type SandboxArray, type SandboxClosure, type SandboxValue } from "../va
 export type ArrayMethodName = "map" | "filter" | "find" | "findIndex" | "findLast" | "findLastIndex" | "some" | "every" | "reduce" | "reduceRight" | "forEach" | "flatMap" | "flat" | "includes" | "indexOf" | "lastIndexOf" | "join" | "slice" | "concat" | "splice" | "fill" | "copyWithin" | "at" | "sort" | "reverse" | "toSorted" | "toReversed" | "toSpliced" | "with" | "push" | "pop" | "shift" | "unshift";
 export type ArrayMethodOptions = {
     budget: Budget;
-    callClosure: (closure: SandboxClosure, args: readonly SandboxValue[], stack: readonly string[]) => Promise<SandboxValue>;
+    callClosure: (closure: SandboxClosure, args: readonly SandboxValue[], stack: readonly string[], thisValue?: SandboxValue) => Promise<SandboxValue>;
 };
 export declare function getArrayMember(value: SandboxArray, property: string | number, options: ArrayMethodOptions): SandboxValue | undefined;
 export declare function isArrayMethodName(property: string | number): property is ArrayMethodName;

package/packages/safejs/dist/interp/methods/array.js

@@ -96,30 +96,30 @@ export async function callArrayMethod(value, methodName, args, options, stack =
 async function callArrayMethodUnlocked(value, methodName, args, options, stack) {
     switch (methodName) {
         case "map":
-            return budgetProducedValue(await mapArray(value, getRequiredCallback(methodName, args[0]), options, stack), options.budget);
+            return budgetProducedValue(await mapArray(value, getRequiredCallback(methodName, args[0]), options, stack, args[1]), options.budget);
         case "filter":
-            return budgetProducedValue(await filterArray(value, getRequiredCallback(methodName, args[0]), options, stack), options.budget);
+            return budgetProducedValue(await filterArray(value, getRequiredCallback(methodName, args[0]), options, stack, args[1]), options.budget);
         case "find":
-            return budgetProducedValue(await findInArray(value, getRequiredCallback(methodName, args[0]), options, stack), options.budget);
+            return budgetProducedValue(await findInArray(value, getRequiredCallback(methodName, args[0]), options, stack, args[1]), options.budget);
         case "findIndex":
-            return await findIndexInArray(value, getRequiredCallback(methodName, args[0]), options, stack);
+            return await findIndexInArray(value, getRequiredCallback(methodName, args[0]), options, stack, args[1]);
         case "findLast":
-            return budgetProducedValue(await findLastInArray(value, getRequiredCallback(methodName, args[0]), options, stack), options.budget);
+            return budgetProducedValue(await findLastInArray(value, getRequiredCallback(methodName, args[0]), options, stack, args[1]), options.budget);
         case "findLastIndex":
-            return await findLastIndexInArray(value, getRequiredCallback(methodName, args[0]), options, stack);
+            return await findLastIndexInArray(value, getRequiredCallback(methodName, args[0]), options, stack, args[1]);
         case "some":
-            return await someInArray(value, getRequiredCallback(methodName, args[0]), options, stack);
+            return await someInArray(value, getRequiredCallback(methodName, args[0]), options, stack, args[1]);
         case "every":
-            return await everyInArray(value, getRequiredCallback(methodName, args[0]), options, stack);
+            return await everyInArray(value, getRequiredCallback(methodName, args[0]), options, stack, args[1]);
         case "reduce":
             return budgetProducedValue(await reduceArray(value, getRequiredCallback(methodName, args[0]), args.length > 1, args[1], options, stack), options.budget);
         case "reduceRight":
             return budgetProducedValue(await reduceRightArray(value, getRequiredCallback(methodName, args[0]), args.length > 1, args[1], options, stack), options.budget);
         case "forEach":
-            await forEachArray(value, getRequiredCallback(methodName, args[0]), options, stack);
+            await forEachArray(value, getRequiredCallback(methodName, args[0]), options, stack, args[1]);
             return undefined;
         case "flatMap":
-            return budgetProducedValue(await flatMapArray(value, getRequiredCallback(methodName, args[0]), options, stack), options.budget);
+            return budgetProducedValue(await flatMapArray(value, getRequiredCallback(methodName, args[0]), options, stack, args[1]), options.budget);
         case "flat":
             return budgetProducedValue(flattenArray(value, toIntegerOrInfinity(args[0] ?? 1), options.budget), options.budget);
         case "includes":
@@ -264,83 +264,83 @@ function getRequiredCallback(methodName, value) {
     }
     return value;
 }
-async function mapArray(value, callback, options, stack) {
+async function mapArray(value, callback, options, stack, thisValue) {
     const length = value.length;
     const result = new Array(length);
     for (let index = 0; index < length; index += 1) {
         if (!(index in value)) {
             continue;
         }
-        result[index] = await callArrayCallback(callback, value[index], index, value, options, stack);
+        result[index] = await callArrayCallback(callback, value[index], index, value, options, stack, thisValue);
     }
     return result;
 }
-async function filterArray(value, callback, options, stack) {
+async function filterArray(value, callback, options, stack, thisValue) {
     const length = value.length;
     const result = [];
     for (let index = 0; index < length; index += 1) {
         if (!(index in value)) {
             continue;
         }
-        if (await callArrayCallback(callback, value[index], index, value, options, stack)) {
+        if (await callArrayCallback(callback, value[index], index, value, options, stack, thisValue)) {
             result.push(value[index]);
         }
     }
     return result;
 }
-async function findInArray(value, callback, options, stack) {
-    const index = await findIndexInArray(value, callback, options, stack);
+async function findInArray(value, callback, options, stack, thisValue) {
+    const index = await findIndexInArray(value, callback, options, stack, thisValue);
     return index < 0 ? undefined : value[index];
 }
-async function findIndexInArray(value, callback, options, stack) {
+async function findIndexInArray(value, callback, options, stack, thisValue) {
     const length = value.length;
     for (let index = 0; index < length; index += 1) {
         const entry = index in value ? value[index] : undefined;
-        if (await callArrayCallback(callback, entry, index, value, options, stack)) {
+        if (await callArrayCallback(callback, entry, index, value, options, stack, thisValue)) {
             return index;
         }
     }
     return -1;
 }
-async function findLastInArray(value, callback, options, stack) {
+async function findLastInArray(value, callback, options, stack, thisValue) {
     const length = value.length;
     for (let index = length - 1; index >= 0; index -= 1) {
         const entry = index in value ? value[index] : undefined;
-        if (await callArrayCallback(callback, entry, index, value, options, stack)) {
+        if (await callArrayCallback(callback, entry, index, value, options, stack, thisValue)) {
             return entry;
         }
     }
     return undefined;
 }
-async function findLastIndexInArray(value, callback, options, stack) {
+async function findLastIndexInArray(value, callback, options, stack, thisValue) {
     const length = value.length;
     for (let index = length - 1; index >= 0; index -= 1) {
         const entry = index in value ? value[index] : undefined;
-        if (await callArrayCallback(callback, entry, index, value, options, stack)) {
+        if (await callArrayCallback(callback, entry, index, value, options, stack, thisValue)) {
             return index;
         }
     }
     return -1;
 }
-async function someInArray(value, callback, options, stack) {
+async function someInArray(value, callback, options, stack, thisValue) {
     const length = value.length;
     for (let index = 0; index < length; index += 1) {
         if (!(index in value)) {
             continue;
         }
-        if (await callArrayCallback(callback, value[index], index, value, options, stack)) {
+        if (await callArrayCallback(callback, value[index], index, value, options, stack, thisValue)) {
             return true;
         }
     }
     return false;
 }
-async function everyInArray(value, callback, options, stack) {
+async function everyInArray(value, callback, options, stack, thisValue) {
     const length = value.length;
     for (let index = 0; index < length; index += 1) {
         if (!(index in value)) {
             continue;
         }
-        if (!(await callArrayCallback(callback, value[index], index, value, options, stack))) {
+        if (!(await callArrayCallback(callback, value[index], index, value, options, stack, thisValue))) {
             return false;
         }
     }
@@ -388,23 +388,23 @@ async function reduceFromRight(value, callback, accumulator, startIndex, length,
     }
     return current;
 }
-async function forEachArray(value, callback, options, stack) {
+async function forEachArray(value, callback, options, stack, thisValue) {
     const length = value.length;
     for (let index = 0; index < length; index += 1) {
         if (!(index in value)) {
             continue;
         }
-        await callArrayCallback(callback, value[index], index, value, options, stack);
+        await callArrayCallback(callback, value[index], index, value, options, stack, thisValue);
     }
 }
-async function flatMapArray(value, callback, options, stack) {
+async function flatMapArray(value, callback, options, stack, thisValue) {
     const length = value.length;
     const result = [];
     for (let index = 0; index < length; index += 1) {
         if (!(index in value)) {
             continue;
         }
-        const mapped = await callArrayCallback(callback, value[index], index, value, options, stack);
+        const mapped = await callArrayCallback(callback, value[index], index, value, options, stack, thisValue);
         if (Array.isArray(mapped)) {
             for (let mappedIndex = 0; mappedIndex < mapped.length; mappedIndex += 1) {
                 if (!(mappedIndex in mapped)) {
@@ -477,8 +477,8 @@ async function compareEntries(left, right, comparator, options, stack) {
     const result = Number(await options.callClosure(comparator, [left, right], stack));
     return Number.isNaN(result) ? 0 : result;
 }
-async function callArrayCallback(callback, value, index, array, options, stack) {
-    return options.callClosure(callback, [value, index, array], stack);
+async function callArrayCallback(callback, value, index, array, options, stack, thisValue) {
+    return options.callClosure(callback, [value, index, array], stack, thisValue);
 }
 function findNextDefinedIndex(value, startIndex, direction, length) {
     for (let index = startIndex; direction > 0 ? index < length : index >= 0; index += direction) {

package/packages/safejs/dist/interp/methods/map.d.ts

@@ -3,7 +3,7 @@ import { type SandboxClosure, type SandboxMap, type SandboxValue } from "../valu
 export type MapMethodName = "get" | "set" | "has" | "delete" | "clear" | "forEach" | "keys" | "values" | "entries";
 export type MapMethodOptions = {
     budget: Budget;
-    callClosure: (closure: SandboxClosure, args: readonly SandboxValue[], stack: readonly string[]) => Promise<SandboxValue>;
+    callClosure: (closure: SandboxClosure, args: readonly SandboxValue[], stack: readonly string[], thisValue?: SandboxValue) => Promise<SandboxValue>;
 };
 export declare function isMapMethodName(value: string | number): value is MapMethodName;
 export declare function getMapMember(target: SandboxMap, property: string | number, options: MapMethodOptions): SandboxValue | undefined;

package/packages/safejs/dist/interp/methods/map.js

@@ -57,7 +57,7 @@ export async function callMapMethod(target, methodName, args, options, stack = [
                 const entries = [...target.entries];
                 for (let index = 0; index < entries.length; index += 1) {
                     const [key, value] = entries[index];
-                    await options.callClosure(callback, [value, key, target], stack);
+                    await options.callClosure(callback, [value, key, target], stack, args[1]);
                 }
             }
             finally {

package/packages/safejs/dist/interp/methods/set.d.ts

@@ -3,7 +3,7 @@ import { type SandboxClosure, type SandboxSet, type SandboxValue } from "../valu
 export type SetMethodName = "add" | "has" | "delete" | "clear" | "forEach" | "keys" | "values" | "entries";
 export type SetMethodOptions = {
     budget: Budget;
-    callClosure: (closure: SandboxClosure, args: readonly SandboxValue[], stack: readonly string[]) => Promise<SandboxValue>;
+    callClosure: (closure: SandboxClosure, args: readonly SandboxValue[], stack: readonly string[], thisValue?: SandboxValue) => Promise<SandboxValue>;
 };
 export declare function isSetMethodName(value: string | number): value is SetMethodName;
 export declare function getSetMember(target: SandboxSet, property: string | number, options: SetMethodOptions): SandboxValue | undefined;

package/packages/safejs/dist/interp/methods/set.js

@@ -54,7 +54,7 @@ export async function callSetMethod(target, methodName, args, options, stack = [
                 const values = [...target.values];
                 for (let index = 0; index < values.length; index += 1) {
                     const value = values[index];
-                    await options.callClosure(callback, [value, value, target], stack);
+                    await options.callClosure(callback, [value, value, target], stack, args[1]);
                 }
             }
             finally {

package/packages/safejs/dist/run.js

@@ -1,3 +1,4 @@
+import { AsyncLocalStorage } from "node:async_hooks";
 import { hashSource } from "./parse/hash.js";
 import { withRunResources } from "./interp/resources.js";
 import { createReplayableRandom } from "./random.js";
@@ -41,7 +42,10 @@ export class UnhandledRejectionError extends Error {
 }
 const DEFAULT_MAX_CALL_DEPTH = 1_000;
 export function run(source, options = {}) {
-    const lifecycle = { hostCallbackDepth: 0 };
+    const lifecycle = {
+        hostCallbackDepth: 0,
+        hostCallbackContext: new AsyncLocalStorage()
+    };
     const dumpController = createDumpController(lifecycle);
     const promiseTracker = createSandboxPromiseRejectionTracker();
     let completedSnapshot;
@@ -335,6 +339,7 @@ export function run(source, options = {}) {
                 throw error;
             }
             finally {
+                lifecycle.hostCallbackContext.disable();
                 options.signal?.removeEventListener("abort", captureCancellationSnapshot);
                 leaveInputReplay?.();
                 leaveHostReplay?.();

package/packages/safejs/dist/runner/signal-dump.js

@@ -2,9 +2,8 @@ import { randomUUID } from "node:crypto";
 import { mkdir, rename, rm, writeFile as nodeWriteFile } from "node:fs/promises";
 import { basename, dirname, join } from "node:path";
 import { hasOwnErrorCode } from "../error-codes.js";
-import { dumpCurrent } from "../snapshot/dump.js";
+import { dump } from "../snapshot/dump.js";
 export function attachSignalDumpHandler(result, options = {}) {
-    const dumpResult = options.dumpResult ?? dumpCurrent;
     const signalProcess = options.process ?? process;
     const stderr = options.stderr ?? process.stderr;
     const writeDumpFile = options.writeFile ?? nodeWriteFile;
@@ -19,7 +18,9 @@ export function attachSignalDumpHandler(result, options = {}) {
     }
     async function writeDump(signal) {
         try {
-            const snapshot = await dumpResult(result);
+            const snapshot = options.dumpResult === undefined
+                ? await dump(result, { mode: "replay" })
+                : await options.dumpResult(result);
             if (options.dumpPath !== undefined) {
                 const parentPath = dirname(options.dumpPath);
                 const tempPath = join(parentPath, `.${basename(options.dumpPath)}.${randomUUID()}.tmp`);

package/packages/safejs/dist/snapshot/dump.d.ts

@@ -1,15 +1,18 @@
+import type { AsyncLocalStorage } from "node:async_hooks";
 import type { RunResult, RunSnapshot } from "../run.js";
 export type DumpOptions = {
+    mode?: "capture" | "replay";
     onFailure?: "throw" | "checkpoint";
 };
 type DumpController = {
     fail(error: unknown): void;
     finalize(snapshot: RunSnapshot): void;
     onYield(createSnapshot: () => RunSnapshot): void;
-    requestCurrentSnapshot(): Promise<string>;
+    requestCurrentSnapshot(options?: DumpOptions): Promise<string>;
     requestSnapshot(options?: DumpOptions): Promise<string>;
 };
 export type RunLifecycle = {
+    hostCallbackContext: AsyncLocalStorage<boolean>;
     hostCallbackDepth: number;
 };
 export declare function attachDumpController(result: Promise<RunResult>, controller: DumpController): Promise<RunResult>;

package/packages/safejs/dist/snapshot/dump.js

@@ -46,8 +46,8 @@ export function createDumpController(lifecycle) {
             }
             settlePendingSnapshot(createSnapshot());
         },
-        requestCurrentSnapshot() {
-            assertDumpAllowed();
+        requestCurrentSnapshot(options = {}) {
+            assertDumpAllowed(options);
             if (failed !== undefined) {
                 return Promise.reject(failed.error);
             }
@@ -59,10 +59,10 @@ export function createDumpController(lifecycle) {
                     return Promise.reject(error);
                 }
             }
-            return this.requestSnapshot();
+            return this.requestSnapshot(options);
         },
         requestSnapshot(options = {}) {
-            assertDumpAllowed();
+            assertDumpAllowed(options);
             if (failed !== undefined) {
                 if ((options.onFailure === "checkpoint" ||
                     (options.onFailure === undefined && isDataBudgetError(failed.error))) &&
@@ -88,6 +88,10 @@ export function createDumpController(lifecycle) {
                     return Promise.reject(error);
                 }
             }
+            if (options.mode === "replay" &&
+                (latestSnapshot !== undefined || latestSnapshotFactory !== undefined)) {
+                return this.requestCurrentSnapshot(options);
+            }
             if (pendingRequest !== undefined) {
                 return pendingRequest.promise;
             }
@@ -105,8 +109,9 @@ export function createDumpController(lifecycle) {
             return promise;
         }
     };
-    function assertDumpAllowed() {
-        if ((lifecycle?.hostCallbackDepth ?? 0) > 0) {
+    function assertDumpAllowed(options) {
+        if ((lifecycle?.hostCallbackDepth ?? 0) > 0 &&
+            (options.mode !== "replay" || lifecycle?.hostCallbackContext.getStore() === true)) {
             throw new SandboxError("reentry");
         }
     }