@metamask/snaps-execution-environments 0.24.0

package/dist/common/endowments/network.js

@@ -0,0 +1,362 @@
+"use strict";
+var __classPrivateFieldSet = (this && this.__classPrivateFieldSet) || function (receiver, state, value, kind, f) {
+    if (kind === "m") throw new TypeError("Private method is not writable");
+    if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a setter");
+    if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot write private member to an object whose class did not declare it");
+    return (kind === "a" ? f.call(receiver, value) : f ? f.value = value : state.set(receiver, value)), value;
+};
+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 _ResponseWrapper_ogResponse;
+Object.defineProperty(exports, "__esModule", { value: true });
+const utils_1 = require("../utils");
+/**
+ * This class wraps a Response object.
+ * That way, a teardown process can stop any processes left.
+ */
+class ResponseWrapper {
+    constructor(ogResponse, teardownRef) {
+        _ResponseWrapper_ogResponse.set(this, void 0);
+        __classPrivateFieldSet(this, _ResponseWrapper_ogResponse, ogResponse, "f");
+        this.teardownRef = teardownRef;
+    }
+    get body() {
+        return __classPrivateFieldGet(this, _ResponseWrapper_ogResponse, "f").body;
+    }
+    get bodyUsed() {
+        return __classPrivateFieldGet(this, _ResponseWrapper_ogResponse, "f").bodyUsed;
+    }
+    get headers() {
+        return __classPrivateFieldGet(this, _ResponseWrapper_ogResponse, "f").headers;
+    }
+    get ok() {
+        return __classPrivateFieldGet(this, _ResponseWrapper_ogResponse, "f").ok;
+    }
+    get redirected() {
+        return __classPrivateFieldGet(this, _ResponseWrapper_ogResponse, "f").redirected;
+    }
+    get status() {
+        return __classPrivateFieldGet(this, _ResponseWrapper_ogResponse, "f").status;
+    }
+    get statusText() {
+        return __classPrivateFieldGet(this, _ResponseWrapper_ogResponse, "f").statusText;
+    }
+    get type() {
+        return __classPrivateFieldGet(this, _ResponseWrapper_ogResponse, "f").type;
+    }
+    get url() {
+        return __classPrivateFieldGet(this, _ResponseWrapper_ogResponse, "f").url;
+    }
+    text() {
+        return (0, utils_1.withTeardown)(__classPrivateFieldGet(this, _ResponseWrapper_ogResponse, "f").text(), this);
+    }
+    arrayBuffer() {
+        return (0, utils_1.withTeardown)(__classPrivateFieldGet(this, _ResponseWrapper_ogResponse, "f").arrayBuffer(), this);
+    }
+    blob() {
+        return (0, utils_1.withTeardown)(__classPrivateFieldGet(this, _ResponseWrapper_ogResponse, "f").blob(), this);
+    }
+    clone() {
+        const newResponse = __classPrivateFieldGet(this, _ResponseWrapper_ogResponse, "f").clone();
+        return new ResponseWrapper(newResponse, this.teardownRef);
+    }
+    formData() {
+        return (0, utils_1.withTeardown)(__classPrivateFieldGet(this, _ResponseWrapper_ogResponse, "f").formData(), this);
+    }
+    json() {
+        return (0, utils_1.withTeardown)(__classPrivateFieldGet(this, _ResponseWrapper_ogResponse, "f").json(), this);
+    }
+}
+_ResponseWrapper_ogResponse = new WeakMap();
+/**
+ * Create a network endowment, consisting of a `WebSocket` object and `fetch`
+ * function. This allows us to provide a teardown function, so that we can cancel
+ * any pending requests, connections, streams, etc. that may be open when a snap
+ * is terminated.
+ *
+ * This wraps the original implementations of `WebSocket` and `fetch`,
+ * to ensure that a bad actor cannot get access to the original objects, thus
+ * potentially preventing the network requests from being torn down.
+ *
+ * @returns An object containing a wrapped `WebSocket` class and `fetch`
+ * function, as well as a teardown function.
+ */
+const createNetwork = () => {
+    var _instances, _teardownClose, _createWrapped, _socket, _isTornDown, _events, _onopenOriginal, _onmessageOriginal, _onerrorOriginal, _oncloseOriginal, _a;
+    // Open fetch calls or open body streams or open websockets
+    const openConnections = new Set();
+    // Track last teardown count
+    const teardownRef = { lastTeardown: 0 };
+    // Remove items from openConnections after they were garbage collected
+    const cleanup = new FinalizationRegistry(
+    /* istanbul ignore next: can't test garbage collection without modifying node parameters */
+    (callback) => callback());
+    const _fetch = async (input, init) => {
+        const abortController = new AbortController();
+        if ((init === null || init === void 0 ? void 0 : init.signal) !== null && (init === null || init === void 0 ? void 0 : init.signal) !== undefined) {
+            const originalSignal = init.signal;
+            // Merge abort controllers
+            originalSignal.addEventListener('abort', () => {
+                abortController.abort(originalSignal.reason);
+            }, { once: true });
+        }
+        let res;
+        let openFetchConnection;
+        try {
+            const fetchPromise = fetch(input, Object.assign(Object.assign({}, init), { signal: abortController.signal }));
+            openFetchConnection = {
+                cancel: async () => {
+                    abortController.abort();
+                    try {
+                        await fetchPromise;
+                    }
+                    catch (_a) {
+                        /* do nothing */
+                    }
+                },
+            };
+            openConnections.add(openFetchConnection);
+            res = new ResponseWrapper(await (0, utils_1.withTeardown)(fetchPromise, teardownRef), teardownRef);
+        }
+        finally {
+            if (openFetchConnection !== undefined) {
+                openConnections.delete(openFetchConnection);
+            }
+        }
+        if (res.body !== null) {
+            const body = new WeakRef(res.body);
+            const openBodyConnection = {
+                cancel: 
+                /* istanbul ignore next: see it.todo('can be torn down during body read') test */
+                async () => {
+                    var _a;
+                    try {
+                        await ((_a = body.deref()) === null || _a === void 0 ? void 0 : _a.cancel());
+                    }
+                    catch (_b) {
+                        /* do nothing */
+                    }
+                },
+            };
+            openConnections.add(openBodyConnection);
+            cleanup.register(res.body, 
+            /* istanbul ignore next: can't test garbage collection without modifying node parameters */
+            () => openConnections.delete(openBodyConnection));
+        }
+        return res;
+    };
+    /**
+     * This class wraps a WebSocket object instead of extending it.
+     * That way, a bad actor can't get access to original methods using
+     * `socket.prototype`.
+     *
+     * When modifying this class, ensure that no method calls any other method
+     * from the same class (#socket calls are fine). Otherwise, a bad actor could
+     * replace one of the implementations
+     */
+    const _WebSocket = (_a = class {
+            constructor(url, protocols) {
+                _instances.add(this);
+                _socket.set(this, void 0);
+                /**
+                 * After this is set to true, no new event listeners can be added
+                 */
+                _isTornDown.set(this, false);
+                _events.set(this, {});
+                _onopenOriginal.set(this, null);
+                _onmessageOriginal.set(this, null);
+                _onerrorOriginal.set(this, null);
+                _oncloseOriginal.set(this, null);
+                __classPrivateFieldSet(this, _socket, new WebSocket(url, protocols), "f");
+                // You can't call ref.deref()?.#teardownClose()
+                // But you can capture the close itself
+                const ref = new WeakRef(__classPrivateFieldGet(this, _instances, "m", _teardownClose).bind(this));
+                const openConnection = {
+                    cancel: async () => {
+                        var _a;
+                        try {
+                            await ((_a = ref.deref()) === null || _a === void 0 ? void 0 : _a());
+                        }
+                        catch (_b) {
+                            /* do nothing */
+                        }
+                    },
+                };
+                openConnections.add(openConnection);
+                cleanup.register(this, 
+                /* istanbul ignore next: can't test garbage collection without modifying node parameters */
+                () => openConnections.delete(openConnection));
+            }
+            get onclose() {
+                return __classPrivateFieldGet(this, _oncloseOriginal, "f");
+            }
+            set onclose(callback) {
+                if (__classPrivateFieldGet(this, _isTornDown, "f") !== true) {
+                    __classPrivateFieldSet(this, _oncloseOriginal, callback, "f");
+                    __classPrivateFieldGet(this, _socket, "f").onclose = __classPrivateFieldGet(this, _instances, "m", _createWrapped).call(this, callback);
+                }
+            }
+            get onerror() {
+                return __classPrivateFieldGet(this, _onerrorOriginal, "f");
+            }
+            set onerror(callback) {
+                __classPrivateFieldSet(this, _onerrorOriginal, callback, "f");
+                __classPrivateFieldGet(this, _socket, "f").onerror = __classPrivateFieldGet(this, _instances, "m", _createWrapped).call(this, callback);
+            }
+            get onmessage() {
+                return __classPrivateFieldGet(this, _onmessageOriginal, "f");
+            }
+            set onmessage(callback) {
+                __classPrivateFieldSet(this, _onmessageOriginal, callback, "f");
+                __classPrivateFieldGet(this, _socket, "f").onmessage = __classPrivateFieldGet(this, _instances, "m", _createWrapped).call(this, callback);
+            }
+            get onopen() {
+                return __classPrivateFieldGet(this, _onopenOriginal, "f");
+            }
+            set onopen(callback) {
+                __classPrivateFieldSet(this, _onopenOriginal, callback, "f");
+                __classPrivateFieldGet(this, _socket, "f").onopen = __classPrivateFieldGet(this, _instances, "m", _createWrapped).call(this, callback);
+            }
+            close(code, reason) {
+                __classPrivateFieldGet(this, _socket, "f").close(code, reason);
+            }
+            send(data) {
+                __classPrivateFieldGet(this, _socket, "f").send(data);
+            }
+            get CLOSED() {
+                return __classPrivateFieldGet(this, _socket, "f").CLOSED;
+            }
+            get CLOSING() {
+                return __classPrivateFieldGet(this, _socket, "f").CLOSING;
+            }
+            get CONNECTING() {
+                return __classPrivateFieldGet(this, _socket, "f").CONNECTING;
+            }
+            get OPEN() {
+                return __classPrivateFieldGet(this, _socket, "f").OPEN;
+            }
+            get binaryType() {
+                return __classPrivateFieldGet(this, _socket, "f").binaryType;
+            }
+            set binaryType(value) {
+                __classPrivateFieldGet(this, _socket, "f").binaryType = value;
+            }
+            get bufferedAmount() {
+                var _a;
+                return (_a = __classPrivateFieldGet(this, _socket, "f").bufferedAmount) !== null && _a !== void 0 ? _a : 0;
+            }
+            get extensions() {
+                var _a;
+                return (_a = __classPrivateFieldGet(this, _socket, "f").extensions) !== null && _a !== void 0 ? _a : '';
+            }
+            get protocol() {
+                return __classPrivateFieldGet(this, _socket, "f").protocol;
+            }
+            get readyState() {
+                return __classPrivateFieldGet(this, _socket, "f").readyState;
+            }
+            get url() {
+                return __classPrivateFieldGet(this, _socket, "f").url;
+            }
+            addEventListener(type, listener, options) {
+                if (__classPrivateFieldGet(this, _isTornDown, "f") !== true) {
+                    if (__classPrivateFieldGet(this, _events, "f")[type] === undefined) {
+                        __classPrivateFieldGet(this, _events, "f")[type] = new Map();
+                    }
+                    const wrapped = __classPrivateFieldGet(this, _instances, "m", _createWrapped).call(this, listener);
+                    if (wrapped !== null) {
+                        __classPrivateFieldGet(this, _events, "f")[type].set(listener, wrapped);
+                        __classPrivateFieldGet(this, _socket, "f").addEventListener(type, wrapped, options);
+                    }
+                }
+            }
+            removeEventListener(type, listener, options) {
+                if (__classPrivateFieldGet(this, _events, "f")[type] !== undefined) {
+                    const wrapped = __classPrivateFieldGet(this, _events, "f")[type].get(listener);
+                    if (wrapped !== undefined) {
+                        __classPrivateFieldGet(this, _events, "f")[type].delete(listener);
+                        __classPrivateFieldGet(this, _socket, "f").removeEventListener(type, wrapped, options);
+                    }
+                }
+            }
+            dispatchEvent(event) {
+                // Can't call close prematurely before the teardown finishes
+                if (event.type !== 'close' || __classPrivateFieldGet(this, _isTornDown, "f") !== true) {
+                    return __classPrivateFieldGet(this, _socket, "f").dispatchEvent(event);
+                }
+                return false;
+            }
+        },
+        _socket = new WeakMap(),
+        _isTornDown = new WeakMap(),
+        _events = new WeakMap(),
+        _onopenOriginal = new WeakMap(),
+        _onmessageOriginal = new WeakMap(),
+        _onerrorOriginal = new WeakMap(),
+        _oncloseOriginal = new WeakMap(),
+        _instances = new WeakSet(),
+        _teardownClose = function _teardownClose() {
+            var _a, _b, _c;
+            // We clear all close listeners
+            __classPrivateFieldGet(this, _socket, "f").onclose = null;
+            for (const wrapped of (_b = (_a = __classPrivateFieldGet(this, _events, "f").close) === null || _a === void 0 ? void 0 : _a.values()) !== null && _b !== void 0 ? _b : []) {
+                __classPrivateFieldGet(this, _socket, "f").removeEventListener('close', wrapped);
+            }
+            (_c = __classPrivateFieldGet(this, _events, "f").close) === null || _c === void 0 ? void 0 : _c.clear();
+            // We add our own listener
+            let onClosedResolve;
+            const onClosed = new Promise((r) => (onClosedResolve = r));
+            __classPrivateFieldGet(this, _socket, "f").onclose = () => {
+                onClosedResolve();
+            };
+            // No new listeners can be added after this point
+            __classPrivateFieldSet(this, _isTornDown, true, "f");
+            __classPrivateFieldGet(this, _socket, "f").close();
+            return onClosed;
+        },
+        _createWrapped = function _createWrapped(listener) {
+            if (listener === null) {
+                return null;
+            }
+            return (e) => {
+                // TODO: Should we migrate this to use a wrapper class?
+                const properties = [...(0, utils_1.allProperties)(e)]
+                    .filter(([_, key]) => key !== 'constructor')
+                    .reduce((acc, [obj, key]) => {
+                    const stringKey = key.toString();
+                    const descriptor = Reflect.getOwnPropertyDescriptor(obj, key);
+                    if (typeof (descriptor === null || descriptor === void 0 ? void 0 : descriptor.value) === 'function') {
+                        acc[stringKey] = e[stringKey].bind(this);
+                    }
+                    else {
+                        acc[stringKey] = e[stringKey];
+                    }
+                    return acc;
+                }, {});
+                const event = Object.assign(Object.assign({}, properties), { target: this, currentTarget: this, srcElement: this, ports: [this], source: null, composedPath: () => [this] });
+                listener.apply(this, [event]);
+            };
+        },
+        _a);
+    const teardownFunction = async () => {
+        teardownRef.lastTeardown += 1;
+        const promises = [];
+        openConnections.forEach(({ cancel }) => promises.push(cancel()));
+        openConnections.clear();
+        await Promise.all(promises);
+    };
+    return {
+        fetch: _fetch,
+        WebSocket: _WebSocket,
+        teardownFunction,
+    };
+};
+const endowmentModule = {
+    names: ['fetch', 'WebSocket'],
+    factory: createNetwork,
+};
+exports.default = endowmentModule;
+//# sourceMappingURL=network.js.map

package/dist/common/endowments/network.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"network.js","sourceRoot":"","sources":["../../../src/common/endowments/network.ts"],"names":[],"mappings":";;;;;;;;;;;;;;AAAA,oCAAuD;AAIvD;;;GAGG;AACH,MAAM,eAAe;IAKnB,YAAY,UAAoB,EAAE,WAAqC;QAFvE,8CAAsB;QAGpB,uBAAA,IAAI,+BAAe,UAAU,MAAA,CAAC;QAC9B,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;IACjC,CAAC;IAED,IAAI,IAAI;QACN,OAAO,uBAAA,IAAI,mCAAY,CAAC,IAAI,CAAC;IAC/B,CAAC;IAED,IAAI,QAAQ;QACV,OAAO,uBAAA,IAAI,mCAAY,CAAC,QAAQ,CAAC;IACnC,CAAC;IAED,IAAI,OAAO;QACT,OAAO,uBAAA,IAAI,mCAAY,CAAC,OAAO,CAAC;IAClC,CAAC;IAED,IAAI,EAAE;QACJ,OAAO,uBAAA,IAAI,mCAAY,CAAC,EAAE,CAAC;IAC7B,CAAC;IAED,IAAI,UAAU;QACZ,OAAO,uBAAA,IAAI,mCAAY,CAAC,UAAU,CAAC;IACrC,CAAC;IAED,IAAI,MAAM;QACR,OAAO,uBAAA,IAAI,mCAAY,CAAC,MAAM,CAAC;IACjC,CAAC;IAED,IAAI,UAAU;QACZ,OAAO,uBAAA,IAAI,mCAAY,CAAC,UAAU,CAAC;IACrC,CAAC;IAED,IAAI,IAAI;QACN,OAAO,uBAAA,IAAI,mCAAY,CAAC,IAAI,CAAC;IAC/B,CAAC;IAED,IAAI,GAAG;QACL,OAAO,uBAAA,IAAI,mCAAY,CAAC,GAAG,CAAC;IAC9B,CAAC;IAED,IAAI;QACF,OAAO,IAAA,oBAAY,EAAS,uBAAA,IAAI,mCAAY,CAAC,IAAI,EAAE,EAAE,IAAW,CAAC,CAAC;IACpE,CAAC;IAED,WAAW;QACT,OAAO,IAAA,oBAAY,EACjB,uBAAA,IAAI,mCAAY,CAAC,WAAW,EAAE,EAC9B,IAAW,CACZ,CAAC;IACJ,CAAC;IAED,IAAI;QACF,OAAO,IAAA,oBAAY,EAAO,uBAAA,IAAI,mCAAY,CAAC,IAAI,EAAE,EAAE,IAAW,CAAC,CAAC;IAClE,CAAC;IAED,KAAK;QACH,MAAM,WAAW,GAAG,uBAAA,IAAI,mCAAY,CAAC,KAAK,EAAE,CAAC;QAC7C,OAAO,IAAI,eAAe,CAAC,WAAW,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;IAC5D,CAAC;IAED,QAAQ;QACN,OAAO,IAAA,oBAAY,EAAW,uBAAA,IAAI,mCAAY,CAAC,QAAQ,EAAE,EAAE,IAAW,CAAC,CAAC;IAC1E,CAAC;IAED,IAAI;QACF,OAAO,IAAA,oBAAY,EAAC,uBAAA,IAAI,mCAAY,CAAC,IAAI,EAAE,EAAE,IAAW,CAAC,CAAC;IAC5D,CAAC;CACF;;AAED;;;;;;;;;;;;GAYG;AACH,MAAM,aAAa,GAAG,GAAG,EAAE;;IACzB,2DAA2D;IAC3D,MAAM,eAAe,GAAG,IAAI,GAAG,EAAmC,CAAC;IACnE,4BAA4B;IAC5B,MAAM,WAAW,GAAG,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC;IAExC,sEAAsE;IACtE,MAAM,OAAO,GAAG,IAAI,oBAAoB;IACtC,2FAA2F;IAC3F,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,EAAE,CACzB,CAAC;IAEF,MAAM,MAAM,GAAiB,KAAK,EAChC,KAAkB,EAClB,IAAkB,EACC,EAAE;QACrB,MAAM,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;QAC9C,IAAI,CAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,MAAM,MAAK,IAAI,IAAI,CAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,MAAM,MAAK,SAAS,EAAE;YACvD,MAAM,cAAc,GAAG,IAAI,CAAC,MAAM,CAAC;YACnC,0BAA0B;YAC1B,cAAc,CAAC,gBAAgB,CAC7B,OAAO,EACP,GAAG,EAAE;gBACH,eAAe,CAAC,KAAK,CAAE,cAAsB,CAAC,MAAM,CAAC,CAAC;YACxD,CAAC,EACD,EAAE,IAAI,EAAE,IAAI,EAAE,CACf,CAAC;SACH;QAED,IAAI,GAAa,CAAC;QAClB,IAAI,mBAAgE,CAAC;QACrE,IAAI;YACF,MAAM,YAAY,GAAG,KAAK,CAAC,KAAK,kCAC3B,IAAI,KACP,MAAM,EAAE,eAAe,CAAC,MAAM,IAC9B,CAAC;YAEH,mBAAmB,GAAG;gBACpB,MAAM,EAAE,KAAK,IAAI,EAAE;oBACjB,eAAe,CAAC,KAAK,EAAE,CAAC;oBACxB,IAAI;wBACF,MAAM,YAAY,CAAC;qBACpB;oBAAC,WAAM;wBACN,gBAAgB;qBACjB;gBACH,CAAC;aACF,CAAC;YACF,eAAe,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC;YAEzC,GAAG,GAAG,IAAI,eAAe,CACvB,MAAM,IAAA,oBAAY,EAAC,YAAY,EAAE,WAAW,CAAC,EAC7C,WAAW,CACZ,CAAC;SACH;gBAAS;YACR,IAAI,mBAAmB,KAAK,SAAS,EAAE;gBACrC,eAAe,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC;aAC7C;SACF;QAED,IAAI,GAAG,CAAC,IAAI,KAAK,IAAI,EAAE;YACrB,MAAM,IAAI,GAAG,IAAI,OAAO,CAAiB,GAAG,CAAC,IAAI,CAAC,CAAC;YAEnD,MAAM,kBAAkB,GAAG;gBACzB,MAAM;gBACJ,iFAAiF;gBACjF,KAAK,IAAI,EAAE;;oBACT,IAAI;wBACF,MAAM,CAAA,MAAA,IAAI,CAAC,KAAK,EAAE,0CAAE,MAAM,EAAE,CAAA,CAAC;qBAC9B;oBAAC,WAAM;wBACN,gBAAgB;qBACjB;gBACH,CAAC;aACJ,CAAC;YACF,eAAe,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC;YACxC,OAAO,CAAC,QAAQ,CACd,GAAG,CAAC,IAAI;YACR,2FAA2F;YAC3F,GAAG,EAAE,CAAC,eAAe,CAAC,MAAM,CAAC,kBAAkB,CAAC,CACjD,CAAC;SACH;QACD,OAAO,GAAG,CAAC;IACb,CAAC,CAAC;IAEF;;;;;;;;OAQG;IACH,MAAM,UAAU,SAAG;YACjB,YAAY,GAAiB,EAAE,SAA6B;;gBA2O5D,0BAAmB;gBAEnB;;mBAEG;gBACH,sBAAc,KAAK,EAAC;gBAEpB,kBAAqE,EAAE,EAAC;gBAExE,0BAA4C,IAAI,EAAC;gBAEjD,6BAA+C,IAAI,EAAC;gBAEpD,2BAA6C,IAAI,EAAC;gBAElD,2BAA6C,IAAI,EAAC;gBAzPhD,uBAAA,IAAI,WAAW,IAAI,SAAS,CAAC,GAAG,EAAE,SAAS,CAAC,MAAA,CAAC;gBAE7C,+CAA+C;gBAC/C,uCAAuC;gBACvC,MAAM,GAAG,GAAG,IAAI,OAAO,CAAC,uBAAA,IAAI,kCAAe,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;gBACxD,MAAM,cAAc,GAAG;oBACrB,MAAM,EAAE,KAAK,IAAI,EAAE;;wBACjB,IAAI;4BACF,MAAM,CAAA,MAAA,GAAG,CAAC,KAAK,EAAE,2CAAI,CAAA,CAAC;yBACvB;wBAAC,WAAM;4BACN,gBAAgB;yBACjB;oBACH,CAAC;iBACF,CAAC;gBACF,eAAe,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;gBACpC,OAAO,CAAC,QAAQ,CACd,IAAI;gBACJ,2FAA2F;gBAC3F,GAAG,EAAE,CAAC,eAAe,CAAC,MAAM,CAAC,cAAc,CAAC,CAC7C,CAAC;YACJ,CAAC;YAED,IAAI,OAAO;gBACT,OAAO,uBAAA,IAAI,wBAAiB,CAAC;YAC/B,CAAC;YAED,IAAI,OAAO,CAAC,QAAkC;gBAC5C,IAAI,uBAAA,IAAI,mBAAY,KAAK,IAAI,EAAE;oBAC7B,uBAAA,IAAI,oBAAoB,QAAQ,MAAA,CAAC;oBACjC,uBAAA,IAAI,eAAQ,CAAC,OAAO,GAAG,uBAAA,IAAI,kCAAe,MAAnB,IAAI,EAAgB,QAAQ,CAAC,CAAC;iBACtD;YACH,CAAC;YAED,IAAI,OAAO;gBACT,OAAO,uBAAA,IAAI,wBAAiB,CAAC;YAC/B,CAAC;YAED,IAAI,OAAO,CAAC,QAAsD;gBAChE,uBAAA,IAAI,oBAAoB,QAAQ,MAAA,CAAC;gBACjC,uBAAA,IAAI,eAAQ,CAAC,OAAO,GAAG,uBAAA,IAAI,kCAAe,MAAnB,IAAI,EAAgB,QAAQ,CAAC,CAAC;YACvD,CAAC;YAED,IAAI,SAAS;gBACX,OAAO,uBAAA,IAAI,0BAAmB,CAAC;YACjC,CAAC;YAED,IAAI,SAAS,CACX,QAAkE;gBAElE,uBAAA,IAAI,sBAAsB,QAAQ,MAAA,CAAC;gBACnC,uBAAA,IAAI,eAAQ,CAAC,SAAS,GAAG,uBAAA,IAAI,kCAAe,MAAnB,IAAI,EAAgB,QAAQ,CAAC,CAAC;YACzD,CAAC;YAED,IAAI,MAAM;gBACR,OAAO,uBAAA,IAAI,uBAAgB,CAAC;YAC9B,CAAC;YAED,IAAI,MAAM,CAAC,QAAsD;gBAC/D,uBAAA,IAAI,mBAAmB,QAAQ,MAAA,CAAC;gBAChC,uBAAA,IAAI,eAAQ,CAAC,MAAM,GAAG,uBAAA,IAAI,kCAAe,MAAnB,IAAI,EAAgB,QAAQ,CAAC,CAAC;YACtD,CAAC;YAED,KAAK,CAAC,IAAa,EAAE,MAAe;gBAClC,uBAAA,IAAI,eAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;YACnC,CAAC;YAED,IAAI,CAAC,IAAuD;gBAC1D,uBAAA,IAAI,eAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC1B,CAAC;YAED,IAAI,MAAM;gBACR,OAAO,uBAAA,IAAI,eAAQ,CAAC,MAAM,CAAC;YAC7B,CAAC;YAED,IAAI,OAAO;gBACT,OAAO,uBAAA,IAAI,eAAQ,CAAC,OAAO,CAAC;YAC9B,CAAC;YAED,IAAI,UAAU;gBACZ,OAAO,uBAAA,IAAI,eAAQ,CAAC,UAAU,CAAC;YACjC,CAAC;YAED,IAAI,IAAI;gBACN,OAAO,uBAAA,IAAI,eAAQ,CAAC,IAAI,CAAC;YAC3B,CAAC;YAED,IAAI,UAAU;gBACZ,OAAO,uBAAA,IAAI,eAAQ,CAAC,UAAU,CAAC;YACjC,CAAC;YAED,IAAI,UAAU,CAAC,KAAiB;gBAC9B,uBAAA,IAAI,eAAQ,CAAC,UAAU,GAAG,KAAK,CAAC;YAClC,CAAC;YAED,IAAI,cAAc;;gBAChB,OAAO,MAAA,uBAAA,IAAI,eAAQ,CAAC,cAAc,mCAAI,CAAC,CAAC;YAC1C,CAAC;YAED,IAAI,UAAU;;gBACZ,OAAO,MAAA,uBAAA,IAAI,eAAQ,CAAC,UAAU,mCAAI,EAAE,CAAC;YACvC,CAAC;YAED,IAAI,QAAQ;gBACV,OAAO,uBAAA,IAAI,eAAQ,CAAC,QAAQ,CAAC;YAC/B,CAAC;YAED,IAAI,UAAU;gBACZ,OAAO,uBAAA,IAAI,eAAQ,CAAC,UAAU,CAAC;YACjC,CAAC;YAED,IAAI,GAAG;gBACL,OAAO,uBAAA,IAAI,eAAQ,CAAC,GAAG,CAAC;YAC1B,CAAC;YAcD,gBAAgB,CAAC,IAAS,EAAE,QAAa,EAAE,OAAa;gBACtD,IAAI,uBAAA,IAAI,mBAAY,KAAK,IAAI,EAAE;oBAC7B,IAAI,uBAAA,IAAI,eAAQ,CAAC,IAAI,CAAC,KAAK,SAAS,EAAE;wBACpC,uBAAA,IAAI,eAAQ,CAAC,IAAI,CAAC,GAAG,IAAI,GAAG,EAAE,CAAC;qBAChC;oBACD,MAAM,OAAO,GAAG,uBAAA,IAAI,kCAAe,MAAnB,IAAI,EAAgB,QAAQ,CAAC,CAAC;oBAC9C,IAAI,OAAO,KAAK,IAAI,EAAE;wBACpB,uBAAA,IAAI,eAAQ,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;wBAC1C,uBAAA,IAAI,eAAQ,CAAC,gBAAgB,CAAC,IAAI,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;qBACvD;iBACF;YACH,CAAC;YAcD,mBAAmB,CAAC,IAAS,EAAE,QAAa,EAAE,OAAa;gBACzD,IAAI,uBAAA,IAAI,eAAQ,CAAC,IAAI,CAAC,KAAK,SAAS,EAAE;oBACpC,MAAM,OAAO,GAAG,uBAAA,IAAI,eAAQ,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;oBACjD,IAAI,OAAO,KAAK,SAAS,EAAE;wBACzB,uBAAA,IAAI,eAAQ,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;wBACpC,uBAAA,IAAI,eAAQ,CAAC,mBAAmB,CAAC,IAAW,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;qBACjE;iBACF;YACH,CAAC;YAED,aAAa,CAAC,KAAY;gBACxB,4DAA4D;gBAC5D,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO,IAAI,uBAAA,IAAI,mBAAY,KAAK,IAAI,EAAE;oBACvD,OAAO,uBAAA,IAAI,eAAQ,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;iBAC1C;gBACD,OAAO,KAAK,CAAC;YACf,CAAC;SAmFF;;;;;;;;;;;YAhFG,+BAA+B;YAC/B,uBAAA,IAAI,eAAQ,CAAC,OAAO,GAAG,IAAI,CAAC;YAC5B,KAAK,MAAM,OAAO,IAAI,MAAA,MAAA,uBAAA,IAAI,eAAQ,CAAC,KAAK,0CAAE,MAAM,EAAE,mCAAI,EAAE,EAAE;gBACxD,uBAAA,IAAI,eAAQ,CAAC,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;aACpD;YACD,MAAA,uBAAA,IAAI,eAAQ,CAAC,KAAK,0CAAE,KAAK,EAAE,CAAC;YAE5B,0BAA0B;YAC1B,IAAI,eAAoB,CAAC;YACzB,MAAM,QAAQ,GAAG,IAAI,OAAO,CAAO,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,eAAe,GAAG,CAAC,CAAC,CAAC,CAAC;YACjE,uBAAA,IAAI,eAAQ,CAAC,OAAO,GAAG,GAAG,EAAE;gBAC1B,eAAe,EAAE,CAAC;YACpB,CAAC,CAAC;YAEF,iDAAiD;YACjD,uBAAA,IAAI,eAAe,IAAI,MAAA,CAAC;YAExB,uBAAA,IAAI,eAAQ,CAAC,KAAK,EAAE,CAAC;YAErB,OAAO,QAAQ,CAAC;QAClB,CAAC;iDAWC,QAAkC;YAElC,IAAI,QAAQ,KAAK,IAAI,EAAE;gBACrB,OAAO,IAAI,CAAC;aACb;YAED,OAAO,CAAC,CAAC,EAAE,EAAE;gBACX,uDAAuD;gBACvD,MAAM,UAAU,GAAG,CAAC,GAAG,IAAA,qBAAa,EAAC,CAAC,CAAC,CAAC;qBACrC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,EAAE,CAAC,GAAG,KAAK,aAAa,CAAC;qBAC3C,MAAM,CAAsB,CAAC,GAAG,EAAE,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,EAAE;oBAC/C,MAAM,SAAS,GAAG,GAAG,CAAC,QAAQ,EAAE,CAAC;oBACjC,MAAM,UAAU,GAAG,OAAO,CAAC,wBAAwB,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;oBAC9D,IAAI,OAAO,CAAA,UAAU,aAAV,UAAU,uBAAV,UAAU,CAAE,KAAK,CAAA,KAAK,UAAU,EAAE;wBAC3C,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;qBAC1C;yBAAM;wBACL,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC;qBAC/B;oBACD,OAAO,GAAG,CAAC;gBACb,CAAC,EAAE,EAAE,CAAC,CAAC;gBACT,MAAM,KAAK,mCACN,UAAU,KACb,MAAM,EAAE,IAAI,EACZ,aAAa,EAAE,IAAI,EACnB,UAAU,EAAE,IAAI,EAChB,KAAK,EAAE,CAAC,IAAI,CAAC,EACb,MAAM,EAAE,IAAI,EACZ,YAAY,EAAE,GAAG,EAAE,CAAC,CAAC,IAAI,CAAC,GAC3B,CAAC;gBACF,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC;YAChC,CAAC,CAAC;QACJ,CAAC;WAkBF,CAAC;IAEF,MAAM,gBAAgB,GAAG,KAAK,IAAI,EAAE;QAClC,WAAW,CAAC,YAAY,IAAI,CAAC,CAAC;QAC9B,MAAM,QAAQ,GAAoB,EAAE,CAAC;QACrC,eAAe,CAAC,OAAO,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QACjE,eAAe,CAAC,KAAK,EAAE,CAAC;QACxB,MAAM,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IAC9B,CAAC,CAAC;IAEF,OAAO;QACL,KAAK,EAAE,MAAM;QACb,SAAS,EAAE,UAAU;QACrB,gBAAgB;KACjB,CAAC;AACJ,CAAC,CAAC;AAEF,MAAM,eAAe,GAAG;IACtB,KAAK,EAAE,CAAC,OAAO,EAAE,WAAW,CAAU;IACtC,OAAO,EAAE,aAAa;CACvB,CAAC;AACF,kBAAe,eAAe,CAAC","sourcesContent":["import { allProperties, withTeardown } from '../utils';\n\ntype WebSocketCallback = (this: WebSocket, ev: any) => any;\n\n/**\n * This class wraps a Response object.\n * That way, a teardown process can stop any processes left.\n */\nclass ResponseWrapper implements Response {\n  private readonly teardownRef: { lastTeardown: number };\n\n  #ogResponse: Response;\n\n  constructor(ogResponse: Response, teardownRef: { lastTeardown: number }) {\n    this.#ogResponse = ogResponse;\n    this.teardownRef = teardownRef;\n  }\n\n  get body(): ReadableStream<Uint8Array> | null {\n    return this.#ogResponse.body;\n  }\n\n  get bodyUsed() {\n    return this.#ogResponse.bodyUsed;\n  }\n\n  get headers() {\n    return this.#ogResponse.headers;\n  }\n\n  get ok() {\n    return this.#ogResponse.ok;\n  }\n\n  get redirected() {\n    return this.#ogResponse.redirected;\n  }\n\n  get status() {\n    return this.#ogResponse.status;\n  }\n\n  get statusText() {\n    return this.#ogResponse.statusText;\n  }\n\n  get type() {\n    return this.#ogResponse.type;\n  }\n\n  get url() {\n    return this.#ogResponse.url;\n  }\n\n  text() {\n    return withTeardown<string>(this.#ogResponse.text(), this as any);\n  }\n\n  arrayBuffer(): Promise<ArrayBuffer> {\n    return withTeardown<ArrayBuffer>(\n      this.#ogResponse.arrayBuffer(),\n      this as any,\n    );\n  }\n\n  blob(): Promise<Blob> {\n    return withTeardown<Blob>(this.#ogResponse.blob(), this as any);\n  }\n\n  clone(): Response {\n    const newResponse = this.#ogResponse.clone();\n    return new ResponseWrapper(newResponse, this.teardownRef);\n  }\n\n  formData(): Promise<FormData> {\n    return withTeardown<FormData>(this.#ogResponse.formData(), this as any);\n  }\n\n  json(): Promise<any> {\n    return withTeardown(this.#ogResponse.json(), this as any);\n  }\n}\n\n/**\n * Create a network endowment, consisting of a `WebSocket` object and `fetch`\n * function. This allows us to provide a teardown function, so that we can cancel\n * any pending requests, connections, streams, etc. that may be open when a snap\n * is terminated.\n *\n * This wraps the original implementations of `WebSocket` and `fetch`,\n * to ensure that a bad actor cannot get access to the original objects, thus\n * potentially preventing the network requests from being torn down.\n *\n * @returns An object containing a wrapped `WebSocket` class and `fetch`\n * function, as well as a teardown function.\n */\nconst createNetwork = () => {\n  // Open fetch calls or open body streams or open websockets\n  const openConnections = new Set<{ cancel: () => Promise<void> }>();\n  // Track last teardown count\n  const teardownRef = { lastTeardown: 0 };\n\n  // Remove items from openConnections after they were garbage collected\n  const cleanup = new FinalizationRegistry<() => void>(\n    /* istanbul ignore next: can't test garbage collection without modifying node parameters */\n    (callback) => callback(),\n  );\n\n  const _fetch: typeof fetch = async (\n    input: RequestInfo,\n    init?: RequestInit,\n  ): Promise<Response> => {\n    const abortController = new AbortController();\n    if (init?.signal !== null && init?.signal !== undefined) {\n      const originalSignal = init.signal;\n      // Merge abort controllers\n      originalSignal.addEventListener(\n        'abort',\n        () => {\n          abortController.abort((originalSignal as any).reason);\n        },\n        { once: true },\n      );\n    }\n\n    let res: Response;\n    let openFetchConnection: { cancel: () => Promise<void> } | undefined;\n    try {\n      const fetchPromise = fetch(input, {\n        ...init,\n        signal: abortController.signal,\n      });\n\n      openFetchConnection = {\n        cancel: async () => {\n          abortController.abort();\n          try {\n            await fetchPromise;\n          } catch {\n            /* do nothing */\n          }\n        },\n      };\n      openConnections.add(openFetchConnection);\n\n      res = new ResponseWrapper(\n        await withTeardown(fetchPromise, teardownRef),\n        teardownRef,\n      );\n    } finally {\n      if (openFetchConnection !== undefined) {\n        openConnections.delete(openFetchConnection);\n      }\n    }\n\n    if (res.body !== null) {\n      const body = new WeakRef<ReadableStream>(res.body);\n\n      const openBodyConnection = {\n        cancel:\n          /* istanbul ignore next: see it.todo('can be torn down during body read') test */\n          async () => {\n            try {\n              await body.deref()?.cancel();\n            } catch {\n              /* do nothing */\n            }\n          },\n      };\n      openConnections.add(openBodyConnection);\n      cleanup.register(\n        res.body,\n        /* istanbul ignore next: can't test garbage collection without modifying node parameters */\n        () => openConnections.delete(openBodyConnection),\n      );\n    }\n    return res;\n  };\n\n  /**\n   * This class wraps a WebSocket object instead of extending it.\n   * That way, a bad actor can't get access to original methods using\n   * `socket.prototype`.\n   *\n   * When modifying this class, ensure that no method calls any other method\n   * from the same class (#socket calls are fine). Otherwise, a bad actor could\n   * replace one of the implementations\n   */\n  const _WebSocket = class implements WebSocket {\n    constructor(url: string | URL, protocols?: string | string[]) {\n      this.#socket = new WebSocket(url, protocols);\n\n      // You can't call ref.deref()?.#teardownClose()\n      // But you can capture the close itself\n      const ref = new WeakRef(this.#teardownClose.bind(this));\n      const openConnection = {\n        cancel: async () => {\n          try {\n            await ref.deref()?.();\n          } catch {\n            /* do nothing */\n          }\n        },\n      };\n      openConnections.add(openConnection);\n      cleanup.register(\n        this,\n        /* istanbul ignore next: can't test garbage collection without modifying node parameters */\n        () => openConnections.delete(openConnection),\n      );\n    }\n\n    get onclose(): WebSocketCallback | null {\n      return this.#oncloseOriginal;\n    }\n\n    set onclose(callback: WebSocketCallback | null) {\n      if (this.#isTornDown !== true) {\n        this.#oncloseOriginal = callback;\n        this.#socket.onclose = this.#createWrapped(callback);\n      }\n    }\n\n    get onerror(): ((this: WebSocket, ev: Event) => any) | null {\n      return this.#onerrorOriginal;\n    }\n\n    set onerror(callback: ((this: WebSocket, ev: Event) => any) | null) {\n      this.#onerrorOriginal = callback;\n      this.#socket.onerror = this.#createWrapped(callback);\n    }\n\n    get onmessage(): ((this: WebSocket, ev: MessageEvent<any>) => any) | null {\n      return this.#onmessageOriginal;\n    }\n\n    set onmessage(\n      callback: ((this: WebSocket, ev: MessageEvent<any>) => any) | null,\n    ) {\n      this.#onmessageOriginal = callback;\n      this.#socket.onmessage = this.#createWrapped(callback);\n    }\n\n    get onopen(): ((this: WebSocket, ev: Event) => any) | null {\n      return this.#onopenOriginal;\n    }\n\n    set onopen(callback: ((this: WebSocket, ev: Event) => any) | null) {\n      this.#onopenOriginal = callback;\n      this.#socket.onopen = this.#createWrapped(callback);\n    }\n\n    close(code?: number, reason?: string): void {\n      this.#socket.close(code, reason);\n    }\n\n    send(data: string | Blob | ArrayBufferView | ArrayBufferLike): void {\n      this.#socket.send(data);\n    }\n\n    get CLOSED(): number {\n      return this.#socket.CLOSED;\n    }\n\n    get CLOSING(): number {\n      return this.#socket.CLOSING;\n    }\n\n    get CONNECTING(): number {\n      return this.#socket.CONNECTING;\n    }\n\n    get OPEN(): number {\n      return this.#socket.OPEN;\n    }\n\n    get binaryType(): BinaryType {\n      return this.#socket.binaryType;\n    }\n\n    set binaryType(value: BinaryType) {\n      this.#socket.binaryType = value;\n    }\n\n    get bufferedAmount(): number {\n      return this.#socket.bufferedAmount ?? 0;\n    }\n\n    get extensions(): string {\n      return this.#socket.extensions ?? '';\n    }\n\n    get protocol(): string {\n      return this.#socket.protocol;\n    }\n\n    get readyState(): number {\n      return this.#socket.readyState;\n    }\n\n    get url(): string {\n      return this.#socket.url;\n    }\n\n    addEventListener<K extends keyof WebSocketEventMap>(\n      type: K,\n      listener: (this: WebSocket, ev: WebSocketEventMap[K]) => any,\n      options?: boolean | AddEventListenerOptions,\n    ): void;\n\n    addEventListener(\n      type: string,\n      listener: EventListenerOrEventListenerObject,\n      options?: boolean | AddEventListenerOptions,\n    ): void;\n\n    addEventListener(type: any, listener: any, options?: any): void {\n      if (this.#isTornDown !== true) {\n        if (this.#events[type] === undefined) {\n          this.#events[type] = new Map();\n        }\n        const wrapped = this.#createWrapped(listener);\n        if (wrapped !== null) {\n          this.#events[type].set(listener, wrapped);\n          this.#socket.addEventListener(type, wrapped, options);\n        }\n      }\n    }\n\n    removeEventListener<K extends keyof WebSocketEventMap>(\n      type: K,\n      listener: (this: WebSocket, ev: WebSocketEventMap[K]) => any,\n      options?: boolean | EventListenerOptions,\n    ): void;\n\n    removeEventListener(\n      type: string,\n      listener: EventListenerOrEventListenerObject,\n      options?: boolean | EventListenerOptions,\n    ): void;\n\n    removeEventListener(type: any, listener: any, options?: any): void {\n      if (this.#events[type] !== undefined) {\n        const wrapped = this.#events[type].get(listener);\n        if (wrapped !== undefined) {\n          this.#events[type].delete(listener);\n          this.#socket.removeEventListener(type as any, wrapped, options);\n        }\n      }\n    }\n\n    dispatchEvent(event: Event): boolean {\n      // Can't call close prematurely before the teardown finishes\n      if (event.type !== 'close' || this.#isTornDown !== true) {\n        return this.#socket.dispatchEvent(event);\n      }\n      return false;\n    }\n\n    #teardownClose() {\n      // We clear all close listeners\n      this.#socket.onclose = null;\n      for (const wrapped of this.#events.close?.values() ?? []) {\n        this.#socket.removeEventListener('close', wrapped);\n      }\n      this.#events.close?.clear();\n\n      // We add our own listener\n      let onClosedResolve: any;\n      const onClosed = new Promise<void>((r) => (onClosedResolve = r));\n      this.#socket.onclose = () => {\n        onClosedResolve();\n      };\n\n      // No new listeners can be added after this point\n      this.#isTornDown = true;\n\n      this.#socket.close();\n\n      return onClosed;\n    }\n\n    #createWrapped(listener: WebSocketCallback): WebSocketCallback;\n\n    #createWrapped(listener: null): null;\n\n    #createWrapped(\n      listener: WebSocketCallback | null,\n    ): WebSocketCallback | null;\n\n    #createWrapped(\n      listener: WebSocketCallback | null,\n    ): WebSocketCallback | null {\n      if (listener === null) {\n        return null;\n      }\n\n      return (e) => {\n        // TODO: Should we migrate this to use a wrapper class?\n        const properties = [...allProperties(e)]\n          .filter(([_, key]) => key !== 'constructor')\n          .reduce<Record<string, any>>((acc, [obj, key]) => {\n            const stringKey = key.toString();\n            const descriptor = Reflect.getOwnPropertyDescriptor(obj, key);\n            if (typeof descriptor?.value === 'function') {\n              acc[stringKey] = e[stringKey].bind(this);\n            } else {\n              acc[stringKey] = e[stringKey];\n            }\n            return acc;\n          }, {});\n        const event = {\n          ...properties,\n          target: this,\n          currentTarget: this,\n          srcElement: this,\n          ports: [this],\n          source: null,\n          composedPath: () => [this],\n        };\n        listener.apply(this, [event]);\n      };\n    }\n\n    #socket: WebSocket;\n\n    /**\n     * After this is set to true, no new event listeners can be added\n     */\n    #isTornDown = false;\n\n    #events: Record<string, Map<WebSocketCallback, WebSocketCallback>> = {};\n\n    #onopenOriginal: WebSocketCallback | null = null;\n\n    #onmessageOriginal: WebSocketCallback | null = null;\n\n    #onerrorOriginal: WebSocketCallback | null = null;\n\n    #oncloseOriginal: WebSocketCallback | null = null;\n  };\n\n  const teardownFunction = async () => {\n    teardownRef.lastTeardown += 1;\n    const promises: Promise<void>[] = [];\n    openConnections.forEach(({ cancel }) => promises.push(cancel()));\n    openConnections.clear();\n    await Promise.all(promises);\n  };\n\n  return {\n    fetch: _fetch,\n    WebSocket: _WebSocket,\n    teardownFunction,\n  };\n};\n\nconst endowmentModule = {\n  names: ['fetch', 'WebSocket'] as const,\n  factory: createNetwork,\n};\nexport default endowmentModule;\n"]}

package/dist/common/endowments/timeout.d.ts

@@ -0,0 +1,9 @@
+declare const endowmentModule: {
+    names: readonly ["setTimeout", "clearTimeout"];
+    factory: () => {
+        readonly setTimeout: (handler: TimerHandler, timeout?: number | undefined) => unknown;
+        readonly clearTimeout: (handle: unknown) => void;
+        readonly teardownFunction: () => void;
+    };
+};
+export default endowmentModule;

package/dist/common/endowments/timeout.js

@@ -0,0 +1,50 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+/**
+ * Creates a pair of `setTimeout` and `clearTimeout` functions attenuated such
+ * that:
+ * - `setTimeout` throws if its "handler" parameter is not a function.
+ * - `clearTimeout` only clears timeouts created by its sibling `setTimeout`,
+ * or else no-ops.
+ *
+ * @returns An object with the attenuated `setTimeout` and `clearTimeout`
+ * functions.
+ */
+const createTimeout = () => {
+    const registeredHandles = new Map();
+    const _setTimeout = (handler, timeout) => {
+        if (typeof handler !== 'function') {
+            throw new Error(`The timeout handler must be a function. Received: ${typeof handler}`);
+        }
+        const handle = Object.freeze({});
+        const platformHandle = setTimeout(() => {
+            registeredHandles.delete(handle);
+            handler();
+        }, timeout);
+        registeredHandles.set(handle, platformHandle);
+        return handle;
+    };
+    const _clearTimeout = (handle) => {
+        const platformHandle = registeredHandles.get(handle);
+        if (platformHandle !== undefined) {
+            clearTimeout(platformHandle);
+            registeredHandles.delete(handle);
+        }
+    };
+    const teardownFunction = () => {
+        for (const handle of registeredHandles.keys()) {
+            _clearTimeout(handle);
+        }
+    };
+    return {
+        setTimeout: _setTimeout,
+        clearTimeout: _clearTimeout,
+        teardownFunction,
+    };
+};
+const endowmentModule = {
+    names: ['setTimeout', 'clearTimeout'],
+    factory: createTimeout,
+};
+exports.default = endowmentModule;
+//# sourceMappingURL=timeout.js.map

package/dist/common/endowments/timeout.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"timeout.js","sourceRoot":"","sources":["../../../src/common/endowments/timeout.ts"],"names":[],"mappings":";;AAAA;;;;;;;;;GASG;AACH,MAAM,aAAa,GAAG,GAAG,EAAE;IACzB,MAAM,iBAAiB,GAAG,IAAI,GAAG,EAAoB,CAAC;IAEtD,MAAM,WAAW,GAAG,CAAC,OAAqB,EAAE,OAAgB,EAAW,EAAE;QACvE,IAAI,OAAO,OAAO,KAAK,UAAU,EAAE;YACjC,MAAM,IAAI,KAAK,CACb,qDAAqD,OAAO,OAAO,EAAE,CACtE,CAAC;SACH;QAED,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QACjC,MAAM,cAAc,GAAG,UAAU,CAAC,GAAG,EAAE;YACrC,iBAAiB,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YACjC,OAAO,EAAE,CAAC;QACZ,CAAC,EAAE,OAAO,CAAC,CAAC;QAEZ,iBAAiB,CAAC,GAAG,CAAC,MAAM,EAAE,cAAc,CAAC,CAAC;QAC9C,OAAO,MAAM,CAAC;IAChB,CAAC,CAAC;IAEF,MAAM,aAAa,GAAG,CAAC,MAAe,EAAQ,EAAE;QAC9C,MAAM,cAAc,GAAG,iBAAiB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACrD,IAAI,cAAc,KAAK,SAAS,EAAE;YAChC,YAAY,CAAC,cAAqB,CAAC,CAAC;YACpC,iBAAiB,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;SAClC;IACH,CAAC,CAAC;IAEF,MAAM,gBAAgB,GAAG,GAAS,EAAE;QAClC,KAAK,MAAM,MAAM,IAAI,iBAAiB,CAAC,IAAI,EAAE,EAAE;YAC7C,aAAa,CAAC,MAAM,CAAC,CAAC;SACvB;IACH,CAAC,CAAC;IAEF,OAAO;QACL,UAAU,EAAE,WAAW;QACvB,YAAY,EAAE,aAAa;QAC3B,gBAAgB;KACR,CAAC;AACb,CAAC,CAAC;AAEF,MAAM,eAAe,GAAG;IACtB,KAAK,EAAE,CAAC,YAAY,EAAE,cAAc,CAAU;IAC9C,OAAO,EAAE,aAAa;CACvB,CAAC;AACF,kBAAe,eAAe,CAAC","sourcesContent":["/**\n * Creates a pair of `setTimeout` and `clearTimeout` functions attenuated such\n * that:\n * - `setTimeout` throws if its \"handler\" parameter is not a function.\n * - `clearTimeout` only clears timeouts created by its sibling `setTimeout`,\n * or else no-ops.\n *\n * @returns An object with the attenuated `setTimeout` and `clearTimeout`\n * functions.\n */\nconst createTimeout = () => {\n  const registeredHandles = new Map<unknown, unknown>();\n\n  const _setTimeout = (handler: TimerHandler, timeout?: number): unknown => {\n    if (typeof handler !== 'function') {\n      throw new Error(\n        `The timeout handler must be a function. Received: ${typeof handler}`,\n      );\n    }\n\n    const handle = Object.freeze({});\n    const platformHandle = setTimeout(() => {\n      registeredHandles.delete(handle);\n      handler();\n    }, timeout);\n\n    registeredHandles.set(handle, platformHandle);\n    return handle;\n  };\n\n  const _clearTimeout = (handle: unknown): void => {\n    const platformHandle = registeredHandles.get(handle);\n    if (platformHandle !== undefined) {\n      clearTimeout(platformHandle as any);\n      registeredHandles.delete(handle);\n    }\n  };\n\n  const teardownFunction = (): void => {\n    for (const handle of registeredHandles.keys()) {\n      _clearTimeout(handle);\n    }\n  };\n\n  return {\n    setTimeout: _setTimeout,\n    clearTimeout: _clearTimeout,\n    teardownFunction,\n  } as const;\n};\n\nconst endowmentModule = {\n  names: ['setTimeout', 'clearTimeout'] as const,\n  factory: createTimeout,\n};\nexport default endowmentModule;\n"]}

package/dist/common/globalEvents.d.ts

@@ -0,0 +1,19 @@
+/// <reference types="node" />
+/**
+ * Adds an event listener platform agnostically, trying both `globalThis.addEventListener` and `globalThis.process.on`
+ *
+ * @param event - The event to listen for.
+ * @param listener - The listener to be triggered when the event occurs.
+ * @returns The result of the platform agnostic operation if any.
+ * @throws If none of the platform options are present.
+ */
+export declare function addEventListener(event: string, listener: (...args: any[]) => void): void | NodeJS.Process;
+/**
+ * Removes an event listener platform agnostically, trying both `globalThis.removeEventListener` and `globalThis.process.removeListener`
+ *
+ * @param event - The event to remove the listener for.
+ * @param listener - The currently attached listener.
+ * @returns The result of the platform agnostic operation if any.
+ * @throws If none of the platform options are present.
+ */
+export declare function removeEventListener(event: string, listener: (...args: any[]) => void): void | NodeJS.Process;

package/dist/common/globalEvents.js

@@ -0,0 +1,47 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.removeEventListener = exports.addEventListener = void 0;
+const globalObject_1 = require("./globalObject");
+/**
+ * Adds an event listener platform agnostically, trying both `globalThis.addEventListener` and `globalThis.process.on`
+ *
+ * @param event - The event to listen for.
+ * @param listener - The listener to be triggered when the event occurs.
+ * @returns The result of the platform agnostic operation if any.
+ * @throws If none of the platform options are present.
+ */
+function addEventListener(event, listener) {
+    if ('addEventListener' in globalObject_1.rootRealmGlobal &&
+        typeof globalObject_1.rootRealmGlobal.addEventListener === 'function') {
+        return globalObject_1.rootRealmGlobal.addEventListener(event.toLowerCase(), listener);
+    }
+    if (globalObject_1.rootRealmGlobal.process &&
+        'on' in globalObject_1.rootRealmGlobal.process &&
+        typeof globalObject_1.rootRealmGlobal.process.on === 'function') {
+        return globalObject_1.rootRealmGlobal.process.on(event, listener);
+    }
+    throw new Error('Platform agnostic addEventListener failed');
+}
+exports.addEventListener = addEventListener;
+/**
+ * Removes an event listener platform agnostically, trying both `globalThis.removeEventListener` and `globalThis.process.removeListener`
+ *
+ * @param event - The event to remove the listener for.
+ * @param listener - The currently attached listener.
+ * @returns The result of the platform agnostic operation if any.
+ * @throws If none of the platform options are present.
+ */
+function removeEventListener(event, listener) {
+    if ('removeEventListener' in globalObject_1.rootRealmGlobal &&
+        typeof globalObject_1.rootRealmGlobal.removeEventListener === 'function') {
+        return globalObject_1.rootRealmGlobal.removeEventListener(event.toLowerCase(), listener);
+    }
+    if (globalObject_1.rootRealmGlobal.process &&
+        'removeListener' in globalObject_1.rootRealmGlobal.process &&
+        typeof globalObject_1.rootRealmGlobal.process.removeListener === 'function') {
+        return globalObject_1.rootRealmGlobal.process.removeListener(event, listener);
+    }
+    throw new Error('Platform agnostic removeEventListener failed');
+}
+exports.removeEventListener = removeEventListener;
+//# sourceMappingURL=globalEvents.js.map

package/dist/common/globalEvents.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"globalEvents.js","sourceRoot":"","sources":["../../src/common/globalEvents.ts"],"names":[],"mappings":";;;AAAA,iDAAiD;AAEjD;;;;;;;GAOG;AACH,SAAgB,gBAAgB,CAC9B,KAAa,EACb,QAAkC;IAElC,IACE,kBAAkB,IAAI,8BAAe;QACrC,OAAO,8BAAe,CAAC,gBAAgB,KAAK,UAAU,EACtD;QACA,OAAO,8BAAe,CAAC,gBAAgB,CAAC,KAAK,CAAC,WAAW,EAAE,EAAE,QAAQ,CAAC,CAAC;KACxE;IAED,IACE,8BAAe,CAAC,OAAO;QACvB,IAAI,IAAI,8BAAe,CAAC,OAAO;QAC/B,OAAO,8BAAe,CAAC,OAAO,CAAC,EAAE,KAAK,UAAU,EAChD;QACA,OAAO,8BAAe,CAAC,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;KACpD;IAED,MAAM,IAAI,KAAK,CAAC,2CAA2C,CAAC,CAAC;AAC/D,CAAC;AApBD,4CAoBC;AAED;;;;;;;GAOG;AACH,SAAgB,mBAAmB,CACjC,KAAa,EACb,QAAkC;IAElC,IACE,qBAAqB,IAAI,8BAAe;QACxC,OAAO,8BAAe,CAAC,mBAAmB,KAAK,UAAU,EACzD;QACA,OAAO,8BAAe,CAAC,mBAAmB,CAAC,KAAK,CAAC,WAAW,EAAE,EAAE,QAAQ,CAAC,CAAC;KAC3E;IAED,IACE,8BAAe,CAAC,OAAO;QACvB,gBAAgB,IAAI,8BAAe,CAAC,OAAO;QAC3C,OAAO,8BAAe,CAAC,OAAO,CAAC,cAAc,KAAK,UAAU,EAC5D;QACA,OAAO,8BAAe,CAAC,OAAO,CAAC,cAAc,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;KAChE;IAED,MAAM,IAAI,KAAK,CAAC,8CAA8C,CAAC,CAAC;AAClE,CAAC;AApBD,kDAoBC","sourcesContent":["import { rootRealmGlobal } from './globalObject';\n\n/**\n * Adds an event listener platform agnostically, trying both `globalThis.addEventListener` and `globalThis.process.on`\n *\n * @param event - The event to listen for.\n * @param listener - The listener to be triggered when the event occurs.\n * @returns The result of the platform agnostic operation if any.\n * @throws If none of the platform options are present.\n */\nexport function addEventListener(\n  event: string,\n  listener: (...args: any[]) => void,\n) {\n  if (\n    'addEventListener' in rootRealmGlobal &&\n    typeof rootRealmGlobal.addEventListener === 'function'\n  ) {\n    return rootRealmGlobal.addEventListener(event.toLowerCase(), listener);\n  }\n\n  if (\n    rootRealmGlobal.process &&\n    'on' in rootRealmGlobal.process &&\n    typeof rootRealmGlobal.process.on === 'function'\n  ) {\n    return rootRealmGlobal.process.on(event, listener);\n  }\n\n  throw new Error('Platform agnostic addEventListener failed');\n}\n\n/**\n * Removes an event listener platform agnostically, trying both `globalThis.removeEventListener` and `globalThis.process.removeListener`\n *\n * @param event - The event to remove the listener for.\n * @param listener - The currently attached listener.\n * @returns The result of the platform agnostic operation if any.\n * @throws If none of the platform options are present.\n */\nexport function removeEventListener(\n  event: string,\n  listener: (...args: any[]) => void,\n) {\n  if (\n    'removeEventListener' in rootRealmGlobal &&\n    typeof rootRealmGlobal.removeEventListener === 'function'\n  ) {\n    return rootRealmGlobal.removeEventListener(event.toLowerCase(), listener);\n  }\n\n  if (\n    rootRealmGlobal.process &&\n    'removeListener' in rootRealmGlobal.process &&\n    typeof rootRealmGlobal.process.removeListener === 'function'\n  ) {\n    return rootRealmGlobal.process.removeListener(event, listener);\n  }\n\n  throw new Error('Platform agnostic removeEventListener failed');\n}\n"]}

package/dist/common/globalObject.d.ts

@@ -0,0 +1,9 @@
+/**
+ * A platform-agnostic alias for the root realm global object.
+ */
+declare const rootRealmGlobal: typeof globalThis;
+/**
+ * The string literal corresponding to the name of the root realm global object.
+ */
+declare const rootRealmGlobalName: string;
+export { rootRealmGlobal, rootRealmGlobalName };

package/dist/common/globalObject.js

@@ -0,0 +1,48 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.rootRealmGlobalName = exports.rootRealmGlobal = void 0;
+var GlobalObjectNames;
+(function (GlobalObjectNames) {
+    // The globalThis entry is incorrectly identified as shadowing the global
+    // globalThis.
+    // eslint-disable-next-line @typescript-eslint/no-shadow
+    GlobalObjectNames["globalThis"] = "globalThis";
+    GlobalObjectNames["global"] = "global";
+    GlobalObjectNames["self"] = "self";
+    GlobalObjectNames["window"] = "window";
+})(GlobalObjectNames || (GlobalObjectNames = {}));
+let _rootRealmGlobal;
+let _rootRealmGlobalName;
+/* istanbul ignore next */
+/* eslint-disable no-negated-condition */
+if (typeof globalThis !== 'undefined') {
+    _rootRealmGlobal = globalThis;
+    _rootRealmGlobalName = GlobalObjectNames.globalThis;
+}
+else if (typeof self !== 'undefined') {
+    _rootRealmGlobal = self;
+    _rootRealmGlobalName = GlobalObjectNames.self;
+}
+else if (typeof window !== 'undefined') {
+    _rootRealmGlobal = window;
+    _rootRealmGlobalName = GlobalObjectNames.window;
+}
+else if (typeof global !== 'undefined') {
+    _rootRealmGlobal = global;
+    _rootRealmGlobalName = GlobalObjectNames.global;
+}
+else {
+    throw new Error('Unknown realm type; failed to identify global object.');
+}
+/* eslint-enable no-negated-condition */
+/**
+ * A platform-agnostic alias for the root realm global object.
+ */
+const rootRealmGlobal = _rootRealmGlobal;
+exports.rootRealmGlobal = rootRealmGlobal;
+/**
+ * The string literal corresponding to the name of the root realm global object.
+ */
+const rootRealmGlobalName = _rootRealmGlobalName;
+exports.rootRealmGlobalName = rootRealmGlobalName;
+//# sourceMappingURL=globalObject.js.map

package/dist/common/globalObject.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"globalObject.js","sourceRoot":"","sources":["../../src/common/globalObject.ts"],"names":[],"mappings":";;;AAAA,IAAK,iBAQJ;AARD,WAAK,iBAAiB;IACpB,yEAAyE;IACzE,cAAc;IACd,wDAAwD;IACxD,8CAAyB,CAAA;IACzB,sCAAiB,CAAA;IACjB,kCAAa,CAAA;IACb,sCAAiB,CAAA;AACnB,CAAC,EARI,iBAAiB,KAAjB,iBAAiB,QAQrB;AAED,IAAI,gBAAmC,CAAC;AACxC,IAAI,oBAA4B,CAAC;AAEjC,0BAA0B;AAC1B,yCAAyC;AACzC,IAAI,OAAO,UAAU,KAAK,WAAW,EAAE;IACrC,gBAAgB,GAAG,UAAU,CAAC;IAC9B,oBAAoB,GAAG,iBAAiB,CAAC,UAAU,CAAC;CACrD;KAAM,IAAI,OAAO,IAAI,KAAK,WAAW,EAAE;IACtC,gBAAgB,GAAG,IAAI,CAAC;IACxB,oBAAoB,GAAG,iBAAiB,CAAC,IAAI,CAAC;CAC/C;KAAM,IAAI,OAAO,MAAM,KAAK,WAAW,EAAE;IACxC,gBAAgB,GAAG,MAAM,CAAC;IAC1B,oBAAoB,GAAG,iBAAiB,CAAC,MAAM,CAAC;CACjD;KAAM,IAAI,OAAO,MAAM,KAAK,WAAW,EAAE;IACxC,gBAAgB,GAAG,MAAM,CAAC;IAC1B,oBAAoB,GAAG,iBAAiB,CAAC,MAAM,CAAC;CACjD;KAAM;IACL,MAAM,IAAI,KAAK,CAAC,uDAAuD,CAAC,CAAC;CAC1E;AACD,wCAAwC;AAExC;;GAEG;AACH,MAAM,eAAe,GAAG,gBAAgB,CAAC;AAOhC,0CAAe;AALxB;;GAEG;AACH,MAAM,mBAAmB,GAAG,oBAAoB,CAAC;AAEvB,kDAAmB","sourcesContent":["enum GlobalObjectNames {\n  // The globalThis entry is incorrectly identified as shadowing the global\n  // globalThis.\n  // eslint-disable-next-line @typescript-eslint/no-shadow\n  globalThis = 'globalThis',\n  global = 'global',\n  self = 'self',\n  window = 'window',\n}\n\nlet _rootRealmGlobal: typeof globalThis;\nlet _rootRealmGlobalName: string;\n\n/* istanbul ignore next */\n/* eslint-disable no-negated-condition */\nif (typeof globalThis !== 'undefined') {\n  _rootRealmGlobal = globalThis;\n  _rootRealmGlobalName = GlobalObjectNames.globalThis;\n} else if (typeof self !== 'undefined') {\n  _rootRealmGlobal = self;\n  _rootRealmGlobalName = GlobalObjectNames.self;\n} else if (typeof window !== 'undefined') {\n  _rootRealmGlobal = window;\n  _rootRealmGlobalName = GlobalObjectNames.window;\n} else if (typeof global !== 'undefined') {\n  _rootRealmGlobal = global;\n  _rootRealmGlobalName = GlobalObjectNames.global;\n} else {\n  throw new Error('Unknown realm type; failed to identify global object.');\n}\n/* eslint-enable no-negated-condition */\n\n/**\n * A platform-agnostic alias for the root realm global object.\n */\nconst rootRealmGlobal = _rootRealmGlobal;\n\n/**\n * The string literal corresponding to the name of the root realm global object.\n */\nconst rootRealmGlobalName = _rootRealmGlobalName;\n\nexport { rootRealmGlobal, rootRealmGlobalName };\n"]}

package/dist/common/keyring.d.ts

@@ -0,0 +1,12 @@
+import { SnapKeyring } from '@metamask/snaps-types';
+import { Json, JsonRpcNotification, JsonRpcRequest } from '@metamask/utils';
+/**
+ * Wraps a SnapKeyring class, returning a handler that can route requests to the exposed functions by the class.
+ *
+ * @param notify - The BaseSnapExecutor notify function, used for event communication.
+ * @param keyring - The SnapKeyring instance.
+ * @returns A handler function.
+ */
+export declare function wrapKeyring(notify: (requestObject: Omit<JsonRpcNotification<Record<string, Json> | Json[]>, 'jsonrpc'>) => void, keyring?: SnapKeyring): ({ request }: {
+    request: JsonRpcRequest<Json[]>;
+}) => unknown;