import { ExtraWebSocket, State } from '@src/extra-websocket.js'
import { delay } from 'extra-promise'
import { pass } from '@blackglory/prelude'
import { AbortController, timeoutSignal, raceAbortSignals } from 'extra-abort'
import { waitForFunction } from '@blackglory/wait-for'

export function autoReconnect(
  ws: ExtraWebSocket
, reconnectInterval: number = 0
, connectTimeout?: number
): () => void {
  const controller = new AbortController()

  // Make sure the error listener is added, prevent crashes due to uncaught errors.
  const removeErrorListener = ws.on('error', pass)
  let removeCloseListener = ws.once('close', closeListener)

  return () => {
    controller.abort()

    removeCloseListener()
    removeErrorListener()
  }

  async function closeListener(): Promise<void> {
    while (true) {
      if (controller.signal.aborted) return

      await delay(reconnectInterval)
      if (controller.signal.aborted) return

      try {
        await waitForFunction(() => ws.getState() === State.Closed)
        await ws.connect(
          raceAbortSignals([
            connectTimeout && timeoutSignal(connectTimeout)
          , controller.signal
          ])
        )

        removeCloseListener()
        removeCloseListener = ws.once('close', closeListener)

        break
      } catch {
        pass()
      }
    }
  }
}