@poe-platform/safe-js 0.1.802 → 0.1.804

package/dist/safe-js/chunks/{chunk-HFRBLE6Y.js → chunk-36PBWSDR.js}

@@ -7,15 +7,15 @@ import {
 } from "./chunk-ONWFPHUI.js";
 import {
   restore
-} from "./chunk-XOMGWE37.js";
+} from "./chunk-CVPZFRCC.js";
 import {
   hashParsedAst,
   hashSource
-} from "./chunk-PMOXSLM5.js";
+} from "./chunk-MJH73KJI.js";
 import {
   attachDumpController,
   createDumpController
-} from "./chunk-XV5OYKND.js";
+} from "./chunk-VLRTLEDI.js";
 import {
   DUMP_FORMAT_VERSION,
   EXECUTION_SEMANTICS,
@@ -107,7 +107,7 @@ import {
   wrapCallerInjectedBindings,
   wrapCancelableBindings,
   wrapHostConstructor
-} from "./chunk-5RLD4TVI.js";
+} from "./chunk-BKNQTNHP.js";
 import {
   Budget,
   CompileScope,
@@ -122,16 +122,16 @@ import {
   parseSourceModule,
   replaceErrorStack,
   safeAddEvent
-} from "./chunk-A3LXCGMJ.js";
+} from "./chunk-EQUBWQJN.js";
 import {
   dirname
-} from "./chunk-MMN3IESA.js";
+} from "./chunk-KMBCLXER.js";
 import {
   AsyncLocalStorage,
   createHostCallbackContext,
   hostFs,
   types
-} from "./chunk-JCHJGIQA.js";
+} from "./chunk-VVOPJZRG.js";
 
 // packages/safe-js/src/modules/source-graph.ts
 var ambiguous = /* @__PURE__ */ Symbol("ambiguous module export");
@@ -2102,7 +2102,7 @@ var UnsnapshotableValueError = class extends Error {
 };
 
 // packages/safe-js/src/snapshot/replay-inputs.ts
-function prepareReplayInputs(current, saved, preparePromise, onCapabilityRestored, compilation, onInputSymbols) {
+function prepareReplayInputs(current, saved, preparePromise, onCapabilityRestored, compilation, onInputSymbols, restoredPromiseOrder = []) {
   const identities = /* @__PURE__ */ new WeakMap();
   const capabilities = /* @__PURE__ */ new Map();
   const promises = /* @__PURE__ */ new Map();
@@ -2197,6 +2197,8 @@ function prepareReplayInputs(current, saved, preparePromise, onCapabilityRestore
     return isSandboxClosure(value) ? value : void 0;
   };
   const memo = { nodes: context.nodes, values: /* @__PURE__ */ new Map() };
+  const restoredRootPromises = /* @__PURE__ */ new Set();
+  const restoredNamespacePromises = /* @__PURE__ */ new Map();
   if (saved !== void 0) validateSnapshotData(saved);
   const snapshot = saved === void 0 ? encodeReplayData(current, {
     context,
@@ -2214,6 +2216,7 @@ function prepareReplayInputs(current, saved, preparePromise, onCapabilityRestore
         resolveCapability,
         resolvePromise: (id) => {
           readCapability(id);
+          restoredRootPromises.add(id);
           return createSandboxPromise(Promise.resolve(void 0), { trackReplay: false });
         }
       }, validationScope);
@@ -2222,7 +2225,8 @@ function prepareReplayInputs(current, saved, preparePromise, onCapabilityRestore
         if (snapshot.namespaceRoots === null || typeof snapshot.namespaceRoots !== "object" || Array.isArray(snapshot.namespaceRoots))
           throw new TypeError("Invalid replay namespace roots.");
         const validationCapability = createSandboxClosure({ call: () => void 0 });
-        for (const root of Object.values(snapshot.namespaceRoots)) {
+        for (const [name, root] of Object.entries(snapshot.namespaceRoots)) {
+          const ids = /* @__PURE__ */ new Set();
           const namespace = decodeReplayData({ root, nodes: snapshot.nodes }, {
             resolveCapability: (id) => {
               readCapability(id);
@@ -2230,10 +2234,12 @@ function prepareReplayInputs(current, saved, preparePromise, onCapabilityRestore
             },
             resolvePromise: (id) => {
               readCapability(id);
+              ids.add(id);
               return createSandboxPromise(Promise.resolve(void 0), { trackReplay: false });
             }
           }, validationScope);
           if (!isSandboxModuleNamespace(namespace)) throw new TypeError("Invalid replay module namespace.");
+          restoredNamespacePromises.set(name, ids);
         }
       }
     } finally {
@@ -2256,6 +2262,7 @@ function prepareReplayInputs(current, saved, preparePromise, onCapabilityRestore
       promises.set(id, preparePromise(isSandboxPromise(value) ? value : void 0, id));
     return promises.get(id);
   };
+  for (const id of restoredPromiseOrder) if (restoredRootPromises.has(id)) resolvePromise(id);
   if (inputPromises.size > 0) {
     memo.values.clear();
     for (const [id, symbol] of inputSymbols) memo.values.set(id, symbol);
@@ -2286,6 +2293,10 @@ function prepareReplayInputs(current, saved, preparePromise, onCapabilityRestore
     captureNamespace,
     prepareNamespace: (namespace, name) => {
       captureNamespace(namespace, name);
+      const ids = restoredNamespacePromises.get(name);
+      if (ids !== void 0) {
+        for (const id of restoredPromiseOrder) if (ids.has(id)) resolvePromise(id);
+      }
       return decodeReplayData(
         { root: snapshot.namespaceRoots[name], nodes: snapshot.nodes },
         { memo, resolveCapability, resolvePromise, onCapabilityRestored },
@@ -2328,7 +2339,7 @@ function run(source, options = {}) {
         throw new TypeError("Source resolution requires sourceType: 'module'.");
       if (options.sourceRoot !== void 0) {
         if (options.sourceResolver !== void 0) throw new TypeError("Specify sourceRoot or sourceResolver, not both.");
-        const { createRootedSourceResolver } = await import("./source-files-BSDM22Y2.js");
+        const { createRootedSourceResolver } = await import("./source-files-A2WSOO6G.js");
         const { sourceRoot, ...resolved } = options;
         const sourceResolver = await createRootedSourceResolver(sourceRoot);
         return runWithExtensions(source, { ...resolved, filename: await sourceResolver.entryId(options.filename), sourceResolver }, jobs);
@@ -2492,7 +2503,8 @@ function run(source, options = {}) {
             prepareInputPromise,
             hostCalls.rebindHostCapability.bind(hostCalls),
             compilation,
-            hostCalls.registerInputSymbols.bind(hostCalls)
+            hostCalls.registerInputSymbols.bind(hostCalls),
+            hostCalls.inputPromiseOrder()
           );
           leaveInputReplay = () => {
             budget.setRetainedDataUsage(initialInputs, 0);

package/dist/safe-js/chunks/{chunk-CK3YK2EC.js → chunk-6AO2T2VZ.js}

@@ -1,6 +1,6 @@
 import {
   deepCopyToSandbox
-} from "./chunk-5RLD4TVI.js";
+} from "./chunk-BKNQTNHP.js";
 
 // packages/safe-js/src/modules/harness.ts
 function makeHarnessModule(frontmatter, meta) {

package/dist/safe-js/chunks/{chunk-5RLD4TVI.js → chunk-BKNQTNHP.js}

@@ -41,13 +41,13 @@ import {
   sandboxErrorTypes,
   syntaxDiagnostics,
   templateRealms
-} from "./chunk-A3LXCGMJ.js";
+} from "./chunk-EQUBWQJN.js";
 import {
   AsyncLocalStorage,
   cloneSharedBufferWrapper,
   setImmediate,
   types
-} from "./chunk-JCHJGIQA.js";
+} from "./chunk-VVOPJZRG.js";
 
 // packages/safe-js/src/interp/native-promise-properties.ts
 var admittedSymbols = /* @__PURE__ */ new WeakMap();
@@ -8173,7 +8173,9 @@ function createInterpretedClosure(node, context, evaluateNode2, homeObject, init
               }
             });
             const value = "hasValue" in result && result.hasValue ? result.value : void 0;
-            reconcileCompiledValues(context.budget, [...scope.retainedDataRoots(), value], compilation, parent, [value]);
+            if (context.budget.limits.dataSize !== void 0 || compilation.tickets.size > 0) {
+              reconcileCompiledValues(context.budget, [...scope.retainedDataRoots(), value], compilation, parent, [value]);
+            }
             if (result.kind === "error") throw result.error;
             if (result.kind === "throw") throw result.value;
             return value;
@@ -8422,13 +8424,15 @@ async function executeClosure(node, args, thisValue, context, evaluateNode2) {
       scope
     });
     const value = "hasValue" in result && result.hasValue ? result.value : void 0;
-    reconcileCompiledValues(
-      context.budget,
-      [...scope.retainedDataRoots(), value],
-      compilation,
-      parent,
-      [value]
-    );
+    if (context.budget.limits.dataSize !== void 0 || compilation.tickets.size > 0) {
+      reconcileCompiledValues(
+        context.budget,
+        [...scope.retainedDataRoots(), value],
+        compilation,
+        parent,
+        [value]
+      );
+    }
     if (result.kind === "error") {
       throw result.error;
     }
@@ -25665,7 +25669,6 @@ async function evaluateNode(node, context, onReference) {
   const evaluationContext = compilation === void 0 ? context : {
     ...context,
     compilation,
-    deferDataScans: context.deferDataScans || node.type === "ForStatement" || node.type === "ForInStatement" || node.type === "ForOfStatement" || node.type === "WhileStatement" || node.type === "DoWhileStatement",
     get generatorResume() {
       return context.generatorResume;
     },
@@ -25675,16 +25678,17 @@ async function evaluateNode(node, context, onReference) {
   };
   try {
     const result = node.type === "Identifier" && onReference !== void 0 ? await evaluateIdentifier(node, evaluationContext, onReference) : await handler(node, evaluationContext);
-    if (!context.deferDataScans || context.budget.limits.dataSize !== void 0 || (compilation?.tickets.size ?? 0) > 0) {
-      reconcileDataBudget(
-        context.budget,
-        context.stats,
-        context.scope,
-        "hasValue" in result && result.hasValue ? result.value : void 0,
-        compilation,
-        context.compilation
-      );
-    }
+    const transient = "hasValue" in result && result.hasValue ? result.value : void 0;
+    reconcileDataBudget(
+      context.budget,
+      context.stats,
+      context.scope,
+      transient,
+      compilation,
+      context.compilation,
+      true,
+      handler === evaluatePrimitiveLiteral ? typeof transient === "string" ? transient.length : 8 : void 0
+    );
     if (result.kind === "break" && result.label !== void 0 && "labels" in node && node.labels?.includes(result.label))
       return { kind: "normal", hasValue: context.evalCompletion === true && result.hasValue, value: context.evalCompletion ? result.value : void 0 };
     return result;
@@ -25709,14 +25713,20 @@ async function evaluateNode(node, context, onReference) {
       context.scope,
       completion.value,
       compilation,
-      context.compilation
+      context.compilation,
+      true
     );
     return completion;
   } finally {
     compilation?.dispose();
   }
 }
-function reconcileDataBudget(budget, stats, scope, transient, compilation, parent) {
+function reconcileDataBudget(budget, stats, scope, transient, compilation, parent, nodeCheckpoint = false, growth) {
+  if (nodeCheckpoint && (compilation?.tickets.size ?? 0) === 0) {
+    const limit = budget.limits.dataSize;
+    if (limit === void 0 || limit === Infinity) return;
+    if (growth !== void 0 && budget.currentDataSize + growth <= limit) return;
+  }
   reconcileCompiledValues(
     budget,
     [...scope.retainedDataRoots(), transient],
@@ -45183,6 +45193,9 @@ var HostCallJournal = class {
       ...outcome === void 0 ? {} : { outcome: copyOutcome(outcome) }
     }));
   }
+  inputPromiseOrder() {
+    return this.records.filter((record4) => record4.moduleId === "<inputs>").map((record4) => record4.operation);
+  }
   dispose() {
     this.disposed = true;
     for (const cancel of this.pendingReconciliations) cancel();

package/dist/safe-js/chunks/{chunk-YCFOANSU.js → chunk-BYDDUKS3.js}

@@ -7,7 +7,7 @@ import {
 import {
   declareBudgetedHostOperation,
   declareHostOperation
-} from "./chunk-5RLD4TVI.js";
+} from "./chunk-BKNQTNHP.js";
 import {
   basename,
   dirname,
@@ -15,14 +15,14 @@ import {
   relative,
   resolve,
   sep
-} from "./chunk-MMN3IESA.js";
+} from "./chunk-KMBCLXER.js";
 import {
   accessDeniedSystemError,
   constants,
   hostFs,
   hostPlatform,
   inspect
-} from "./chunk-JCHJGIQA.js";
+} from "./chunk-VVOPJZRG.js";
 
 // packages/safe-js/src/modules/fs.ts
 import { createFsBridge } from "@poe-platform/safe-fs/core";