bluebot/core/bluebotStream.js

A bitcoin trading bot for auto trading at various exchanges

// Small writable stream wrapper that
// passes data to all `candleConsumers`.

var Writable = require('stream').Writable;
var _ = require('lodash');

var util = require('./util');
var env = util.bluebotEnv();
var mode = util.bluebotMode();

var BlueBot = function(candleConsumers) {
  this.candleConsumers = candleConsumers;
  Writable.call(this, {objectMode: true});

  this.finalize = _.bind(this.finalize, this);
}

BlueBot.prototype = Object.create(Writable.prototype, {
  constructor: { value: BlueBot }
});

BlueBot.prototype._write = function(chunk, encoding, _done) {
  var done = _.after(this.candleConsumers.length, _done);
  _.each(this.candleConsumers, function(c) {
    c.processCandle(chunk, done);
  });
}

BlueBot.prototype.finalize = function() {
  var tradingMethod = _.find(
    this.candleConsumers,
    c => c.meta.name === 'Trading Advisor'
  );

  if(!tradingMethod)
    return this.shutdown();

  tradingMethod.finish(this.shutdown.bind(this));
}

BlueBot.prototype.shutdown = function() {
  _.each(this.candleConsumers, function(c) {
    if(c.finalize)
      c.finalize();
  });

  if(env === 'child-process')
    process.exit(0);
}

module.exports = BlueBot;