import type { MessageEvent } from 'isows'

import { WebSocketRequestError } from '../../errors/request.js'
import {
  type Socket,
  type SocketRpcClient,
  getSocketRpcClient,
} from './socket.js'

export async function getWebSocketRpcClient(
  url: string,
): Promise<SocketRpcClient<WebSocket>> {
  return getSocketRpcClient({
    async getSocket({ onResponse }) {
      const WebSocket = await import('isows').then((module) => module.WebSocket)
      const socket = new WebSocket(url)

      function onClose() {
        socket.removeEventListener('close', onClose)
        socket.removeEventListener('message', onMessage)
      }
      function onMessage({ data }: MessageEvent) {
        onResponse(JSON.parse(data))
      }

      // Setup event listeners for RPC & subscription responses.
      socket.addEventListener('close', onClose)
      socket.addEventListener('message', onMessage)

      // 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()
        },
        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))
        },
      } as Socket<WebSocket>)
    },
    url,
  })
}