import crypto from 'node:crypto'
import ReconnectingWebSocket from 'reconnecting-websocket'
import WebSocket from 'ws'
import type { Event } from './types/event/Event.ts'

let ws: ReconnectingWebSocket.default | null = null
let timeout: NodeJS.Timeout | null = null

export function initializeWebSocket({
  ip,
  accessToken,
  callback,
}: {
  ip: string
  accessToken: string
  callback: (o: Event) => void | Promise<void>
}) {
  // https://github.com/pladaria/reconnecting-websocket/issues/196
  ws = new (ReconnectingWebSocket as any)(`wss://${ip}:8443/v1`, [], {
    minReconnectionDelay: 10,
    maxReconnectionDelay: 10000,
    maxRetries: Number.MAX_SAFE_INTEGER,
    WebSocket: class extends WebSocket {
      constructor(url: string, protocols: string | string[]) {
        super(url, protocols, {
          headers: {
            authorization: `Bearer ${accessToken}`,
          },
          rejectUnauthorized: false,
        })
      }
    },
    debug: process.env['NODE_ENV'] === 'development',
  }) as ReconnectingWebSocket.default

  ws.addEventListener('message', (message) => {
    callback(JSON.parse(String(message.data)))
  })

  timeout = setInterval(() => {
    ws?.send(
      JSON.stringify({
        id: crypto.randomUUID(),
        specversion: '1.1.0',
        source: `urn:lpgera:dirigera`,
        time: new Date().toISOString(),
        type: 'ping',
        data: null,
      })
    )
  }, 30000)
}

export function closeWebSocket() {
  ws?.close()
  if (timeout) {
    clearInterval(timeout)
  }
}