poe-code 11.0.22 → 11.0.23

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

@@ -27,7 +27,7 @@ import {
   validateMigrationSemantics,
   validateSnapshotData,
   validateSnapshotMigration
-} from "./chunk-IB5DSFBQ.js";
+} from "./chunk-3627FMJF.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-LRGHMX22.js.map

package/packages/safejs/dist/cli.js

@@ -20,7 +20,7 @@ import {
   parseMcpConfig,
   splitFrontmatter,
   withBrokenPipeGuard
-} from "./chunks/chunk-K3DCXHEE.js";
+} from "./chunks/chunk-LRGHMX22.js";
 import {
   Budget,
   SandboxError,
@@ -32,7 +32,7 @@ import {
   replaceErrorStack,
   restore,
   run
-} from "./chunks/chunk-IB5DSFBQ.js";
+} from "./chunks/chunk-3627FMJF.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-3627FMJF.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-LRGHMX22.js";
 import {
   Budget,
   FileSnapshotBackend,
@@ -49,7 +49,7 @@ import {
   registerPendingHostCallPolicy,
   restore,
   run
-} from "./chunks/chunk-IB5DSFBQ.js";
+} from "./chunks/chunk-3627FMJF.js";
 import "./chunks/chunk-4YIDX44G.js";
 
 // packages/safejs/src/runner/run-harness.ts

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/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");
         }
     }