@metamask/snaps-execution-environments 6.7.1 → 6.8.0

package/dist/common/BaseSnapExecutor.cjs

@@ -1,10 +1,4 @@
 "use strict";
-var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (receiver, state, kind, f) {
-    if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a getter");
-    if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot read private member from an object whose class did not declare it");
-    return kind === "m" ? f : kind === "a" ? f.call(receiver) : f ? f.value : state.get(receiver);
-};
-var _BaseSnapExecutor_instances, _BaseSnapExecutor_write, _BaseSnapExecutor_notify, _BaseSnapExecutor_respond;
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.BaseSnapExecutor = void 0;
 // eslint-disable-next-line @typescript-eslint/triple-slash-reference, spaced-comment
@@ -56,9 +50,14 @@ const EXECUTION_ENVIRONMENT_METHODS = {
     },
 };
 class BaseSnapExecutor {
+    snapData;
+    commandStream;
+    rpcStream;
+    methods;
+    snapErrorHandler;
+    snapPromiseErrorHandler;
+    lastTeardown = 0;
     constructor(commandStream, rpcStream) {
-        _BaseSnapExecutor_instances.add(this);
-        this.lastTeardown = 0;
         this.snapData = new Map();
         this.commandStream = commandStream;
         this.commandStream.on('data', (data) => {
@@ -102,7 +101,7 @@ class BaseSnapExecutor {
             shouldIncludeStack: false,
         });
         const errorData = (0, snaps_sdk_1.getErrorData)(serializedError);
-        __classPrivateFieldGet(this, _BaseSnapExecutor_instances, "m", _BaseSnapExecutor_notify).call(this, {
+        this.#notify({
             method: 'UnhandledError',
             params: {
                 error: {
@@ -123,7 +122,7 @@ class BaseSnapExecutor {
                 (0, superstruct_1.is)(message.id, utils_1.JsonRpcIdStruct)) {
                 // Instead of throwing, we directly respond with an error.
                 // We can only do this if the message ID is still valid.
-                await __classPrivateFieldGet(this, _BaseSnapExecutor_instances, "m", _BaseSnapExecutor_write).call(this, {
+                await this.#write({
                     error: (0, rpc_errors_1.serializeError)(rpc_errors_1.rpcErrors.internal('JSON-RPC requests must be JSON serializable objects.')),
                     id: message.id,
                     jsonrpc: '2.0',
@@ -136,7 +135,7 @@ class BaseSnapExecutor {
         }
         const { id, method, params } = message;
         if (!(0, utils_1.hasProperty)(EXECUTION_ENVIRONMENT_METHODS, method)) {
-            await __classPrivateFieldGet(this, _BaseSnapExecutor_instances, "m", _BaseSnapExecutor_respond).call(this, id, {
+            await this.#respond(id, {
                 error: rpc_errors_1.rpcErrors
                     .methodNotFound({
                     data: {
@@ -152,7 +151,7 @@ class BaseSnapExecutor {
         const paramsAsArray = (0, sortParams_1.sortParamKeys)(methodObject.params, params);
         const [error] = (0, superstruct_1.validate)(paramsAsArray, methodObject.struct);
         if (error) {
-            await __classPrivateFieldGet(this, _BaseSnapExecutor_instances, "m", _BaseSnapExecutor_respond).call(this, id, {
+            await this.#respond(id, {
                 error: rpc_errors_1.rpcErrors
                     .invalidParams({
                     message: `Invalid parameters for method "${method}": ${error.message}.`,
@@ -167,16 +166,56 @@ class BaseSnapExecutor {
         }
         try {
             const result = await this.methods[method](...paramsAsArray);
-            await __classPrivateFieldGet(this, _BaseSnapExecutor_instances, "m", _BaseSnapExecutor_respond).call(this, id, { result });
+            await this.#respond(id, { result });
         }
         catch (rpcError) {
-            await __classPrivateFieldGet(this, _BaseSnapExecutor_instances, "m", _BaseSnapExecutor_respond).call(this, id, {
+            await this.#respond(id, {
                 error: (0, rpc_errors_1.serializeError)(rpcError, {
                     fallbackError,
                 }),
             });
         }
     }
+    // Awaitable function that writes back to the command stream
+    // To prevent snap execution from blocking writing we wrap in a promise
+    // and await it before continuing execution
+    async #write(chunk) {
+        return new Promise((resolve, reject) => {
+            this.commandStream.write(chunk, (error) => {
+                if (error) {
+                    reject(error);
+                    return;
+                }
+                resolve();
+            });
+        });
+    }
+    async #notify(notification) {
+        if (!(0, utils_2.isValidResponse)(notification)) {
+            throw rpc_errors_1.rpcErrors.internal('JSON-RPC notifications must be JSON serializable objects smaller than 64 MB.');
+        }
+        await this.#write({
+            ...notification,
+            jsonrpc: '2.0',
+        });
+    }
+    async #respond(id, response) {
+        if (!(0, utils_2.isValidResponse)(response)) {
+            // Instead of throwing, we directly respond with an error.
+            // This prevents an issue where we wouldn't respond when errors were non-serializable
+            await this.#write({
+                error: (0, rpc_errors_1.serializeError)(rpc_errors_1.rpcErrors.internal('JSON-RPC responses must be JSON serializable objects smaller than 64 MB.')),
+                id,
+                jsonrpc: '2.0',
+            });
+            return;
+        }
+        await this.#write({
+            ...response,
+            id,
+            jsonrpc: '2.0',
+        });
+    }
     /**
      * Attempts to evaluate a snap in SES. Generates APIs for the snap. May throw
      * on errors.
@@ -216,7 +255,7 @@ class BaseSnapExecutor {
                 ethereum,
                 snapId,
                 endowments: _endowments,
-                notify: __classPrivateFieldGet(this, _BaseSnapExecutor_instances, "m", _BaseSnapExecutor_notify).bind(this),
+                notify: this.#notify.bind(this),
             });
             // !!! Ensure that this is the only place the data is being set.
             // Other methods access the object value and mutate its properties.
@@ -296,7 +335,7 @@ class BaseSnapExecutor {
             const sanitizedArgs = (0, utils_2.sanitizeRequestArguments)(args);
             (0, utils_2.assertSnapOutboundRequest)(sanitizedArgs);
             return await (0, utils_2.withTeardown)((async () => {
-                await __classPrivateFieldGet(this, _BaseSnapExecutor_instances, "m", _BaseSnapExecutor_notify).call(this, {
+                await this.#notify({
                     method: 'OutboundRequest',
                     params: { source: 'snap.request' },
                 });
@@ -304,7 +343,7 @@ class BaseSnapExecutor {
                     return await originalRequest(sanitizedArgs);
                 }
                 finally {
-                    await __classPrivateFieldGet(this, _BaseSnapExecutor_instances, "m", _BaseSnapExecutor_notify).call(this, {
+                    await this.#notify({
                         method: 'OutboundResponse',
                         params: { source: 'snap.request' },
                     });
@@ -326,7 +365,7 @@ class BaseSnapExecutor {
             const sanitizedArgs = (0, utils_2.sanitizeRequestArguments)(args);
             (0, utils_2.assertEthereumOutboundRequest)(sanitizedArgs);
             return await (0, utils_2.withTeardown)((async () => {
-                await __classPrivateFieldGet(this, _BaseSnapExecutor_instances, "m", _BaseSnapExecutor_notify).call(this, {
+                await this.#notify({
                     method: 'OutboundRequest',
                     params: { source: 'ethereum.request' },
                 });
@@ -334,7 +373,7 @@ class BaseSnapExecutor {
                     return await originalRequest(sanitizedArgs);
                 }
                 finally {
-                    await __classPrivateFieldGet(this, _BaseSnapExecutor_instances, "m", _BaseSnapExecutor_notify).call(this, {
+                    await this.#notify({
                         method: 'OutboundResponse',
                         params: { source: 'ethereum.request' },
                     });
@@ -394,43 +433,4 @@ class BaseSnapExecutor {
     }
 }
 exports.BaseSnapExecutor = BaseSnapExecutor;
-_BaseSnapExecutor_instances = new WeakSet(), _BaseSnapExecutor_write = 
-// Awaitable function that writes back to the command stream
-// To prevent snap execution from blocking writing we wrap in a promise
-// and await it before continuing execution
-async function _BaseSnapExecutor_write(chunk) {
-    return new Promise((resolve, reject) => {
-        this.commandStream.write(chunk, (error) => {
-            if (error) {
-                reject(error);
-                return;
-            }
-            resolve();
-        });
-    });
-}, _BaseSnapExecutor_notify = async function _BaseSnapExecutor_notify(notification) {
-    if (!(0, utils_2.isValidResponse)(notification)) {
-        throw rpc_errors_1.rpcErrors.internal('JSON-RPC notifications must be JSON serializable objects smaller than 64 MB.');
-    }
-    await __classPrivateFieldGet(this, _BaseSnapExecutor_instances, "m", _BaseSnapExecutor_write).call(this, {
-        ...notification,
-        jsonrpc: '2.0',
-    });
-}, _BaseSnapExecutor_respond = async function _BaseSnapExecutor_respond(id, response) {
-    if (!(0, utils_2.isValidResponse)(response)) {
-        // Instead of throwing, we directly respond with an error.
-        // This prevents an issue where we wouldn't respond when errors were non-serializable
-        await __classPrivateFieldGet(this, _BaseSnapExecutor_instances, "m", _BaseSnapExecutor_write).call(this, {
-            error: (0, rpc_errors_1.serializeError)(rpc_errors_1.rpcErrors.internal('JSON-RPC responses must be JSON serializable objects smaller than 64 MB.')),
-            id,
-            jsonrpc: '2.0',
-        });
-        return;
-    }
-    await __classPrivateFieldGet(this, _BaseSnapExecutor_instances, "m", _BaseSnapExecutor_write).call(this, {
-        ...response,
-        id,
-        jsonrpc: '2.0',
-    });
-};
 //# sourceMappingURL=BaseSnapExecutor.cjs.map

package/dist/common/BaseSnapExecutor.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"BaseSnapExecutor.cjs","sourceRoot":"","sources":["../../src/common/BaseSnapExecutor.ts"],"names":[],"mappings":";;;;;;;;;AAAA,qFAAqF;AACrF,gEAAgE;AAChE,+DAAoE;AAEpE,yEAAqE;AACrE,qDAA6E;AAE7E,mDAAmD;AAMnD,uDAO+B;AAC/B,uDAAqD;AAOrD,2CAMyB;AAGzB,4CAAiC;AAEjC,6CAA6D;AAC7D,uDAAgD;AAChD,qDAAuE;AACvE,iDAA6C;AAC7C,uCAOiB;AACjB,iDAKsB;AAYtB,MAAM,aAAa,GAAG;IACpB,IAAI,EAAE,uBAAU,CAAC,GAAG,CAAC,QAAQ;IAC7B,OAAO,EAAE,6BAA6B;CACvC,CAAC;AAEF,MAAM,cAAc,GAAG,sBAAS;KAC7B,QAAQ,CAAO;IACd,OAAO,EAAE,sBAAsB;CAChC,CAAC;KACD,SAAS,EAAE,CAAC;AAUf;;;;GAIG;AACH,MAAM,6BAA6B,GAAG;IACpC,IAAI,EAAE;QACJ,MAAM,EAAE,uCAA0B;QAClC,MAAM,EAAE,EAAE;KACX;IACD,WAAW,EAAE;QACX,MAAM,EAAE,8CAAiC;QACzC,MAAM,EAAE,CAAC,QAAQ,EAAE,YAAY,EAAE,YAAY,CAAC;KAC/C;IACD,SAAS,EAAE;QACT,MAAM,EAAE,4CAA+B;QACvC,MAAM,EAAE,EAAE;KACX;IACD,OAAO,EAAE;QACP,MAAM,EAAE,0CAA6B;QACrC,MAAM,EAAE,CAAC,QAAQ,EAAE,SAAS,EAAE,QAAQ,EAAE,SAAS,CAAC;KACnD;CACF,CAAC;AAQF,MAAa,gBAAgB;IAe3B,YAAsB,aAAqB,EAAE,SAAiB;;QAFtD,iBAAY,GAAG,CAAC,CAAC;QAGvB,IAAI,CAAC,QAAQ,GAAG,IAAI,GAAG,EAAE,CAAC;QAC1B,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;QACnC,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE;YACrC,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBAC1C,qCAAqC;gBACrC,IAAA,sBAAQ,EAAC,KAAK,CAAC,CAAC;YAClB,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAE3B,IAAI,CAAC,OAAO,GAAG,IAAA,0CAA+B,EAC5C,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,EACzB,KAAK,EAAE,MAAM,EAAE,WAAW,EAAE,IAAI,EAAE,EAAE;YAClC,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YACvC,uEAAuE;YACvE,mBAAmB;YACnB,MAAM,OAAO,GAAG,IAAI,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;YAC3C,MAAM,EAAE,QAAQ,EAAE,GAAG,0BAAY,CAAC,WAAW,CAAC,CAAC;YAE/C,IAAA,cAAM,EACJ,CAAC,QAAQ,IAAI,OAAO,KAAK,SAAS,EAClC,MAAM,WAAW,+BAA+B,MAAM,EAAE,EACxD,sBAAS,CAAC,kBAAkB,CAC7B,CAAC;YAEF,kEAAkE;YAClE,eAAe;YACf,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,OAAO,IAAI,CAAC;YACd,CAAC;YAED,IAAI,MAAM,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,MAAM,EAAE,GAAG,EAAE;YACxD,mCAAmC;YACnC,OAAO,CAAC,IAAW,CAAC,CACrB,CAAC;YAEF,0EAA0E;YAC1E,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;gBACzB,MAAM,GAAG,IAAI,CAAC;YAChB,CAAC;YAED,uEAAuE;YACvE,IAAI,CAAC;gBACH,OAAO,IAAA,mBAAW,EAAC,MAAM,CAAC,CAAC;YAC7B,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,sBAAS,CAAC,QAAQ,CACtB,yCAAyC,KAAK,CAAC,OAAO,CAAC,OAAO,CAC5D,sBAAsB,EACtB,EAAE,CACH,EAAE,CACJ,CAAC;YACJ,CAAC;QACH,CAAC,EACD,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAC5B,CAAC;IACJ,CAAC;IAEO,YAAY,CAAC,KAAc,EAAE,IAA0B;QAC7D,MAAM,eAAe,GAAG,IAAA,2BAAc,EAAC,KAAK,EAAE;YAC5C,aAAa,EAAE,cAAc;YAC7B,kBAAkB,EAAE,KAAK;SAC1B,CAAC,CAAC;QAEH,MAAM,SAAS,GAAG,IAAA,wBAAY,EAAC,eAAe,CAAC,CAAC;QAEhD,uBAAA,IAAI,6DAAQ,MAAZ,IAAI,EAAS;YACX,MAAM,EAAE,gBAAgB;YACxB,MAAM,EAAE;gBACN,KAAK,EAAE;oBACL,GAAG,eAAe;oBAClB,IAAI,EAAE;wBACJ,GAAG,SAAS;wBACZ,GAAG,IAAI;qBACR;iBACF;aACF;SACF,CAAC,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,EAAE;YACvB,IAAA,sBAAQ,EAAC,WAAW,CAAC,CAAC;QACxB,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAAC,OAAuB;QACpD,IAAI,CAAC,IAAA,wBAAgB,EAAC,OAAO,CAAC,EAAE,CAAC;YAC/B,IACE,IAAA,mBAAW,EAAC,OAAO,EAAE,IAAI,CAAC;gBAC1B,IAAA,gBAAE,EAAE,OAAsC,CAAC,EAAE,EAAE,uBAAe,CAAC,EAC/D,CAAC;gBACD,0DAA0D;gBAC1D,wDAAwD;gBACxD,MAAM,uBAAA,IAAI,4DAAO,MAAX,IAAI,EAAQ;oBAChB,KAAK,EAAE,IAAA,2BAAc,EACnB,sBAAS,CAAC,QAAQ,CAChB,sDAAsD,CACvD,CACF;oBACD,EAAE,EAAG,OAAsC,CAAC,EAAE;oBAC9C,OAAO,EAAE,KAAK;iBACf,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,IAAA,qBAAO,EACL,4EAA4E,CAC7E,CAAC;YACJ,CAAC;YACD,OAAO;QACT,CAAC;QAED,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC;QAEvC,IAAI,CAAC,IAAA,mBAAW,EAAC,6BAA6B,EAAE,MAAM,CAAC,EAAE,CAAC;YACxD,MAAM,uBAAA,IAAI,8DAAS,MAAb,IAAI,EAAU,EAAE,EAAE;gBACtB,KAAK,EAAE,sBAAS;qBACb,cAAc,CAAC;oBACd,IAAI,EAAE;wBACJ,MAAM;qBACP;iBACF,CAAC;qBACD,SAAS,EAAE;aACf,CAAC,CAAC;YACH,OAAO;QACT,CAAC;QAED,MAAM,YAAY,GAAG,6BAA6B,CAAC,MAAuB,CAAC,CAAC;QAE5E,yCAAyC;QACzC,MAAM,aAAa,GAAG,IAAA,0BAAa,EAAC,YAAY,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAEjE,MAAM,CAAC,KAAK,CAAC,GAAG,IAAA,sBAAQ,EAAW,aAAa,EAAE,YAAY,CAAC,MAAM,CAAC,CAAC;QACvE,IAAI,KAAK,EAAE,CAAC;YACV,MAAM,uBAAA,IAAI,8DAAS,MAAb,IAAI,EAAU,EAAE,EAAE;gBACtB,KAAK,EAAE,sBAAS;qBACb,aAAa,CAAC;oBACb,OAAO,EAAE,kCAAkC,MAAM,MAAM,KAAK,CAAC,OAAO,GAAG;oBACvE,IAAI,EAAE;wBACJ,MAAM;wBACN,MAAM,EAAE,aAAa;qBACtB;iBACF,CAAC;qBACD,SAAS,EAAE;aACf,CAAC,CAAC;YACH,OAAO;QACT,CAAC;QAED,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAO,IAAI,CAAC,OAAe,CAAC,MAAM,CAAC,CAAC,GAAG,aAAa,CAAC,CAAC;YACrE,MAAM,uBAAA,IAAI,8DAAS,MAAb,IAAI,EAAU,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;QACtC,CAAC;QAAC,OAAO,QAAQ,EAAE,CAAC;YAClB,MAAM,uBAAA,IAAI,8DAAS,MAAb,IAAI,EAAU,EAAE,EAAE;gBACtB,KAAK,EAAE,IAAA,2BAAc,EAAC,QAAQ,EAAE;oBAC9B,aAAa;iBACd,CAAC;aACH,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAqDD;;;;;;;OAOG;IACO,KAAK,CAAC,SAAS,CACvB,MAAc,EACd,UAAkB,EAClB,WAAqB;QAErB,IAAA,aAAG,EAAC,kBAAkB,MAAM,cAAc,CAAC,CAAC;QAC5C,IAAI,IAAI,CAAC,uBAAuB,EAAE,CAAC;YACjC,IAAA,kCAAmB,EAAC,oBAAoB,EAAE,IAAI,CAAC,uBAAuB,CAAC,CAAC;QAC1E,CAAC;QAED,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC1B,IAAA,kCAAmB,EAAC,OAAO,EAAE,IAAI,CAAC,gBAAgB,CAAC,CAAC;QACtD,CAAC;QAED,IAAI,CAAC,gBAAgB,GAAG,CAAC,KAAiB,EAAE,EAAE;YAC5C,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;QAC7C,CAAC,CAAC;QAEF,IAAI,CAAC,uBAAuB,GAAG,CAAC,KAA4B,EAAE,EAAE;YAC9D,IAAI,CAAC,YAAY,CAAC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,EAAE;gBAC/D,MAAM;aACP,CAAC,CAAC;QACL,CAAC,CAAC;QAEF,MAAM,QAAQ,GAAG,IAAI,gCAAc,CAAC,IAAI,CAAC,SAAS,EAAE;YAClD,iBAAiB,EAAE,mBAAmB;YACtC,aAAa,EAAE,CAAC,IAAA,yCAAuB,GAAE,CAAC;SAC3C,CAAC,CAAC;QAEH,MAAM,QAAQ,CAAC,UAAU,EAAE,CAAC;QAE5B,MAAM,IAAI,GAAG,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QAC7C,MAAM,QAAQ,GAAG,IAAI,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC;QACtD,wFAAwF;QACxF,MAAM,UAAU,GAAQ,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;QAExC,IAAI,CAAC;YACH,MAAM,EAAE,UAAU,EAAE,QAAQ,EAAE,iBAAiB,EAAE,GAAG,IAAA,6BAAgB,EAAC;gBACnE,IAAI;gBACJ,QAAQ;gBACR,MAAM;gBACN,UAAU,EAAE,WAAW;gBACvB,MAAM,EAAE,uBAAA,IAAI,6DAAQ,CAAC,IAAI,CAAC,IAAI,CAAC;aAChC,CAAC,CAAC;YAEH,gEAAgE;YAChE,mEAAmE;YACnE,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,EAAE;gBACxB,YAAY,EAAE,iBAAiB;gBAC/B,kBAAkB,EAAE,IAAI,GAAG,EAAE;gBAC7B,OAAO,EAAE,EAAE;aACZ,CAAC,CAAC;YAEH,IAAA,+BAAgB,EAAC,oBAAoB,EAAE,IAAI,CAAC,uBAAuB,CAAC,CAAC;YACrE,IAAA,+BAAgB,EAAC,OAAO,EAAE,IAAI,CAAC,gBAAgB,CAAC,CAAC;YAEjD,MAAM,WAAW,GAAG,IAAI,WAAW,CAAC;gBAClC,GAAG,UAAU;gBACb,MAAM,EAAE,UAAU;gBAClB,OAAO,EAAE,UAAU,CAAC,OAAO;aAC5B,CAAC,CAAC;YAEH,qEAAqE;YACrE,kEAAkE;YAClE,EAAE;YACF,wDAAwD;YACxD,yCAAyC;YACzC,WAAW,CAAC,UAAU,CAAC,IAAI,GAAG,WAAW,CAAC,UAAU,CAAC;YACrD,WAAW,CAAC,UAAU,CAAC,MAAM,GAAG,WAAW,CAAC,UAAU,CAAC;YACvD,WAAW,CAAC,UAAU,CAAC,MAAM,GAAG,WAAW,CAAC,UAAU,CAAC;YAEvD,MAAM,IAAI,CAAC,oBAAoB,CAAC,MAAM,EAAE,GAAG,EAAE;gBAC3C,WAAW,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;gBACjC,IAAI,CAAC,mBAAmB,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;YAC/C,CAAC,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;YAExB,MAAM,CAAC,KAAK,CAAC,GAAG,IAAA,yBAAW,EAAC,KAAK,CAAC,CAAC;YACnC,MAAM,sBAAS,CAAC,QAAQ,CAAC;gBACvB,OAAO,EAAE,6BAA6B,MAAM,MAAM,KAAK,CAAC,OAAO,EAAE;gBACjE,IAAI,EAAE;oBACJ,KAAK,EAAE,KAAK,CAAC,SAAS,EAAE;iBACzB;aACF,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED;;;OAGG;IACO,WAAW;QACnB,uCAAuC;QACvC,+DAA+D;QAC/D,6CAA6C;QAC7C,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAC7B,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC,CACnE,CAAC;QACF,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;IACxB,CAAC;IAEO,mBAAmB,CAAC,MAAc,EAAE,UAAe;QACzD,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACvC,sDAAsD;QACtD,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,OAAO;QACT,CAAC;QAED,IAAI,CAAC,OAAO,GAAG,+BAAiB,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,UAAU,EAAE,EAAE;YAC1D,MAAM,UAAU,GAAG,UAAU,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;YAClD,MAAM,EAAE,SAAS,EAAE,GAAG,0BAAY,CAAC,UAAU,CAAC,CAAC;YAC/C,IAAI,SAAS,CAAC,UAAU,CAAC,EAAE,CAAC;gBAC1B,OAAO,EAAE,GAAG,GAAG,EAAE,CAAC,UAAU,CAAC,EAAE,UAAU,EAAE,CAAC;YAC9C,CAAC;YACD,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,EAAE,CAAC,CAAC;QAEP,qDAAqD;QACrD,IAAA,cAAM,EAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,4BAA4B,CAAC,CAAC;IAC7E,CAAC;IAED;;;;;OAKG;IACK,gBAAgB,CAAC,QAAwB;QAC/C,MAAM,eAAe,GAAG,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAExD,MAAM,OAAO,GAAG,KAAK,EAAE,IAAsB,EAAE,EAAE;YAC/C,MAAM,aAAa,GAAG,IAAA,gCAAwB,EAAC,IAAI,CAAC,CAAC;YACrD,IAAA,iCAAyB,EAAC,aAAa,CAAC,CAAC;YACzC,OAAO,MAAM,IAAA,oBAAY,EACvB,CAAC,KAAK,IAAI,EAAE;gBACV,MAAM,uBAAA,IAAI,6DAAQ,MAAZ,IAAI,EAAS;oBACjB,MAAM,EAAE,iBAAiB;oBACzB,MAAM,EAAE,EAAE,MAAM,EAAE,cAAc,EAAE;iBACnC,CAAC,CAAC;gBACH,IAAI,CAAC;oBACH,OAAO,MAAM,eAAe,CAAC,aAAa,CAAC,CAAC;gBAC9C,CAAC;wBAAS,CAAC;oBACT,MAAM,uBAAA,IAAI,6DAAQ,MAAZ,IAAI,EAAS;wBACjB,MAAM,EAAE,kBAAkB;wBAC1B,MAAM,EAAE,EAAE,MAAM,EAAE,cAAc,EAAE;qBACnC,CAAC,CAAC;gBACL,CAAC;YACH,CAAC,CAAC,EAAE,EACJ,IAAW,CACZ,CAAC;QACJ,CAAC,CAAC;QAEF,MAAM,eAAe,GAAG,IAAA,2BAAmB,EAAC,OAAO,CAAkB,CAAC;QAEtE,OAAO,MAAM,CAAC,eAAe,CAAC,CAAC;IACjC,CAAC;IAED;;;;;OAKG;IACK,qBAAqB,CAAC,QAAwB;QACpD,MAAM,eAAe,GAAG,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAExD,MAAM,OAAO,GAAG,KAAK,EAAE,IAAsB,EAAE,EAAE;YAC/C,MAAM,aAAa,GAAG,IAAA,gCAAwB,EAAC,IAAI,CAAC,CAAC;YACrD,IAAA,qCAA6B,EAAC,aAAa,CAAC,CAAC;YAC7C,OAAO,MAAM,IAAA,oBAAY,EACvB,CAAC,KAAK,IAAI,EAAE;gBACV,MAAM,uBAAA,IAAI,6DAAQ,MAAZ,IAAI,EAAS;oBACjB,MAAM,EAAE,iBAAiB;oBACzB,MAAM,EAAE,EAAE,MAAM,EAAE,kBAAkB,EAAE;iBACvC,CAAC,CAAC;gBACH,IAAI,CAAC;oBACH,OAAO,MAAM,eAAe,CAAC,aAAa,CAAC,CAAC;gBAC9C,CAAC;wBAAS,CAAC;oBACT,MAAM,uBAAA,IAAI,6DAAQ,MAAZ,IAAI,EAAS;wBACjB,MAAM,EAAE,kBAAkB;wBAC1B,MAAM,EAAE,EAAE,MAAM,EAAE,kBAAkB,EAAE;qBACvC,CAAC,CAAC;gBACL,CAAC;YACH,CAAC,CAAC,EAAE,EACJ,IAAW,CACZ,CAAC;QACJ,CAAC,CAAC;QAEF,MAAM,mBAAmB,GAAG,IAAA,2BAAmB,EAAC,OAAO,CAAC,CAAC;QAEzD,OAAO,MAAM,CAAC,mBAAmB,CAAC,CAAC;IACrC,CAAC;IAED;;;;OAIG;IACK,UAAU,CAAC,MAAc;QAC/B,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IAC/B,CAAC;IAED;;;;;;;;;;OAUG;IACK,KAAK,CAAC,oBAAoB,CAChC,MAAc,EACd,QAAwC;QAExC,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACvC,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;YACvB,MAAM,sBAAS,CAAC,QAAQ,CACtB,iDAAiD,MAAM,IAAI,CAC5D,CAAC;QACJ,CAAC;QAED,IAAI,IAAgB,CAAC;QACrB,MAAM,WAAW,GAAG,IAAI,OAAO,CAC7B,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,CACZ,CAAC,IAAI,GAAG,GAAG,EAAE,CACX,MAAM;QACJ,kEAAkE;QAClE,sBAAS,CAAC,QAAQ,CAChB,aAAa,MAAM,yCAAyC,CAC7D,CACF,CAAC,CACP,CAAC;QAEF,oEAAoE;QACpE,MAAM,cAAc,GAAG,EAAE,IAAI,EAAE,IAAK,EAAE,CAAC;QAEvC,IAAI,CAAC;YACH,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;YAC5C,8CAA8C;YAC9C,oEAAoE;YACpE,uCAAuC;YACvC,OAAO,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,EAAE,WAAW,CAAC,CAAC,CAAC;QACvD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,8BAAgB,CAAC,KAAK,CAAC,CAAC;QACpC,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;YAE/C,IAAI,IAAI,CAAC,kBAAkB,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;gBACvC,IAAI,CAAC,YAAY,IAAI,CAAC,CAAC;gBACvB,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;YAC5B,CAAC;QACH,CAAC;IACH,CAAC;CACF;AAteD,4CAseC;;AA5TC,4DAA4D;AAC5D,uEAAuE;AACvE,2CAA2C;AAC3C,KAAK,kCAAQ,KAAW;IACtB,OAAO,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QAC3C,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;YACxC,IAAI,KAAK,EAAE,CAAC;gBACV,MAAM,CAAC,KAAK,CAAC,CAAC;gBACd,OAAO;YACT,CAAC;YACD,OAAO,EAAE,CAAC;QACZ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,6BAED,KAAK,mCAAS,YAAkD;IAC9D,IAAI,CAAC,IAAA,uBAAe,EAAC,YAAY,CAAC,EAAE,CAAC;QACnC,MAAM,sBAAS,CAAC,QAAQ,CACtB,8EAA8E,CAC/E,CAAC;IACJ,CAAC;IAED,MAAM,uBAAA,IAAI,4DAAO,MAAX,IAAI,EAAQ;QAChB,GAAG,YAAY;QACf,OAAO,EAAE,KAAK;KACf,CAAC,CAAC;AACL,CAAC,8BAED,KAAK,oCAAU,EAAa,EAAE,QAAiC;IAC7D,IAAI,CAAC,IAAA,uBAAe,EAAC,QAAQ,CAAC,EAAE,CAAC;QAC/B,0DAA0D;QAC1D,qFAAqF;QACrF,MAAM,uBAAA,IAAI,4DAAO,MAAX,IAAI,EAAQ;YAChB,KAAK,EAAE,IAAA,2BAAc,EACnB,sBAAS,CAAC,QAAQ,CAChB,0EAA0E,CAC3E,CACF;YACD,EAAE;YACF,OAAO,EAAE,KAAK;SACf,CAAC,CAAC;QACH,OAAO;IACT,CAAC;IAED,MAAM,uBAAA,IAAI,4DAAO,MAAX,IAAI,EAAQ;QAChB,GAAG,QAAQ;QACX,EAAE;QACF,OAAO,EAAE,KAAK;KACf,CAAC,CAAC;AACL,CAAC","sourcesContent":["// eslint-disable-next-line @typescript-eslint/triple-slash-reference, spaced-comment\n/// <reference path=\"../../../../node_modules/ses/types.d.ts\" />\nimport { createIdRemapMiddleware } from '@metamask/json-rpc-engine';\nimport type { RequestArguments } from '@metamask/providers';\nimport { StreamProvider } from '@metamask/providers/stream-provider';\nimport { errorCodes, rpcErrors, serializeError } from '@metamask/rpc-errors';\nimport type { SnapsProvider } from '@metamask/snaps-sdk';\nimport { getErrorData } from '@metamask/snaps-sdk';\nimport type {\n  SnapExports,\n  HandlerType,\n  SnapExportsParameters,\n} from '@metamask/snaps-utils';\nimport {\n  SNAP_EXPORT_NAMES,\n  logError,\n  SNAP_EXPORTS,\n  WrappedSnapError,\n  unwrapError,\n  logInfo,\n} from '@metamask/snaps-utils';\nimport { validate, is } from '@metamask/superstruct';\nimport type {\n  JsonRpcNotification,\n  JsonRpcId,\n  JsonRpcRequest,\n  Json,\n} from '@metamask/utils';\nimport {\n  assert,\n  isJsonRpcRequest,\n  hasProperty,\n  getSafeJson,\n  JsonRpcIdStruct,\n} from '@metamask/utils';\nimport type { Duplex } from 'readable-stream';\n\nimport { log } from '../logging';\nimport type { CommandMethodsMapping } from './commands';\nimport { getCommandMethodImplementations } from './commands';\nimport { createEndowments } from './endowments';\nimport { addEventListener, removeEventListener } from './globalEvents';\nimport { sortParamKeys } from './sortParams';\nimport {\n  assertEthereumOutboundRequest,\n  assertSnapOutboundRequest,\n  sanitizeRequestArguments,\n  proxyStreamProvider,\n  withTeardown,\n  isValidResponse,\n} from './utils';\nimport {\n  ExecuteSnapRequestArgumentsStruct,\n  PingRequestArgumentsStruct,\n  SnapRpcRequestArgumentsStruct,\n  TerminateRequestArgumentsStruct,\n} from './validation';\n\ntype EvaluationData = {\n  stop: () => void;\n};\n\ntype SnapData = {\n  exports: SnapExports;\n  runningEvaluations: Set<EvaluationData>;\n  idleTeardown: () => Promise<void>;\n};\n\nconst fallbackError = {\n  code: errorCodes.rpc.internal,\n  message: 'Execution Environment Error',\n};\n\nconst unhandledError = rpcErrors\n  .internal<Json>({\n    message: 'Unhandled Snap Error',\n  })\n  .serialize();\n\nexport type InvokeSnapArgs = Omit<SnapExportsParameters[0], 'chainId'>;\n\nexport type InvokeSnap = (\n  target: string,\n  handler: HandlerType,\n  args: InvokeSnapArgs | undefined,\n) => Promise<Json>;\n\n/**\n * The supported methods in the execution environment. The validator checks the\n * incoming JSON-RPC request, and the `params` property is used for sorting the\n * parameters, if they are an object.\n */\nconst EXECUTION_ENVIRONMENT_METHODS = {\n  ping: {\n    struct: PingRequestArgumentsStruct,\n    params: [],\n  },\n  executeSnap: {\n    struct: ExecuteSnapRequestArgumentsStruct,\n    params: ['snapId', 'sourceCode', 'endowments'],\n  },\n  terminate: {\n    struct: TerminateRequestArgumentsStruct,\n    params: [],\n  },\n  snapRpc: {\n    struct: SnapRpcRequestArgumentsStruct,\n    params: ['target', 'handler', 'origin', 'request'],\n  },\n};\n\ntype Methods = typeof EXECUTION_ENVIRONMENT_METHODS;\n\nexport type NotifyFunction = (\n  notification: Omit<JsonRpcNotification, 'jsonrpc'>,\n) => Promise<void>;\n\nexport class BaseSnapExecutor {\n  private readonly snapData: Map<string, SnapData>;\n\n  private readonly commandStream: Duplex;\n\n  private readonly rpcStream: Duplex;\n\n  private readonly methods: CommandMethodsMapping;\n\n  private snapErrorHandler?: (event: ErrorEvent) => void;\n\n  private snapPromiseErrorHandler?: (event: PromiseRejectionEvent) => void;\n\n  private lastTeardown = 0;\n\n  protected constructor(commandStream: Duplex, rpcStream: Duplex) {\n    this.snapData = new Map();\n    this.commandStream = commandStream;\n    this.commandStream.on('data', (data) => {\n      this.onCommandRequest(data).catch((error) => {\n        // TODO: Decide how to handle errors.\n        logError(error);\n      });\n    });\n    this.rpcStream = rpcStream;\n\n    this.methods = getCommandMethodImplementations(\n      this.startSnap.bind(this),\n      async (target, handlerType, args) => {\n        const data = this.snapData.get(target);\n        // We're capturing the handler in case someone modifies the data object\n        // before the call.\n        const handler = data?.exports[handlerType];\n        const { required } = SNAP_EXPORTS[handlerType];\n\n        assert(\n          !required || handler !== undefined,\n          `No ${handlerType} handler exported for snap \"${target}`,\n          rpcErrors.methodNotSupported,\n        );\n\n        // Certain handlers are not required. If they are not exported, we\n        // return null.\n        if (!handler) {\n          return null;\n        }\n\n        let result = await this.executeInSnapContext(target, () =>\n          // TODO: fix handler args type cast\n          handler(args as any),\n        );\n\n        // The handler might not return anything, but undefined is not valid JSON.\n        if (result === undefined) {\n          result = null;\n        }\n\n        // /!\\ Always return only sanitized JSON to prevent security flaws. /!\\\n        try {\n          return getSafeJson(result);\n        } catch (error) {\n          throw rpcErrors.internal(\n            `Received non-JSON-serializable value: ${error.message.replace(\n              /^Assertion failed: /u,\n              '',\n            )}`,\n          );\n        }\n      },\n      this.onTerminate.bind(this),\n    );\n  }\n\n  private errorHandler(error: unknown, data: Record<string, Json>) {\n    const serializedError = serializeError(error, {\n      fallbackError: unhandledError,\n      shouldIncludeStack: false,\n    });\n\n    const errorData = getErrorData(serializedError);\n\n    this.#notify({\n      method: 'UnhandledError',\n      params: {\n        error: {\n          ...serializedError,\n          data: {\n            ...errorData,\n            ...data,\n          },\n        },\n      },\n    }).catch((notifyError) => {\n      logError(notifyError);\n    });\n  }\n\n  private async onCommandRequest(message: JsonRpcRequest) {\n    if (!isJsonRpcRequest(message)) {\n      if (\n        hasProperty(message, 'id') &&\n        is((message as Pick<JsonRpcRequest, 'id'>).id, JsonRpcIdStruct)\n      ) {\n        // Instead of throwing, we directly respond with an error.\n        // We can only do this if the message ID is still valid.\n        await this.#write({\n          error: serializeError(\n            rpcErrors.internal(\n              'JSON-RPC requests must be JSON serializable objects.',\n            ),\n          ),\n          id: (message as Pick<JsonRpcRequest, 'id'>).id,\n          jsonrpc: '2.0',\n        });\n      } else {\n        logInfo(\n          'Command stream received a non-JSON-RPC request, and was unable to respond.',\n        );\n      }\n      return;\n    }\n\n    const { id, method, params } = message;\n\n    if (!hasProperty(EXECUTION_ENVIRONMENT_METHODS, method)) {\n      await this.#respond(id, {\n        error: rpcErrors\n          .methodNotFound({\n            data: {\n              method,\n            },\n          })\n          .serialize(),\n      });\n      return;\n    }\n\n    const methodObject = EXECUTION_ENVIRONMENT_METHODS[method as keyof Methods];\n\n    // support params by-name and by-position\n    const paramsAsArray = sortParamKeys(methodObject.params, params);\n\n    const [error] = validate<any, any>(paramsAsArray, methodObject.struct);\n    if (error) {\n      await this.#respond(id, {\n        error: rpcErrors\n          .invalidParams({\n            message: `Invalid parameters for method \"${method}\": ${error.message}.`,\n            data: {\n              method,\n              params: paramsAsArray,\n            },\n          })\n          .serialize(),\n      });\n      return;\n    }\n\n    try {\n      const result = await (this.methods as any)[method](...paramsAsArray);\n      await this.#respond(id, { result });\n    } catch (rpcError) {\n      await this.#respond(id, {\n        error: serializeError(rpcError, {\n          fallbackError,\n        }),\n      });\n    }\n  }\n\n  // Awaitable function that writes back to the command stream\n  // To prevent snap execution from blocking writing we wrap in a promise\n  // and await it before continuing execution\n  async #write(chunk: Json) {\n    return new Promise<void>((resolve, reject) => {\n      this.commandStream.write(chunk, (error) => {\n        if (error) {\n          reject(error);\n          return;\n        }\n        resolve();\n      });\n    });\n  }\n\n  async #notify(notification: Omit<JsonRpcNotification, 'jsonrpc'>) {\n    if (!isValidResponse(notification)) {\n      throw rpcErrors.internal(\n        'JSON-RPC notifications must be JSON serializable objects smaller than 64 MB.',\n      );\n    }\n\n    await this.#write({\n      ...notification,\n      jsonrpc: '2.0',\n    });\n  }\n\n  async #respond(id: JsonRpcId, response: Record<string, unknown>) {\n    if (!isValidResponse(response)) {\n      // Instead of throwing, we directly respond with an error.\n      // This prevents an issue where we wouldn't respond when errors were non-serializable\n      await this.#write({\n        error: serializeError(\n          rpcErrors.internal(\n            'JSON-RPC responses must be JSON serializable objects smaller than 64 MB.',\n          ),\n        ),\n        id,\n        jsonrpc: '2.0',\n      });\n      return;\n    }\n\n    await this.#write({\n      ...response,\n      id,\n      jsonrpc: '2.0',\n    });\n  }\n\n  /**\n   * Attempts to evaluate a snap in SES. Generates APIs for the snap. May throw\n   * on errors.\n   *\n   * @param snapId - The id of the snap.\n   * @param sourceCode - The source code of the snap, in IIFE format.\n   * @param _endowments - An array of the names of the endowments.\n   */\n  protected async startSnap(\n    snapId: string,\n    sourceCode: string,\n    _endowments: string[],\n  ): Promise<void> {\n    log(`Starting snap '${snapId}' in worker.`);\n    if (this.snapPromiseErrorHandler) {\n      removeEventListener('unhandledrejection', this.snapPromiseErrorHandler);\n    }\n\n    if (this.snapErrorHandler) {\n      removeEventListener('error', this.snapErrorHandler);\n    }\n\n    this.snapErrorHandler = (error: ErrorEvent) => {\n      this.errorHandler(error.error, { snapId });\n    };\n\n    this.snapPromiseErrorHandler = (error: PromiseRejectionEvent) => {\n      this.errorHandler(error instanceof Error ? error : error.reason, {\n        snapId,\n      });\n    };\n\n    const provider = new StreamProvider(this.rpcStream, {\n      jsonRpcStreamName: 'metamask-provider',\n      rpcMiddleware: [createIdRemapMiddleware()],\n    });\n\n    await provider.initialize();\n\n    const snap = this.createSnapGlobal(provider);\n    const ethereum = this.createEIP1193Provider(provider);\n    // We specifically use any type because the Snap can modify the object any way they want\n    const snapModule: any = { exports: {} };\n\n    try {\n      const { endowments, teardown: endowmentTeardown } = createEndowments({\n        snap,\n        ethereum,\n        snapId,\n        endowments: _endowments,\n        notify: this.#notify.bind(this),\n      });\n\n      // !!! Ensure that this is the only place the data is being set.\n      // Other methods access the object value and mutate its properties.\n      this.snapData.set(snapId, {\n        idleTeardown: endowmentTeardown,\n        runningEvaluations: new Set(),\n        exports: {},\n      });\n\n      addEventListener('unhandledRejection', this.snapPromiseErrorHandler);\n      addEventListener('error', this.snapErrorHandler);\n\n      const compartment = new Compartment({\n        ...endowments,\n        module: snapModule,\n        exports: snapModule.exports,\n      });\n\n      // All of those are JavaScript runtime specific and self referential,\n      // but we add them for compatibility sake with external libraries.\n      //\n      // We can't do that in the injected globals object above\n      // because SES creates its own globalThis\n      compartment.globalThis.self = compartment.globalThis;\n      compartment.globalThis.global = compartment.globalThis;\n      compartment.globalThis.window = compartment.globalThis;\n\n      await this.executeInSnapContext(snapId, () => {\n        compartment.evaluate(sourceCode);\n        this.registerSnapExports(snapId, snapModule);\n      });\n    } catch (error) {\n      this.removeSnap(snapId);\n\n      const [cause] = unwrapError(error);\n      throw rpcErrors.internal({\n        message: `Error while running snap '${snapId}': ${cause.message}`,\n        data: {\n          cause: cause.serialize(),\n        },\n      });\n    }\n  }\n\n  /**\n   * Cancels all running evaluations of all snaps and clears all snap data.\n   * NOTE:** Should only be called in response to the `terminate` RPC command.\n   */\n  protected onTerminate() {\n    // `stop()` tears down snap endowments.\n    // Teardown will also be run for each snap as soon as there are\n    // no more running evaluations for that snap.\n    this.snapData.forEach((data) =>\n      data.runningEvaluations.forEach((evaluation) => evaluation.stop()),\n    );\n    this.snapData.clear();\n  }\n\n  private registerSnapExports(snapId: string, snapModule: any) {\n    const data = this.snapData.get(snapId);\n    // Somebody deleted the snap before we could register.\n    if (!data) {\n      return;\n    }\n\n    data.exports = SNAP_EXPORT_NAMES.reduce((acc, exportName) => {\n      const snapExport = snapModule.exports[exportName];\n      const { validator } = SNAP_EXPORTS[exportName];\n      if (validator(snapExport)) {\n        return { ...acc, [exportName]: snapExport };\n      }\n      return acc;\n    }, {});\n\n    // If the Snap has no valid exports after this, fail.\n    assert(Object.keys(data.exports).length > 0, 'Snap has no valid exports.');\n  }\n\n  /**\n   * Instantiates a snap API object (i.e. `globalThis.snap`).\n   *\n   * @param provider - A StreamProvider connected to MetaMask.\n   * @returns The snap provider object.\n   */\n  private createSnapGlobal(provider: StreamProvider): SnapsProvider {\n    const originalRequest = provider.request.bind(provider);\n\n    const request = async (args: RequestArguments) => {\n      const sanitizedArgs = sanitizeRequestArguments(args);\n      assertSnapOutboundRequest(sanitizedArgs);\n      return await withTeardown(\n        (async () => {\n          await this.#notify({\n            method: 'OutboundRequest',\n            params: { source: 'snap.request' },\n          });\n          try {\n            return await originalRequest(sanitizedArgs);\n          } finally {\n            await this.#notify({\n              method: 'OutboundResponse',\n              params: { source: 'snap.request' },\n            });\n          }\n        })(),\n        this as any,\n      );\n    };\n\n    const snapGlobalProxy = proxyStreamProvider(request) as SnapsProvider;\n\n    return harden(snapGlobalProxy);\n  }\n\n  /**\n   * Instantiates an EIP-1193 Ethereum provider object (i.e. `globalThis.ethereum`).\n   *\n   * @param provider - A StreamProvider connected to MetaMask.\n   * @returns The EIP-1193 Ethereum provider object.\n   */\n  private createEIP1193Provider(provider: StreamProvider): StreamProvider {\n    const originalRequest = provider.request.bind(provider);\n\n    const request = async (args: RequestArguments) => {\n      const sanitizedArgs = sanitizeRequestArguments(args);\n      assertEthereumOutboundRequest(sanitizedArgs);\n      return await withTeardown(\n        (async () => {\n          await this.#notify({\n            method: 'OutboundRequest',\n            params: { source: 'ethereum.request' },\n          });\n          try {\n            return await originalRequest(sanitizedArgs);\n          } finally {\n            await this.#notify({\n              method: 'OutboundResponse',\n              params: { source: 'ethereum.request' },\n            });\n          }\n        })(),\n        this as any,\n      );\n    };\n\n    const streamProviderProxy = proxyStreamProvider(request);\n\n    return harden(streamProviderProxy);\n  }\n\n  /**\n   * Removes the snap with the given name.\n   *\n   * @param snapId - The id of the snap to remove.\n   */\n  private removeSnap(snapId: string): void {\n    this.snapData.delete(snapId);\n  }\n\n  /**\n   * Calls the specified executor function in the context of the specified snap.\n   * Essentially, this means that the operation performed by the executor is\n   * counted as an evaluation of the specified snap. When the count of running\n   * evaluations of a snap reaches zero, its endowments are torn down.\n   *\n   * @param snapId - The id of the snap whose context to execute in.\n   * @param executor - The function that will be executed in the snap's context.\n   * @returns The executor's return value.\n   * @template Result - The return value of the executor.\n   */\n  private async executeInSnapContext<Result>(\n    snapId: string,\n    executor: () => Promise<Result> | Result,\n  ): Promise<Result> {\n    const data = this.snapData.get(snapId);\n    if (data === undefined) {\n      throw rpcErrors.internal(\n        `Tried to execute in context of unknown snap: \"${snapId}\".`,\n      );\n    }\n\n    let stop: () => void;\n    const stopPromise = new Promise<never>(\n      (_, reject) =>\n        (stop = () =>\n          reject(\n            // TODO(rekmarks): Specify / standardize error code for this case.\n            rpcErrors.internal(\n              `The snap \"${snapId}\" has been terminated during execution.`,\n            ),\n          )),\n    );\n\n    // eslint-disable-next-line @typescript-eslint/no-non-null-assertion\n    const evaluationData = { stop: stop! };\n\n    try {\n      data.runningEvaluations.add(evaluationData);\n      // Notice that we have to await this executor.\n      // If we didn't, we would decrease the amount of running evaluations\n      // before the promise actually resolves\n      return await Promise.race([executor(), stopPromise]);\n    } catch (error) {\n      throw new WrappedSnapError(error);\n    } finally {\n      data.runningEvaluations.delete(evaluationData);\n\n      if (data.runningEvaluations.size === 0) {\n        this.lastTeardown += 1;\n        await data.idleTeardown();\n      }\n    }\n  }\n}\n"]}
+{"version":3,"file":"BaseSnapExecutor.cjs","sourceRoot":"","sources":["../../src/common/BaseSnapExecutor.ts"],"names":[],"mappings":";;;AAAA,qFAAqF;AACrF,gEAAgE;AAChE,+DAAoE;AAEpE,yEAAqE;AACrE,qDAA6E;AAE7E,mDAAmD;AAMnD,uDAO+B;AAC/B,uDAAqD;AAOrD,2CAMyB;AAGzB,4CAAiC;AAEjC,6CAA6D;AAC7D,uDAAgD;AAChD,qDAAuE;AACvE,iDAA6C;AAC7C,uCAOiB;AACjB,iDAKsB;AAYtB,MAAM,aAAa,GAAG;IACpB,IAAI,EAAE,uBAAU,CAAC,GAAG,CAAC,QAAQ;IAC7B,OAAO,EAAE,6BAA6B;CACvC,CAAC;AAEF,MAAM,cAAc,GAAG,sBAAS;KAC7B,QAAQ,CAAO;IACd,OAAO,EAAE,sBAAsB;CAChC,CAAC;KACD,SAAS,EAAE,CAAC;AAUf;;;;GAIG;AACH,MAAM,6BAA6B,GAAG;IACpC,IAAI,EAAE;QACJ,MAAM,EAAE,uCAA0B;QAClC,MAAM,EAAE,EAAE;KACX;IACD,WAAW,EAAE;QACX,MAAM,EAAE,8CAAiC;QACzC,MAAM,EAAE,CAAC,QAAQ,EAAE,YAAY,EAAE,YAAY,CAAC;KAC/C;IACD,SAAS,EAAE;QACT,MAAM,EAAE,4CAA+B;QACvC,MAAM,EAAE,EAAE;KACX;IACD,OAAO,EAAE;QACP,MAAM,EAAE,0CAA6B;QACrC,MAAM,EAAE,CAAC,QAAQ,EAAE,SAAS,EAAE,QAAQ,EAAE,SAAS,CAAC;KACnD;CACF,CAAC;AAQF,MAAa,gBAAgB;IACV,QAAQ,CAAwB;IAEhC,aAAa,CAAS;IAEtB,SAAS,CAAS;IAElB,OAAO,CAAwB;IAExC,gBAAgB,CAA+B;IAE/C,uBAAuB,CAA0C;IAEjE,YAAY,GAAG,CAAC,CAAC;IAEzB,YAAsB,aAAqB,EAAE,SAAiB;QAC5D,IAAI,CAAC,QAAQ,GAAG,IAAI,GAAG,EAAE,CAAC;QAC1B,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;QACnC,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE;YACrC,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBAC1C,qCAAqC;gBACrC,IAAA,sBAAQ,EAAC,KAAK,CAAC,CAAC;YAClB,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAE3B,IAAI,CAAC,OAAO,GAAG,IAAA,0CAA+B,EAC5C,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,EACzB,KAAK,EAAE,MAAM,EAAE,WAAW,EAAE,IAAI,EAAE,EAAE;YAClC,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YACvC,uEAAuE;YACvE,mBAAmB;YACnB,MAAM,OAAO,GAAG,IAAI,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;YAC3C,MAAM,EAAE,QAAQ,EAAE,GAAG,0BAAY,CAAC,WAAW,CAAC,CAAC;YAE/C,IAAA,cAAM,EACJ,CAAC,QAAQ,IAAI,OAAO,KAAK,SAAS,EAClC,MAAM,WAAW,+BAA+B,MAAM,EAAE,EACxD,sBAAS,CAAC,kBAAkB,CAC7B,CAAC;YAEF,kEAAkE;YAClE,eAAe;YACf,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,OAAO,IAAI,CAAC;YACd,CAAC;YAED,IAAI,MAAM,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,MAAM,EAAE,GAAG,EAAE;YACxD,mCAAmC;YACnC,OAAO,CAAC,IAAW,CAAC,CACrB,CAAC;YAEF,0EAA0E;YAC1E,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;gBACzB,MAAM,GAAG,IAAI,CAAC;YAChB,CAAC;YAED,uEAAuE;YACvE,IAAI,CAAC;gBACH,OAAO,IAAA,mBAAW,EAAC,MAAM,CAAC,CAAC;YAC7B,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,sBAAS,CAAC,QAAQ,CACtB,yCAAyC,KAAK,CAAC,OAAO,CAAC,OAAO,CAC5D,sBAAsB,EACtB,EAAE,CACH,EAAE,CACJ,CAAC;YACJ,CAAC;QACH,CAAC,EACD,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAC5B,CAAC;IACJ,CAAC;IAEO,YAAY,CAAC,KAAc,EAAE,IAA0B;QAC7D,MAAM,eAAe,GAAG,IAAA,2BAAc,EAAC,KAAK,EAAE;YAC5C,aAAa,EAAE,cAAc;YAC7B,kBAAkB,EAAE,KAAK;SAC1B,CAAC,CAAC;QAEH,MAAM,SAAS,GAAG,IAAA,wBAAY,EAAC,eAAe,CAAC,CAAC;QAEhD,IAAI,CAAC,OAAO,CAAC;YACX,MAAM,EAAE,gBAAgB;YACxB,MAAM,EAAE;gBACN,KAAK,EAAE;oBACL,GAAG,eAAe;oBAClB,IAAI,EAAE;wBACJ,GAAG,SAAS;wBACZ,GAAG,IAAI;qBACR;iBACF;aACF;SACF,CAAC,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,EAAE;YACvB,IAAA,sBAAQ,EAAC,WAAW,CAAC,CAAC;QACxB,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAAC,OAAuB;QACpD,IAAI,CAAC,IAAA,wBAAgB,EAAC,OAAO,CAAC,EAAE,CAAC;YAC/B,IACE,IAAA,mBAAW,EAAC,OAAO,EAAE,IAAI,CAAC;gBAC1B,IAAA,gBAAE,EAAE,OAAsC,CAAC,EAAE,EAAE,uBAAe,CAAC,EAC/D,CAAC;gBACD,0DAA0D;gBAC1D,wDAAwD;gBACxD,MAAM,IAAI,CAAC,MAAM,CAAC;oBAChB,KAAK,EAAE,IAAA,2BAAc,EACnB,sBAAS,CAAC,QAAQ,CAChB,sDAAsD,CACvD,CACF;oBACD,EAAE,EAAG,OAAsC,CAAC,EAAE;oBAC9C,OAAO,EAAE,KAAK;iBACf,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,IAAA,qBAAO,EACL,4EAA4E,CAC7E,CAAC;YACJ,CAAC;YACD,OAAO;QACT,CAAC;QAED,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC;QAEvC,IAAI,CAAC,IAAA,mBAAW,EAAC,6BAA6B,EAAE,MAAM,CAAC,EAAE,CAAC;YACxD,MAAM,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE;gBACtB,KAAK,EAAE,sBAAS;qBACb,cAAc,CAAC;oBACd,IAAI,EAAE;wBACJ,MAAM;qBACP;iBACF,CAAC;qBACD,SAAS,EAAE;aACf,CAAC,CAAC;YACH,OAAO;QACT,CAAC;QAED,MAAM,YAAY,GAAG,6BAA6B,CAAC,MAAuB,CAAC,CAAC;QAE5E,yCAAyC;QACzC,MAAM,aAAa,GAAG,IAAA,0BAAa,EAAC,YAAY,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAEjE,MAAM,CAAC,KAAK,CAAC,GAAG,IAAA,sBAAQ,EAAW,aAAa,EAAE,YAAY,CAAC,MAAM,CAAC,CAAC;QACvE,IAAI,KAAK,EAAE,CAAC;YACV,MAAM,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE;gBACtB,KAAK,EAAE,sBAAS;qBACb,aAAa,CAAC;oBACb,OAAO,EAAE,kCAAkC,MAAM,MAAM,KAAK,CAAC,OAAO,GAAG;oBACvE,IAAI,EAAE;wBACJ,MAAM;wBACN,MAAM,EAAE,aAAa;qBACtB;iBACF,CAAC;qBACD,SAAS,EAAE;aACf,CAAC,CAAC;YACH,OAAO;QACT,CAAC;QAED,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAO,IAAI,CAAC,OAAe,CAAC,MAAM,CAAC,CAAC,GAAG,aAAa,CAAC,CAAC;YACrE,MAAM,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;QACtC,CAAC;QAAC,OAAO,QAAQ,EAAE,CAAC;YAClB,MAAM,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE;gBACtB,KAAK,EAAE,IAAA,2BAAc,EAAC,QAAQ,EAAE;oBAC9B,aAAa;iBACd,CAAC;aACH,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,4DAA4D;IAC5D,uEAAuE;IACvE,2CAA2C;IAC3C,KAAK,CAAC,MAAM,CAAC,KAAW;QACtB,OAAO,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YAC3C,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;gBACxC,IAAI,KAAK,EAAE,CAAC;oBACV,MAAM,CAAC,KAAK,CAAC,CAAC;oBACd,OAAO;gBACT,CAAC;gBACD,OAAO,EAAE,CAAC;YACZ,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,YAAkD;QAC9D,IAAI,CAAC,IAAA,uBAAe,EAAC,YAAY,CAAC,EAAE,CAAC;YACnC,MAAM,sBAAS,CAAC,QAAQ,CACtB,8EAA8E,CAC/E,CAAC;QACJ,CAAC;QAED,MAAM,IAAI,CAAC,MAAM,CAAC;YAChB,GAAG,YAAY;YACf,OAAO,EAAE,KAAK;SACf,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,EAAa,EAAE,QAAiC;QAC7D,IAAI,CAAC,IAAA,uBAAe,EAAC,QAAQ,CAAC,EAAE,CAAC;YAC/B,0DAA0D;YAC1D,qFAAqF;YACrF,MAAM,IAAI,CAAC,MAAM,CAAC;gBAChB,KAAK,EAAE,IAAA,2BAAc,EACnB,sBAAS,CAAC,QAAQ,CAChB,0EAA0E,CAC3E,CACF;gBACD,EAAE;gBACF,OAAO,EAAE,KAAK;aACf,CAAC,CAAC;YACH,OAAO;QACT,CAAC;QAED,MAAM,IAAI,CAAC,MAAM,CAAC;YAChB,GAAG,QAAQ;YACX,EAAE;YACF,OAAO,EAAE,KAAK;SACf,CAAC,CAAC;IACL,CAAC;IAED;;;;;;;OAOG;IACO,KAAK,CAAC,SAAS,CACvB,MAAc,EACd,UAAkB,EAClB,WAAqB;QAErB,IAAA,aAAG,EAAC,kBAAkB,MAAM,cAAc,CAAC,CAAC;QAC5C,IAAI,IAAI,CAAC,uBAAuB,EAAE,CAAC;YACjC,IAAA,kCAAmB,EAAC,oBAAoB,EAAE,IAAI,CAAC,uBAAuB,CAAC,CAAC;QAC1E,CAAC;QAED,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC1B,IAAA,kCAAmB,EAAC,OAAO,EAAE,IAAI,CAAC,gBAAgB,CAAC,CAAC;QACtD,CAAC;QAED,IAAI,CAAC,gBAAgB,GAAG,CAAC,KAAiB,EAAE,EAAE;YAC5C,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;QAC7C,CAAC,CAAC;QAEF,IAAI,CAAC,uBAAuB,GAAG,CAAC,KAA4B,EAAE,EAAE;YAC9D,IAAI,CAAC,YAAY,CAAC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,EAAE;gBAC/D,MAAM;aACP,CAAC,CAAC;QACL,CAAC,CAAC;QAEF,MAAM,QAAQ,GAAG,IAAI,gCAAc,CAAC,IAAI,CAAC,SAAS,EAAE;YAClD,iBAAiB,EAAE,mBAAmB;YACtC,aAAa,EAAE,CAAC,IAAA,yCAAuB,GAAE,CAAC;SAC3C,CAAC,CAAC;QAEH,MAAM,QAAQ,CAAC,UAAU,EAAE,CAAC;QAE5B,MAAM,IAAI,GAAG,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QAC7C,MAAM,QAAQ,GAAG,IAAI,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC;QACtD,wFAAwF;QACxF,MAAM,UAAU,GAAQ,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;QAExC,IAAI,CAAC;YACH,MAAM,EAAE,UAAU,EAAE,QAAQ,EAAE,iBAAiB,EAAE,GAAG,IAAA,6BAAgB,EAAC;gBACnE,IAAI;gBACJ,QAAQ;gBACR,MAAM;gBACN,UAAU,EAAE,WAAW;gBACvB,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC;aAChC,CAAC,CAAC;YAEH,gEAAgE;YAChE,mEAAmE;YACnE,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,EAAE;gBACxB,YAAY,EAAE,iBAAiB;gBAC/B,kBAAkB,EAAE,IAAI,GAAG,EAAE;gBAC7B,OAAO,EAAE,EAAE;aACZ,CAAC,CAAC;YAEH,IAAA,+BAAgB,EAAC,oBAAoB,EAAE,IAAI,CAAC,uBAAuB,CAAC,CAAC;YACrE,IAAA,+BAAgB,EAAC,OAAO,EAAE,IAAI,CAAC,gBAAgB,CAAC,CAAC;YAEjD,MAAM,WAAW,GAAG,IAAI,WAAW,CAAC;gBAClC,GAAG,UAAU;gBACb,MAAM,EAAE,UAAU;gBAClB,OAAO,EAAE,UAAU,CAAC,OAAO;aAC5B,CAAC,CAAC;YAEH,qEAAqE;YACrE,kEAAkE;YAClE,EAAE;YACF,wDAAwD;YACxD,yCAAyC;YACzC,WAAW,CAAC,UAAU,CAAC,IAAI,GAAG,WAAW,CAAC,UAAU,CAAC;YACrD,WAAW,CAAC,UAAU,CAAC,MAAM,GAAG,WAAW,CAAC,UAAU,CAAC;YACvD,WAAW,CAAC,UAAU,CAAC,MAAM,GAAG,WAAW,CAAC,UAAU,CAAC;YAEvD,MAAM,IAAI,CAAC,oBAAoB,CAAC,MAAM,EAAE,GAAG,EAAE;gBAC3C,WAAW,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;gBACjC,IAAI,CAAC,mBAAmB,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;YAC/C,CAAC,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;YAExB,MAAM,CAAC,KAAK,CAAC,GAAG,IAAA,yBAAW,EAAC,KAAK,CAAC,CAAC;YACnC,MAAM,sBAAS,CAAC,QAAQ,CAAC;gBACvB,OAAO,EAAE,6BAA6B,MAAM,MAAM,KAAK,CAAC,OAAO,EAAE;gBACjE,IAAI,EAAE;oBACJ,KAAK,EAAE,KAAK,CAAC,SAAS,EAAE;iBACzB;aACF,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED;;;OAGG;IACO,WAAW;QACnB,uCAAuC;QACvC,+DAA+D;QAC/D,6CAA6C;QAC7C,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAC7B,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC,CACnE,CAAC;QACF,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;IACxB,CAAC;IAEO,mBAAmB,CAAC,MAAc,EAAE,UAAe;QACzD,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACvC,sDAAsD;QACtD,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,OAAO;QACT,CAAC;QAED,IAAI,CAAC,OAAO,GAAG,+BAAiB,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,UAAU,EAAE,EAAE;YAC1D,MAAM,UAAU,GAAG,UAAU,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;YAClD,MAAM,EAAE,SAAS,EAAE,GAAG,0BAAY,CAAC,UAAU,CAAC,CAAC;YAC/C,IAAI,SAAS,CAAC,UAAU,CAAC,EAAE,CAAC;gBAC1B,OAAO,EAAE,GAAG,GAAG,EAAE,CAAC,UAAU,CAAC,EAAE,UAAU,EAAE,CAAC;YAC9C,CAAC;YACD,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,EAAE,CAAC,CAAC;QAEP,qDAAqD;QACrD,IAAA,cAAM,EAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,4BAA4B,CAAC,CAAC;IAC7E,CAAC;IAED;;;;;OAKG;IACK,gBAAgB,CAAC,QAAwB;QAC/C,MAAM,eAAe,GAAG,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAExD,MAAM,OAAO,GAAG,KAAK,EAAE,IAAsB,EAAE,EAAE;YAC/C,MAAM,aAAa,GAAG,IAAA,gCAAwB,EAAC,IAAI,CAAC,CAAC;YACrD,IAAA,iCAAyB,EAAC,aAAa,CAAC,CAAC;YACzC,OAAO,MAAM,IAAA,oBAAY,EACvB,CAAC,KAAK,IAAI,EAAE;gBACV,MAAM,IAAI,CAAC,OAAO,CAAC;oBACjB,MAAM,EAAE,iBAAiB;oBACzB,MAAM,EAAE,EAAE,MAAM,EAAE,cAAc,EAAE;iBACnC,CAAC,CAAC;gBACH,IAAI,CAAC;oBACH,OAAO,MAAM,eAAe,CAAC,aAAa,CAAC,CAAC;gBAC9C,CAAC;wBAAS,CAAC;oBACT,MAAM,IAAI,CAAC,OAAO,CAAC;wBACjB,MAAM,EAAE,kBAAkB;wBAC1B,MAAM,EAAE,EAAE,MAAM,EAAE,cAAc,EAAE;qBACnC,CAAC,CAAC;gBACL,CAAC;YACH,CAAC,CAAC,EAAE,EACJ,IAAW,CACZ,CAAC;QACJ,CAAC,CAAC;QAEF,MAAM,eAAe,GAAG,IAAA,2BAAmB,EAAC,OAAO,CAAkB,CAAC;QAEtE,OAAO,MAAM,CAAC,eAAe,CAAC,CAAC;IACjC,CAAC;IAED;;;;;OAKG;IACK,qBAAqB,CAAC,QAAwB;QACpD,MAAM,eAAe,GAAG,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAExD,MAAM,OAAO,GAAG,KAAK,EAAE,IAAsB,EAAE,EAAE;YAC/C,MAAM,aAAa,GAAG,IAAA,gCAAwB,EAAC,IAAI,CAAC,CAAC;YACrD,IAAA,qCAA6B,EAAC,aAAa,CAAC,CAAC;YAC7C,OAAO,MAAM,IAAA,oBAAY,EACvB,CAAC,KAAK,IAAI,EAAE;gBACV,MAAM,IAAI,CAAC,OAAO,CAAC;oBACjB,MAAM,EAAE,iBAAiB;oBACzB,MAAM,EAAE,EAAE,MAAM,EAAE,kBAAkB,EAAE;iBACvC,CAAC,CAAC;gBACH,IAAI,CAAC;oBACH,OAAO,MAAM,eAAe,CAAC,aAAa,CAAC,CAAC;gBAC9C,CAAC;wBAAS,CAAC;oBACT,MAAM,IAAI,CAAC,OAAO,CAAC;wBACjB,MAAM,EAAE,kBAAkB;wBAC1B,MAAM,EAAE,EAAE,MAAM,EAAE,kBAAkB,EAAE;qBACvC,CAAC,CAAC;gBACL,CAAC;YACH,CAAC,CAAC,EAAE,EACJ,IAAW,CACZ,CAAC;QACJ,CAAC,CAAC;QAEF,MAAM,mBAAmB,GAAG,IAAA,2BAAmB,EAAC,OAAO,CAAC,CAAC;QAEzD,OAAO,MAAM,CAAC,mBAAmB,CAAC,CAAC;IACrC,CAAC;IAED;;;;OAIG;IACK,UAAU,CAAC,MAAc;QAC/B,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IAC/B,CAAC;IAED;;;;;;;;;;OAUG;IACK,KAAK,CAAC,oBAAoB,CAChC,MAAc,EACd,QAAwC;QAExC,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACvC,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;YACvB,MAAM,sBAAS,CAAC,QAAQ,CACtB,iDAAiD,MAAM,IAAI,CAC5D,CAAC;QACJ,CAAC;QAED,IAAI,IAAgB,CAAC;QACrB,MAAM,WAAW,GAAG,IAAI,OAAO,CAC7B,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,CACZ,CAAC,IAAI,GAAG,GAAG,EAAE,CACX,MAAM;QACJ,kEAAkE;QAClE,sBAAS,CAAC,QAAQ,CAChB,aAAa,MAAM,yCAAyC,CAC7D,CACF,CAAC,CACP,CAAC;QAEF,oEAAoE;QACpE,MAAM,cAAc,GAAG,EAAE,IAAI,EAAE,IAAK,EAAE,CAAC;QAEvC,IAAI,CAAC;YACH,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;YAC5C,8CAA8C;YAC9C,oEAAoE;YACpE,uCAAuC;YACvC,OAAO,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,EAAE,WAAW,CAAC,CAAC,CAAC;QACvD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,8BAAgB,CAAC,KAAK,CAAC,CAAC;QACpC,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;YAE/C,IAAI,IAAI,CAAC,kBAAkB,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;gBACvC,IAAI,CAAC,YAAY,IAAI,CAAC,CAAC;gBACvB,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;YAC5B,CAAC;QACH,CAAC;IACH,CAAC;CACF;AAteD,4CAseC","sourcesContent":["// eslint-disable-next-line @typescript-eslint/triple-slash-reference, spaced-comment\n/// <reference path=\"../../../../node_modules/ses/types.d.ts\" />\nimport { createIdRemapMiddleware } from '@metamask/json-rpc-engine';\nimport type { RequestArguments } from '@metamask/providers';\nimport { StreamProvider } from '@metamask/providers/stream-provider';\nimport { errorCodes, rpcErrors, serializeError } from '@metamask/rpc-errors';\nimport type { SnapsProvider } from '@metamask/snaps-sdk';\nimport { getErrorData } from '@metamask/snaps-sdk';\nimport type {\n  SnapExports,\n  HandlerType,\n  SnapExportsParameters,\n} from '@metamask/snaps-utils';\nimport {\n  SNAP_EXPORT_NAMES,\n  logError,\n  SNAP_EXPORTS,\n  WrappedSnapError,\n  unwrapError,\n  logInfo,\n} from '@metamask/snaps-utils';\nimport { validate, is } from '@metamask/superstruct';\nimport type {\n  JsonRpcNotification,\n  JsonRpcId,\n  JsonRpcRequest,\n  Json,\n} from '@metamask/utils';\nimport {\n  assert,\n  isJsonRpcRequest,\n  hasProperty,\n  getSafeJson,\n  JsonRpcIdStruct,\n} from '@metamask/utils';\nimport type { Duplex } from 'readable-stream';\n\nimport { log } from '../logging';\nimport type { CommandMethodsMapping } from './commands';\nimport { getCommandMethodImplementations } from './commands';\nimport { createEndowments } from './endowments';\nimport { addEventListener, removeEventListener } from './globalEvents';\nimport { sortParamKeys } from './sortParams';\nimport {\n  assertEthereumOutboundRequest,\n  assertSnapOutboundRequest,\n  sanitizeRequestArguments,\n  proxyStreamProvider,\n  withTeardown,\n  isValidResponse,\n} from './utils';\nimport {\n  ExecuteSnapRequestArgumentsStruct,\n  PingRequestArgumentsStruct,\n  SnapRpcRequestArgumentsStruct,\n  TerminateRequestArgumentsStruct,\n} from './validation';\n\ntype EvaluationData = {\n  stop: () => void;\n};\n\ntype SnapData = {\n  exports: SnapExports;\n  runningEvaluations: Set<EvaluationData>;\n  idleTeardown: () => Promise<void>;\n};\n\nconst fallbackError = {\n  code: errorCodes.rpc.internal,\n  message: 'Execution Environment Error',\n};\n\nconst unhandledError = rpcErrors\n  .internal<Json>({\n    message: 'Unhandled Snap Error',\n  })\n  .serialize();\n\nexport type InvokeSnapArgs = Omit<SnapExportsParameters[0], 'chainId'>;\n\nexport type InvokeSnap = (\n  target: string,\n  handler: HandlerType,\n  args: InvokeSnapArgs | undefined,\n) => Promise<Json>;\n\n/**\n * The supported methods in the execution environment. The validator checks the\n * incoming JSON-RPC request, and the `params` property is used for sorting the\n * parameters, if they are an object.\n */\nconst EXECUTION_ENVIRONMENT_METHODS = {\n  ping: {\n    struct: PingRequestArgumentsStruct,\n    params: [],\n  },\n  executeSnap: {\n    struct: ExecuteSnapRequestArgumentsStruct,\n    params: ['snapId', 'sourceCode', 'endowments'],\n  },\n  terminate: {\n    struct: TerminateRequestArgumentsStruct,\n    params: [],\n  },\n  snapRpc: {\n    struct: SnapRpcRequestArgumentsStruct,\n    params: ['target', 'handler', 'origin', 'request'],\n  },\n};\n\ntype Methods = typeof EXECUTION_ENVIRONMENT_METHODS;\n\nexport type NotifyFunction = (\n  notification: Omit<JsonRpcNotification, 'jsonrpc'>,\n) => Promise<void>;\n\nexport class BaseSnapExecutor {\n  private readonly snapData: Map<string, SnapData>;\n\n  private readonly commandStream: Duplex;\n\n  private readonly rpcStream: Duplex;\n\n  private readonly methods: CommandMethodsMapping;\n\n  private snapErrorHandler?: (event: ErrorEvent) => void;\n\n  private snapPromiseErrorHandler?: (event: PromiseRejectionEvent) => void;\n\n  private lastTeardown = 0;\n\n  protected constructor(commandStream: Duplex, rpcStream: Duplex) {\n    this.snapData = new Map();\n    this.commandStream = commandStream;\n    this.commandStream.on('data', (data) => {\n      this.onCommandRequest(data).catch((error) => {\n        // TODO: Decide how to handle errors.\n        logError(error);\n      });\n    });\n    this.rpcStream = rpcStream;\n\n    this.methods = getCommandMethodImplementations(\n      this.startSnap.bind(this),\n      async (target, handlerType, args) => {\n        const data = this.snapData.get(target);\n        // We're capturing the handler in case someone modifies the data object\n        // before the call.\n        const handler = data?.exports[handlerType];\n        const { required } = SNAP_EXPORTS[handlerType];\n\n        assert(\n          !required || handler !== undefined,\n          `No ${handlerType} handler exported for snap \"${target}`,\n          rpcErrors.methodNotSupported,\n        );\n\n        // Certain handlers are not required. If they are not exported, we\n        // return null.\n        if (!handler) {\n          return null;\n        }\n\n        let result = await this.executeInSnapContext(target, () =>\n          // TODO: fix handler args type cast\n          handler(args as any),\n        );\n\n        // The handler might not return anything, but undefined is not valid JSON.\n        if (result === undefined) {\n          result = null;\n        }\n\n        // /!\\ Always return only sanitized JSON to prevent security flaws. /!\\\n        try {\n          return getSafeJson(result);\n        } catch (error) {\n          throw rpcErrors.internal(\n            `Received non-JSON-serializable value: ${error.message.replace(\n              /^Assertion failed: /u,\n              '',\n            )}`,\n          );\n        }\n      },\n      this.onTerminate.bind(this),\n    );\n  }\n\n  private errorHandler(error: unknown, data: Record<string, Json>) {\n    const serializedError = serializeError(error, {\n      fallbackError: unhandledError,\n      shouldIncludeStack: false,\n    });\n\n    const errorData = getErrorData(serializedError);\n\n    this.#notify({\n      method: 'UnhandledError',\n      params: {\n        error: {\n          ...serializedError,\n          data: {\n            ...errorData,\n            ...data,\n          },\n        },\n      },\n    }).catch((notifyError) => {\n      logError(notifyError);\n    });\n  }\n\n  private async onCommandRequest(message: JsonRpcRequest) {\n    if (!isJsonRpcRequest(message)) {\n      if (\n        hasProperty(message, 'id') &&\n        is((message as Pick<JsonRpcRequest, 'id'>).id, JsonRpcIdStruct)\n      ) {\n        // Instead of throwing, we directly respond with an error.\n        // We can only do this if the message ID is still valid.\n        await this.#write({\n          error: serializeError(\n            rpcErrors.internal(\n              'JSON-RPC requests must be JSON serializable objects.',\n            ),\n          ),\n          id: (message as Pick<JsonRpcRequest, 'id'>).id,\n          jsonrpc: '2.0',\n        });\n      } else {\n        logInfo(\n          'Command stream received a non-JSON-RPC request, and was unable to respond.',\n        );\n      }\n      return;\n    }\n\n    const { id, method, params } = message;\n\n    if (!hasProperty(EXECUTION_ENVIRONMENT_METHODS, method)) {\n      await this.#respond(id, {\n        error: rpcErrors\n          .methodNotFound({\n            data: {\n              method,\n            },\n          })\n          .serialize(),\n      });\n      return;\n    }\n\n    const methodObject = EXECUTION_ENVIRONMENT_METHODS[method as keyof Methods];\n\n    // support params by-name and by-position\n    const paramsAsArray = sortParamKeys(methodObject.params, params);\n\n    const [error] = validate<any, any>(paramsAsArray, methodObject.struct);\n    if (error) {\n      await this.#respond(id, {\n        error: rpcErrors\n          .invalidParams({\n            message: `Invalid parameters for method \"${method}\": ${error.message}.`,\n            data: {\n              method,\n              params: paramsAsArray,\n            },\n          })\n          .serialize(),\n      });\n      return;\n    }\n\n    try {\n      const result = await (this.methods as any)[method](...paramsAsArray);\n      await this.#respond(id, { result });\n    } catch (rpcError) {\n      await this.#respond(id, {\n        error: serializeError(rpcError, {\n          fallbackError,\n        }),\n      });\n    }\n  }\n\n  // Awaitable function that writes back to the command stream\n  // To prevent snap execution from blocking writing we wrap in a promise\n  // and await it before continuing execution\n  async #write(chunk: Json) {\n    return new Promise<void>((resolve, reject) => {\n      this.commandStream.write(chunk, (error) => {\n        if (error) {\n          reject(error);\n          return;\n        }\n        resolve();\n      });\n    });\n  }\n\n  async #notify(notification: Omit<JsonRpcNotification, 'jsonrpc'>) {\n    if (!isValidResponse(notification)) {\n      throw rpcErrors.internal(\n        'JSON-RPC notifications must be JSON serializable objects smaller than 64 MB.',\n      );\n    }\n\n    await this.#write({\n      ...notification,\n      jsonrpc: '2.0',\n    });\n  }\n\n  async #respond(id: JsonRpcId, response: Record<string, unknown>) {\n    if (!isValidResponse(response)) {\n      // Instead of throwing, we directly respond with an error.\n      // This prevents an issue where we wouldn't respond when errors were non-serializable\n      await this.#write({\n        error: serializeError(\n          rpcErrors.internal(\n            'JSON-RPC responses must be JSON serializable objects smaller than 64 MB.',\n          ),\n        ),\n        id,\n        jsonrpc: '2.0',\n      });\n      return;\n    }\n\n    await this.#write({\n      ...response,\n      id,\n      jsonrpc: '2.0',\n    });\n  }\n\n  /**\n   * Attempts to evaluate a snap in SES. Generates APIs for the snap. May throw\n   * on errors.\n   *\n   * @param snapId - The id of the snap.\n   * @param sourceCode - The source code of the snap, in IIFE format.\n   * @param _endowments - An array of the names of the endowments.\n   */\n  protected async startSnap(\n    snapId: string,\n    sourceCode: string,\n    _endowments: string[],\n  ): Promise<void> {\n    log(`Starting snap '${snapId}' in worker.`);\n    if (this.snapPromiseErrorHandler) {\n      removeEventListener('unhandledrejection', this.snapPromiseErrorHandler);\n    }\n\n    if (this.snapErrorHandler) {\n      removeEventListener('error', this.snapErrorHandler);\n    }\n\n    this.snapErrorHandler = (error: ErrorEvent) => {\n      this.errorHandler(error.error, { snapId });\n    };\n\n    this.snapPromiseErrorHandler = (error: PromiseRejectionEvent) => {\n      this.errorHandler(error instanceof Error ? error : error.reason, {\n        snapId,\n      });\n    };\n\n    const provider = new StreamProvider(this.rpcStream, {\n      jsonRpcStreamName: 'metamask-provider',\n      rpcMiddleware: [createIdRemapMiddleware()],\n    });\n\n    await provider.initialize();\n\n    const snap = this.createSnapGlobal(provider);\n    const ethereum = this.createEIP1193Provider(provider);\n    // We specifically use any type because the Snap can modify the object any way they want\n    const snapModule: any = { exports: {} };\n\n    try {\n      const { endowments, teardown: endowmentTeardown } = createEndowments({\n        snap,\n        ethereum,\n        snapId,\n        endowments: _endowments,\n        notify: this.#notify.bind(this),\n      });\n\n      // !!! Ensure that this is the only place the data is being set.\n      // Other methods access the object value and mutate its properties.\n      this.snapData.set(snapId, {\n        idleTeardown: endowmentTeardown,\n        runningEvaluations: new Set(),\n        exports: {},\n      });\n\n      addEventListener('unhandledRejection', this.snapPromiseErrorHandler);\n      addEventListener('error', this.snapErrorHandler);\n\n      const compartment = new Compartment({\n        ...endowments,\n        module: snapModule,\n        exports: snapModule.exports,\n      });\n\n      // All of those are JavaScript runtime specific and self referential,\n      // but we add them for compatibility sake with external libraries.\n      //\n      // We can't do that in the injected globals object above\n      // because SES creates its own globalThis\n      compartment.globalThis.self = compartment.globalThis;\n      compartment.globalThis.global = compartment.globalThis;\n      compartment.globalThis.window = compartment.globalThis;\n\n      await this.executeInSnapContext(snapId, () => {\n        compartment.evaluate(sourceCode);\n        this.registerSnapExports(snapId, snapModule);\n      });\n    } catch (error) {\n      this.removeSnap(snapId);\n\n      const [cause] = unwrapError(error);\n      throw rpcErrors.internal({\n        message: `Error while running snap '${snapId}': ${cause.message}`,\n        data: {\n          cause: cause.serialize(),\n        },\n      });\n    }\n  }\n\n  /**\n   * Cancels all running evaluations of all snaps and clears all snap data.\n   * NOTE:** Should only be called in response to the `terminate` RPC command.\n   */\n  protected onTerminate() {\n    // `stop()` tears down snap endowments.\n    // Teardown will also be run for each snap as soon as there are\n    // no more running evaluations for that snap.\n    this.snapData.forEach((data) =>\n      data.runningEvaluations.forEach((evaluation) => evaluation.stop()),\n    );\n    this.snapData.clear();\n  }\n\n  private registerSnapExports(snapId: string, snapModule: any) {\n    const data = this.snapData.get(snapId);\n    // Somebody deleted the snap before we could register.\n    if (!data) {\n      return;\n    }\n\n    data.exports = SNAP_EXPORT_NAMES.reduce((acc, exportName) => {\n      const snapExport = snapModule.exports[exportName];\n      const { validator } = SNAP_EXPORTS[exportName];\n      if (validator(snapExport)) {\n        return { ...acc, [exportName]: snapExport };\n      }\n      return acc;\n    }, {});\n\n    // If the Snap has no valid exports after this, fail.\n    assert(Object.keys(data.exports).length > 0, 'Snap has no valid exports.');\n  }\n\n  /**\n   * Instantiates a snap API object (i.e. `globalThis.snap`).\n   *\n   * @param provider - A StreamProvider connected to MetaMask.\n   * @returns The snap provider object.\n   */\n  private createSnapGlobal(provider: StreamProvider): SnapsProvider {\n    const originalRequest = provider.request.bind(provider);\n\n    const request = async (args: RequestArguments) => {\n      const sanitizedArgs = sanitizeRequestArguments(args);\n      assertSnapOutboundRequest(sanitizedArgs);\n      return await withTeardown(\n        (async () => {\n          await this.#notify({\n            method: 'OutboundRequest',\n            params: { source: 'snap.request' },\n          });\n          try {\n            return await originalRequest(sanitizedArgs);\n          } finally {\n            await this.#notify({\n              method: 'OutboundResponse',\n              params: { source: 'snap.request' },\n            });\n          }\n        })(),\n        this as any,\n      );\n    };\n\n    const snapGlobalProxy = proxyStreamProvider(request) as SnapsProvider;\n\n    return harden(snapGlobalProxy);\n  }\n\n  /**\n   * Instantiates an EIP-1193 Ethereum provider object (i.e. `globalThis.ethereum`).\n   *\n   * @param provider - A StreamProvider connected to MetaMask.\n   * @returns The EIP-1193 Ethereum provider object.\n   */\n  private createEIP1193Provider(provider: StreamProvider): StreamProvider {\n    const originalRequest = provider.request.bind(provider);\n\n    const request = async (args: RequestArguments) => {\n      const sanitizedArgs = sanitizeRequestArguments(args);\n      assertEthereumOutboundRequest(sanitizedArgs);\n      return await withTeardown(\n        (async () => {\n          await this.#notify({\n            method: 'OutboundRequest',\n            params: { source: 'ethereum.request' },\n          });\n          try {\n            return await originalRequest(sanitizedArgs);\n          } finally {\n            await this.#notify({\n              method: 'OutboundResponse',\n              params: { source: 'ethereum.request' },\n            });\n          }\n        })(),\n        this as any,\n      );\n    };\n\n    const streamProviderProxy = proxyStreamProvider(request);\n\n    return harden(streamProviderProxy);\n  }\n\n  /**\n   * Removes the snap with the given name.\n   *\n   * @param snapId - The id of the snap to remove.\n   */\n  private removeSnap(snapId: string): void {\n    this.snapData.delete(snapId);\n  }\n\n  /**\n   * Calls the specified executor function in the context of the specified snap.\n   * Essentially, this means that the operation performed by the executor is\n   * counted as an evaluation of the specified snap. When the count of running\n   * evaluations of a snap reaches zero, its endowments are torn down.\n   *\n   * @param snapId - The id of the snap whose context to execute in.\n   * @param executor - The function that will be executed in the snap's context.\n   * @returns The executor's return value.\n   * @template Result - The return value of the executor.\n   */\n  private async executeInSnapContext<Result>(\n    snapId: string,\n    executor: () => Promise<Result> | Result,\n  ): Promise<Result> {\n    const data = this.snapData.get(snapId);\n    if (data === undefined) {\n      throw rpcErrors.internal(\n        `Tried to execute in context of unknown snap: \"${snapId}\".`,\n      );\n    }\n\n    let stop: () => void;\n    const stopPromise = new Promise<never>(\n      (_, reject) =>\n        (stop = () =>\n          reject(\n            // TODO(rekmarks): Specify / standardize error code for this case.\n            rpcErrors.internal(\n              `The snap \"${snapId}\" has been terminated during execution.`,\n            ),\n          )),\n    );\n\n    // eslint-disable-next-line @typescript-eslint/no-non-null-assertion\n    const evaluationData = { stop: stop! };\n\n    try {\n      data.runningEvaluations.add(evaluationData);\n      // Notice that we have to await this executor.\n      // If we didn't, we would decrease the amount of running evaluations\n      // before the promise actually resolves\n      return await Promise.race([executor(), stopPromise]);\n    } catch (error) {\n      throw new WrappedSnapError(error);\n    } finally {\n      data.runningEvaluations.delete(evaluationData);\n\n      if (data.runningEvaluations.size === 0) {\n        this.lastTeardown += 1;\n        await data.idleTeardown();\n      }\n    }\n  }\n}\n"]}

package/dist/common/BaseSnapExecutor.mjs

@@ -1,9 +1,3 @@
-var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (receiver, state, kind, f) {
-    if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a getter");
-    if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot read private member from an object whose class did not declare it");
-    return kind === "m" ? f : kind === "a" ? f.call(receiver) : f ? f.value : state.get(receiver);
-};
-var _BaseSnapExecutor_instances, _BaseSnapExecutor_write, _BaseSnapExecutor_notify, _BaseSnapExecutor_respond;
 // eslint-disable-next-line @typescript-eslint/triple-slash-reference, spaced-comment
 /// <reference path="../../../../node_modules/ses/types.d.ts" />
 import { createIdRemapMiddleware } from "@metamask/json-rpc-engine";
@@ -53,9 +47,14 @@ const EXECUTION_ENVIRONMENT_METHODS = {
     },
 };
 export class BaseSnapExecutor {
+    snapData;
+    commandStream;
+    rpcStream;
+    methods;
+    snapErrorHandler;
+    snapPromiseErrorHandler;
+    lastTeardown = 0;
     constructor(commandStream, rpcStream) {
-        _BaseSnapExecutor_instances.add(this);
-        this.lastTeardown = 0;
         this.snapData = new Map();
         this.commandStream = commandStream;
         this.commandStream.on('data', (data) => {
@@ -99,7 +98,7 @@ export class BaseSnapExecutor {
             shouldIncludeStack: false,
         });
         const errorData = getErrorData(serializedError);
-        __classPrivateFieldGet(this, _BaseSnapExecutor_instances, "m", _BaseSnapExecutor_notify).call(this, {
+        this.#notify({
             method: 'UnhandledError',
             params: {
                 error: {
@@ -120,7 +119,7 @@ export class BaseSnapExecutor {
                 is(message.id, JsonRpcIdStruct)) {
                 // Instead of throwing, we directly respond with an error.
                 // We can only do this if the message ID is still valid.
-                await __classPrivateFieldGet(this, _BaseSnapExecutor_instances, "m", _BaseSnapExecutor_write).call(this, {
+                await this.#write({
                     error: serializeError(rpcErrors.internal('JSON-RPC requests must be JSON serializable objects.')),
                     id: message.id,
                     jsonrpc: '2.0',
@@ -133,7 +132,7 @@ export class BaseSnapExecutor {
         }
         const { id, method, params } = message;
         if (!hasProperty(EXECUTION_ENVIRONMENT_METHODS, method)) {
-            await __classPrivateFieldGet(this, _BaseSnapExecutor_instances, "m", _BaseSnapExecutor_respond).call(this, id, {
+            await this.#respond(id, {
                 error: rpcErrors
                     .methodNotFound({
                     data: {
@@ -149,7 +148,7 @@ export class BaseSnapExecutor {
         const paramsAsArray = sortParamKeys(methodObject.params, params);
         const [error] = validate(paramsAsArray, methodObject.struct);
         if (error) {
-            await __classPrivateFieldGet(this, _BaseSnapExecutor_instances, "m", _BaseSnapExecutor_respond).call(this, id, {
+            await this.#respond(id, {
                 error: rpcErrors
                     .invalidParams({
                     message: `Invalid parameters for method "${method}": ${error.message}.`,
@@ -164,16 +163,56 @@ export class BaseSnapExecutor {
         }
         try {
             const result = await this.methods[method](...paramsAsArray);
-            await __classPrivateFieldGet(this, _BaseSnapExecutor_instances, "m", _BaseSnapExecutor_respond).call(this, id, { result });
+            await this.#respond(id, { result });
         }
         catch (rpcError) {
-            await __classPrivateFieldGet(this, _BaseSnapExecutor_instances, "m", _BaseSnapExecutor_respond).call(this, id, {
+            await this.#respond(id, {
                 error: serializeError(rpcError, {
                     fallbackError,
                 }),
             });
         }
     }
+    // Awaitable function that writes back to the command stream
+    // To prevent snap execution from blocking writing we wrap in a promise
+    // and await it before continuing execution
+    async #write(chunk) {
+        return new Promise((resolve, reject) => {
+            this.commandStream.write(chunk, (error) => {
+                if (error) {
+                    reject(error);
+                    return;
+                }
+                resolve();
+            });
+        });
+    }
+    async #notify(notification) {
+        if (!isValidResponse(notification)) {
+            throw rpcErrors.internal('JSON-RPC notifications must be JSON serializable objects smaller than 64 MB.');
+        }
+        await this.#write({
+            ...notification,
+            jsonrpc: '2.0',
+        });
+    }
+    async #respond(id, response) {
+        if (!isValidResponse(response)) {
+            // Instead of throwing, we directly respond with an error.
+            // This prevents an issue where we wouldn't respond when errors were non-serializable
+            await this.#write({
+                error: serializeError(rpcErrors.internal('JSON-RPC responses must be JSON serializable objects smaller than 64 MB.')),
+                id,
+                jsonrpc: '2.0',
+            });
+            return;
+        }
+        await this.#write({
+            ...response,
+            id,
+            jsonrpc: '2.0',
+        });
+    }
     /**
      * Attempts to evaluate a snap in SES. Generates APIs for the snap. May throw
      * on errors.
@@ -213,7 +252,7 @@ export class BaseSnapExecutor {
                 ethereum,
                 snapId,
                 endowments: _endowments,
-                notify: __classPrivateFieldGet(this, _BaseSnapExecutor_instances, "m", _BaseSnapExecutor_notify).bind(this),
+                notify: this.#notify.bind(this),
             });
             // !!! Ensure that this is the only place the data is being set.
             // Other methods access the object value and mutate its properties.
@@ -293,7 +332,7 @@ export class BaseSnapExecutor {
             const sanitizedArgs = sanitizeRequestArguments(args);
             assertSnapOutboundRequest(sanitizedArgs);
             return await withTeardown((async () => {
-                await __classPrivateFieldGet(this, _BaseSnapExecutor_instances, "m", _BaseSnapExecutor_notify).call(this, {
+                await this.#notify({
                     method: 'OutboundRequest',
                     params: { source: 'snap.request' },
                 });
@@ -301,7 +340,7 @@ export class BaseSnapExecutor {
                     return await originalRequest(sanitizedArgs);
                 }
                 finally {
-                    await __classPrivateFieldGet(this, _BaseSnapExecutor_instances, "m", _BaseSnapExecutor_notify).call(this, {
+                    await this.#notify({
                         method: 'OutboundResponse',
                         params: { source: 'snap.request' },
                     });
@@ -323,7 +362,7 @@ export class BaseSnapExecutor {
             const sanitizedArgs = sanitizeRequestArguments(args);
             assertEthereumOutboundRequest(sanitizedArgs);
             return await withTeardown((async () => {
-                await __classPrivateFieldGet(this, _BaseSnapExecutor_instances, "m", _BaseSnapExecutor_notify).call(this, {
+                await this.#notify({
                     method: 'OutboundRequest',
                     params: { source: 'ethereum.request' },
                 });
@@ -331,7 +370,7 @@ export class BaseSnapExecutor {
                     return await originalRequest(sanitizedArgs);
                 }
                 finally {
-                    await __classPrivateFieldGet(this, _BaseSnapExecutor_instances, "m", _BaseSnapExecutor_notify).call(this, {
+                    await this.#notify({
                         method: 'OutboundResponse',
                         params: { source: 'ethereum.request' },
                     });
@@ -390,43 +429,4 @@ export class BaseSnapExecutor {
         }
     }
 }
-_BaseSnapExecutor_instances = new WeakSet(), _BaseSnapExecutor_write = 
-// Awaitable function that writes back to the command stream
-// To prevent snap execution from blocking writing we wrap in a promise
-// and await it before continuing execution
-async function _BaseSnapExecutor_write(chunk) {
-    return new Promise((resolve, reject) => {
-        this.commandStream.write(chunk, (error) => {
-            if (error) {
-                reject(error);
-                return;
-            }
-            resolve();
-        });
-    });
-}, _BaseSnapExecutor_notify = async function _BaseSnapExecutor_notify(notification) {
-    if (!isValidResponse(notification)) {
-        throw rpcErrors.internal('JSON-RPC notifications must be JSON serializable objects smaller than 64 MB.');
-    }
-    await __classPrivateFieldGet(this, _BaseSnapExecutor_instances, "m", _BaseSnapExecutor_write).call(this, {
-        ...notification,
-        jsonrpc: '2.0',
-    });
-}, _BaseSnapExecutor_respond = async function _BaseSnapExecutor_respond(id, response) {
-    if (!isValidResponse(response)) {
-        // Instead of throwing, we directly respond with an error.
-        // This prevents an issue where we wouldn't respond when errors were non-serializable
-        await __classPrivateFieldGet(this, _BaseSnapExecutor_instances, "m", _BaseSnapExecutor_write).call(this, {
-            error: serializeError(rpcErrors.internal('JSON-RPC responses must be JSON serializable objects smaller than 64 MB.')),
-            id,
-            jsonrpc: '2.0',
-        });
-        return;
-    }
-    await __classPrivateFieldGet(this, _BaseSnapExecutor_instances, "m", _BaseSnapExecutor_write).call(this, {
-        ...response,
-        id,
-        jsonrpc: '2.0',
-    });
-};
 //# sourceMappingURL=BaseSnapExecutor.mjs.map