@poe-platform/safe-js 0.1.215 → 0.1.217

package/dist/safe-js/chunks/{chunk-VJZDZJ6B.js → chunk-42FLUHBI.js}

@@ -27,7 +27,7 @@ import {
   validateMigrationSemantics,
   validateSnapshotData,
   validateSnapshotMigration
-} from "./chunk-XEABTBIE.js";
+} from "./chunk-F2O4RZOG.js";
 
 // packages/safe-js/src/migrate.ts
 import { createHash } from "node:crypto";
@@ -8249,4 +8249,4 @@ export {
   parseMcpConfig,
   makeMcpModule
 };
-//# sourceMappingURL=chunk-VJZDZJ6B.js.map
+//# sourceMappingURL=chunk-42FLUHBI.js.map

package/dist/safe-js/chunks/{chunk-XEABTBIE.js → chunk-F2O4RZOG.js}

@@ -191,6 +191,7 @@ var Budget = class {
   retainedValueSources = /* @__PURE__ */ new Map();
   compileGeneration = 0;
   activeCompileOwner;
+  defaultCompileOwner;
   compileUses = 0;
   provisionalScopes = 0;
   compileTickets = /* @__PURE__ */ new Map();
@@ -273,7 +274,7 @@ var Budget = class {
       throw new SandboxError("reentry");
     }
     if (reset) this.reset();
-    const selected = owner ?? Object.freeze({ budget: this, generation: this.compileGeneration });
+    const selected = owner ?? (this.defaultCompileOwner ??= Object.freeze({ budget: this, generation: this.compileGeneration }));
     this.activeCompileOwner = selected;
     this.compileUses += 1;
     let released = false;
@@ -391,6 +392,7 @@ var Budget = class {
   reset() {
     if (this.compileUses !== 0) throw new SandboxError("reentry");
     this.compileGeneration += 1;
+    this.defaultCompileOwner = void 0;
     this.provisionalScopes = 0;
     this.compileTickets.clear();
     this.completedCompileTickets.clear();
@@ -8536,7 +8538,7 @@ function captureGuestHeapNode(value, encode) {
       ...value.state !== "suspended" || origin2.expressionStates === void 0 ? {} : {
         expressionStates: Object.fromEntries([...origin2.expressionStates].map(([id, expression]) => [
           String(id),
-          expression.kind === "binary" ? { kind: "binary", left: encode(expression.left) } : { kind: "array", values: encode(expression.values), index: expression.index }
+          expression.kind === "binary" ? { kind: "binary", left: encode(expression.left) } : expression.kind === "array" ? { kind: "array", values: encode(expression.values), index: expression.index } : expression.kind === "array-call" ? { kind: "array-call", target: encode(expression.target), method: expression.method, args: encode(expression.args), index: expression.index } : { kind: expression.kind, callee: encode(expression.callee), thisValue: encode(expression.thisValue), args: encode(expression.args), index: expression.index }
         ]))
       },
       sent: channel.sent.map((completion) => ({ type: completion.type, value: encode(completion.value) })),
@@ -9846,7 +9848,7 @@ function typeTag(value, builtinOnly = false) {
 
 // packages/safe-js/src/interp/resume-target.ts
 function containsResumeTarget(node, targetNodeIds) {
-  if (node.nodeId !== void 0 && targetNodeIds.has(node.nodeId)) {
+  if (targetNodeIds === void 0 ? node.type === "YieldExpression" : node.nodeId !== void 0 && targetNodeIds.has(node.nodeId)) {
     return true;
   }
   if (node.type === "ArrowFunctionExpression" || node.type === "FunctionDeclaration" || node.type === "FunctionExpression") {
@@ -18073,6 +18075,12 @@ function createPatternContext(context, scope = context.scope, evaluate = evaluat
   };
 }
 async function evaluateCallExpression(node, context) {
+  const restored = context.generatorResume === void 0 || node.nodeId === void 0 ? void 0 : context.restoredGeneratorExpressionStates?.get(node.nodeId);
+  if (restored?.kind === "call") return evaluateResolvedCallExpression(node, restored.callee, context, restored.thisValue);
+  if (restored?.kind === "array-call") {
+    if (!Array.isArray(restored.target) || !isArrayMethodName(restored.method)) throw new TypeError("Invalid array call continuation.");
+    return evaluateArrayMethodCall(node, restored.target, restored.method, context);
+  }
   if (node.callee.type === "Super") {
     const construction = context.functionEnvironment?.construction;
     if (construction === void 0)
@@ -18093,12 +18101,15 @@ async function evaluateCallExpression(node, context) {
   return evaluateResolvedCallExpression(node, callee.value, context);
 }
 async function evaluateNewExpression(node, context) {
-  const callee = await evaluateNode(node.callee, context);
+  const restored = context.generatorResume === void 0 || node.nodeId === void 0 ? void 0 : context.restoredGeneratorExpressionStates?.get(node.nodeId);
+  const callee = restored?.kind === "new" ? { kind: "normal", value: restored.callee } : await evaluateNode(node.callee, context);
   if (callee.kind !== "normal") {
     return callee;
   }
   const name = getConstructorName(node.callee);
-  const args = await evaluateCallArguments(node.arguments, context);
+  const call = createCallContinuation(node, callee.value, context);
+  context = call.context;
+  const args = await evaluateCallArguments(node.arguments, context, call.state);
   if (!args.ok) {
     return args.result;
   }
@@ -18231,6 +18242,12 @@ async function evaluateMemberCallExpression(node, context) {
       ...reference,
       property: await toPropertyKey(reference.property, context.budget, createCoercionContext(context))
     };
+    if (context.generatorYield !== void 0 && node.arguments.some((argument) => containsResumeTarget(argument.type === "SpreadElement" ? argument.argument : argument))) {
+      if (Array.isArray(member.object) && !hasExplicitSandboxPrototype(member.object) && typeof member.property !== "symbol" && isArrayMethodName(member.property) && !Object.hasOwn(member.object, member.property))
+        return evaluateArrayMethodCall(node, member.object, member.property, context);
+      const receiver = member.superReceiver?.value ?? member.object;
+      return evaluateResolvedCallExpression(node, await getPropertyValue(member.object, member.property, context, receiver), context, receiver);
+    }
     if (member.superReceiver !== void 0)
       return evaluateResolvedCallExpression(node, await getPropertyValue(member.object, member.property, context, member.superReceiver.value), context, member.superReceiver.value);
     if (typeof member.property === "symbol")
@@ -18395,7 +18412,21 @@ async function evaluateStringMethodCall(node, target, methodName, context) {
   }
 }
 async function evaluateArrayMethodCall(node, target, methodName, context) {
-  const args = await evaluateCallArguments(node.arguments, context);
+  const restored = context.generatorResume === void 0 || node.nodeId === void 0 ? void 0 : context.restoredGeneratorExpressionStates?.get(node.nodeId);
+  if (restored !== void 0 && (restored.kind !== "array-call" || !Array.isArray(restored.args)))
+    throw new TypeError("Invalid array call continuation.");
+  const state = {
+    kind: "array-call",
+    target,
+    method: methodName,
+    args: restored?.kind === "array-call" ? restored.args : [],
+    index: restored?.kind === "array-call" ? restored.index : 0
+  };
+  if (context.generatorYield !== void 0 && node.nodeId !== void 0) context = {
+    ...context,
+    generatorExpressionStates: new Map([...context.generatorExpressionStates ?? [], [node.nodeId, state]])
+  };
+  const args = await evaluateCallArguments(node.arguments, context, state);
   if (!args.ok) {
     return args.result;
   }
@@ -18941,7 +18972,9 @@ async function evaluateResolvedCallExpression(node, callee, context, thisValue =
       value: void 0
     };
   }
-  const args = await evaluateCallArguments(node.arguments, context);
+  const call = createCallContinuation(node, callee, context, thisValue);
+  context = call.context;
+  const args = await evaluateCallArguments(node.arguments, context, call.state);
   if (!args.ok) {
     return args.result;
   }
@@ -19041,11 +19074,30 @@ async function invokeSandboxClosure(callee, args, context, stack, span, thisValu
     leaveCall();
   }
 }
-async function evaluateCallArguments(args, context) {
-  const values = [];
+function createCallContinuation(node, callee, context, thisValue = void 0) {
+  const kind = node.type === "NewExpression" ? "new" : "call";
+  const restored = context.generatorResume === void 0 || node.nodeId === void 0 ? void 0 : context.restoredGeneratorExpressionStates?.get(node.nodeId);
+  if (restored !== void 0 && (restored.kind !== "call" && restored.kind !== "new" || restored.kind !== kind || !Array.isArray(restored.args)))
+    throw new TypeError("Invalid call expression continuation.");
+  const state = {
+    kind,
+    callee,
+    thisValue,
+    args: restored?.kind === "call" || restored?.kind === "new" ? restored.args : [],
+    index: restored?.kind === "call" || restored?.kind === "new" ? restored.index : 0
+  };
+  return { state, context: context.generatorYield === void 0 || node.nodeId === void 0 ? context : {
+    ...context,
+    generatorExpressionStates: new Map([...context.generatorExpressionStates ?? [], [node.nodeId, state]])
+  } };
+}
+async function evaluateCallArguments(args, context, continuation) {
+  const values = continuation?.args ?? [];
   const release = retainValues(context.budget, () => values);
   try {
-    for (const arg of args) {
+    for (let index = continuation?.index ?? 0; index < args.length; index++) {
+      if (continuation !== void 0) continuation.index = index;
+      const arg = args[index];
       if (arg.type === "SpreadElement") {
         const spreadValues = await evaluateSpreadElement(arg, context);
         if (!spreadValues.ok) {
@@ -22307,6 +22359,16 @@ function validateGuestHeapNode(raw, heap) {
           fields(expression, ["kind", "values", "index"]);
           integer(expression.index);
           reference(expression.values, ["array"]);
+        } else if (expression.kind === "call" || expression.kind === "new") {
+          fields(expression, ["kind", "callee", "thisValue", "args", "index"]);
+          integer(expression.index);
+          reference(expression.args, ["array"]);
+        } else if (expression.kind === "array-call") {
+          fields(expression, ["kind", "target", "method", "args", "index"]);
+          if (typeof expression.method !== "string") throw new TypeError("Invalid array method.");
+          integer(expression.index);
+          reference(expression.args, ["array"]);
+          reference(expression.target, ["array"]);
         } else throw new TypeError("Invalid expression continuation.");
       }
     }
@@ -22411,12 +22473,17 @@ function validateGuestFunctionAst(record3, origin) {
       yieldExpressions = frame.expressions;
     }
     for (const [key, value] of Object.entries(node)) {
-      if (node.type === "ArrayExpression" && key === "elements" && typeof node.nodeId === "number" && Array.isArray(value)) {
+      if ((node.type === "ArrayExpression" && key === "elements" || (node.type === "CallExpression" || node.type === "NewExpression") && key === "arguments") && typeof node.nodeId === "number" && Array.isArray(value)) {
+        const kind = node.type === "ArrayExpression" ? "array" : node.type === "CallExpression" ? "call" : "new";
         value.forEach((element, index) => pending.push({
           value: element,
           blocks,
           finalizers: frame.finalizers,
-          expressions: new Map([...frame.expressions, [node.nodeId, { kind: "array", index }]])
+          expressions: new Map([...frame.expressions, [node.nodeId, {
+            kind,
+            index,
+            member: kind === "call" && node.callee?.type === "MemberExpression"
+          }]])
         }));
         continue;
       }
@@ -22448,7 +22515,8 @@ function validateGuestFunctionAst(record3, origin) {
     throw new TypeError("Invalid generator AST identity: missing expression continuation");
   for (const [id, expression] of expressions) {
     const expected = yieldExpressions?.get(Number(id));
-    if (expected === void 0 || expected.kind !== expression.kind || expected.kind === "array" && expected.index !== expression.index)
+    const compatibleKind = expected?.kind === expression.kind || expected?.kind === "call" && expected.member === true && expression.kind === "array-call";
+    if (expected === void 0 || !compatibleKind || expected.kind !== "binary" && expected.index !== expression.index)
       throw new TypeError("Invalid generator AST identity: unrelated expression continuation");
   }
 }
@@ -38633,4 +38701,4 @@ export {
   FileSnapshotBackend,
   run
 };
-//# sourceMappingURL=chunk-XEABTBIE.js.map
+//# sourceMappingURL=chunk-F2O4RZOG.js.map