soyuka/relieve

const TCPEE = require('tcpee')
const util = require('util')
const EE = require('events')

function TCPEEGroup(options) {
  if(!(this instanceof TCPEEGroup)) { return new TCPEEGroup(options) }

  this.options = options
  this.clients = new Map()
  this._ready = false

  EE.call(this, options)
}

util.inherits(TCPEEGroup, EE)

TCPEEGroup.prototype.add = function(sock) {
  let tcpee = new TCPEE(sock, this.options)

  tcpee.client.once('close', () => {
    this.clients.delete(tcpee.$TCPEE_IDENTITY)
  })

  tcpee.on('error', function(error, stack) {
    console.error('Got an error on the remote TCPEE socket')
    console.error(error, stack)
  })

  return new Promise((resolve, reject) => {
    tcpee.send('$TCPEE_IDENTITY')
    tcpee.once('$TCPEE_IDENTITY', (identity) => {
      tcpee.$TCPEE_IDENTITY = identity
      this.clients.set(identity, tcpee)
      this.emit('$TCPEE_ADD:'+identity, tcpee)
      resolve(tcpee)
    })
  })
}

TCPEEGroup.prototype.get = function(identity) {
  return this.clients.get(identity)
}

TCPEEGroup.prototype.ready = function() {
  this._ready = true
  this.emit('ready')
}

TCPEEGroup.prototype.destroy = function() {
  for (let [key, value] of this.clients) {
    value.client.destroy()
  }
}

module.exports = TCPEEGroup