@metamask/snaps-controllers 13.0.0 → 13.1.0

package/dist/snaps/constants.cjs

@@ -41,5 +41,6 @@ exports.CLIENT_ONLY_HANDLERS = Object.freeze([
     snaps_utils_1.HandlerType.OnAssetsLookup,
     snaps_utils_1.HandlerType.OnAssetsConversion,
     snaps_utils_1.HandlerType.OnAssetHistoricalPrice,
+    snaps_utils_1.HandlerType.OnWebSocketEvent,
 ]);
 //# sourceMappingURL=constants.cjs.map

package/dist/snaps/constants.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"constants.cjs","sourceRoot":"","sources":["../../src/snaps/constants.ts"],"names":[],"mappings":";;;AAAA,mEAA6D;AAC7D,uDAAoD;AAEpD,gEAAgE;AACnD,QAAA,mBAAmB,GAAG,MAAM,CAAC,MAAM,CAAC;IAC/C,aAAa;IACb,kBAAkB;IAClB,aAAa;IACb,gBAAgB;IAChB,kCAAc,CAAC,OAAO;IACtB,kCAAc,CAAC,QAAQ;IACvB,kCAAc,CAAC,cAAc;IAC7B,kCAAc,CAAC,gBAAgB;IAC/B,kCAAc,CAAC,kBAAkB;IACjC,kCAAc,CAAC,gBAAgB;CAChC,CAAC,CAAC;AAEU,QAAA,wCAAwC,GAAG;IACtD,SAAS,EAAE,QAAiB;IAC5B,MAAM,EAAE;QACN,UAAU,EAAE,KAAM;KACnB;CACF,CAAC;AAEF;;GAEG;AACU,QAAA,sBAAsB,GAAG,GAAG,CAAC;AAE1C,+DAA+D;AAClD,QAAA,eAAe,GAAG,UAAU,CAAC;AAE1C,+DAA+D;AAClD,QAAA,oBAAoB,GAAG,MAAM,CAAC,MAAM,CAAC;IAChD,yBAAW,CAAC,eAAe;IAC3B,yBAAW,CAAC,WAAW;IACvB,yBAAW,CAAC,aAAa;IACzB,yBAAW,CAAC,SAAS;IACrB,yBAAW,CAAC,YAAY;IACxB,yBAAW,CAAC,UAAU;IACtB,yBAAW,CAAC,cAAc;IAC1B,yBAAW,CAAC,WAAW;IACvB,yBAAW,CAAC,cAAc;IAC1B,yBAAW,CAAC,kBAAkB;IAC9B,yBAAW,CAAC,sBAAsB;CACnC,CAAC,CAAC","sourcesContent":["import { SnapEndowments } from '@metamask/snaps-rpc-methods';\nimport { HandlerType } from '@metamask/snaps-utils';\n\n// These permissions are allowed without being on the allowlist.\nexport const ALLOWED_PERMISSIONS = Object.freeze([\n  'snap_dialog',\n  'snap_manageState',\n  'snap_notify',\n  'snap_getLocale',\n  SnapEndowments.Cronjob,\n  SnapEndowments.HomePage,\n  SnapEndowments.LifecycleHooks,\n  SnapEndowments.EthereumProvider,\n  SnapEndowments.TransactionInsight,\n  SnapEndowments.SignatureInsight,\n]);\n\nexport const LEGACY_ENCRYPTION_KEY_DERIVATION_OPTIONS = {\n  algorithm: 'PBKDF2' as const,\n  params: {\n    iterations: 10_000,\n  },\n};\n\n/**\n * The timeout for debouncing state updates.\n */\nexport const STATE_DEBOUNCE_TIMEOUT = 500;\n\n// The origin used to indicate requests coming from the client.\nexport const METAMASK_ORIGIN = 'metamask';\n\n// These handlers are only allowed to be invoked by the client.\nexport const CLIENT_ONLY_HANDLERS = Object.freeze([\n  HandlerType.OnClientRequest,\n  HandlerType.OnSignature,\n  HandlerType.OnTransaction,\n  HandlerType.OnCronjob,\n  HandlerType.OnNameLookup,\n  HandlerType.OnHomePage,\n  HandlerType.OnSettingsPage,\n  HandlerType.OnUserInput,\n  HandlerType.OnAssetsLookup,\n  HandlerType.OnAssetsConversion,\n  HandlerType.OnAssetHistoricalPrice,\n]);\n"]}
+{"version":3,"file":"constants.cjs","sourceRoot":"","sources":["../../src/snaps/constants.ts"],"names":[],"mappings":";;;AAAA,mEAA6D;AAC7D,uDAAoD;AAEpD,gEAAgE;AACnD,QAAA,mBAAmB,GAAG,MAAM,CAAC,MAAM,CAAC;IAC/C,aAAa;IACb,kBAAkB;IAClB,aAAa;IACb,gBAAgB;IAChB,kCAAc,CAAC,OAAO;IACtB,kCAAc,CAAC,QAAQ;IACvB,kCAAc,CAAC,cAAc;IAC7B,kCAAc,CAAC,gBAAgB;IAC/B,kCAAc,CAAC,kBAAkB;IACjC,kCAAc,CAAC,gBAAgB;CAChC,CAAC,CAAC;AAEU,QAAA,wCAAwC,GAAG;IACtD,SAAS,EAAE,QAAiB;IAC5B,MAAM,EAAE;QACN,UAAU,EAAE,KAAM;KACnB;CACF,CAAC;AAEF;;GAEG;AACU,QAAA,sBAAsB,GAAG,GAAG,CAAC;AAE1C,+DAA+D;AAClD,QAAA,eAAe,GAAG,UAAU,CAAC;AAE1C,+DAA+D;AAClD,QAAA,oBAAoB,GAAG,MAAM,CAAC,MAAM,CAAC;IAChD,yBAAW,CAAC,eAAe;IAC3B,yBAAW,CAAC,WAAW;IACvB,yBAAW,CAAC,aAAa;IACzB,yBAAW,CAAC,SAAS;IACrB,yBAAW,CAAC,YAAY;IACxB,yBAAW,CAAC,UAAU;IACtB,yBAAW,CAAC,cAAc;IAC1B,yBAAW,CAAC,WAAW;IACvB,yBAAW,CAAC,cAAc;IAC1B,yBAAW,CAAC,kBAAkB;IAC9B,yBAAW,CAAC,sBAAsB;IAClC,yBAAW,CAAC,gBAAgB;CAC7B,CAAC,CAAC","sourcesContent":["import { SnapEndowments } from '@metamask/snaps-rpc-methods';\nimport { HandlerType } from '@metamask/snaps-utils';\n\n// These permissions are allowed without being on the allowlist.\nexport const ALLOWED_PERMISSIONS = Object.freeze([\n  'snap_dialog',\n  'snap_manageState',\n  'snap_notify',\n  'snap_getLocale',\n  SnapEndowments.Cronjob,\n  SnapEndowments.HomePage,\n  SnapEndowments.LifecycleHooks,\n  SnapEndowments.EthereumProvider,\n  SnapEndowments.TransactionInsight,\n  SnapEndowments.SignatureInsight,\n]);\n\nexport const LEGACY_ENCRYPTION_KEY_DERIVATION_OPTIONS = {\n  algorithm: 'PBKDF2' as const,\n  params: {\n    iterations: 10_000,\n  },\n};\n\n/**\n * The timeout for debouncing state updates.\n */\nexport const STATE_DEBOUNCE_TIMEOUT = 500;\n\n// The origin used to indicate requests coming from the client.\nexport const METAMASK_ORIGIN = 'metamask';\n\n// These handlers are only allowed to be invoked by the client.\nexport const CLIENT_ONLY_HANDLERS = Object.freeze([\n  HandlerType.OnClientRequest,\n  HandlerType.OnSignature,\n  HandlerType.OnTransaction,\n  HandlerType.OnCronjob,\n  HandlerType.OnNameLookup,\n  HandlerType.OnHomePage,\n  HandlerType.OnSettingsPage,\n  HandlerType.OnUserInput,\n  HandlerType.OnAssetsLookup,\n  HandlerType.OnAssetsConversion,\n  HandlerType.OnAssetHistoricalPrice,\n  HandlerType.OnWebSocketEvent,\n]);\n"]}

package/dist/snaps/constants.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"constants.d.cts","sourceRoot":"","sources":["../../src/snaps/constants.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,WAAW,EAAE,8BAA8B;AAGpD,eAAO,MAAM,mBAAmB,mBAW9B,CAAC;AAEH,eAAO,MAAM,wCAAwC;;;;;CAKpD,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,sBAAsB,MAAM,CAAC;AAG1C,eAAO,MAAM,eAAe,aAAa,CAAC;AAG1C,eAAO,MAAM,oBAAoB,wBAY/B,CAAC"}
+{"version":3,"file":"constants.d.cts","sourceRoot":"","sources":["../../src/snaps/constants.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,WAAW,EAAE,8BAA8B;AAGpD,eAAO,MAAM,mBAAmB,mBAW9B,CAAC;AAEH,eAAO,MAAM,wCAAwC;;;;;CAKpD,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,sBAAsB,MAAM,CAAC;AAG1C,eAAO,MAAM,eAAe,aAAa,CAAC;AAG1C,eAAO,MAAM,oBAAoB,wBAa/B,CAAC"}

package/dist/snaps/constants.d.mts.map

@@ -1 +1 @@
-{"version":3,"file":"constants.d.mts","sourceRoot":"","sources":["../../src/snaps/constants.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,WAAW,EAAE,8BAA8B;AAGpD,eAAO,MAAM,mBAAmB,mBAW9B,CAAC;AAEH,eAAO,MAAM,wCAAwC;;;;;CAKpD,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,sBAAsB,MAAM,CAAC;AAG1C,eAAO,MAAM,eAAe,aAAa,CAAC;AAG1C,eAAO,MAAM,oBAAoB,wBAY/B,CAAC"}
+{"version":3,"file":"constants.d.mts","sourceRoot":"","sources":["../../src/snaps/constants.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,WAAW,EAAE,8BAA8B;AAGpD,eAAO,MAAM,mBAAmB,mBAW9B,CAAC;AAEH,eAAO,MAAM,wCAAwC;;;;;CAKpD,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,sBAAsB,MAAM,CAAC;AAG1C,eAAO,MAAM,eAAe,aAAa,CAAC;AAG1C,eAAO,MAAM,oBAAoB,wBAa/B,CAAC"}

package/dist/snaps/constants.mjs

@@ -38,5 +38,6 @@ export const CLIENT_ONLY_HANDLERS = Object.freeze([
     HandlerType.OnAssetsLookup,
     HandlerType.OnAssetsConversion,
     HandlerType.OnAssetHistoricalPrice,
+    HandlerType.OnWebSocketEvent,
 ]);
 //# sourceMappingURL=constants.mjs.map

package/dist/snaps/constants.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"constants.mjs","sourceRoot":"","sources":["../../src/snaps/constants.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,oCAAoC;AAC7D,OAAO,EAAE,WAAW,EAAE,8BAA8B;AAEpD,gEAAgE;AAChE,MAAM,CAAC,MAAM,mBAAmB,GAAG,MAAM,CAAC,MAAM,CAAC;IAC/C,aAAa;IACb,kBAAkB;IAClB,aAAa;IACb,gBAAgB;IAChB,cAAc,CAAC,OAAO;IACtB,cAAc,CAAC,QAAQ;IACvB,cAAc,CAAC,cAAc;IAC7B,cAAc,CAAC,gBAAgB;IAC/B,cAAc,CAAC,kBAAkB;IACjC,cAAc,CAAC,gBAAgB;CAChC,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,wCAAwC,GAAG;IACtD,SAAS,EAAE,QAAiB;IAC5B,MAAM,EAAE;QACN,UAAU,EAAE,KAAM;KACnB;CACF,CAAC;AAEF;;GAEG;AACH,MAAM,CAAC,MAAM,sBAAsB,GAAG,GAAG,CAAC;AAE1C,+DAA+D;AAC/D,MAAM,CAAC,MAAM,eAAe,GAAG,UAAU,CAAC;AAE1C,+DAA+D;AAC/D,MAAM,CAAC,MAAM,oBAAoB,GAAG,MAAM,CAAC,MAAM,CAAC;IAChD,WAAW,CAAC,eAAe;IAC3B,WAAW,CAAC,WAAW;IACvB,WAAW,CAAC,aAAa;IACzB,WAAW,CAAC,SAAS;IACrB,WAAW,CAAC,YAAY;IACxB,WAAW,CAAC,UAAU;IACtB,WAAW,CAAC,cAAc;IAC1B,WAAW,CAAC,WAAW;IACvB,WAAW,CAAC,cAAc;IAC1B,WAAW,CAAC,kBAAkB;IAC9B,WAAW,CAAC,sBAAsB;CACnC,CAAC,CAAC","sourcesContent":["import { SnapEndowments } from '@metamask/snaps-rpc-methods';\nimport { HandlerType } from '@metamask/snaps-utils';\n\n// These permissions are allowed without being on the allowlist.\nexport const ALLOWED_PERMISSIONS = Object.freeze([\n  'snap_dialog',\n  'snap_manageState',\n  'snap_notify',\n  'snap_getLocale',\n  SnapEndowments.Cronjob,\n  SnapEndowments.HomePage,\n  SnapEndowments.LifecycleHooks,\n  SnapEndowments.EthereumProvider,\n  SnapEndowments.TransactionInsight,\n  SnapEndowments.SignatureInsight,\n]);\n\nexport const LEGACY_ENCRYPTION_KEY_DERIVATION_OPTIONS = {\n  algorithm: 'PBKDF2' as const,\n  params: {\n    iterations: 10_000,\n  },\n};\n\n/**\n * The timeout for debouncing state updates.\n */\nexport const STATE_DEBOUNCE_TIMEOUT = 500;\n\n// The origin used to indicate requests coming from the client.\nexport const METAMASK_ORIGIN = 'metamask';\n\n// These handlers are only allowed to be invoked by the client.\nexport const CLIENT_ONLY_HANDLERS = Object.freeze([\n  HandlerType.OnClientRequest,\n  HandlerType.OnSignature,\n  HandlerType.OnTransaction,\n  HandlerType.OnCronjob,\n  HandlerType.OnNameLookup,\n  HandlerType.OnHomePage,\n  HandlerType.OnSettingsPage,\n  HandlerType.OnUserInput,\n  HandlerType.OnAssetsLookup,\n  HandlerType.OnAssetsConversion,\n  HandlerType.OnAssetHistoricalPrice,\n]);\n"]}
+{"version":3,"file":"constants.mjs","sourceRoot":"","sources":["../../src/snaps/constants.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,oCAAoC;AAC7D,OAAO,EAAE,WAAW,EAAE,8BAA8B;AAEpD,gEAAgE;AAChE,MAAM,CAAC,MAAM,mBAAmB,GAAG,MAAM,CAAC,MAAM,CAAC;IAC/C,aAAa;IACb,kBAAkB;IAClB,aAAa;IACb,gBAAgB;IAChB,cAAc,CAAC,OAAO;IACtB,cAAc,CAAC,QAAQ;IACvB,cAAc,CAAC,cAAc;IAC7B,cAAc,CAAC,gBAAgB;IAC/B,cAAc,CAAC,kBAAkB;IACjC,cAAc,CAAC,gBAAgB;CAChC,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,wCAAwC,GAAG;IACtD,SAAS,EAAE,QAAiB;IAC5B,MAAM,EAAE;QACN,UAAU,EAAE,KAAM;KACnB;CACF,CAAC;AAEF;;GAEG;AACH,MAAM,CAAC,MAAM,sBAAsB,GAAG,GAAG,CAAC;AAE1C,+DAA+D;AAC/D,MAAM,CAAC,MAAM,eAAe,GAAG,UAAU,CAAC;AAE1C,+DAA+D;AAC/D,MAAM,CAAC,MAAM,oBAAoB,GAAG,MAAM,CAAC,MAAM,CAAC;IAChD,WAAW,CAAC,eAAe;IAC3B,WAAW,CAAC,WAAW;IACvB,WAAW,CAAC,aAAa;IACzB,WAAW,CAAC,SAAS;IACrB,WAAW,CAAC,YAAY;IACxB,WAAW,CAAC,UAAU;IACtB,WAAW,CAAC,cAAc;IAC1B,WAAW,CAAC,WAAW;IACvB,WAAW,CAAC,cAAc;IAC1B,WAAW,CAAC,kBAAkB;IAC9B,WAAW,CAAC,sBAAsB;IAClC,WAAW,CAAC,gBAAgB;CAC7B,CAAC,CAAC","sourcesContent":["import { SnapEndowments } from '@metamask/snaps-rpc-methods';\nimport { HandlerType } from '@metamask/snaps-utils';\n\n// These permissions are allowed without being on the allowlist.\nexport const ALLOWED_PERMISSIONS = Object.freeze([\n  'snap_dialog',\n  'snap_manageState',\n  'snap_notify',\n  'snap_getLocale',\n  SnapEndowments.Cronjob,\n  SnapEndowments.HomePage,\n  SnapEndowments.LifecycleHooks,\n  SnapEndowments.EthereumProvider,\n  SnapEndowments.TransactionInsight,\n  SnapEndowments.SignatureInsight,\n]);\n\nexport const LEGACY_ENCRYPTION_KEY_DERIVATION_OPTIONS = {\n  algorithm: 'PBKDF2' as const,\n  params: {\n    iterations: 10_000,\n  },\n};\n\n/**\n * The timeout for debouncing state updates.\n */\nexport const STATE_DEBOUNCE_TIMEOUT = 500;\n\n// The origin used to indicate requests coming from the client.\nexport const METAMASK_ORIGIN = 'metamask';\n\n// These handlers are only allowed to be invoked by the client.\nexport const CLIENT_ONLY_HANDLERS = Object.freeze([\n  HandlerType.OnClientRequest,\n  HandlerType.OnSignature,\n  HandlerType.OnTransaction,\n  HandlerType.OnCronjob,\n  HandlerType.OnNameLookup,\n  HandlerType.OnHomePage,\n  HandlerType.OnSettingsPage,\n  HandlerType.OnUserInput,\n  HandlerType.OnAssetsLookup,\n  HandlerType.OnAssetsConversion,\n  HandlerType.OnAssetHistoricalPrice,\n  HandlerType.OnWebSocketEvent,\n]);\n"]}

package/dist/snaps/index.cjs

@@ -14,6 +14,7 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
     for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
 };
 Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./constants.cjs"), exports);
 __exportStar(require("./location/index.cjs"), exports);
 __exportStar(require("./SnapController.cjs"), exports);
 __exportStar(require("./selectors.cjs"), exports);

package/dist/snaps/index.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"index.cjs","sourceRoot":"","sources":["../../src/snaps/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,uDAA2B;AAC3B,uDAAiC;AACjC,kDAA4B;AAC5B,uDAA2B","sourcesContent":["export * from './location';\nexport * from './SnapController';\nexport * from './selectors';\nexport * from './registry';\n"]}
+{"version":3,"file":"index.cjs","sourceRoot":"","sources":["../../src/snaps/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,kDAA4B;AAC5B,uDAA2B;AAC3B,uDAAiC;AACjC,kDAA4B;AAC5B,uDAA2B","sourcesContent":["export * from './constants';\nexport * from './location';\nexport * from './SnapController';\nexport * from './selectors';\nexport * from './registry';\n"]}

package/dist/snaps/index.d.cts

@@ -1,3 +1,4 @@
+export * from "./constants.cjs";
 export * from "./location/index.cjs";
 export * from "./SnapController.cjs";
 export * from "./selectors.cjs";

package/dist/snaps/index.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.cts","sourceRoot":"","sources":["../../src/snaps/index.ts"],"names":[],"mappings":"AAAA,qCAA2B;AAC3B,qCAAiC;AACjC,gCAA4B;AAC5B,qCAA2B"}
+{"version":3,"file":"index.d.cts","sourceRoot":"","sources":["../../src/snaps/index.ts"],"names":[],"mappings":"AAAA,gCAA4B;AAC5B,qCAA2B;AAC3B,qCAAiC;AACjC,gCAA4B;AAC5B,qCAA2B"}

package/dist/snaps/index.d.mts

@@ -1,3 +1,4 @@
+export * from "./constants.mjs";
 export * from "./location/index.mjs";
 export * from "./SnapController.mjs";
 export * from "./selectors.mjs";

package/dist/snaps/index.d.mts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.mts","sourceRoot":"","sources":["../../src/snaps/index.ts"],"names":[],"mappings":"AAAA,qCAA2B;AAC3B,qCAAiC;AACjC,gCAA4B;AAC5B,qCAA2B"}
+{"version":3,"file":"index.d.mts","sourceRoot":"","sources":["../../src/snaps/index.ts"],"names":[],"mappings":"AAAA,gCAA4B;AAC5B,qCAA2B;AAC3B,qCAAiC;AACjC,gCAA4B;AAC5B,qCAA2B"}

package/dist/snaps/index.mjs

@@ -1,3 +1,4 @@
+export * from "./constants.mjs";
 export * from "./location/index.mjs";
 export * from "./SnapController.mjs";
 export * from "./selectors.mjs";

package/dist/snaps/index.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"index.mjs","sourceRoot":"","sources":["../../src/snaps/index.ts"],"names":[],"mappings":"AAAA,qCAA2B;AAC3B,qCAAiC;AACjC,gCAA4B;AAC5B,qCAA2B","sourcesContent":["export * from './location';\nexport * from './SnapController';\nexport * from './selectors';\nexport * from './registry';\n"]}
+{"version":3,"file":"index.mjs","sourceRoot":"","sources":["../../src/snaps/index.ts"],"names":[],"mappings":"AAAA,gCAA4B;AAC5B,qCAA2B;AAC3B,qCAAiC;AACjC,gCAA4B;AAC5B,qCAA2B","sourcesContent":["export * from './constants';\nexport * from './location';\nexport * from './SnapController';\nexport * from './selectors';\nexport * from './registry';\n"]}

package/dist/utils.d.cts

@@ -232,7 +232,7 @@ export declare function fetchSnap(snapId: SnapId, location: SnapLocation): Promi
             type: string;
             url: string;
         } | undefined;
-        initialConnections?: Record<string & URL, {}> | undefined;
+        initialConnections?: Record<string, {}> | undefined;
         platformVersion?: import("@metamask/utils").SemVerVersion | undefined;
         $schema?: string | undefined;
     }>;

package/dist/utils.d.mts

@@ -232,7 +232,7 @@ export declare function fetchSnap(snapId: SnapId, location: SnapLocation): Promi
             type: string;
             url: string;
         } | undefined;
-        initialConnections?: Record<string & URL, {}> | undefined;
+        initialConnections?: Record<string, {}> | undefined;
         platformVersion?: import("@metamask/utils").SemVerVersion | undefined;
         $schema?: string | undefined;
     }>;

package/dist/websocket/WebSocketService.cjs

@@ -0,0 +1,194 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.WebSocketService = void 0;
+const rpc_errors_1 = require("@metamask/rpc-errors");
+const snaps_utils_1 = require("@metamask/snaps-utils");
+const utils_1 = require("@metamask/utils");
+const nanoid_1 = require("nanoid");
+const snaps_1 = require("../snaps/index.cjs");
+const serviceName = 'WebSocketService';
+class WebSocketService {
+    #messenger;
+    #sockets;
+    constructor({ messenger }) {
+        this.#messenger = messenger;
+        this.#sockets = new Map();
+        this.#messenger.registerActionHandler(`${serviceName}:open`, async (...args) => this.#open(...args));
+        this.#messenger.registerActionHandler(`${serviceName}:close`, (...args) => this.#close(...args));
+        this.#messenger.registerActionHandler(`${serviceName}:sendMessage`, async (...args) => this.#sendMessage(...args));
+        this.#messenger.registerActionHandler(`${serviceName}:getAll`, (...args) => this.#getAll(...args));
+        this.#messenger.subscribe('SnapController:snapUpdated', (snap) => {
+            this.#closeAll(snap.id);
+        });
+        this.#messenger.subscribe('SnapController:snapUninstalled', (snap) => {
+            this.#closeAll(snap.id);
+        });
+        // Due to local Snaps not currently emitting snapUninstalled we also have to
+        // listen to snapInstalled.
+        this.#messenger.subscribe('SnapController:snapInstalled', (snap) => {
+            this.#closeAll(snap.id);
+        });
+    }
+    /**
+     * Get information about a given WebSocket connection with an ID.
+     *
+     * @param snapId - The Snap ID.
+     * @param id - The identifier for the WebSocket connection.
+     * @returns Information abou the WebSocket connection.
+     * @throws If the WebSocket connection cannot be found.
+     */
+    #get(snapId, id) {
+        const socket = this.#sockets.get(id);
+        (0, utils_1.assert)(socket && socket.snapId === snapId, `Socket with ID "${id}" not found.`);
+        return socket;
+    }
+    /**
+     * Check whether a given Snap ID already has an open connection for a URL and protocol.
+     *
+     * @param snapId - The Snap ID.
+     * @param url - The URL.
+     * @param protocols - A protocols parameter.
+     * @returns True if a matching connection already exists, otherwise false.
+     */
+    #exists(snapId, url, protocols) {
+        return this.#getAll(snapId).some((socket) => socket.url === url && (0, snaps_utils_1.isEqual)(socket.protocols, protocols));
+    }
+    /**
+     * Handle sending a specific WebSocketEvent to a Snap.
+     *
+     * @param snapId - The Snap ID.
+     * @param event - The WebSocketEvent.
+     */
+    #handleEvent(snapId, event) {
+        this.#messenger
+            .call('SnapController:handleRequest', {
+            origin: snaps_1.METAMASK_ORIGIN,
+            snapId,
+            handler: snaps_utils_1.HandlerType.OnWebSocketEvent,
+            request: { method: '', params: { event } },
+        })
+            .catch((error) => {
+            (0, snaps_utils_1.logError)(`An error occurred while handling a WebSocket message for Snap "${snapId}":`, error);
+        });
+    }
+    /**
+     * Open a WebSocket connection.
+     *
+     * @param snapId - The Snap ID.
+     * @param url - The URL for the WebSocket connection.
+     * @param protocols - An optional parameter for protocols.
+     * @returns The identifier for the opened connection.
+     * @throws If the connection fails.
+     */
+    async #open(snapId, url, protocols = []) {
+        (0, utils_1.assert)(!this.#exists(snapId, url, protocols), `An open WebSocket connection to ${url} already exists.`);
+        const parsedUrl = new URL(url);
+        const { origin } = parsedUrl;
+        const id = (0, nanoid_1.nanoid)();
+        // eslint-disable-next-line no-restricted-globals
+        const socket = new WebSocket(url, protocols);
+        socket.binaryType = 'arraybuffer';
+        const { promise, resolve, reject } = (0, utils_1.createDeferredPromise)();
+        socket.addEventListener('open', () => {
+            resolve();
+            this.#handleEvent(snapId, {
+                type: 'open',
+                id,
+                origin,
+            });
+        });
+        socket.addEventListener('close', (event) => {
+            this.#handleEvent(snapId, {
+                type: 'close',
+                id,
+                origin,
+                code: event.code,
+                reason: event.reason,
+                // wasClean is not available on mobile.
+                wasClean: event.wasClean ?? null,
+            });
+        });
+        const errorListener = () => {
+            reject(rpc_errors_1.rpcErrors.resourceUnavailable('An error occurred while opening the WebSocket.'));
+        };
+        socket.addEventListener('error', errorListener);
+        socket.addEventListener('message', (event) => {
+            const isText = typeof event.data === 'string';
+            const data = isText
+                ? { type: 'text', message: event.data }
+                : {
+                    type: 'binary',
+                    // We request that the WebSocket gives us an array buffer.
+                    message: Array.from(new Uint8Array(event.data)),
+                };
+            this.#handleEvent(snapId, {
+                type: 'message',
+                id,
+                origin,
+                data,
+            });
+        });
+        this.#sockets.set(id, {
+            id,
+            snapId,
+            url,
+            protocols,
+            socket,
+            openPromise: promise,
+        });
+        await promise;
+        socket.removeEventListener('error', errorListener);
+        return id;
+    }
+    /**
+     * Close a given WebSocket connection.
+     *
+     * @param snapId - The Snap ID.
+     * @param id - The identifier for the WebSocket connection.
+     */
+    #close(snapId, id) {
+        const { socket } = this.#get(snapId, id);
+        socket.close();
+        this.#sockets.delete(id);
+    }
+    /**
+     * Close all open connections for a given Snap ID.
+     *
+     * @param snapId - The Snap ID.
+     */
+    #closeAll(snapId) {
+        for (const socket of this.#getAll(snapId)) {
+            this.#close(snapId, socket.id);
+        }
+    }
+    /**
+     * Send a message from a given Snap ID to a WebSocket connection.
+     *
+     * @param snapId - The Snap ID.
+     * @param id - The identifier for the WebSocket connection.
+     * @param data - The message to send.
+     */
+    async #sendMessage(snapId, id, data) {
+        const { socket, openPromise } = this.#get(snapId, id);
+        await openPromise;
+        const wrappedData = Array.isArray(data) ? new Uint8Array(data) : data;
+        socket.send(wrappedData);
+    }
+    /**
+     * Get a list of all open WebSocket connections for a Snap ID.
+     *
+     * @param snapId - The Snap ID.
+     * @returns A list of WebSocket connections.
+     */
+    #getAll(snapId) {
+        return [...this.#sockets.values()]
+            .filter((socket) => socket.snapId === snapId)
+            .map((socket) => ({
+            id: socket.id,
+            url: socket.url,
+            protocols: socket.protocols,
+        }));
+    }
+}
+exports.WebSocketService = WebSocketService;
+//# sourceMappingURL=WebSocketService.cjs.map

package/dist/websocket/WebSocketService.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"WebSocketService.cjs","sourceRoot":"","sources":["../../src/websocket/WebSocketService.ts"],"names":[],"mappings":";;;AACA,qDAAiD;AAMjD,uDAAuE;AACvE,2CAAgE;AAChE,mCAAgC;AAQhC,8CAA2C;AAE3C,MAAM,WAAW,GAAG,kBAAkB,CAAC;AAiEvC,MAAa,gBAAgB;IAClB,UAAU,CAA4B;IAEtC,QAAQ,CAA8B;IAE/C,YAAY,EAAE,SAAS,EAAwB;QAC7C,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC;QAC5B,IAAI,CAAC,QAAQ,GAAG,IAAI,GAAG,EAAE,CAAC;QAE1B,IAAI,CAAC,UAAU,CAAC,qBAAqB,CACnC,GAAG,WAAW,OAAO,EACrB,KAAK,EAAE,GAAG,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,CACvC,CAAC;QAEF,IAAI,CAAC,UAAU,CAAC,qBAAqB,CAAC,GAAG,WAAW,QAAQ,EAAE,CAAC,GAAG,IAAI,EAAE,EAAE,CACxE,IAAI,CAAC,MAAM,CAAC,GAAG,IAAI,CAAC,CACrB,CAAC;QAEF,IAAI,CAAC,UAAU,CAAC,qBAAqB,CACnC,GAAG,WAAW,cAAc,EAC5B,KAAK,EAAE,GAAG,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG,IAAI,CAAC,CAC9C,CAAC;QAEF,IAAI,CAAC,UAAU,CAAC,qBAAqB,CAAC,GAAG,WAAW,SAAS,EAAE,CAAC,GAAG,IAAI,EAAE,EAAE,CACzE,IAAI,CAAC,OAAO,CAAC,GAAG,IAAI,CAAC,CACtB,CAAC;QAEF,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,4BAA4B,EAAE,CAAC,IAAI,EAAE,EAAE;YAC/D,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC1B,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,gCAAgC,EAAE,CAAC,IAAI,EAAE,EAAE;YACnE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC1B,CAAC,CAAC,CAAC;QAEH,4EAA4E;QAC5E,2BAA2B;QAC3B,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,8BAA8B,EAAE,CAAC,IAAI,EAAE,EAAE;YACjE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC1B,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;;;;;OAOG;IACH,IAAI,CAAC,MAAc,EAAE,EAAU;QAC7B,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAErC,IAAA,cAAM,EACJ,MAAM,IAAI,MAAM,CAAC,MAAM,KAAK,MAAM,EAClC,mBAAmB,EAAE,cAAc,CACpC,CAAC;QAEF,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;;;;;OAOG;IACH,OAAO,CAAC,MAAc,EAAE,GAAW,EAAE,SAAmB;QACtD,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAC9B,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,GAAG,KAAK,GAAG,IAAI,IAAA,qBAAO,EAAC,MAAM,CAAC,SAAS,EAAE,SAAS,CAAC,CACvE,CAAC;IACJ,CAAC;IAED;;;;;OAKG;IACH,YAAY,CAAC,MAAc,EAAE,KAAqB;QAChD,IAAI,CAAC,UAAU;aACZ,IAAI,CAAC,8BAA8B,EAAE;YACpC,MAAM,EAAE,uBAAe;YACvB,MAAM;YACN,OAAO,EAAE,yBAAW,CAAC,gBAAgB;YACrC,OAAO,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,KAAK,EAAE,EAAE;SAC3C,CAAC;aACD,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;YACf,IAAA,sBAAQ,EACN,kEAAkE,MAAM,IAAI,EAC5E,KAAK,CACN,CAAC;QACJ,CAAC,CAAC,CAAC;IACP,CAAC;IAED;;;;;;;;OAQG;IACH,KAAK,CAAC,KAAK,CAAC,MAAc,EAAE,GAAW,EAAE,YAAsB,EAAE;QAC/D,IAAA,cAAM,EACJ,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,EAAE,SAAS,CAAC,EACrC,mCAAmC,GAAG,kBAAkB,CACzD,CAAC;QAEF,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC,CAAC;QAC/B,MAAM,EAAE,MAAM,EAAE,GAAG,SAAS,CAAC;QAE7B,MAAM,EAAE,GAAG,IAAA,eAAM,GAAE,CAAC;QAEpB,iDAAiD;QACjD,MAAM,MAAM,GAAG,IAAI,SAAS,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;QAC7C,MAAM,CAAC,UAAU,GAAG,aAAa,CAAC;QAElC,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,IAAA,6BAAqB,GAAE,CAAC;QAE7D,MAAM,CAAC,gBAAgB,CAAC,MAAM,EAAE,GAAG,EAAE;YACnC,OAAO,EAAE,CAAC;YACV,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE;gBACxB,IAAI,EAAE,MAAM;gBACZ,EAAE;gBACF,MAAM;aACP,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;YACzC,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE;gBACxB,IAAI,EAAE,OAAO;gBACb,EAAE;gBACF,MAAM;gBACN,IAAI,EAAE,KAAK,CAAC,IAAI;gBAChB,MAAM,EAAE,KAAK,CAAC,MAAM;gBACpB,uCAAuC;gBACvC,QAAQ,EAAE,KAAK,CAAC,QAAQ,IAAI,IAAI;aACjC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,aAAa,GAAG,GAAG,EAAE;YACzB,MAAM,CACJ,sBAAS,CAAC,mBAAmB,CAC3B,gDAAgD,CACjD,CACF,CAAC;QACJ,CAAC,CAAC;QAEF,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC;QAEhD,MAAM,CAAC,gBAAgB,CAAC,SAAS,EAAE,CAAC,KAAK,EAAE,EAAE;YAC3C,MAAM,MAAM,GAAG,OAAO,KAAK,CAAC,IAAI,KAAK,QAAQ,CAAC;YAC9C,MAAM,IAAI,GAAG,MAAM;gBACjB,CAAC,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,OAAO,EAAE,KAAK,CAAC,IAAI,EAAE;gBAChD,CAAC,CAAC;oBACE,IAAI,EAAE,QAAiB;oBACvB,0DAA0D;oBAC1D,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;iBAChD,CAAC;YAEN,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE;gBACxB,IAAI,EAAE,SAAS;gBACf,EAAE;gBACF,MAAM;gBACN,IAAI;aACL,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,EAAE;YACpB,EAAE;YACF,MAAM;YACN,GAAG;YACH,SAAS;YACT,MAAM;YACN,WAAW,EAAE,OAAO;SACrB,CAAC,CAAC;QAEH,MAAM,OAAO,CAAC;QAEd,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC;QAEnD,OAAO,EAAE,CAAC;IACZ,CAAC;IAED;;;;;OAKG;IACH,MAAM,CAAC,MAAc,EAAE,EAAU;QAC/B,MAAM,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;QAEzC,MAAM,CAAC,KAAK,EAAE,CAAC;QAEf,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IAC3B,CAAC;IAED;;;;OAIG;IACH,SAAS,CAAC,MAAc;QACtB,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;YAC1C,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,EAAE,CAAC,CAAC;QACjC,CAAC;IACH,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,YAAY,CAAC,MAAc,EAAE,EAAU,EAAE,IAAuB;QACpE,MAAM,EAAE,MAAM,EAAE,WAAW,EAAE,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;QAEtD,MAAM,WAAW,CAAC;QAElB,MAAM,WAAW,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;QAEtE,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IAC3B,CAAC;IAED;;;;;OAKG;IACH,OAAO,CAAC,MAAc;QACpB,OAAO,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC;aAC/B,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,KAAK,MAAM,CAAC;aAC5C,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;YAChB,EAAE,EAAE,MAAM,CAAC,EAAE;YACb,GAAG,EAAE,MAAM,CAAC,GAAG;YACf,SAAS,EAAE,MAAM,CAAC,SAAS;SAC5B,CAAC,CAAC,CAAC;IACR,CAAC;CACF;AArPD,4CAqPC","sourcesContent":["import type { RestrictedMessenger } from '@metamask/base-controller';\nimport { rpcErrors } from '@metamask/rpc-errors';\nimport type {\n  GetWebSocketsResult,\n  SnapId,\n  WebSocketEvent,\n} from '@metamask/snaps-sdk';\nimport { HandlerType, isEqual, logError } from '@metamask/snaps-utils';\nimport { assert, createDeferredPromise } from '@metamask/utils';\nimport { nanoid } from 'nanoid';\n\nimport type {\n  HandleSnapRequest,\n  SnapInstalled,\n  SnapUninstalled,\n  SnapUpdated,\n} from '../snaps';\nimport { METAMASK_ORIGIN } from '../snaps';\n\nconst serviceName = 'WebSocketService';\n\nexport type WebSocketServiceOpenAction = {\n  type: `${typeof serviceName}:open`;\n  handler: (\n    snapId: SnapId,\n    url: string,\n    protocols?: string[],\n  ) => Promise<string>;\n};\n\nexport type WebSocketServiceCloseAction = {\n  type: `${typeof serviceName}:close`;\n  handler: (snapId: SnapId, id: string) => void;\n};\n\nexport type WebSocketServiceSendMessageAction = {\n  type: `${typeof serviceName}:sendMessage`;\n  handler: (\n    snapId: SnapId,\n    id: string,\n    data: string | number[],\n  ) => Promise<void>;\n};\n\nexport type WebSocketServiceGetAllAction = {\n  type: `${typeof serviceName}:getAll`;\n  handler: (snapId: SnapId) => GetWebSocketsResult;\n};\n\nexport type WebSocketServiceActions =\n  | WebSocketServiceOpenAction\n  | WebSocketServiceCloseAction\n  | WebSocketServiceSendMessageAction\n  | WebSocketServiceGetAllAction;\n\nexport type WebSocketServiceAllowedActions = HandleSnapRequest;\n\nexport type WebSocketServiceEvents =\n  | SnapUninstalled\n  | SnapUpdated\n  | SnapInstalled;\n\nexport type WebSocketServiceMessenger = RestrictedMessenger<\n  'WebSocketService',\n  WebSocketServiceActions | WebSocketServiceAllowedActions,\n  WebSocketServiceEvents,\n  WebSocketServiceAllowedActions['type'],\n  WebSocketServiceEvents['type']\n>;\n\ntype WebSocketServiceArgs = {\n  messenger: WebSocketServiceMessenger;\n};\n\ntype InternalSocket = {\n  id: string;\n  snapId: SnapId;\n  url: string;\n  protocols: string[];\n  openPromise: Promise<void>;\n  // eslint-disable-next-line no-restricted-globals\n  socket: WebSocket;\n};\n\nexport class WebSocketService {\n  readonly #messenger: WebSocketServiceMessenger;\n\n  readonly #sockets: Map<string, InternalSocket>;\n\n  constructor({ messenger }: WebSocketServiceArgs) {\n    this.#messenger = messenger;\n    this.#sockets = new Map();\n\n    this.#messenger.registerActionHandler(\n      `${serviceName}:open`,\n      async (...args) => this.#open(...args),\n    );\n\n    this.#messenger.registerActionHandler(`${serviceName}:close`, (...args) =>\n      this.#close(...args),\n    );\n\n    this.#messenger.registerActionHandler(\n      `${serviceName}:sendMessage`,\n      async (...args) => this.#sendMessage(...args),\n    );\n\n    this.#messenger.registerActionHandler(`${serviceName}:getAll`, (...args) =>\n      this.#getAll(...args),\n    );\n\n    this.#messenger.subscribe('SnapController:snapUpdated', (snap) => {\n      this.#closeAll(snap.id);\n    });\n\n    this.#messenger.subscribe('SnapController:snapUninstalled', (snap) => {\n      this.#closeAll(snap.id);\n    });\n\n    // Due to local Snaps not currently emitting snapUninstalled we also have to\n    // listen to snapInstalled.\n    this.#messenger.subscribe('SnapController:snapInstalled', (snap) => {\n      this.#closeAll(snap.id);\n    });\n  }\n\n  /**\n   * Get information about a given WebSocket connection with an ID.\n   *\n   * @param snapId - The Snap ID.\n   * @param id - The identifier for the WebSocket connection.\n   * @returns Information abou the WebSocket connection.\n   * @throws If the WebSocket connection cannot be found.\n   */\n  #get(snapId: SnapId, id: string) {\n    const socket = this.#sockets.get(id);\n\n    assert(\n      socket && socket.snapId === snapId,\n      `Socket with ID \"${id}\" not found.`,\n    );\n\n    return socket;\n  }\n\n  /**\n   * Check whether a given Snap ID already has an open connection for a URL and protocol.\n   *\n   * @param snapId - The Snap ID.\n   * @param url - The URL.\n   * @param protocols - A protocols parameter.\n   * @returns True if a matching connection already exists, otherwise false.\n   */\n  #exists(snapId: SnapId, url: string, protocols: string[]) {\n    return this.#getAll(snapId).some(\n      (socket) => socket.url === url && isEqual(socket.protocols, protocols),\n    );\n  }\n\n  /**\n   * Handle sending a specific WebSocketEvent to a Snap.\n   *\n   * @param snapId - The Snap ID.\n   * @param event - The WebSocketEvent.\n   */\n  #handleEvent(snapId: SnapId, event: WebSocketEvent) {\n    this.#messenger\n      .call('SnapController:handleRequest', {\n        origin: METAMASK_ORIGIN,\n        snapId,\n        handler: HandlerType.OnWebSocketEvent,\n        request: { method: '', params: { event } },\n      })\n      .catch((error) => {\n        logError(\n          `An error occurred while handling a WebSocket message for Snap \"${snapId}\":`,\n          error,\n        );\n      });\n  }\n\n  /**\n   * Open a WebSocket connection.\n   *\n   * @param snapId - The Snap ID.\n   * @param url - The URL for the WebSocket connection.\n   * @param protocols - An optional parameter for protocols.\n   * @returns The identifier for the opened connection.\n   * @throws If the connection fails.\n   */\n  async #open(snapId: SnapId, url: string, protocols: string[] = []) {\n    assert(\n      !this.#exists(snapId, url, protocols),\n      `An open WebSocket connection to ${url} already exists.`,\n    );\n\n    const parsedUrl = new URL(url);\n    const { origin } = parsedUrl;\n\n    const id = nanoid();\n\n    // eslint-disable-next-line no-restricted-globals\n    const socket = new WebSocket(url, protocols);\n    socket.binaryType = 'arraybuffer';\n\n    const { promise, resolve, reject } = createDeferredPromise();\n\n    socket.addEventListener('open', () => {\n      resolve();\n      this.#handleEvent(snapId, {\n        type: 'open',\n        id,\n        origin,\n      });\n    });\n\n    socket.addEventListener('close', (event) => {\n      this.#handleEvent(snapId, {\n        type: 'close',\n        id,\n        origin,\n        code: event.code,\n        reason: event.reason,\n        // wasClean is not available on mobile.\n        wasClean: event.wasClean ?? null,\n      });\n    });\n\n    const errorListener = () => {\n      reject(\n        rpcErrors.resourceUnavailable(\n          'An error occurred while opening the WebSocket.',\n        ),\n      );\n    };\n\n    socket.addEventListener('error', errorListener);\n\n    socket.addEventListener('message', (event) => {\n      const isText = typeof event.data === 'string';\n      const data = isText\n        ? { type: 'text' as const, message: event.data }\n        : {\n            type: 'binary' as const,\n            // We request that the WebSocket gives us an array buffer.\n            message: Array.from(new Uint8Array(event.data)),\n          };\n\n      this.#handleEvent(snapId, {\n        type: 'message',\n        id,\n        origin,\n        data,\n      });\n    });\n\n    this.#sockets.set(id, {\n      id,\n      snapId,\n      url,\n      protocols,\n      socket,\n      openPromise: promise,\n    });\n\n    await promise;\n\n    socket.removeEventListener('error', errorListener);\n\n    return id;\n  }\n\n  /**\n   * Close a given WebSocket connection.\n   *\n   * @param snapId - The Snap ID.\n   * @param id - The identifier for the WebSocket connection.\n   */\n  #close(snapId: SnapId, id: string) {\n    const { socket } = this.#get(snapId, id);\n\n    socket.close();\n\n    this.#sockets.delete(id);\n  }\n\n  /**\n   * Close all open connections for a given Snap ID.\n   *\n   * @param snapId - The Snap ID.\n   */\n  #closeAll(snapId: SnapId) {\n    for (const socket of this.#getAll(snapId)) {\n      this.#close(snapId, socket.id);\n    }\n  }\n\n  /**\n   * Send a message from a given Snap ID to a WebSocket connection.\n   *\n   * @param snapId - The Snap ID.\n   * @param id - The identifier for the WebSocket connection.\n   * @param data - The message to send.\n   */\n  async #sendMessage(snapId: SnapId, id: string, data: string | number[]) {\n    const { socket, openPromise } = this.#get(snapId, id);\n\n    await openPromise;\n\n    const wrappedData = Array.isArray(data) ? new Uint8Array(data) : data;\n\n    socket.send(wrappedData);\n  }\n\n  /**\n   * Get a list of all open WebSocket connections for a Snap ID.\n   *\n   * @param snapId - The Snap ID.\n   * @returns A list of WebSocket connections.\n   */\n  #getAll(snapId: SnapId) {\n    return [...this.#sockets.values()]\n      .filter((socket) => socket.snapId === snapId)\n      .map((socket) => ({\n        id: socket.id,\n        url: socket.url,\n        protocols: socket.protocols,\n      }));\n  }\n}\n"]}

package/dist/websocket/WebSocketService.d.cts

@@ -0,0 +1,33 @@
+import type { RestrictedMessenger } from "@metamask/base-controller";
+import type { GetWebSocketsResult, SnapId } from "@metamask/snaps-sdk";
+import type { HandleSnapRequest, SnapInstalled, SnapUninstalled, SnapUpdated } from "../snaps/index.cjs";
+declare const serviceName = "WebSocketService";
+export type WebSocketServiceOpenAction = {
+    type: `${typeof serviceName}:open`;
+    handler: (snapId: SnapId, url: string, protocols?: string[]) => Promise<string>;
+};
+export type WebSocketServiceCloseAction = {
+    type: `${typeof serviceName}:close`;
+    handler: (snapId: SnapId, id: string) => void;
+};
+export type WebSocketServiceSendMessageAction = {
+    type: `${typeof serviceName}:sendMessage`;
+    handler: (snapId: SnapId, id: string, data: string | number[]) => Promise<void>;
+};
+export type WebSocketServiceGetAllAction = {
+    type: `${typeof serviceName}:getAll`;
+    handler: (snapId: SnapId) => GetWebSocketsResult;
+};
+export type WebSocketServiceActions = WebSocketServiceOpenAction | WebSocketServiceCloseAction | WebSocketServiceSendMessageAction | WebSocketServiceGetAllAction;
+export type WebSocketServiceAllowedActions = HandleSnapRequest;
+export type WebSocketServiceEvents = SnapUninstalled | SnapUpdated | SnapInstalled;
+export type WebSocketServiceMessenger = RestrictedMessenger<'WebSocketService', WebSocketServiceActions | WebSocketServiceAllowedActions, WebSocketServiceEvents, WebSocketServiceAllowedActions['type'], WebSocketServiceEvents['type']>;
+type WebSocketServiceArgs = {
+    messenger: WebSocketServiceMessenger;
+};
+export declare class WebSocketService {
+    #private;
+    constructor({ messenger }: WebSocketServiceArgs);
+}
+export {};
+//# sourceMappingURL=WebSocketService.d.cts.map

package/dist/websocket/WebSocketService.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"WebSocketService.d.cts","sourceRoot":"","sources":["../../src/websocket/WebSocketService.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,mBAAmB,EAAE,kCAAkC;AAErE,OAAO,KAAK,EACV,mBAAmB,EACnB,MAAM,EAEP,4BAA4B;AAK7B,OAAO,KAAK,EACV,iBAAiB,EACjB,aAAa,EACb,eAAe,EACf,WAAW,EACZ,2BAAiB;AAGlB,QAAA,MAAM,WAAW,qBAAqB,CAAC;AAEvC,MAAM,MAAM,0BAA0B,GAAG;IACvC,IAAI,EAAE,GAAG,OAAO,WAAW,OAAO,CAAC;IACnC,OAAO,EAAE,CACP,MAAM,EAAE,MAAM,EACd,GAAG,EAAE,MAAM,EACX,SAAS,CAAC,EAAE,MAAM,EAAE,KACjB,OAAO,CAAC,MAAM,CAAC,CAAC;CACtB,CAAC;AAEF,MAAM,MAAM,2BAA2B,GAAG;IACxC,IAAI,EAAE,GAAG,OAAO,WAAW,QAAQ,CAAC;IACpC,OAAO,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,KAAK,IAAI,CAAC;CAC/C,CAAC;AAEF,MAAM,MAAM,iCAAiC,GAAG;IAC9C,IAAI,EAAE,GAAG,OAAO,WAAW,cAAc,CAAC;IAC1C,OAAO,EAAE,CACP,MAAM,EAAE,MAAM,EACd,EAAE,EAAE,MAAM,EACV,IAAI,EAAE,MAAM,GAAG,MAAM,EAAE,KACpB,OAAO,CAAC,IAAI,CAAC,CAAC;CACpB,CAAC;AAEF,MAAM,MAAM,4BAA4B,GAAG;IACzC,IAAI,EAAE,GAAG,OAAO,WAAW,SAAS,CAAC;IACrC,OAAO,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,mBAAmB,CAAC;CAClD,CAAC;AAEF,MAAM,MAAM,uBAAuB,GAC/B,0BAA0B,GAC1B,2BAA2B,GAC3B,iCAAiC,GACjC,4BAA4B,CAAC;AAEjC,MAAM,MAAM,8BAA8B,GAAG,iBAAiB,CAAC;AAE/D,MAAM,MAAM,sBAAsB,GAC9B,eAAe,GACf,WAAW,GACX,aAAa,CAAC;AAElB,MAAM,MAAM,yBAAyB,GAAG,mBAAmB,CACzD,kBAAkB,EAClB,uBAAuB,GAAG,8BAA8B,EACxD,sBAAsB,EACtB,8BAA8B,CAAC,MAAM,CAAC,EACtC,sBAAsB,CAAC,MAAM,CAAC,CAC/B,CAAC;AAEF,KAAK,oBAAoB,GAAG;IAC1B,SAAS,EAAE,yBAAyB,CAAC;CACtC,CAAC;AAYF,qBAAa,gBAAgB;;gBAKf,EAAE,SAAS,EAAE,EAAE,oBAAoB;CAgPhD"}

package/dist/websocket/WebSocketService.d.mts

@@ -0,0 +1,33 @@
+import type { RestrictedMessenger } from "@metamask/base-controller";
+import type { GetWebSocketsResult, SnapId } from "@metamask/snaps-sdk";
+import type { HandleSnapRequest, SnapInstalled, SnapUninstalled, SnapUpdated } from "../snaps/index.mjs";
+declare const serviceName = "WebSocketService";
+export type WebSocketServiceOpenAction = {
+    type: `${typeof serviceName}:open`;
+    handler: (snapId: SnapId, url: string, protocols?: string[]) => Promise<string>;
+};
+export type WebSocketServiceCloseAction = {
+    type: `${typeof serviceName}:close`;
+    handler: (snapId: SnapId, id: string) => void;
+};
+export type WebSocketServiceSendMessageAction = {
+    type: `${typeof serviceName}:sendMessage`;
+    handler: (snapId: SnapId, id: string, data: string | number[]) => Promise<void>;
+};
+export type WebSocketServiceGetAllAction = {
+    type: `${typeof serviceName}:getAll`;
+    handler: (snapId: SnapId) => GetWebSocketsResult;
+};
+export type WebSocketServiceActions = WebSocketServiceOpenAction | WebSocketServiceCloseAction | WebSocketServiceSendMessageAction | WebSocketServiceGetAllAction;
+export type WebSocketServiceAllowedActions = HandleSnapRequest;
+export type WebSocketServiceEvents = SnapUninstalled | SnapUpdated | SnapInstalled;
+export type WebSocketServiceMessenger = RestrictedMessenger<'WebSocketService', WebSocketServiceActions | WebSocketServiceAllowedActions, WebSocketServiceEvents, WebSocketServiceAllowedActions['type'], WebSocketServiceEvents['type']>;
+type WebSocketServiceArgs = {
+    messenger: WebSocketServiceMessenger;
+};
+export declare class WebSocketService {
+    #private;
+    constructor({ messenger }: WebSocketServiceArgs);
+}
+export {};
+//# sourceMappingURL=WebSocketService.d.mts.map

package/dist/websocket/WebSocketService.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"WebSocketService.d.mts","sourceRoot":"","sources":["../../src/websocket/WebSocketService.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,mBAAmB,EAAE,kCAAkC;AAErE,OAAO,KAAK,EACV,mBAAmB,EACnB,MAAM,EAEP,4BAA4B;AAK7B,OAAO,KAAK,EACV,iBAAiB,EACjB,aAAa,EACb,eAAe,EACf,WAAW,EACZ,2BAAiB;AAGlB,QAAA,MAAM,WAAW,qBAAqB,CAAC;AAEvC,MAAM,MAAM,0BAA0B,GAAG;IACvC,IAAI,EAAE,GAAG,OAAO,WAAW,OAAO,CAAC;IACnC,OAAO,EAAE,CACP,MAAM,EAAE,MAAM,EACd,GAAG,EAAE,MAAM,EACX,SAAS,CAAC,EAAE,MAAM,EAAE,KACjB,OAAO,CAAC,MAAM,CAAC,CAAC;CACtB,CAAC;AAEF,MAAM,MAAM,2BAA2B,GAAG;IACxC,IAAI,EAAE,GAAG,OAAO,WAAW,QAAQ,CAAC;IACpC,OAAO,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,KAAK,IAAI,CAAC;CAC/C,CAAC;AAEF,MAAM,MAAM,iCAAiC,GAAG;IAC9C,IAAI,EAAE,GAAG,OAAO,WAAW,cAAc,CAAC;IAC1C,OAAO,EAAE,CACP,MAAM,EAAE,MAAM,EACd,EAAE,EAAE,MAAM,EACV,IAAI,EAAE,MAAM,GAAG,MAAM,EAAE,KACpB,OAAO,CAAC,IAAI,CAAC,CAAC;CACpB,CAAC;AAEF,MAAM,MAAM,4BAA4B,GAAG;IACzC,IAAI,EAAE,GAAG,OAAO,WAAW,SAAS,CAAC;IACrC,OAAO,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,mBAAmB,CAAC;CAClD,CAAC;AAEF,MAAM,MAAM,uBAAuB,GAC/B,0BAA0B,GAC1B,2BAA2B,GAC3B,iCAAiC,GACjC,4BAA4B,CAAC;AAEjC,MAAM,MAAM,8BAA8B,GAAG,iBAAiB,CAAC;AAE/D,MAAM,MAAM,sBAAsB,GAC9B,eAAe,GACf,WAAW,GACX,aAAa,CAAC;AAElB,MAAM,MAAM,yBAAyB,GAAG,mBAAmB,CACzD,kBAAkB,EAClB,uBAAuB,GAAG,8BAA8B,EACxD,sBAAsB,EACtB,8BAA8B,CAAC,MAAM,CAAC,EACtC,sBAAsB,CAAC,MAAM,CAAC,CAC/B,CAAC;AAEF,KAAK,oBAAoB,GAAG;IAC1B,SAAS,EAAE,yBAAyB,CAAC;CACtC,CAAC;AAYF,qBAAa,gBAAgB;;gBAKf,EAAE,SAAS,EAAE,EAAE,oBAAoB;CAgPhD"}

package/dist/websocket/WebSocketService.mjs

@@ -0,0 +1,190 @@
+import { rpcErrors } from "@metamask/rpc-errors";
+import { HandlerType, isEqual, logError } from "@metamask/snaps-utils";
+import { assert, createDeferredPromise } from "@metamask/utils";
+import { nanoid } from "nanoid";
+import { METAMASK_ORIGIN } from "../snaps/index.mjs";
+const serviceName = 'WebSocketService';
+export class WebSocketService {
+    #messenger;
+    #sockets;
+    constructor({ messenger }) {
+        this.#messenger = messenger;
+        this.#sockets = new Map();
+        this.#messenger.registerActionHandler(`${serviceName}:open`, async (...args) => this.#open(...args));
+        this.#messenger.registerActionHandler(`${serviceName}:close`, (...args) => this.#close(...args));
+        this.#messenger.registerActionHandler(`${serviceName}:sendMessage`, async (...args) => this.#sendMessage(...args));
+        this.#messenger.registerActionHandler(`${serviceName}:getAll`, (...args) => this.#getAll(...args));
+        this.#messenger.subscribe('SnapController:snapUpdated', (snap) => {
+            this.#closeAll(snap.id);
+        });
+        this.#messenger.subscribe('SnapController:snapUninstalled', (snap) => {
+            this.#closeAll(snap.id);
+        });
+        // Due to local Snaps not currently emitting snapUninstalled we also have to
+        // listen to snapInstalled.
+        this.#messenger.subscribe('SnapController:snapInstalled', (snap) => {
+            this.#closeAll(snap.id);
+        });
+    }
+    /**
+     * Get information about a given WebSocket connection with an ID.
+     *
+     * @param snapId - The Snap ID.
+     * @param id - The identifier for the WebSocket connection.
+     * @returns Information abou the WebSocket connection.
+     * @throws If the WebSocket connection cannot be found.
+     */
+    #get(snapId, id) {
+        const socket = this.#sockets.get(id);
+        assert(socket && socket.snapId === snapId, `Socket with ID "${id}" not found.`);
+        return socket;
+    }
+    /**
+     * Check whether a given Snap ID already has an open connection for a URL and protocol.
+     *
+     * @param snapId - The Snap ID.
+     * @param url - The URL.
+     * @param protocols - A protocols parameter.
+     * @returns True if a matching connection already exists, otherwise false.
+     */
+    #exists(snapId, url, protocols) {
+        return this.#getAll(snapId).some((socket) => socket.url === url && isEqual(socket.protocols, protocols));
+    }
+    /**
+     * Handle sending a specific WebSocketEvent to a Snap.
+     *
+     * @param snapId - The Snap ID.
+     * @param event - The WebSocketEvent.
+     */
+    #handleEvent(snapId, event) {
+        this.#messenger
+            .call('SnapController:handleRequest', {
+            origin: METAMASK_ORIGIN,
+            snapId,
+            handler: HandlerType.OnWebSocketEvent,
+            request: { method: '', params: { event } },
+        })
+            .catch((error) => {
+            logError(`An error occurred while handling a WebSocket message for Snap "${snapId}":`, error);
+        });
+    }
+    /**
+     * Open a WebSocket connection.
+     *
+     * @param snapId - The Snap ID.
+     * @param url - The URL for the WebSocket connection.
+     * @param protocols - An optional parameter for protocols.
+     * @returns The identifier for the opened connection.
+     * @throws If the connection fails.
+     */
+    async #open(snapId, url, protocols = []) {
+        assert(!this.#exists(snapId, url, protocols), `An open WebSocket connection to ${url} already exists.`);
+        const parsedUrl = new URL(url);
+        const { origin } = parsedUrl;
+        const id = nanoid();
+        // eslint-disable-next-line no-restricted-globals
+        const socket = new WebSocket(url, protocols);
+        socket.binaryType = 'arraybuffer';
+        const { promise, resolve, reject } = createDeferredPromise();
+        socket.addEventListener('open', () => {
+            resolve();
+            this.#handleEvent(snapId, {
+                type: 'open',
+                id,
+                origin,
+            });
+        });
+        socket.addEventListener('close', (event) => {
+            this.#handleEvent(snapId, {
+                type: 'close',
+                id,
+                origin,
+                code: event.code,
+                reason: event.reason,
+                // wasClean is not available on mobile.
+                wasClean: event.wasClean ?? null,
+            });
+        });
+        const errorListener = () => {
+            reject(rpcErrors.resourceUnavailable('An error occurred while opening the WebSocket.'));
+        };
+        socket.addEventListener('error', errorListener);
+        socket.addEventListener('message', (event) => {
+            const isText = typeof event.data === 'string';
+            const data = isText
+                ? { type: 'text', message: event.data }
+                : {
+                    type: 'binary',
+                    // We request that the WebSocket gives us an array buffer.
+                    message: Array.from(new Uint8Array(event.data)),
+                };
+            this.#handleEvent(snapId, {
+                type: 'message',
+                id,
+                origin,
+                data,
+            });
+        });
+        this.#sockets.set(id, {
+            id,
+            snapId,
+            url,
+            protocols,
+            socket,
+            openPromise: promise,
+        });
+        await promise;
+        socket.removeEventListener('error', errorListener);
+        return id;
+    }
+    /**
+     * Close a given WebSocket connection.
+     *
+     * @param snapId - The Snap ID.
+     * @param id - The identifier for the WebSocket connection.
+     */
+    #close(snapId, id) {
+        const { socket } = this.#get(snapId, id);
+        socket.close();
+        this.#sockets.delete(id);
+    }
+    /**
+     * Close all open connections for a given Snap ID.
+     *
+     * @param snapId - The Snap ID.
+     */
+    #closeAll(snapId) {
+        for (const socket of this.#getAll(snapId)) {
+            this.#close(snapId, socket.id);
+        }
+    }
+    /**
+     * Send a message from a given Snap ID to a WebSocket connection.
+     *
+     * @param snapId - The Snap ID.
+     * @param id - The identifier for the WebSocket connection.
+     * @param data - The message to send.
+     */
+    async #sendMessage(snapId, id, data) {
+        const { socket, openPromise } = this.#get(snapId, id);
+        await openPromise;
+        const wrappedData = Array.isArray(data) ? new Uint8Array(data) : data;
+        socket.send(wrappedData);
+    }
+    /**
+     * Get a list of all open WebSocket connections for a Snap ID.
+     *
+     * @param snapId - The Snap ID.
+     * @returns A list of WebSocket connections.
+     */
+    #getAll(snapId) {
+        return [...this.#sockets.values()]
+            .filter((socket) => socket.snapId === snapId)
+            .map((socket) => ({
+            id: socket.id,
+            url: socket.url,
+            protocols: socket.protocols,
+        }));
+    }
+}
+//# sourceMappingURL=WebSocketService.mjs.map