bot18/node_modules/bitfinex-api-node/lib/model.js

A high-frequency cryptocurrency trading bot by Zenbot creator @carlos8f

'use strict'

const { EventEmitter } = require('events')
const { isArray } = Array

class Model extends EventEmitter {
  constructor (data = {}, fields = {}, boolFields = [], fieldKeys = []) {
    super()

    this._fields = fields
    this._boolFields = boolFields
    this._fieldKeys = fieldKeys

    if (Array.isArray(data)) {
      return Object.assign(this, this.constructor.unserialize(data))
    }

    Object.assign(this, data)
  }

  serialize () {
    const arr = []
    let i, key

    for (let j = 0; j < this._fieldKeys.length; j += 1) {
      key = this._fieldKeys[j]
      i = this._fields[key]
      arr[i] = this[key]

      if (this._boolFields.indexOf(key) !== -1) {
        arr[i] = arr[i] ? 1 : 0
      }
    }

    return arr
  }

  /**
   * @return {Object} pojo
   */
  toJS () {
    const arr = this.serialize()

    return Model.unserialize(
      arr, this._fields, this._boolFields, this._fieldKeys
    )
  }

  static unserialize (data, fields, boolFields, fieldKeys) {
    if (isArray(data) && isArray(data[0])) {
      return data.map(m => {
        return Model.unserialize(m, fields, boolFields, fieldKeys)
      })
    }

    const obj = {}

    fieldKeys.forEach((key) => {
      if ((fields[key] + '').length === 0) return

      if (isArray(data)) {
        obj[key] = data[fields[key]]
      } else {
        obj[key] = data[key]
      }

      // TODO: Convert boolFields to Object to speed lookup
      if (boolFields.indexOf('key') !== -1) {
        obj[key] = obj[key] === 1
      }
    })

    return obj
  }
}

module.exports = Model