const io = require('socket.io-client')

class Socket {
  constructor (config = {}) {
    if (typeof config === 'string') {
      // TODO remove this check in v2.0
      this.socket = io(`${config}`)
    } else {
      if (config.restURL) {
        this.socket = io(`${config.restURL}`)
      } else {
        const restURL = 'https://rest.bitcoin.com'
        this.socket = io(`${restURL}`)
      }

      if (config.callback) config.callback()
    }
  }

  listen (endpoint, cb) {
    this.socket.emit(endpoint)

    if (endpoint === 'blocks') this.socket.on('blocks', msg => cb(msg))
    else if (endpoint === 'transactions') { this.socket.on('transactions', msg => cb(msg)) }
  }
}

module.exports = Socket