neko/src/component/utils/reconnecter.ts

import EventEmitter from 'eventemitter3'

export interface ReconnecterAbstractEvents {
  connect: () => void
  disconnect: (error?: Error) => void
}

export abstract class ReconnecterAbstract extends EventEmitter<ReconnecterAbstractEvents> {
  constructor() {
    super()

    if (this.constructor == ReconnecterAbstract) {
      throw new Error("Abstract classes can't be instantiated.")
    }
  }

  public async connect() {
    throw new Error("Method 'connect()' must be implemented.")
  }

  public async disconnect() {
    throw new Error("Method 'disconnect()' must be implemented.")
  }
}

export interface ReconnecterEvents {
  open: () => void
  connect: () => void
  disconnect: () => void
  close: (error?: Error) => void
}

export interface ReconnecterConfig {
  max_reconnects: number
  timeout_ms: number
  backoff_ms: number
}

export class Reconnecter extends EventEmitter<ReconnecterEvents> {
  private _conn: ReconnecterAbstract
  private _config: ReconnecterConfig
  private _timeout: number | undefined

  private _open = false
  private _connected = false
  private _total_reconnects = 0
  private _last_connected: Date | undefined

  constructor(conn: ReconnecterAbstract, config?: ReconnecterConfig) {
    super()

    this._conn = conn
    this._config = {
      max_reconnects: 10,
      timeout_ms: 1500,
      backoff_ms: 750,
      ...config,
    }

    this._conn.on('connect', this.onConnect)
    this._conn.on('disconnect', this.onDisconnect)
  }

  private onConnect() {
    if (this._timeout) {
      window.clearTimeout(this._timeout)
      this._timeout = undefined
    }

    this._connected = true

    if (this._open) {
      this._last_connected = new Date()
      this.emit('connect')
    } else {
      this._conn.disconnect()
    }
  }

  private onDisconnect() {
    if (this._timeout) {
      window.clearTimeout(this._timeout)
      this._timeout = undefined
    }

    this._connected = false

    if (this._open) {
      this.emit('disconnect')
      this.reconnect()
    }
  }

  public get isOpen(): boolean {
    return this._open
  }

  public get isConnected(): boolean {
    return this._connected
  }

  public get totalReconnects(): number {
    return this._total_reconnects
  }

  public get lastConnected(): Date | undefined {
    return this._last_connected
  }

  public get config(): ReconnecterConfig {
    return { ...this._config }
  }

  public set config(conf: ReconnecterConfig) {
    this._config = { ...conf }

    if (this._config.max_reconnects > this._total_reconnects) {
      this.close(new Error('reconnection config changed'))
    }
  }

  public open(): void {
    if (this._open) {
      throw new Error('connection is already open')
    }

    this._open = true
    this.emit('open')

    this._conn.connect()
    this._timeout = window.setTimeout(() => {
      this._conn.disconnect()
      this.reconnect()
    }, this._config.timeout_ms)
  }

  public close(error?: Error): void {
    if (!this._open) {
      throw new Error('connection is already closed')
    }

    this._open = false
    this._last_connected = undefined
    this.emit('close', error)

    if (this._connected) {
      this._conn.disconnect()
    }
  }

  public reconnect(): void {
    if (this._connected) {
      throw new Error('connection is already connected')
    }

    this._total_reconnects++

    if (this._config.max_reconnects < this._total_reconnects || this._total_reconnects < 0) {
      setTimeout(this._conn.connect, this._config.backoff_ms)
    } else {
      this.close(new Error('reconnection failed'))
    }
  }

  public destroy() {
    this._conn.off('connect', this.onConnect)
    this._conn.off('disconnect', this.onDisconnect)
  }
}