/**
 * A WebSocket implementation that wraps the browser native WebSocket
 * interface and exposes a similar interface to the Node.js WebSocket
 */
export class BrowserSocket {
    #callbacks = new Set<Callback>()
    #ws: WebSocket

    constructor (wsUrl: string, _opts: unknown) {
        this.#ws = new globalThis.WebSocket(wsUrl)
        this.#ws.onmessage = this.handleMessage.bind(this)
    }

    handleMessage (event: MessageEvent) {
        for (const callback of this.#callbacks) {
            callback(event.data)
        }
    }

    send (data: string) {
        this.#ws.send(data)
    }

    on (event: string, callback: Callback) {
        if (event === 'open') {
            this.#ws.onopen = callback
        } else if (event === 'close') {
            this.#ws.onclose = callback
        } else if (event === 'error') {
            this.#ws.onerror = callback
        } else {
            this.#callbacks.add(callback)
        }

        return this
    }

    off (_event: string, callback: Callback) {
        this.#callbacks.delete(callback)
        return this
    }

    close () {
        this.#ws.close()
    }
}

type Callback = (data: unknown, reason?: unknown) => void