@poe-platform/safe-js 0.1.383 → 0.1.385

package/dist/safe-js/chunks/{chunk-XIEVJEAG.js → chunk-4M2UASWA.js}

@@ -7272,12 +7272,16 @@ function copySandboxArgumentProperties(source, target, copyValue) {
 }
 
 // packages/safe-js/src/interp/collection-iterator.ts
+var collectionIteratorPrototypes = /* @__PURE__ */ new WeakMap();
 var states = /* @__PURE__ */ new WeakMap();
 function isSandboxCollectionIterator(value) {
   return typeof value === "object" && value !== null && states.has(value);
 }
-function createSandboxCollectionIterator(collection, method) {
-  return restoreSandboxCollectionIterator({ collection, collectionKind: collection.kind, method, index: 0, exhausted: false });
+function createSandboxCollectionIterator(collection, method, budget) {
+  const iterator = restoreSandboxCollectionIterator({ collection, collectionKind: collection.kind, method, index: 0, exhausted: false });
+  const prototype = budget === void 0 ? void 0 : collectionIteratorPrototypes.get(budget)?.[collection.kind];
+  if (prototype !== void 0) setSandboxPrototype(iterator, prototype, budget);
+  return iterator;
 }
 function restoreSandboxCollectionIterator(snapshot, target = /* @__PURE__ */ Object.create(null)) {
   const { collection, collectionKind, method, index, exhausted } = snapshot;
@@ -9787,7 +9791,7 @@ async function acquireSandboxIterator(value, budget, context, asyncProtocol = fa
       if (!hasExplicitSandboxPrototype(value) && value.async === true === asyncProtocol && getSandboxPropertyDescriptor(value, "next", budget) !== void 0)
         return guestIterator(value, await context.getProperty(value, "next"), asyncProtocol, budget, context, signal);
     }
-    if (!asyncProtocol && Array.isArray(value) && getSandboxPrototype(value, budget) !== null) return void 0;
+    if (!asyncProtocol && (typeof value === "string" ? lookupTarget !== void 0 : (Array.isArray(value) || isSandboxBox(value) || isSandboxMap(value) || isSandboxSet(value) || isSandboxCollectionIterator(value)) && (hasExplicitSandboxPrototype(value) || getSandboxPrototype(value, budget) !== null))) return void 0;
     if (!asyncProtocol) return getSandboxIterator(value, budget, context);
     if (isSandboxGenerator(value) && value.async && !hasExplicitSandboxPrototype(value))
       return getSandboxAsyncIterator(value, budget, context, signal);
@@ -9981,6 +9985,8 @@ function getSandboxIterator(value, budget, context) {
       }
     };
   }
+  if (isSandboxCollectionIterator(value) && hasExplicitSandboxPrototype(value))
+    return context?.getProperty === void 0 ? guestProtocolAdapter(value, budget ?? new Budget(), context, false) : void 0;
   if (isSandboxCollectionIterator(value))
     return {
       next: () => nextCollectionIterator(value, budget),
@@ -16087,7 +16093,7 @@ async function callMapMethod(target, methodName, args, options, stack = []) {
     case "keys":
     case "values":
     case "entries":
-      return createSandboxCollectionIterator(target, methodName);
+      return createSandboxCollectionIterator(target, methodName, options.budget);
   }
 }
 
@@ -16188,7 +16194,7 @@ async function callSetMethod(target, methodName, args, options, stack = []) {
     case "keys":
     case "values":
     case "entries":
-      return createSandboxCollectionIterator(target, methodName);
+      return createSandboxCollectionIterator(target, methodName, options.budget);
   }
 }
 
@@ -18689,7 +18695,7 @@ function getPropertyValue(target, property, context, receiver = target) {
   if (isSandboxMap(target)) return getSandboxPrototype(target, context.budget) === null ? getMapMember(target, property, createMapMethodOptions(context)) : void 0;
   if (isSandboxSet(target)) return getSandboxPrototype(target, context.budget) === null ? getSetMember(target, property, createSetMethodOptions(context)) : void 0;
   if (isSandboxCollectionIterator(target))
-    return getCollectionIteratorMember(target, property, context.budget);
+    return hasExplicitSandboxPrototype(target) ? void 0 : getCollectionIteratorMember(target, property, context.budget);
   if (isSandboxGenerator(target)) return hasExplicitSandboxPrototype(target) ? void 0 : getGeneratorMember(target, property, context.budget);
   if (isSandboxClosure(target)) return getClosureMemberValue(target, property, context);
   if (isSandboxPromise(target)) return hasExplicitSandboxPrototype(target) ? void 0 : getPromiseMember(property, context.budget);
@@ -22023,6 +22029,18 @@ function captureGuestHeapNode(value, encode) {
     const cursor = stringIteratorState(value);
     return { kind: "string-iterator", input: encode(cursor.input), index: cursor.index, state: captureObjectState(value, encode) };
   }
+  if (isSandboxCollectionIterator(value) && hasGuestObjectState(value)) {
+    const cursor = snapshotCollectionIterator(value);
+    return {
+      kind: "guest-collection-iterator",
+      collectionKind: cursor.collectionKind,
+      method: cursor.method,
+      collection: encode(cursor.collection),
+      index: cursor.index,
+      exhausted: cursor.exhausted,
+      state: captureObjectState(value, encode)
+    };
+  }
   if (isRawJson(value)) return { kind: "raw-json", text: value.rawJSON };
   const intrinsic = getIntrinsicIdentity(value);
   if (intrinsic !== void 0) {
@@ -25208,6 +25226,34 @@ function installCollectionPrototypes(budget, mapConstructor, setConstructor) {
     for (const method of /* @__PURE__ */ new Set([...Object.values(methods2), size, species])) registerIntrinsicFunction(budget, method);
   }
 }
+function installCollectionIteratorPrototypes(budget) {
+  for (const name of ["Map", "Set"]) {
+    const iteratorPrototype = /* @__PURE__ */ Object.create(null);
+    const kind = name === "Map" ? "map" : "set";
+    const next = createSandboxClosure({
+      guest: true,
+      sandbox: true,
+      name: "next",
+      length: 0,
+      call: (_args, context) => {
+        const receiver = context?.thisValue;
+        if (!isSandboxCollectionIterator(receiver) || collectionIteratorState(receiver).collectionKind !== kind)
+          throw new TypeError("Iterator next requires a matching collection iterator receiver.");
+        return nextCollectionIterator(receiver, budget);
+      }
+    });
+    setSandboxPrototype(iteratorPrototype, resolveIntrinsicIdentity(budget, '["%IteratorPrototype%"]'));
+    Object.defineProperties(iteratorPrototype, {
+      next: { value: next, writable: true, configurable: true },
+      [Symbol.toStringTag]: { value: `${name} Iterator`, configurable: true }
+    });
+    const installed = collectionIteratorPrototypes.get(budget) ?? {};
+    installed[kind] = iteratorPrototype;
+    collectionIteratorPrototypes.set(budget, installed);
+    registerBuiltinIdentities(budget, { [`%${name}IteratorPrototype%`]: iteratorPrototype });
+    registerIntrinsicObject(budget, iteratorPrototype);
+  }
+}
 
 // packages/safe-js/src/interp/globals/collections.ts
 var mapConstructors = /* @__PURE__ */ new WeakSet();
@@ -26902,6 +26948,7 @@ function createBuiltinBindings(options) {
     ...createPromiseGlobals(options),
     ...createRegexGlobals(options)
   };
+  installCollectionIteratorPrototypes(options.budget);
   createFunctionPrototype(options.budget, options.functionHasInstance);
   if (options.typedArrayPrototypes !== false) createNumericTypedArrayPrototypes(options.budget, bindings);
   if (options.errorPrototypes !== false) createErrorPrototypes(options.budget, bindings);
@@ -26969,7 +27016,7 @@ function validateGuestHeapNode(raw, heap, maxArrayLength = 4294967295) {
     createRawJson(node.text);
     return true;
   }
-  if (!["intrinsic", "bound-function", "guest-function", "guest-class", "guest-generator", "scope-frame", "guest-object", "guest-array", "array-iterator", "string-iterator", "map", "set"].includes(String(node.kind))) return false;
+  if (!["intrinsic", "bound-function", "guest-function", "guest-class", "guest-generator", "scope-frame", "guest-object", "guest-array", "array-iterator", "string-iterator", "guest-collection-iterator", "map", "set"].includes(String(node.kind))) return false;
   const reference = (value, kinds) => {
     const ref = record3(value);
     fields(ref, ["kind", "id"]);
@@ -27027,6 +27074,15 @@ function validateGuestHeapNode(raw, heap, maxArrayLength = 4294967295) {
     fields(node, ["kind", "id"], ["state"]);
     if (typeof node.id !== "string" || !intrinsicCatalogue().has(node.id)) throw new TypeError("Unknown intrinsic identity.");
     if (Object.hasOwn(node, "state")) state(node.state);
+  } else if (node.kind === "guest-collection-iterator") {
+    fields(node, ["kind", "collectionKind", "method", "collection", "index", "exhausted", "state"]);
+    if (!["map", "set"].includes(String(node.collectionKind)) || !["keys", "values", "entries"].includes(String(node.method)) || typeof node.exhausted !== "boolean")
+      throw new TypeError("Invalid collection iterator state.");
+    integer(node.index);
+    if (absent(node.collection)) {
+      if (!node.exhausted || node.index !== 0) throw new TypeError("Invalid exhausted collection iterator.");
+    } else reference(node.collection, [String(node.collectionKind)]);
+    state(node.state);
   } else if (node.kind === "string-iterator") {
     fields(node, ["kind", "input", "index", "state"]);
     const index = integer(node.index);
@@ -29043,6 +29099,7 @@ function measureSandboxData(values, options = {}) {
       for (const [key, descriptor] of Object.entries(Object.getOwnPropertyDescriptors(value))) {
         usage += key.length + 1;
         if ("value" in descriptor) visit(descriptor.value, depth + 1);
+        else for (const closure of retainedAccessorClosures(descriptor)) visit(closure, depth + 1);
       }
       return;
     }
@@ -29246,8 +29303,8 @@ function copyToSandbox(value, state, path = "<root>", cloneSandboxCollections =
     return copy;
   }
   if (isSandboxCollectionIterator(value)) {
-    if (hasGuestObjectState(value)) throw new TypeError("Guest prototype links and custom descriptors cannot be copied as data.");
     if (!cloneSandboxCollections) return value;
+    if (hasGuestObjectState(value)) throw new TypeError("Guest prototype links and custom descriptors cannot be copied as data.");
     const existing = state.seen.get(value);
     if (existing !== void 0) return existing;
     const snapshot = snapshotCollectionIterator(value);
@@ -44091,4 +44148,4 @@ export {
   FileSnapshotBackend,
   run
 };
-//# sourceMappingURL=chunk-XIEVJEAG.js.map
+//# sourceMappingURL=chunk-4M2UASWA.js.map