bluebot/web/routes/import.js

A bitcoin trading bot for auto trading at various exchanges

const _ = require('lodash');
const promisify = require('tiny-promisify');
const pipelineRunner = promisify(require('../../core/workers/pipeline/parent'));

const cache = require('../state/cache');
const broadcast = cache.get('broadcast');
const importManager = cache.get('imports');

const base = require('./baseConfig');

// starts an import
// requires a post body with a config object
module.exports = function *() {
  let mode = 'importer';

  let config = {}

  _.merge(config, base, this.request.body);

  let importId = (Math.random() + '').slice(3);

  let errored = false;

  console.log('Import', importId, 'started');

  pipelineRunner(mode, config, (err, event) => {
    if(errored)
      return;

    if(err) {
      errored = true;
      console.error('RECEIVED ERROR IN IMPORT', importId);
      console.error(err);
      importManager.delete(importId);
      return broadcast({
        type: 'import_error',
        import_id: importId,
        error: err
      });
    }

    if(!event)
      return;

    // update local cache
    importManager.update(importId, {
      latest: event.latest,
      done: event.done
    });

    // emit update over ws
    let wsEvent = {
      type: 'import_update',
      import_id: importId,
      updates: {
        latest: event.latest,
        done: event.done
      }
    }
    broadcast(wsEvent);
  });

  let daterange = this.request.body.importer.daterange;

  const _import = {
    watch: config.watch,
    id: importId,
    latest: '',
    from: daterange.from,
    to: daterange.to
  }

  importManager.add(_import);
  this.body = _import;
}