@poe-platform/safe-js 0.1.255 → 0.1.257

package/dist/safe-js/chunks/{chunk-PRWJXI5K.js → chunk-A2FDFKJ6.js}

@@ -7765,6 +7765,7 @@ var prototypes = /* @__PURE__ */ new WeakMap();
 var intrinsicPrototypes = /* @__PURE__ */ new WeakMap();
 var boxedPrototypes = /* @__PURE__ */ new WeakMap();
 var regexPrototypes = /* @__PURE__ */ new WeakMap();
+var collectionPrototypes = /* @__PURE__ */ new WeakMap();
 var initialRegexDescriptors = /* @__PURE__ */ new WeakMap();
 var intrinsicPrototypeRoots = /* @__PURE__ */ new WeakMap();
 var intrinsicConstructors = /* @__PURE__ */ new WeakMap();
@@ -7834,6 +7835,12 @@ function installRegexPrototype(budget, prototype, constructor) {
   initialRegexDescriptors.set(budget, Object.getOwnPropertyDescriptors(prototype));
   registerIntrinsicPrototype(budget, prototype, constructor);
 }
+function installCollectionPrototype(budget, name, prototype, constructor) {
+  let state = collectionPrototypes.get(budget);
+  if (state === void 0) collectionPrototypes.set(budget, state = /* @__PURE__ */ new Map());
+  state.set(name, prototype);
+  registerIntrinsicPrototype(budget, prototype, constructor);
+}
 function hasRegexPropertyOverride(value, keys, budget) {
   const initial = initialRegexDescriptors.get(budget);
   const prototype = regexPrototypes.get(budget);
@@ -7854,7 +7861,10 @@ function isIntrinsicFunction(value) {
 function registerIntrinsicPrototype(budget, prototype, constructor) {
   if (constructor.name === void 0) throw new TypeError("Intrinsic constructors require an installation name.");
   registerBuiltinIdentities(budget, { [constructor.name]: constructor });
-  const methods = [prototype, materializeFunctionProperties(constructor)].flatMap((owner) => Reflect.ownKeys(owner).map((key) => Object.getOwnPropertyDescriptor(owner, key)?.value)).filter(isGuestClosure);
+  const methods = [prototype, materializeFunctionProperties(constructor)].flatMap((owner) => Reflect.ownKeys(owner).flatMap((key) => {
+    const descriptor = Object.getOwnPropertyDescriptor(owner, key);
+    return [descriptor.value, ...retainedAccessorClosures(descriptor)];
+  })).filter(isGuestClosure);
   trackIntrinsicState(budget, prototype, constructor, [prototype, constructor, ...methods]);
 }
 function registerIntrinsicFunction(budget, closure) {
@@ -7909,11 +7919,14 @@ function releaseObjectPrototype(budget) {
   intrinsicPrototypeRoots.delete(budget);
   boxedPrototypes.delete(budget);
   regexPrototypes.delete(budget);
+  collectionPrototypes.delete(budget);
   initialRegexDescriptors.delete(budget);
   intrinsicPrototypes.delete(budget);
 }
 function getSandboxPrototype(value, budget) {
   if (prototypes.has(value)) return prototypes.get(value) ?? null;
+  if (budget !== void 0 && (isSandboxMap(value) || isSandboxSet(value)))
+    return collectionPrototypes.get(budget)?.get(isSandboxMap(value) ? "Map" : "Set") ?? null;
   if (budget !== void 0 && isSandboxRegex(value)) return regexPrototypes.get(budget) ?? null;
   if (budget !== void 0 && isSandboxBox(value)) {
     const prototype = getBoxedPrototype(boxedValue(value), budget);
@@ -7933,8 +7946,8 @@ function getSandboxPropertyDescriptor(value, key, budget) {
     return hostProperties === void 0 ? void 0 : Object.getOwnPropertyDescriptor(hostProperties, key);
   let current = value;
   let depth = 0;
-  while (typeof current === "object" && current !== null && (Array.isArray(current) || isSandboxDate(current) || isSandboxRegex(current) || isPrototypeRecord(current))) {
-    const properties = isGuestClosure(current) ? getGuestFunctionProperties(current) : isSandboxRegex(current) ? getRegexProperties(current) : current;
+  while (typeof current === "object" && current !== null && (Array.isArray(current) || isSandboxDate(current) || isSandboxRegex(current) || isSandboxMap(current) || isSandboxSet(current) || isPrototypeRecord(current))) {
+    const properties = isGuestClosure(current) ? getGuestFunctionProperties(current) : isSandboxRegex(current) ? getRegexProperties(current) : isSandboxMap(current) || isSandboxSet(current) ? void 0 : current;
     const descriptor = properties === void 0 ? void 0 : Object.getOwnPropertyDescriptor(properties, key);
     if (descriptor !== void 0) return descriptor;
     current = getSandboxPrototype(current, budget);
@@ -12403,6 +12416,8 @@ function createObjectArrayGlobals(options) {
         getPrototypeOf: createSandboxClosure({
           sandbox: true,
           call: ([value]) => {
+            if (isSandboxMap(value) || isSandboxSet(value))
+              return getSandboxPrototype(value, options.budget);
             if (isSandboxDate(value))
               return getDatePrototype(value, options.budget, options.compileOwner);
             if (value !== null && value !== void 0 && typeof value !== "object")
@@ -14734,12 +14749,90 @@ function isObjectLike(value) {
   return typeof value === "object" && value !== null;
 }
 
+// packages/safe-js/src/interp/globals/collection-prototypes.ts
+function installCollectionPrototypes(budget, mapConstructor, setConstructor) {
+  const mapMethods = Object.fromEntries([...mapMethodNames].map((name) => [name, createSandboxClosure({
+    guest: true,
+    sandbox: true,
+    name,
+    length: Map.prototype[name].length,
+    call: (args, context) => {
+      const receiver = context?.thisValue;
+      if (!isSandboxMap(receiver)) throw new TypeError(`Map.prototype.${name} requires a Map receiver.`);
+      return callMapMethod(receiver, name, args, {
+        budget,
+        callClosure: (closure, values, _stack, thisValue) => invokeBuiltinClosure(closure, values, budget, context, thisValue)
+      }, context?.stack ?? []);
+    }
+  })]));
+  const setMethods = Object.fromEntries([...setMethodNames].filter((name) => name !== "keys").map((name) => [name, createSandboxClosure({
+    guest: true,
+    sandbox: true,
+    name,
+    length: Set.prototype[name].length,
+    call: (args, context) => {
+      const receiver = context?.thisValue;
+      if (!isSandboxSet(receiver)) throw new TypeError(`Set.prototype.${name} requires a Set receiver.`);
+      return callSetMethod(receiver, name, args, {
+        budget,
+        callClosure: (closure, values, _stack, thisValue) => invokeBuiltinClosure(closure, values, budget, context, thisValue)
+      }, context?.stack ?? []);
+    }
+  })]));
+  setMethods.keys = setMethods.values;
+  const mapSize = createSandboxClosure({
+    guest: true,
+    sandbox: true,
+    name: "get size",
+    length: 0,
+    call: (_args, context) => {
+      if (!isSandboxMap(context?.thisValue)) throw new TypeError("Map.prototype.size requires a Map receiver.");
+      return context.thisValue.entries.size;
+    }
+  });
+  const setSize = createSandboxClosure({
+    guest: true,
+    sandbox: true,
+    name: "get size",
+    length: 0,
+    call: (_args, context) => {
+      if (!isSandboxSet(context?.thisValue)) throw new TypeError("Set.prototype.size requires a Set receiver.");
+      return context.thisValue.values.size;
+    }
+  });
+  for (const { name, constructor, methods, size, iterator } of [
+    { name: "Map", constructor: mapConstructor, methods: mapMethods, size: mapSize, iterator: mapMethods.entries },
+    { name: "Set", constructor: setConstructor, methods: setMethods, size: setSize, iterator: setMethods.values }
+  ]) {
+    const prototype = /* @__PURE__ */ Object.create(null);
+    const species = createSandboxClosure({
+      guest: true,
+      sandbox: true,
+      name: "get [Symbol.species]",
+      length: 0,
+      call: (_args, context) => context?.thisValue
+    });
+    Object.defineProperty(materializeFunctionProperties(constructor), "prototype", { value: prototype, writable: false });
+    Object.defineProperty(materializeFunctionProperties(constructor), Symbol.species, { get: accessorAdapter(species, "get"), configurable: true });
+    Object.defineProperty(prototype, "constructor", { value: constructor, writable: true, configurable: true });
+    for (const [key, method] of Object.entries(methods))
+      Object.defineProperty(prototype, key, { value: method, writable: true, configurable: true });
+    Object.defineProperty(prototype, "size", { get: accessorAdapter(size, "get"), configurable: true });
+    Object.defineProperty(prototype, Symbol.toStringTag, { value: name, configurable: true });
+    Object.defineProperty(prototype, Symbol.iterator, { value: iterator, writable: true, configurable: true });
+    installCollectionPrototype(budget, name, prototype, constructor);
+    for (const method of /* @__PURE__ */ new Set([...Object.values(methods), size, species])) registerIntrinsicFunction(budget, method);
+  }
+}
+
 // packages/safe-js/src/interp/globals/collections.ts
 var mapConstructors = /* @__PURE__ */ new WeakSet();
 var setConstructors = /* @__PURE__ */ new WeakSet();
 function createCollectionGlobals(options) {
   const mapConstructor = createSandboxClosure({
     sandbox: true,
+    guest: true,
+    length: 0,
     call: () => {
       throw new TypeError("Constructor Map requires 'new'.");
     },
@@ -14752,11 +14845,16 @@ function createCollectionGlobals(options) {
         budget: options.budget,
         context,
         retainedValues: () => [key, value],
-        append: (entry) => {
+        append: (entry, adder) => {
           if (typeof entry !== "object" || entry === null)
             throw new TypeError("Map constructor requires entry objects.");
           const store = (entryValue) => {
             value = entryValue;
+            if (adder !== void 0)
+              return invokeBuiltinClosure(adder, [key, value], options.budget, context, map).then(() => {
+                key = value = void 0;
+                return 0;
+              });
             const added = map.entries.has(key) ? 0 : 1;
             const growth = added + (options.budget.limits.dataSize === void 0 ? 0 : measureSandboxData([key, value]));
             options.budget.allocateCollectionEntries(map.entries.size + added);
@@ -14778,6 +14876,8 @@ function createCollectionGlobals(options) {
   });
   const setConstructor = createSandboxClosure({
     sandbox: true,
+    guest: true,
+    length: 0,
     call: () => {
       throw new TypeError("Constructor Set requires 'new'.");
     },
@@ -14787,7 +14887,9 @@ function createCollectionGlobals(options) {
         name: "Set",
         budget: options.budget,
         context,
-        append: (value) => {
+        append: (value, adder) => {
+          if (adder !== void 0)
+            return invokeBuiltinClosure(adder, [value], options.budget, context, set).then(() => 0);
           const added = set.values.has(value) ? 0 : 1;
           const growth = added + (options.budget.limits.dataSize === void 0 ? 0 : measureSandboxData([value]));
           options.budget.allocateCollectionEntries(set.values.size + added);
@@ -14800,6 +14902,7 @@ function createCollectionGlobals(options) {
   });
   mapConstructors.add(mapConstructor);
   setConstructors.add(setConstructor);
+  installCollectionPrototypes(options.budget, mapConstructor, setConstructor);
   return { Map: mapConstructor, Set: setConstructor };
 }
 function isSandboxMapConstructor(value) {
@@ -14817,15 +14920,17 @@ function populateCollection(source, collection, {
 }) {
   budget.allocateCollectionEntries(0);
   if (source === void 0 || source === null) return collection;
-  const populate = (iterator2) => {
-    if (iterator2 === void 0) throw new TypeError(`${name} constructor requires an iterable.`);
+  let adder;
+  const populate = (iterator) => {
+    if (iterator === void 0) throw new TypeError(`${name} constructor requires an iterable.`);
     let entry;
     let failure;
     const retained = {};
     budget.setRetainedValues(retained, () => [
       source,
-      iterator2.retainedValue,
+      iterator.retainedValue,
       collection,
+      adder,
       entry,
       failure,
       ...retainedValues?.() ?? []
@@ -14837,13 +14942,13 @@ function populateCollection(source, collection, {
         if (!isFatalSandboxError(error) && isFatalSandboxError(closeError)) throw closeError;
         throw error;
       };
-      if (iterator2.getOperation !== void 0)
-        return closeIterator(iterator2, true).then(() => {
+      if (iterator.getOperation !== void 0)
+        return closeIterator(iterator, true).then(() => {
           throw error;
         }, rethrow);
       try {
-        const closing = iterator2.return?.();
-        if (iterator2.generator || iterator2.asynchronous)
+        const closing = iterator.return?.();
+        if (iterator.generator || iterator.asynchronous)
           return Promise.resolve(closing).then(() => {
             throw error;
           }, rethrow);
@@ -14859,7 +14964,7 @@ function populateCollection(source, collection, {
       entry = result.value;
       try {
         budget.visitNode();
-        const growth = append(entry);
+        const growth = append(entry, adder);
         if (growth instanceof Promise)
           return growth.then((size) => {
             entry = void 0;
@@ -14873,16 +14978,16 @@ function populateCollection(source, collection, {
       }
       return false;
     };
-    if (iterator2.generator || iterator2.asynchronous || context !== void 0) {
+    if (iterator.generator || iterator.asynchronous || context !== void 0 || adder !== void 0) {
       return (async () => {
         try {
           checkData(collection, 0, true);
           while (true) {
-            const result = await iterator2.next();
+            const result = await iterator.next();
             if (typeof result !== "object" || result === null)
               throw new TypeError("Iterator result must be an object.");
-            if ((await readIteratorResult(iterator2, result, "done")).value) break;
-            const value = (await readIteratorResult(iterator2, result, "value")).value;
+            if ((await readIteratorResult(iterator, result, "done")).value) break;
+            const value = (await readIteratorResult(iterator, result, "value")).value;
             await consume({ done: false, value });
           }
           checkData(collection, 0, true);
@@ -14894,7 +14999,7 @@ function populateCollection(source, collection, {
     }
     try {
       checkData(collection, 0, true);
-      while (!consume(iterator2.next())) {
+      while (!consume(iterator.next())) {
       }
       checkData(collection, 0, true);
       return collection;
@@ -14902,8 +15007,16 @@ function populateCollection(source, collection, {
       budget.setRetainedValues(retained, void 0);
     }
   };
-  const iterator = context === void 0 ? getSandboxIterator(source, budget) : acquireSandboxIterator(source, budget, context);
-  return iterator instanceof Promise ? iterator.then(populate) : populate(iterator);
+  const method = name === "Map" ? "set" : "add";
+  const start = (candidate2) => {
+    if (!isSandboxClosure(candidate2)) throw new TypeError(`${name} constructor adder must be callable.`);
+    if (getIntrinsicIdentity(candidate2) !== JSON.stringify([name, "prototype", method])) adder = candidate2;
+    const iterator = context === void 0 ? getSandboxIterator(source, budget) : acquireSandboxIterator(source, budget, context);
+    return iterator instanceof Promise ? iterator.then(populate) : populate(iterator);
+  };
+  const descriptor = getSandboxPropertyDescriptor(collection, method, budget);
+  const candidate = context?.getProperty !== void 0 ? context.getProperty(collection, method) : descriptor === void 0 ? void 0 : readPropertyDescriptor(descriptor, collection, context);
+  return candidate instanceof Promise ? candidate.then(start) : start(candidate);
 }
 
 // packages/safe-js/src/interp/methods/collection-iterator.ts
@@ -17070,8 +17183,8 @@ function getPropertyValue(target, property, context, receiver = target) {
   if (isFloat32Array(target)) return getFloat32Member(target, property, context.budget);
   if (isSandboxDate(target))
     return getDateMember(property, context.budget, context.compilation?.owner);
-  if (isSandboxMap(target)) return getMapMember(target, property, createMapMethodOptions(context));
-  if (isSandboxSet(target)) return getSetMember(target, property, createSetMethodOptions(context));
+  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);
   if (isSandboxGenerator(target)) return getGeneratorMember(target, property, context.budget);
@@ -17301,6 +17414,8 @@ async function evaluateMemberCallExpression(node, context) {
       return evaluateResolvedCallExpression(node, await getPropertyValue(member.object, member.property, context, member.superReceiver.value), context, member.superReceiver.value);
     if (typeof member.property === "symbol")
       return evaluateResolvedCallExpression(node, await getPropertyValue(member.object, member.property, context), context, member.object);
+    if ((isSandboxMap(member.object) || isSandboxSet(member.object)) && getSandboxPrototype(member.object, context.budget) !== null)
+      return evaluateResolvedCallExpression(node, await getPropertyValue(member.object, member.property, context), context, member.object);
     if (Array.isArray(member.object) && hasExplicitSandboxPrototype(member.object))
       return evaluateResolvedCallExpression(node, await getPropertyValue(member.object, member.property, context), context, member.object);
     if ((typeof member.object === "string" || typeof member.object === "number" || typeof member.object === "bigint" || typeof member.object === "boolean" || typeof member.object === "symbol") && getBoxedPrototype(member.object, context.budget) !== void 0) {
@@ -17693,7 +17808,7 @@ function hasSandboxProperty(value, key, context) {
   while (typeof current === "object" && current !== null) {
     if (isGuestHostObject(current)) return typeof key === "symbol" ? false : hasHostObjectMember(current, String(key));
     if (hasOwnSandboxProperty(current, key, false)) return true;
-    if (!isSandboxRegex(current) && !((isGuestClosure(current) || Array.isArray(current)) && hasExplicitSandboxPrototype(current)) && (Array.isArray(current) || !isPlainSandboxObject(current) || isSandboxDate(current) || isFloat32Array(current) || isSandboxGenerator(current) || isSandboxCollectionIterator(current) || isSandboxRegExpIterator(current))) {
+    if (!isSandboxRegex(current) && !isSandboxMap(current) && !isSandboxSet(current) && !((isGuestClosure(current) || Array.isArray(current)) && hasExplicitSandboxPrototype(current)) && (Array.isArray(current) || !isPlainSandboxObject(current) || isSandboxDate(current) || isFloat32Array(current) || isSandboxGenerator(current) || isSandboxCollectionIterator(current) || isSandboxRegExpIterator(current))) {
       return getPropertyValue(current, key, context) !== void 0;
     }
     current = getSandboxPrototype(current, context.budget);
@@ -20764,6 +20879,9 @@ async function acquireSandboxIterator(value, budget, context, asyncProtocol = fa
     return iterator2 === void 0 ? void 0 : asyncFromSyncIterator(iterator2, budget, signal);
   }
   const factory = await context.getProperty(value, key);
+  if (!asyncProtocol && (isSandboxMap(value) || isSandboxSet(value)) && typeof factory === "object" && factory !== null && getIntrinsicIdentity(factory) === JSON.stringify(isSandboxMap(value) ? ["Map", "prototype", "entries"] : ["Set", "prototype", "values"])) {
+    return getSandboxIterator(value, budget, context);
+  }
   if (factory === null || factory === void 0) {
     if (!asyncProtocol) return void 0;
     const iterator2 = await acquireSandboxIterator(value, budget, context);
@@ -39697,4 +39815,4 @@ export {
   FileSnapshotBackend,
   run
 };
-//# sourceMappingURL=chunk-PRWJXI5K.js.map
+//# sourceMappingURL=chunk-A2FDFKJ6.js.map