@owstack/wallet-service/base-service/lib/messagebroker.js

A service for multisignature HD wallets

'use strict';

var owsCommon = require('@owstack/ows-common');
var baseConfig = require('../../config');
var EventEmitter = require('events').EventEmitter;
var log = require('npmlog');
var lodash = owsCommon.deps.lodash;
var $ = require('preconditions').singleton();

log.debug = log.verbose;
log.disableColor();

class MessageBroker extends EventEmitter {
  constructor(config) {
    super();
    var self = this;
    config = config || baseConfig;

    if (config.messageBrokerServer) {
      var url = config.messageBrokerServer.url;
      self.remote = true;
      self.mq = require('socket.io-client').connect(url);
      self.mq.on('connect', function() {});
      self.mq.on('connect_error', function() {
        log.warn('Error connecting to message broker server @ ' + url);
      });

      self.mq.on('msg', function(data) {
        self.emit('msg', data);
      });

      log.info('Using message broker server at ' + url);
    }
  }
};

MessageBroker.isNotificationForMe = function(notification, networkNames) {
  if (!lodash.isArray(networkNames)) {
    networkNames = [networkNames];
  }
  return lodash.includes(networkNames, notification.networkName);
};

MessageBroker.prototype.send = function(data) {
  if (this.remote) {
    this.mq.emit('msg', data);
  } else {
    this.emit('msg', data);
  }
};

MessageBroker.prototype.onMessage = function(handler) {
  this.on('msg', handler);
};

module.exports = MessageBroker;