browser.js

const Base = require('./base')

class WSClient extends Base {
  constructor (endpoint, options = {}) {
    super(options)
    this.client = new WebSocket(endpoint, 'fast-ws')
    this.client.onerror = error => {
      super.emit('error', error)
    }
    this.client.onopen = () => {
      this.connectState = 1
      super.emit('open')
      this._heartbeat = setInterval(() => {
        this.ping()
      }, this.options.pingInterval)
    }
    this.client.onclose = () => {
      super.emit('close')
      this.connectState = -1
      clearInterval(this._heartbeat)
      super.emit('disconnect')
    }
    this.client.onmessage = ({ type, data }) => {
      if (this.connectState !== 2) {
        if (data === '\x00\x02') {
          this.connectState = 2
          super.emit('connect')
        } else {
          super.emit('error', new Error('Server version mismatch.'))
        }
      } else {
        this.incomingPacket(data)
      }
    }
  }

  ping () {
    const autoTerminate = setTimeout(() => this.client.close(), this.options.pingTimeout)
    this.client.send(Base.getPayload(null, 'ping'))
    super.once('pong', () => {
      clearTimeout(autoTerminate)
    })
  }
}

module.exports = WSClient