portfolioManager.js

/*
  The Portfolio class holds data about the portfolio
*/

const _ = require('lodash');
const async = require('async');
const errors = require('./exchangeErrors');
// const EventEmitter = require('events');

class Portfolio {
  constructor(config, api) {
    _.bindAll(this);
    this.config = config;
    this.api = api;
    this.balances = {};
    this.fee = null;
  }

  getBalance(fund) {
    return this.getFund(fund).amount;
  }

  // return the [fund] based on the data we have in memory
  getFund(fund) {
    return _.find(this.balances, function(f) { return f.name === fund});
  }

  // convert into the portfolio expected by the performanceAnalyzer
  convertBalances(asset,currency) { // rename?
    var asset = _.find(this.balances, a => a.name === this.config.asset).amount;
    var currency = _.find(this.balances, a => a.name === this.config.currency).amount;

    return {
      currency,
      asset,
      balance: currency + (asset * this.ticker.bid)
    }
  }

  setBalances(callback) {
    let set = (err, fullPortfolio) => {
      if(err) {
        console.log(err);
        throw new errors.ExchangeError(err);
      }

      // only include the currency/asset of this market
      const balances = [ this.config.currency, this.config.asset ]
        .map(name => {
          let item = _.find(fullPortfolio, {name});

          if(!item) {
            // assume we have 0
            item = { name, amount: 0 };
          }

          return item;
        });

      this.balances = balances;

      if(_.isFunction(callback))
        callback();
    }

    this.api.getPortfolio(set);
  }
  
  setFee(callback) {
    this.api.getFee((err, fee) => {
      if(err)
        throw new errors.ExchangeError(err);

      this.fee = fee;

      if(_.isFunction(callback))
        callback();
    });
  }

  setTicker(ticker) {
    this.ticker = ticker;
  }

}

module.exports = Portfolio