fix: Add missing WS types and allow local URLs

Author: FrederikBolding

packages/snaps-rpc-methods/src/permitted/openWebSocket.test.ts

@@ -88,15 +88,15 @@ describe('snap_openWebSocket', () => {
         id: 1,
         method: 'snap_openWebSocket',
         params: {
-          url: 'ws://metamask.io',
+          url: 'http://metamask.io',
         },
       });
 
       expect(response).toStrictEqual({
         error: {
           code: -32602,
           message:
-            'Invalid params: At path: url -- Expected URL, got "ws://metamask.io"..',
+            'Invalid params: At path: url -- Expected URL, got "http://metamask.io"..',
           stack: expect.any(String),
         },
         id: 1,

packages/snaps-rpc-methods/src/permitted/openWebSocket.ts

@@ -3,6 +3,7 @@ import type { PermittedHandlerExport } from '@metamask/permission-controller';
 import { providerErrors, rpcErrors } from '@metamask/rpc-errors';
 import {
   literal,
+  union,
   type JsonRpcRequest,
   type OpenWebSocketParams,
   type OpenWebSocketResult,
@@ -32,7 +33,7 @@ export type OpenWebSocketMethodHooks = {
 };
 
 const OpenWebSocketParametersStruct = object({
-  url: uri({ protocol: literal('wss:') }),
+  url: uri({ protocol: union([literal('wss:'), literal('ws:')]) }),
   protocols: optional(array(string())),
 });
 

packages/snaps-sdk/src/types/methods/methods.ts

@@ -3,6 +3,10 @@ import type {
   CancelBackgroundEventResult,
 } from './cancel-background-event';
 import type { ClearStateParams, ClearStateResult } from './clear-state';
+import type {
+  CloseWebSocketParams,
+  CloseWebSocketResult,
+} from './close-web-socket';
 import type {
   CreateInterfaceParams,
   CreateInterfaceResult,
@@ -49,6 +53,10 @@ import type {
 } from './get-preferences';
 import type { GetSnapsParams, GetSnapsResult } from './get-snaps';
 import type { GetStateParams, GetStateResult } from './get-state';
+import type {
+  GetWebSocketsParams,
+  GetWebSocketsResult,
+} from './get-web-sockets';
 import type {
   InvokeKeyringParams,
   InvokeKeyringResult,
@@ -64,6 +72,10 @@ import type {
 } from './manage-accounts';
 import type { ManageStateParams, ManageStateResult } from './manage-state';
 import type { NotifyParams, NotifyResult } from './notify';
+import type {
+  OpenWebSocketParams,
+  OpenWebSocketResult,
+} from './open-web-socket';
 import type { RequestSnapsParams, RequestSnapsResult } from './request-snaps';
 import type {
   ResolveInterfaceParams,
@@ -73,6 +85,10 @@ import type {
   ScheduleBackgroundEventParams,
   ScheduleBackgroundEventResult,
 } from './schedule-background-event';
+import type {
+  SendWebSocketMessageParams,
+  SendWebSocketMessageResult,
+} from './send-web-socket-message';
 import type { SetStateParams, SetStateResult } from './set-state';
 import type { TrackEventParams, TrackEventResult } from './track-event';
 import type {
@@ -125,6 +141,13 @@ export type SnapMethods = {
   snap_resolveInterface: [ResolveInterfaceParams, ResolveInterfaceResult];
   snap_setState: [SetStateParams, SetStateResult];
   snap_trackEvent: [TrackEventParams, TrackEventResult];
+  snap_openWebSocket: [OpenWebSocketParams, OpenWebSocketResult];
+  snap_closeWebSocket: [CloseWebSocketParams, CloseWebSocketResult];
+  snap_getWebSockets: [GetWebSocketsParams, GetWebSocketsResult];
+  snap_sendWebSocketMessage: [
+    SendWebSocketMessageParams,
+    SendWebSocketMessageResult,
+  ];
   wallet_getSnaps: [GetSnapsParams, GetSnapsResult];
   wallet_invokeKeyring: [InvokeKeyringParams, InvokeKeyringResult];
   wallet_invokeSnap: [InvokeSnapParams, InvokeSnapResult];