UNPKG

viem

Version:

TypeScript Interface for Ethereum

71 lines 2.95 kB
import { WebSocketRequestError } from '../../errors/request.js'; import { getSocketRpcClient, } from './socket.js'; export async function getWebSocketRpcClient(url, options = {}) { const { keepAlive, reconnect } = options; return getSocketRpcClient({ async getSocket({ onClose, onError, onOpen, onResponse }) { const WebSocket = await import('isows').then((module) => module.WebSocket); const socket = new WebSocket(url); function onClose_() { onClose(); socket.removeEventListener('close', onClose_); socket.removeEventListener('message', onMessage); socket.removeEventListener('error', onError); socket.removeEventListener('open', onOpen); } function onMessage({ data }) { onResponse(JSON.parse(data)); } // Setup event listeners for RPC & subscription responses. socket.addEventListener('close', onClose_); socket.addEventListener('message', onMessage); socket.addEventListener('error', onError); socket.addEventListener('open', onOpen); // Wait for the socket to open. if (socket.readyState === WebSocket.CONNECTING) { await new Promise((resolve, reject) => { if (!socket) return; socket.onopen = resolve; socket.onerror = reject; }); } const { close: close_ } = socket; return Object.assign(socket, { close() { close_.bind(socket)(); onClose(); }, ping() { try { if (socket.readyState === socket.CLOSED || socket.readyState === socket.CLOSING) throw new WebSocketRequestError({ body: {}, url: socket.url, details: 'Socket is closed.', }); socket.send('ping'); } catch (error) { onError(error); } }, request({ body }) { if (socket.readyState === socket.CLOSED || socket.readyState === socket.CLOSING) throw new WebSocketRequestError({ body, url: socket.url, details: 'Socket is closed.', }); return socket.send(JSON.stringify(body)); }, }); }, keepAlive, reconnect, url, }); } //# sourceMappingURL=webSocket.js.map