@poe-platform/safe-js 0.1.105 → 0.1.107

package/dist/safe-js/chunks/{chunk-B365DJ6C.js → chunk-INJEHRIX.js}

@@ -27,7 +27,7 @@ import {
   validateMigrationSemantics,
   validateSnapshotData,
   validateSnapshotMigration
-} from "./chunk-G5X7KIU3.js";
+} from "./chunk-SLQJGSFH.js";
 
 // packages/safe-js/src/migrate.ts
 import { createHash } from "node:crypto";
@@ -8245,4 +8245,4 @@ export {
   parseMcpConfig,
   makeMcpModule
 };
-//# sourceMappingURL=chunk-B365DJ6C.js.map
+//# sourceMappingURL=chunk-INJEHRIX.js.map

package/dist/safe-js/chunks/{chunk-G5X7KIU3.js → chunk-SLQJGSFH.js}

@@ -25362,6 +25362,20 @@ function typeTag(value) {
   return "Object";
 }
 
+// packages/safe-js/src/interp/data-checkpoint.ts
+function createDataCheckpoint(budget, context) {
+  let estimatedDataSize = 0;
+  return (value, growth = 0, force = false) => {
+    const limit = budget.limits.dataSize;
+    if (limit === void 0) return;
+    estimatedDataSize = Math.max(estimatedDataSize, budget.currentDataSize) + growth;
+    if (!force && estimatedDataSize <= limit) return;
+    if (context?.reconcileData !== void 0) context.reconcileData(value);
+    else reconcileCompiledValues(budget, [value], context?.compilation);
+    estimatedDataSize = budget.currentDataSize;
+  };
+}
+
 // packages/safe-js/src/interp/globals/collections.ts
 var mapConstructors = /* @__PURE__ */ new WeakSet();
 var setConstructors = /* @__PURE__ */ new WeakSet();
@@ -25371,13 +25385,7 @@ function createCollectionGlobals(options) {
     call: () => {
       throw new TypeError("Constructor Map requires 'new'.");
     },
-    construct: ([source]) => {
-      const entries = getMapEntries(source);
-      if (entries instanceof Promise) {
-        return entries.then((resolved) => createBudgetedMap(resolved, options.budget));
-      }
-      return createBudgetedMap(entries, options.budget);
-    },
+    construct: ([source], context) => constructMap(source, options.budget, context),
     name: "Map"
   });
   const setConstructor = createSandboxClosure({
@@ -25404,38 +25412,77 @@ function isSandboxMapConstructor(value) {
 function isSandboxSetConstructor(value) {
   return typeof value === "object" && value !== null && setConstructors.has(value);
 }
-function getMapEntries(source) {
-  if (source === void 0 || source === null) {
-    return [];
-  }
-  if (isSandboxMap(source)) {
-    return [...source.entries];
-  }
-  if (Array.isArray(source)) {
-    return validateMapEntries(source);
-  }
+function constructMap(source, budget, context) {
+  const map = createSandboxMap([]);
+  budget.allocateCollectionEntries(0);
+  if (source === void 0 || source === null) return map;
   const iterator = getSandboxIterator(source);
-  if (iterator?.generator !== true) {
-    throw new TypeError("Map constructor argument must be an array of pairs or a Map.");
-  }
-  return collectMapEntries(iterator);
-}
-async function collectMapEntries(iterator) {
-  const sourceEntries = [];
-  while (true) {
-    const result = await iterator.next();
-    if (result.done) break;
-    sourceEntries.push(result.value);
+  if (iterator === void 0) throw new TypeError("Map constructor requires an iterable.");
+  let entry;
+  let key;
+  let value;
+  let failure;
+  const retained = {};
+  budget.setRetainedValues(retained, () => [source, map, entry, key, value, failure]);
+  const checkData = createDataCheckpoint(budget, context);
+  const closeOnThrow = (error) => {
+    failure = isCapturedException(error) ? error.reason : error;
+    const rethrow = (closeError) => {
+      if (!isFatalSandboxError(error) && isFatalSandboxError(closeError)) throw closeError;
+      throw error;
+    };
+    try {
+      const closing = iterator.return?.();
+      if (iterator.generator) return Promise.resolve(closing).then(() => {
+        throw error;
+      }, rethrow);
+    } catch (closeError) {
+      return rethrow(closeError);
+    }
+    throw error;
+  };
+  const consume = (result) => {
+    if (typeof result !== "object" || result === null) throw new TypeError("Iterator result must be an object.");
+    if (result.done) return true;
+    entry = result.value;
+    try {
+      budget.visitNode();
+      if (typeof entry !== "object" || entry === null) throw new TypeError("Map constructor requires entry objects.");
+      key = context?.getProperty !== void 0 ? context.getProperty(entry, 0) : getSandboxDataProperty(entry, 0, budget);
+      value = context?.getProperty !== void 0 ? context.getProperty(entry, 1) : getSandboxDataProperty(entry, 1, budget);
+      const added = map.entries.has(key) ? 0 : 1;
+      const growth = added + (budget.limits.dataSize === void 0 ? 0 : measureSandboxData([key, value]));
+      budget.allocateCollectionEntries(map.entries.size + added);
+      map.entries.set(key, value);
+      entry = key = value = void 0;
+      checkData(map, growth);
+    } catch (error) {
+      return closeOnThrow(error);
+    }
+    return false;
+  };
+  if (iterator.generator) {
+    return (async () => {
+      try {
+        checkData(map, 0, true);
+        while (!await consume(await iterator.next())) {
+        }
+        checkData(map, 0, true);
+        return map;
+      } finally {
+        budget.setRetainedValues(retained, void 0);
+      }
+    })();
   }
-  return validateMapEntries(sourceEntries);
-}
-function validateMapEntries(sourceEntries) {
-  return sourceEntries.map((entry) => {
-    if (!Array.isArray(entry)) {
-      throw new TypeError("Map constructor entries must be arrays.");
+  try {
+    checkData(map, 0, true);
+    while (!consume(iterator.next())) {
     }
-    return [entry[0], entry[1]];
-  });
+    checkData(map, 0, true);
+    return map;
+  } finally {
+    budget.setRetainedValues(retained, void 0);
+  }
 }
 function getSetValues(source) {
   if (source === void 0 || source === null) {
@@ -25461,11 +25508,6 @@ async function collectSetValues(iterator) {
   }
   return values;
 }
-function createBudgetedMap(entries, budget) {
-  const map = createSandboxMap(entries);
-  budget.allocateCollectionEntries(map.entries.size);
-  return map;
-}
 function createBudgetedSet(values, budget) {
   budget.allocateCollectionEntries(new Set(values).size);
   return createSandboxSet(values);
@@ -30819,12 +30861,11 @@ async function arrayFromSandboxValues(args, budget, context) {
         if (Array.isArray(result)) budget.allocateArrayLength(index + 1);
         if (mapFn !== void 0) currentValue = await invokeBuiltinClosure(mapFn, [currentValue, index], budget, context, thisValue);
         const key = String(index);
-        const growth = key.length + 1 + (Array.isArray(result) ? Math.max(0, index + 1 - result.length) : 0) + (typeof currentValue === "string" ? currentValue.length : 0);
-        const graphChanged = typeof currentValue === "object" && currentValue !== null || isSandboxClosure(result);
+        const growth = key.length + 1 + (Array.isArray(result) ? Math.max(0, index + 1 - result.length) : 0) + (budget.limits.dataSize === void 0 ? 0 : measureSandboxData([currentValue]));
         budget.visitNode();
         defineOwnDataProperty(objectProperties(result, true), key, currentValue);
         currentValue = void 0;
-        checkData(result, growth, graphChanged);
+        checkData(result, growth, isSandboxClosure(result));
       } catch (error) {
         await closeOnThrow(error);
       }
@@ -30837,18 +30878,6 @@ async function arrayFromSandboxValues(args, budget, context) {
     budget.setRetainedValues(retained, void 0);
   }
 }
-function createDataCheckpoint(budget, context) {
-  let estimatedDataSize = 0;
-  return (value, growth = 0, force = false) => {
-    const limit = budget.limits.dataSize;
-    if (limit === void 0) return;
-    estimatedDataSize = Math.max(estimatedDataSize, budget.currentDataSize) + growth;
-    if (!force && estimatedDataSize <= limit) return;
-    if (context?.reconcileData !== void 0) context.reconcileData(value);
-    else reconcileCompiledValues(budget, [value], context?.compilation);
-    estimatedDataSize = budget.currentDataSize;
-  };
-}
 function createArrayFromConstructorArgs(args, budget) {
   if (args.length !== 1) {
     return budgetSandboxValue2(Reflect.apply(Array, Array, [...args]), budget);
@@ -33024,4 +33053,4 @@ export {
   FileSnapshotBackend,
   run
 };
-//# sourceMappingURL=chunk-G5X7KIU3.js.map
+//# sourceMappingURL=chunk-SLQJGSFH.js.map