import {Emitter} from "@welshman/lib"
import {normalizeRelayUrl} from "@welshman/util"
import {Socket} from "./Socket.js"
import type {Message} from "./Socket.js"
import {ConnectionEvent} from "./ConnectionEvent.js"
import {ConnectionState} from "./ConnectionState.js"
import {ConnectionStats} from "./ConnectionStats.js"
import {ConnectionAuth} from "./ConnectionAuth.js"
import {ConnectionSender} from "./ConnectionSender.js"

export enum ConnectionStatus {
  Open = "open",
  Closed = "Closed",
}

const {Open, Closed} = ConnectionStatus

export class Connection extends Emitter {
  url: string
  socket: Socket
  sender: ConnectionSender
  state: ConnectionState
  stats: ConnectionStats
  auth: ConnectionAuth
  status = Open

  constructor(url: string) {
    super()

    if (url !== normalizeRelayUrl(url)) {
      console.warn(`Attempted to open connection to non-normalized url ${url}`)
    }

    this.url = url
    this.socket = new Socket(this)
    this.sender = new ConnectionSender(this)
    this.state = new ConnectionState(this)
    this.stats = new ConnectionStats(this)
    this.auth = new ConnectionAuth(this)
    this.setMaxListeners(100)
  }

  emit = (type: ConnectionEvent, ...args: any[]) => super.emit(type, this, ...args)

  send = async (message: Message) => {
    if (this.status !== Open) {
      throw new Error(`Attempted to send message on ${this.status} connection`)
    }

    this.socket.open()
    this.sender.push(message)
  }

  open = () => {
    this.status = Open
    this.socket.open()
    this.sender.worker.resume()
  }

  close = () => {
    this.status = Closed
    this.socket.close()
    this.sender.worker.pause()
  }

  cleanup = () => {
    this.close()
    this.removeAllListeners()
  }
}