const m = require('mithril')
var io = require('./socket')
var socket = io()

class Client {
  constructor(socket) {
    this.socket = socket
    this.isConnected = false

    this.socket.on('disconnect', this.disconnected.bind(this))
    this.socket.on('connect', this.connected.bind(this))
    this.components = new Map()
  }

  registerComponent(component) {
    this.components.set(component, [])
    if (component.ioInit) {
      component.ioInit()
    }
    if (this.isConnected && component.ioConnected) {
      component.ioConnected()
    }
  }

  safeGetComponent(component) {
    if (typeof(component) === 'string') {
      throw new Error('io.on was called without specifying active component')
    }
    let arr = this.components.get(component)
    if (!arr) {
      throw new Error('Registered component was missing but was attempting to add socket listener')
    }
    return arr
  }

  on(component, name, fn) {
    let arr = this.safeGetComponent(component)
    arr.push([name, fn])

    this.socket.on(name, (data) => {
      Promise.resolve().then(() => {
        return fn(data)
      })
      .catch(err => {
        console.error('Error running io handler for ' + name, err)
      })
    })
  }

  emit(name, data) {
    this.socket.emit(name, data)
  }

  unregisterComponent(component) {
    let arr = this.safeGetComponent(component)
    for (let item of arr) {
      this.socket.off(item[0], item[1])
    }
    this.components.delete(component)
  }

  disconnected() {
    this.isConnected = false
    m.redraw()
  }

  connected() {
    this.isConnected = true
    for (let component of this.components) {
      if (component[0].ioConnected) {
        component[0].ioConnected()
      }
    }
    m.redraw()
  }
}

module.exports = new Client(socket)