@poe-platform/safe-js 0.1.387 → 0.1.389

package/dist/safe-js/chunks/{chunk-KSILYYRR.js → chunk-6Y27XODE.js}

@@ -8546,6 +8546,9 @@ function nextStringIterator(value, budget) {
   return { value: entry, done: false };
 }
 
+// packages/safe-js/src/interp/iterator-wrapper.ts
+var iteratorWrapperStates = /* @__PURE__ */ new WeakMap();
+
 // packages/safe-js/src/interp/raw-json.ts
 var rawJsonValues = /* @__PURE__ */ new WeakSet();
 function createRawJson(text) {
@@ -22030,6 +22033,9 @@ function serializeCollectionProperties(value, encode, dataOnly = false) {
 
 // packages/safe-js/src/snapshot/guest-heap.ts
 function captureGuestHeapNode(value, encode) {
+  const wrapper = iteratorWrapperStates.get(value);
+  if (wrapper !== void 0)
+    return { kind: "iterator-wrapper", iterator: encode(wrapper.iterator), next: encode(wrapper.next), state: captureObjectState(value, encode) };
   if (isSandboxModuleNamespace(value))
     return { kind: "module-namespace", entries: Object.keys(value).map((key) => [key, encode(value[key])]) };
   if (isSandboxStringIterator(value)) {
@@ -26347,6 +26353,159 @@ function createGeneratorPrototypes(budget) {
   generatorPrototypes.set(budget, state);
 }
 
+// packages/safe-js/src/interp/globals/iterator-from.ts
+function installIteratorFrom(constructor, budget) {
+  const prototype = /* @__PURE__ */ Object.create(null);
+  setSandboxPrototype(prototype, resolveIntrinsicIdentity(budget, '["%IteratorPrototype%"]'));
+  for (const name of ["next", "return"]) {
+    Object.defineProperty(prototype, name, { writable: true, configurable: true, value: createSandboxClosure({
+      guest: true,
+      sandbox: true,
+      name,
+      length: 0,
+      call: async (_args, context) => {
+        const receiver = context?.thisValue;
+        const state = receiver !== null && typeof receiver === "object" ? iteratorWrapperStates.get(receiver) : void 0;
+        if (state === void 0) throw new TypeError("Iterator wrapper method requires a branded receiver.");
+        const method = name === "next" ? state.next : await read(state.iterator, "return", context);
+        if (name === "return" && (method === void 0 || method === null)) return { value: void 0, done: true };
+        if (!isSandboxClosure(method)) throw new TypeError("Iterator method must be callable.");
+        return invokeBuiltinClosure(method, [], budget, context, state.iterator);
+      }
+    }) });
+  }
+  const from = createSandboxClosure({ guest: true, sandbox: true, name: "from", length: 1, call: async ([input], context) => {
+    if (input === null || typeof input !== "object" && typeof input !== "string")
+      throw new TypeError("Iterator.from requires an object or string.");
+    let iterator = input;
+    let next;
+    const release = retainValues(budget, () => [input, iterator, next]);
+    try {
+      const method = await read(input, Symbol.iterator, context);
+      if (method !== void 0 && method !== null) {
+        if (!isSandboxClosure(method)) throw new TypeError("Symbol.iterator must be callable.");
+        iterator = await invokeBuiltinClosure(method, [], budget, context, input);
+      }
+      if (iterator === null || typeof iterator !== "object") throw new TypeError("Iterator must be an object.");
+      next = await read(iterator, "next", context);
+      if (await ordinaryHasInstance(iterator, constructor, budget, context)) return iterator;
+      const wrapper = /* @__PURE__ */ Object.create(null);
+      iteratorWrapperStates.set(wrapper, { iterator, next });
+      setSandboxPrototype(wrapper, prototype, budget);
+      createDataCheckpoint(budget, context)(wrapper, 0, true);
+      return wrapper;
+    } finally {
+      release();
+    }
+  } });
+  Object.defineProperty(materializeFunctionProperties(constructor), "from", { value: from, writable: true, configurable: true });
+  registerBuiltinIdentities(budget, { Iterator: constructor, "%WrapForValidIteratorPrototype%": prototype });
+  registerIntrinsicFunction(budget, from);
+  registerIntrinsicObject(budget, prototype);
+  async function read(value, key, context) {
+    if (context?.getProperty !== void 0) return context.getProperty(value, key);
+    const descriptor = getSandboxPropertyDescriptor(value, key, budget);
+    return descriptor === void 0 ? void 0 : readPropertyDescriptor(descriptor, value, context);
+  }
+}
+
+// packages/safe-js/src/interp/globals/iterator-consumers.ts
+function installIteratorConsumers(prototype, budget) {
+  for (const name of ["toArray", "reduce", "forEach", "some", "every", "find"]) {
+    Object.defineProperty(prototype, name, { writable: true, configurable: true, value: createSandboxClosure({
+      guest: true,
+      sandbox: true,
+      name,
+      length: name === "toArray" ? 0 : 1,
+      call: async (args, context) => {
+        const receiver = context?.thisValue;
+        if (receiver === null || typeof receiver !== "object") throw new TypeError("Iterator consumer requires an object.");
+        const callback = args[0];
+        let next;
+        let accumulator = args[1];
+        let value;
+        const values = [];
+        const release = retainValues(budget, () => [receiver, callback, next, accumulator, value, values]);
+        const checkpoint = createDataCheckpoint(budget, context);
+        const iterator = {
+          asynchronous: true,
+          next: async () => {
+            if (!isSandboxClosure(next)) throw new TypeError("Iterator next must be callable.");
+            const result = await invokeBuiltinClosure(next, [], budget, context, receiver);
+            if (result === null || typeof result !== "object") throw new TypeError("Iterator result must be an object.");
+            return result;
+          },
+          getOperation: async () => {
+            const method = await read(receiver, "return", context);
+            if (method === void 0 || method === null) return void 0;
+            if (!isSandboxClosure(method)) throw new TypeError("Iterator return must be callable.");
+            return async () => await invokeBuiltinClosure(method, [], budget, context, receiver);
+          }
+        };
+        try {
+          if (name !== "toArray" && !isSandboxClosure(callback)) {
+            await closeIterator(iterator, true);
+            throw new TypeError("Iterator callback must be callable.");
+          }
+          next = await read(receiver, "next", context);
+          let initialized = name !== "reduce" || args.length > 1;
+          let index = 0;
+          while (true) {
+            budget.visitNode();
+            const result = await iterator.next();
+            if (await read(result, "done", context)) {
+              if (!initialized) throw new TypeError("Cannot reduce an empty iterator without an initial value.");
+              if (name === "toArray") return values;
+              if (name === "reduce") return accumulator;
+              if (name === "some") return false;
+              if (name === "every") return true;
+              return void 0;
+            }
+            value = await read(result, "value", context);
+            if (name === "toArray") {
+              values.push(value);
+              checkpoint(values, 0, true);
+            } else if (!initialized) {
+              accumulator = value;
+              initialized = true;
+            } else {
+              let selected;
+              try {
+                selected = await invokeBuiltinClosure(
+                  callback,
+                  name === "reduce" ? [accumulator, value, index] : [value, index],
+                  budget,
+                  context,
+                  void 0
+                );
+              } catch (error) {
+                if (!isFatalSandboxError(error)) await closeIterator(iterator, true);
+                throw error;
+              }
+              if (name === "reduce") accumulator = selected;
+              else if ((name === "some" || name === "find") && selected || name === "every" && !selected) {
+                await closeIterator(iterator);
+                return name === "find" ? value : name === "some";
+              }
+            }
+            value = void 0;
+            index++;
+          }
+        } finally {
+          release();
+        }
+      }
+    }) });
+  }
+  registerBuiltinIdentities(budget, { "%IteratorPrototype%": prototype });
+  registerIntrinsicObject(budget, prototype);
+  async function read(value, key, context) {
+    if (context?.getProperty !== void 0) return context.getProperty(value, key);
+    const descriptor = getSandboxPropertyDescriptor(value, key, budget);
+    return descriptor === void 0 ? void 0 : readPropertyDescriptor(descriptor, value, context);
+  }
+}
+
 // packages/safe-js/src/interp/globals/iterator.ts
 function createIteratorGlobal(budget) {
   const prototype = resolveIntrinsicIdentity(budget, '["%IteratorPrototype%"]');
@@ -26384,6 +26543,8 @@ function createIteratorGlobal(budget) {
   Object.defineProperty(prototype, "constructor", { get: accessorAdapter(getter, "get"), set: accessorAdapter(setter, "set"), configurable: true });
   registerIntrinsicFunction(budget, getter);
   registerIntrinsicFunction(budget, setter);
+  installIteratorFrom(constructor, budget);
+  installIteratorConsumers(prototype, budget);
   registerIntrinsicFunction(budget, constructor);
   registerIntrinsicObject(budget, prototype);
   return constructor;
@@ -27089,7 +27250,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", "guest-collection-iterator", "guest-regexp-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", "iterator-wrapper", "guest-collection-iterator", "guest-regexp-iterator", "map", "set"].includes(String(node.kind))) return false;
   const reference = (value, kinds) => {
     const ref = record3(value);
     fields(ref, ["kind", "id"]);
@@ -27143,7 +27304,12 @@ function validateGuestHeapNode(raw, heap, maxArrayLength = 4294967295) {
     });
     return false;
   }
-  if (node.kind === "intrinsic") {
+  if (node.kind === "iterator-wrapper") {
+    fields(node, ["kind", "iterator", "next", "state"]);
+    const iterator = reference(node.iterator);
+    if (iterator.kind === "symbol" || iterator.kind === "scope-frame") throw new TypeError("Invalid wrapped iterator.");
+    state(node.state);
+  } else if (node.kind === "intrinsic") {
     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);
@@ -29108,6 +29274,11 @@ function measureSandboxData(values, options = {}) {
       }
       return;
     }
+    const wrapperState = iteratorWrapperStates.get(value);
+    if (wrapperState !== void 0) {
+      visit(wrapperState.iterator, depth + 1);
+      visit(wrapperState.next, depth + 1);
+    }
     if (isSandboxDate(value)) usage += 8;
     if (isGuestHostObject(value)) {
       usage += measureHostObjectData(value);
@@ -44237,4 +44408,4 @@ export {
   FileSnapshotBackend,
   run
 };
-//# sourceMappingURL=chunk-KSILYYRR.js.map
+//# sourceMappingURL=chunk-6Y27XODE.js.map