87 lines
1.9 KiB
JavaScript
87 lines
1.9 KiB
JavaScript
/*
|
|
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 |