poe-code 8.0.1 → 9.0.1

package/packages/safejs/dist/run.js

@@ -1,4 +1,5 @@
 import { hashSource } from "./parse/hash.js";
+import { withRunResources } from "./interp/resources.js";
 import { createReplayableRandom } from "./random.js";
 import { attachErrorSpan, describeThrownValue, materializeWrappedErrorCause, replaceErrorStack } from "./error/shape.js";
 import { parseExecutableModule } from "./parse/parser.js";
@@ -43,7 +44,7 @@ export function run(source, options = {}) {
     const lifecycle = { hostCallbackDepth: 0 };
     const dumpController = createDumpController(lifecycle);
     const promiseTracker = createSandboxPromiseRejectionTracker();
-    const result = withSandboxPromiseRejectionTracker(promiseTracker, async () => {
+    const execute = async () => {
         const promiseReplay = new PromiseReplay(options.snapshot?.promiseReplay);
         return promiseReplayContext.run(promiseReplay, async () => {
             const deactivateOtelSink = activateOtelSink(options.otelSink);
@@ -140,7 +141,8 @@ export function run(source, options = {}) {
                     if (!isSandboxClosure(binding))
                         throw new TypeError("Invalid initial promise operation.");
                     return binding.call([]);
-                });
+                }, hostCalls.rebindHostCapability.bind(hostCalls));
+                hostCalls.validateHostCapabilities();
                 budget.setRetainedDataUsage(initialInputs, measureSandboxData([
                     ...Object.values(initialInputs.values.bindings),
                     ...Object.values(initialInputs.values.imports),
@@ -310,7 +312,8 @@ export function run(source, options = {}) {
                 deactivateOtelSink();
             }
         });
-    });
+    };
+    const result = withRunResources(options.signal, () => withSandboxPromiseRejectionTracker(promiseTracker, execute));
     return attachDumpController(result, dumpController);
 }
 async function throwIfReturnedPromiseRejected(result) {

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

@@ -1,5 +1,5 @@
 export declare const DUMP_FORMAT_VERSION = 1;
-export declare const EXECUTION_SEMANTICS = "jobs-v4";
+export declare const EXECUTION_SEMANTICS = "jobs-v5";
 export type DumpableSnapshot = {
     sourceHash: string;
     [key: string]: unknown;

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

@@ -1,5 +1,5 @@
 export const DUMP_FORMAT_VERSION = 1;
-export const EXECUTION_SEMANTICS = "jobs-v4";
+export const EXECUTION_SEMANTICS = "jobs-v5";
 import { assertSnapshotGraphDepth } from "../graph-depth.js";
 import { sandboxErrorTypes } from "../error/shape.js";
 import { getSandboxArgumentEntries, isSandboxArguments } from "../interp/arguments.js";

package/packages/safejs/dist/snapshot/replay-data.d.ts

@@ -61,5 +61,6 @@ export declare function encodeReplayData(value: SandboxValue, options?: {
 export declare function decodeReplayData(input: unknown, options?: {
     resolveCapability?: (id: string) => SandboxClosure | undefined;
     resolvePromise?: (id: string) => SandboxPromise | undefined;
+    onCapabilityRestored?: (original: SandboxClosure, restored: SandboxClosure) => void;
 }): SandboxValue;
 export {};

package/packages/safejs/dist/snapshot/replay-data.js

@@ -185,7 +185,7 @@ export function decodeReplayData(input, options = {}) {
                 restored.set(id, capability);
                 return capability;
             }
-            return createSandboxClosure({
+            const copy = createSandboxClosure({
                 ...capability,
                 boundTarget: capability.boundTarget,
                 cancellationSignal: capability.cancellationSignal,
@@ -202,6 +202,8 @@ export function decodeReplayData(input, options = {}) {
                     return properties;
                 }
             });
+            options.onCapabilityRestored?.(capability, copy);
+            return copy;
         }
         if (kind === "map") {
             const result = createSandboxMap();

package/packages/safejs/dist/snapshot/replay-inputs.d.ts

@@ -1,4 +1,4 @@
-import { type SandboxPromise, type SandboxValue } from "../interp/values.js";
+import { type SandboxClosure, type SandboxPromise, type SandboxValue } from "../interp/values.js";
 import { type ReplayData } from "./replay-data.js";
 export type ReplayInputs = {
     bindings: Record<string, SandboxValue>;
@@ -6,7 +6,7 @@ export type ReplayInputs = {
     entryPointArgs: SandboxValue[] | undefined;
     importMeta: SandboxValue;
 };
-export declare function prepareReplayInputs(current: ReplayInputs, saved?: unknown, preparePromise?: (promise: SandboxPromise | undefined, id: string) => SandboxPromise): {
+export declare function prepareReplayInputs(current: ReplayInputs, saved?: unknown, preparePromise?: (promise: SandboxPromise | undefined, id: string) => SandboxPromise, onCapabilityRestored?: (original: SandboxClosure, restored: SandboxClosure) => void): {
     values: ReplayInputs;
     snapshot: ReplayData;
 };

package/packages/safejs/dist/snapshot/replay-inputs.js

@@ -1,7 +1,7 @@
 import { createSandboxPromise, isSandboxClosure, isSandboxMap, isSandboxSet, isSandboxPromise } from "../interp/values.js";
 import { decodeReplayData, encodeReplayData } from "./replay-data.js";
 const validationPromise = createSandboxPromise(Promise.resolve(undefined));
-export function prepareReplayInputs(current, saved, preparePromise) {
+export function prepareReplayInputs(current, saved, preparePromise, onCapabilityRestored) {
     const identities = new WeakMap();
     const capabilities = new Map();
     const promises = new Map();
@@ -81,6 +81,7 @@ export function prepareReplayInputs(current, saved, preparePromise) {
     assertReplayInputShape(validated);
     const restored = decodeReplayData(saved, {
         resolveCapability,
+        onCapabilityRestored,
         resolvePromise: (id) => {
             const value = readCapability(id);
             if (!promises.has(id) && preparePromise !== undefined)