@poe-platform/safe-js 0.1.723 → 0.1.725

package/dist/safe-js/chunks/{chunk-SWQHZKYM.js → chunk-O2CPPE6M.js}

@@ -1,6 +1,6 @@
 import {
   deepCopyToSandbox
-} from "./chunk-IZ52PAOD.js";
+} from "./chunk-QU2F65UQ.js";
 
 // packages/safe-js/src/modules/harness.ts
 function makeHarnessModule(frontmatter, meta) {
@@ -52,4 +52,4 @@ ${prompt}`;
 export {
   makeHarnessModule
 };
-//# sourceMappingURL=chunk-SWQHZKYM.js.map
+//# sourceMappingURL=chunk-O2CPPE6M.js.map

package/dist/safe-js/chunks/{chunk-M6HUSN7X.js → chunk-PKRGP34X.js}

@@ -1,10 +1,10 @@
 import {
   restore
-} from "./chunk-KHJDAEET.js";
+} from "./chunk-V2ESZF34.js";
 import {
   attachDumpController,
   createDumpController
-} from "./chunk-5MT2LF3W.js";
+} from "./chunk-FYSWUSHN.js";
 import {
   getOwnErrorCode,
   hasOwnErrorCode
@@ -12,7 +12,7 @@ import {
 import {
   hashParsedAst,
   hashSource
-} from "./chunk-Y5ZXQHL6.js";
+} from "./chunk-Y7LSEUWV.js";
 import {
   DUMP_FORMAT_VERSION,
   EXECUTION_SEMANTICS,
@@ -100,7 +100,7 @@ import {
   withSandboxPromiseRejectionTracker,
   wrapCallerInjectedBindings,
   wrapCancelableBindings
-} from "./chunk-IZ52PAOD.js";
+} from "./chunk-QU2F65UQ.js";
 import {
   Budget,
   CompileScope,
@@ -115,7 +115,7 @@ import {
   parseSourceModule,
   replaceErrorStack,
   safeAddEvent
-} from "./chunk-RYVHEOHM.js";
+} from "./chunk-KOO6YGQG.js";
 
 // packages/safe-js/src/realm.ts
 import { AsyncLocalStorage as AsyncLocalStorage2 } from "node:async_hooks";
@@ -751,6 +751,7 @@ var RealmState = class {
   guestReferences = /* @__PURE__ */ new Map();
   retainedOperations = /* @__PURE__ */ new WeakMap();
   nestedOperations = /* @__PURE__ */ new WeakMap();
+  callbackOperations = /* @__PURE__ */ new WeakMap();
   convertedModules = /* @__PURE__ */ new Map();
   nativeConversions = { seen: /* @__PURE__ */ new WeakMap() };
   modules;
@@ -784,15 +785,27 @@ var RealmState = class {
   };
   captureArguments = (operation, args, copy) => {
     const from = this.retainedOperations.get(operation)?.from ?? args.length;
-    const values = copy(args.slice(0, from));
+    const owned = this.callbackOperations.has(operation) ? /* @__PURE__ */ new Map() : void 0;
     const captured = [];
     const rollback = () => {
       for (const reference of captured) {
         revokeGuestReference(reference, this);
         this.guestReferences.delete(reference);
       }
+      for (const callback of owned?.values() ?? []) {
+        revokeGuestCallback(callback, this);
+        this.callbacks.delete(callback);
+      }
     };
     try {
+      const values = copy(args.slice(0, from), owned === void 0 ? void 0 : (closure) => {
+        let callback = owned.get(closure);
+        if (callback === void 0) {
+          callback = this.wrapCallback(closure, true);
+          owned.set(closure, callback);
+        }
+        return callback;
+      });
       for (const value of args.slice(from)) {
         this.checkCollection(
           this.guestReferences.size + 1,
@@ -1053,9 +1066,9 @@ var RealmState = class {
     }
     return object;
   };
-  wrapCallback = (closure) => {
+  wrapCallback = (closure, owned = false) => {
     this.assertOpen();
-    const existing = this.callbackCache.get(closure);
+    const existing = owned ? void 0 : this.callbackCache.get(closure);
     if (existing !== void 0 && this.callbacks.has(existing)) return existing;
     this.checkCollection(this.callbacks.size + 1, this.limits.callbacks, "callback");
     const invokeCallback = this.invokeCallback;
@@ -1063,7 +1076,7 @@ var RealmState = class {
       return invokeCallback(callback, { args, thisValue: this });
     };
     this.callbacks.set(callback, closure);
-    this.callbackCache.set(closure, callback);
+    if (!owned) this.callbackCache.set(closure, callback);
     registerGuestCallback(callback, {
       owner: this,
       closure,
@@ -1221,6 +1234,20 @@ var RealmState = class {
         invokeCallback: this.invokeCallback,
         releaseCallback: this.releaseCallback,
         releaseGuestReference: this.releaseGuestReference,
+        retainCallbackArguments: (operation) => {
+          this.assertOpen();
+          if (!extension.manifest.capabilities?.includes("guest:retain"))
+            throw new TypeError("Retaining callbacks requires the guest:retain grant.");
+          if (typeof operation !== "function")
+            throw new TypeError("Retained callback operation must be a function.");
+          if (this.scope !== void 0)
+            throw new TypeError("Retained callback operations must be registered during setup.");
+          const owner = this.callbackOperations.get(operation);
+          if (owner !== void 0 && owner !== extension)
+            throw new TypeError("Retained callback operation already belongs to another extension.");
+          this.callbackOperations.set(operation, extension);
+          return operation;
+        },
         retainGuestArguments: (operation, from) => {
           this.assertOpen();
           if (!extension.manifest.capabilities?.includes("guest:retain"))
@@ -2798,4 +2825,4 @@ export {
   UnhandledRejectionError,
   run
 };
-//# sourceMappingURL=chunk-M6HUSN7X.js.map
+//# sourceMappingURL=chunk-PKRGP34X.js.map