node-cobinhood/lib/ws/index.js

nodejs cobinhood api client

const Socket = require('./socket')
const utils = require('../utils')
const keys = require('./keys')

modules = require('../import')(__dirname, [
  'orderbook',
  'order',
  'public_trade',
  'ticker',
  'candle',

  'fundingbook',
  'loan_ticker',
  'public_loan',
  'auth_loan',
  'funding',
])

class WS {
  constructor(client) {
    this.key = client.key
    this.socket = new Socket(this)
    this.socket.on('open', () => client.emit('open'))
    this.socket.on('close', () => client.emit('close'))
    this.socket.on('error', (err) => client.emit('error', err))
    this.socket.on('reconnecting', () => client.emit('reconnecting'))
    this.socket.on('reconnected', () => client.emit('reconnected'))
  }
  close() {
    this.socket.close()
  }
  placeLimitOrder(pair, side, price, size, source) {
    return this.socket.placeOrder(pair, '0', side, price, size, '0', source)
  }
  placeMarketOrder(pair, side, size, source) {
    return this.socket.placeOrder(pair, '1', side, '0', size, '0', source)
  }
  placeMarketStopOrder(pair, side, size, stopPrice) {
    return this.socket.placeOrder(pair, '2', side, '0', size, stopPrice, source)
  }
  placeLimitStopOrder(pair, side, price, size, stopPrice) {
    return this.socket.placeOrder(pair, '3', side, price, size, stopPrice, source)
  }
  modifyOrder(orderID, price, size) {
    return this.socket.modifyOrder(orderID, price, size)
  }
  cancelOrder(orderID) {
    return this.socket.cancelOrder(orderID)
  }
}

Object.entries(modules).forEach(([ key, m ]) => {
  let type = m.getType()
  key = key.replace(/^\w/, (c) => c.toUpperCase())
  key = key.replace(/_\w/, (s) => s[1].toUpperCase())
  WS.prototype[`subscribe${key}`] = function(opts, fn) {
    opts = Object.assign({ type: type }, opts)
    let channelKey = m.getChannelKey(opts)
    let preprocessKey = keys.getPreprocessKey(type)
    let promise = Promise.resolve()
    if (this.socket.listenerCount(preprocessKey) === 0) {
      this.socket.on(preprocessKey, m.preprocess.bind(this))
    }
    if (this.socket.listenerCount(channelKey) === 0) {
      promise = this.socket.subscribe(opts)
    }
    this.socket.on(channelKey, fn)
    return (
      promise
      .catch((err) => {
        this.socket.removeListener(channelKey, fn)
        throw err
      })
    )
  }
  WS.prototype[`unsubscribe${key}`] = function(opts, fn) {
    let channelKey = m.getChannelKey(opts)
    let promise = Promise.resolve()
    this.socket.removeListener(channelKey, fn)
    if (this.socket.listenerCount(channelKey) === 0) {
      promise = this.socket.unsubscribe(channelKey)
    }
    return promise
  }
})

module.exports = WS