UNPKG

@metamask/snaps-rpc-methods

Version:
69 lines 2.37 kB
import { providerErrors, rpcErrors } from "@metamask/rpc-errors"; import { union } from "@metamask/snaps-sdk"; import { create, object, number, string, array, StructError } from "@metamask/superstruct"; import { SnapEndowments } from "../endowments/index.mjs"; const hookNames = { hasPermission: true, sendWebSocketMessage: true, }; const SendWebSocketMessageParametersStruct = object({ id: string(), message: union([string(), array(number())]), }); /** * Handler for the `snap_sendWebSocketMessage` method. */ export const sendWebSocketMessageHandler = { methodNames: ['snap_sendWebSocketMessage'], implementation: sendWebSocketMessageImplementation, hookNames, }; /** * The `snap_sendWebSocketMessage` method implementation. * * @param req - The JSON-RPC request object. * @param res - The JSON-RPC response object. * @param _next - The `json-rpc-engine` "next" callback. Not used by this function. * @param end - The `json-rpc-engine` "end" callback. * @param hooks - The RPC method hooks. * @param hooks.hasPermission - The function to check if a snap has the `endowment:network-access` permission. * @param hooks.sendWebSocketMessage - The function to send a WebSocket message. * @returns Nothing. */ function sendWebSocketMessageImplementation(req, res, _next, end, { hasPermission, sendWebSocketMessage }) { if (!hasPermission(SnapEndowments.NetworkAccess)) { return end(providerErrors.unauthorized()); } const { params } = req; try { const { id, message } = getValidatedParams(params); sendWebSocketMessage(id, message); res.result = null; } catch (error) { return end(error); } return end(); } /** * Validates the parameters for the snap_sendWebSocketMessage method. * * @param params - Parameters to validate. * @returns Validated parameters. * @throws Throws RPC error if validation fails. */ function getValidatedParams(params) { try { return create(params, SendWebSocketMessageParametersStruct); } catch (error) { if (error instanceof StructError) { throw rpcErrors.invalidParams({ message: `Invalid params: ${error.message}.`, }); } /* istanbul ignore next */ throw rpcErrors.internal(); } } //# sourceMappingURL=sendWebSocketMessage.mjs.map