chat-service/lib/SocketIOClusterBus.js

'use strict';

var _toConsumableArray2 = require('babel-runtime/helpers/toConsumableArray');

var _toConsumableArray3 = _interopRequireDefault(_toConsumableArray2);

var _slicedToArray2 = require('babel-runtime/helpers/slicedToArray');

var _slicedToArray3 = _interopRequireDefault(_slicedToArray2);

var _toArray2 = require('babel-runtime/helpers/toArray');

var _toArray3 = _interopRequireDefault(_toArray2);

var _getPrototypeOf = require('babel-runtime/core-js/object/get-prototype-of');

var _getPrototypeOf2 = _interopRequireDefault(_getPrototypeOf);

var _classCallCheck2 = require('babel-runtime/helpers/classCallCheck');

var _classCallCheck3 = _interopRequireDefault(_classCallCheck2);

var _createClass2 = require('babel-runtime/helpers/createClass');

var _createClass3 = _interopRequireDefault(_createClass2);

var _possibleConstructorReturn2 = require('babel-runtime/helpers/possibleConstructorReturn');

var _possibleConstructorReturn3 = _interopRequireDefault(_possibleConstructorReturn2);

var _get4 = require('babel-runtime/helpers/get');

var _get5 = _interopRequireDefault(_get4);

var _inherits2 = require('babel-runtime/helpers/inherits');

var _inherits3 = _interopRequireDefault(_inherits2);

function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }

var Promise = require('bluebird');
var _ = require('lodash');
var hasBinary = require('has-binary');

var _require = require('events');

var EventEmitter = _require.EventEmitter;

var SocketIOClusterBus = function (_EventEmitter) {
  (0, _inherits3.default)(SocketIOClusterBus, _EventEmitter);

  function SocketIOClusterBus(server, adapter) {
    (0, _classCallCheck3.default)(this, SocketIOClusterBus);

    var _this = (0, _possibleConstructorReturn3.default)(this, (0, _getPrototypeOf2.default)(SocketIOClusterBus).call(this));

    _this.server = server;
    _this.adapter = adapter;
    _this.channel = 'cluster:bus';
    _this.intenalEvents = ['roomLeaveSocket', 'socketRoomLeft', 'disconnectUserSockets'];
    _this.types = [2, 5];
    _this.injectBusHook();
    return _this;
  }

  (0, _createClass3.default)(SocketIOClusterBus, [{
    key: 'listen',
    value: function listen() {
      var _this2 = this;

      return Promise.fromCallback(function (cb) {
        return _this2.adapter.add(_this2.server.instanceUID, _this2.channel, cb);
      });
    }
  }, {
    key: 'makeSocketRoomLeftName',
    value: function makeSocketRoomLeftName(id, roomName) {
      return 'socketRoomLeft:' + id + ':' + roomName;
    }
  }, {
    key: 'mergeEventName',
    value: function mergeEventName(ev, args) {
      switch (ev) {
        case 'socketRoomLeft':
          var _args = (0, _toArray3.default)(args);

          var id = _args[0];
          var roomName = _args[1];

          var nargs = _args.slice(2);

          var nev = this.makeSocketRoomLeftName(id, roomName);
          return [nev, nargs];
        default:
          return [ev, args];
      }
    }

    // TODO: Use an API from socket.io if(when) it will be available.

  }, {
    key: 'emit',
    value: function emit(ev) {
      for (var _len = arguments.length, args = Array(_len > 1 ? _len - 1 : 0), _key = 1; _key < _len; _key++) {
        args[_key - 1] = arguments[_key];
      }

      var data = [ev].concat(args);
      var packet = { type: hasBinary(args) ? 5 : 2, data: data };
      var opts = { rooms: [this.channel] };
      return this.adapter.broadcast(packet, opts, false);
    }
  }, {
    key: 'onPacket',
    value: function onPacket(packet) {
      var _packet$data = (0, _toArray3.default)(packet.data);

      var ev = _packet$data[0];

      var args = _packet$data.slice(1);

      if (_.includes(this.intenalEvents, ev)) {
        var _get2;

        var _mergeEventName = this.mergeEventName(ev, args);

        var _mergeEventName2 = (0, _slicedToArray3.default)(_mergeEventName, 2);

        var nev = _mergeEventName2[0];
        var nargs = _mergeEventName2[1];

        return (_get2 = (0, _get5.default)((0, _getPrototypeOf2.default)(SocketIOClusterBus.prototype), 'emit', this)).call.apply(_get2, [this, nev].concat((0, _toConsumableArray3.default)(nargs)));
      } else {
        var _get3;

        return (_get3 = (0, _get5.default)((0, _getPrototypeOf2.default)(SocketIOClusterBus.prototype), 'emit', this)).call.apply(_get3, [this, ev].concat((0, _toConsumableArray3.default)(args)));
      }
    }
  }, {
    key: 'broadcastHook',
    value: function broadcastHook(packet, opts) {
      var isBusCahnnel = _.indexOf(opts.rooms, this.channel) >= 0;
      var isBusType = _.indexOf(this.types, packet.type) >= 0;
      if (isBusCahnnel && isBusType) {
        this.onPacket(packet);
      }
    }

    // TODO: Use an API from socket.io if(when) it will be available.

  }, {
    key: 'injectBusHook',
    value: function injectBusHook() {
      var broadcastHook = this.broadcastHook.bind(this);
      var adapter = this.adapter;
      var orig = this.adapter.broadcast;
      adapter.broadcast = function () {
        for (var _len2 = arguments.length, args = Array(_len2), _key2 = 0; _key2 < _len2; _key2++) {
          args[_key2] = arguments[_key2];
        }

        broadcastHook.apply(undefined, args);
        orig.apply(adapter, args);
      };
    }
  }]);
  return SocketIOClusterBus;
}(EventEmitter);

module.exports = SocketIOClusterBus;
//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJzb3VyY2VzIjpbIi4uL3NyYy9Tb2NrZXRJT0NsdXN0ZXJCdXMuanMiXSwibmFtZXMiOltdLCJtYXBwaW5ncyI6IkFBQUE7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7QUFFQSxJQUFNLFVBQVUsUUFBUSxVQUFSLENBQWhCO0FBQ0EsSUFBTSxJQUFJLFFBQVEsUUFBUixDQUFWO0FBQ0EsSUFBTSxZQUFZLFFBQVEsWUFBUixDQUFsQjs7ZUFDeUIsUUFBUSxRQUFSLEM7O0lBQWpCLFksWUFBQSxZOztJQUVGLGtCOzs7QUFFSiw4QkFBYSxNQUFiLEVBQXFCLE9BQXJCLEVBQThCO0FBQUE7O0FBQUE7O0FBRTVCLFVBQUssTUFBTCxHQUFjLE1BQWQ7QUFDQSxVQUFLLE9BQUwsR0FBZSxPQUFmO0FBQ0EsVUFBSyxPQUFMLEdBQWUsYUFBZjtBQUNBLFVBQUssYUFBTCxHQUFxQixDQUFFLGlCQUFGLEVBQ0UsZ0JBREYsRUFFRSx1QkFGRixDQUFyQjtBQUdBLFVBQUssS0FBTCxHQUFhLENBQUUsQ0FBRixFQUFLLENBQUwsQ0FBYjtBQUNBLFVBQUssYUFBTDtBQVQ0QjtBQVU3Qjs7Ozs2QkFFUztBQUFBOztBQUNSLGFBQU8sUUFBUSxZQUFSLENBQXFCLGNBQU07QUFDaEMsZUFBTyxPQUFLLE9BQUwsQ0FBYSxHQUFiLENBQWlCLE9BQUssTUFBTCxDQUFZLFdBQTdCLEVBQTBDLE9BQUssT0FBL0MsRUFBd0QsRUFBeEQsQ0FBUDtBQUNELE9BRk0sQ0FBUDtBQUdEOzs7MkNBRXVCLEUsRUFBSSxRLEVBQVU7QUFDcEMsaUNBQXlCLEVBQXpCLFNBQStCLFFBQS9CO0FBQ0Q7OzttQ0FFZSxFLEVBQUksSSxFQUFNO0FBQ3hCLGNBQVEsRUFBUjtBQUNFLGFBQUssZ0JBQUw7QUFBQSw2Q0FDaUMsSUFEakM7O0FBQUEsY0FDTyxFQURQO0FBQUEsY0FDVyxRQURYOztBQUFBLGNBQ3dCLEtBRHhCOztBQUVFLGNBQUksTUFBTSxLQUFLLHNCQUFMLENBQTRCLEVBQTVCLEVBQWdDLFFBQWhDLENBQVY7QUFDQSxpQkFBTyxDQUFDLEdBQUQsRUFBTSxLQUFOLENBQVA7QUFDRjtBQUNFLGlCQUFPLENBQUMsRUFBRCxFQUFLLElBQUwsQ0FBUDtBQU5KO0FBUUQ7O0FBRUQ7Ozs7eUJBQ00sRSxFQUFhO0FBQUEsd0NBQU4sSUFBTTtBQUFOLFlBQU07QUFBQTs7QUFDakIsVUFBSSxRQUFTLEVBQVQsU0FBZ0IsSUFBaEIsQ0FBSjtBQUNBLFVBQUksU0FBUyxFQUFFLE1BQU8sVUFBVSxJQUFWLElBQWtCLENBQWxCLEdBQXNCLENBQS9CLEVBQW1DLFVBQW5DLEVBQWI7QUFDQSxVQUFJLE9BQU8sRUFBRSxPQUFPLENBQUUsS0FBSyxPQUFQLENBQVQsRUFBWDtBQUNBLGFBQU8sS0FBSyxPQUFMLENBQWEsU0FBYixDQUF1QixNQUF2QixFQUErQixJQUEvQixFQUFxQyxLQUFyQyxDQUFQO0FBQ0Q7Ozs2QkFFUyxNLEVBQVE7QUFBQSxnREFDSSxPQUFPLElBRFg7O0FBQUEsVUFDWCxFQURXOztBQUFBLFVBQ0osSUFESTs7QUFFaEIsVUFBSSxFQUFFLFFBQUYsQ0FBVyxLQUFLLGFBQWhCLEVBQStCLEVBQS9CLENBQUosRUFBd0M7QUFBQTs7QUFBQSw4QkFDbkIsS0FBSyxjQUFMLENBQW9CLEVBQXBCLEVBQXdCLElBQXhCLENBRG1COztBQUFBOztBQUFBLFlBQ2pDLEdBRGlDO0FBQUEsWUFDNUIsS0FENEI7O0FBRXRDLGdKQUFrQixHQUFsQiwwQ0FBMEIsS0FBMUI7QUFDRCxPQUhELE1BR087QUFBQTs7QUFDTCxnSkFBa0IsRUFBbEIsMENBQXlCLElBQXpCO0FBQ0Q7QUFDRjs7O2tDQUVjLE0sRUFBUSxJLEVBQU07QUFDM0IsVUFBSSxlQUFlLEVBQUUsT0FBRixDQUFVLEtBQUssS0FBZixFQUFzQixLQUFLLE9BQTNCLEtBQXVDLENBQTFEO0FBQ0EsVUFBSSxZQUFZLEVBQUUsT0FBRixDQUFVLEtBQUssS0FBZixFQUFzQixPQUFPLElBQTdCLEtBQXNDLENBQXREO0FBQ0EsVUFBSSxnQkFBZ0IsU0FBcEIsRUFBK0I7QUFDN0IsYUFBSyxRQUFMLENBQWMsTUFBZDtBQUNEO0FBQ0Y7O0FBRUQ7Ozs7b0NBQ2lCO0FBQ2YsVUFBSSxnQkFBZ0IsS0FBSyxhQUFMLENBQW1CLElBQW5CLENBQXdCLElBQXhCLENBQXBCO0FBQ0EsVUFBSSxVQUFVLEtBQUssT0FBbkI7QUFDQSxVQUFJLE9BQU8sS0FBSyxPQUFMLENBQWEsU0FBeEI7QUFDQSxjQUFRLFNBQVIsR0FBb0IsWUFBbUI7QUFBQSwyQ0FBTixJQUFNO0FBQU4sY0FBTTtBQUFBOztBQUNyQyx1Q0FBaUIsSUFBakI7QUFDQSxhQUFLLEtBQUwsQ0FBVyxPQUFYLEVBQW9CLElBQXBCO0FBQ0QsT0FIRDtBQUlEOzs7RUF0RThCLFk7O0FBMEVqQyxPQUFPLE9BQVAsR0FBaUIsa0JBQWpCIiwiZmlsZSI6IlNvY2tldElPQ2x1c3RlckJ1cy5qcyIsInNvdXJjZXNDb250ZW50IjpbIid1c2Ugc3RyaWN0J1xuXG5jb25zdCBQcm9taXNlID0gcmVxdWlyZSgnYmx1ZWJpcmQnKVxuY29uc3QgXyA9IHJlcXVpcmUoJ2xvZGFzaCcpXG5jb25zdCBoYXNCaW5hcnkgPSByZXF1aXJlKCdoYXMtYmluYXJ5JylcbmNvbnN0IHsgRXZlbnRFbWl0dGVyIH0gPSByZXF1aXJlKCdldmVudHMnKVxuXG5jbGFzcyBTb2NrZXRJT0NsdXN0ZXJCdXMgZXh0ZW5kcyBFdmVudEVtaXR0ZXIge1xuXG4gIGNvbnN0cnVjdG9yIChzZXJ2ZXIsIGFkYXB0ZXIpIHtcbiAgICBzdXBlcigpXG4gICAgdGhpcy5zZXJ2ZXIgPSBzZXJ2ZXJcbiAgICB0aGlzLmFkYXB0ZXIgPSBhZGFwdGVyXG4gICAgdGhpcy5jaGFubmVsID0gJ2NsdXN0ZXI6YnVzJ1xuICAgIHRoaXMuaW50ZW5hbEV2ZW50cyA9IFsgJ3Jvb21MZWF2ZVNvY2tldCcsXG4gICAgICAgICAgICAgICAgICAgICAgICAgICAnc29ja2V0Um9vbUxlZnQnLFxuICAgICAgICAgICAgICAgICAgICAgICAgICAgJ2Rpc2Nvbm5lY3RVc2VyU29ja2V0cycgXVxuICAgIHRoaXMudHlwZXMgPSBbIDIsIDUgXVxuICAgIHRoaXMuaW5qZWN0QnVzSG9vaygpXG4gIH1cblxuICBsaXN0ZW4gKCkge1xuICAgIHJldHVybiBQcm9taXNlLmZyb21DYWxsYmFjayhjYiA9PiB7XG4gICAgICByZXR1cm4gdGhpcy5hZGFwdGVyLmFkZCh0aGlzLnNlcnZlci5pbnN0YW5jZVVJRCwgdGhpcy5jaGFubmVsLCBjYilcbiAgICB9KVxuICB9XG5cbiAgbWFrZVNvY2tldFJvb21MZWZ0TmFtZSAoaWQsIHJvb21OYW1lKSB7XG4gICAgcmV0dXJuIGBzb2NrZXRSb29tTGVmdDoke2lkfToke3Jvb21OYW1lfWBcbiAgfVxuXG4gIG1lcmdlRXZlbnROYW1lIChldiwgYXJncykge1xuICAgIHN3aXRjaCAoZXYpIHtcbiAgICAgIGNhc2UgJ3NvY2tldFJvb21MZWZ0JzpcbiAgICAgICAgbGV0IFtpZCwgcm9vbU5hbWUsIC4uLm5hcmdzXSA9IGFyZ3NcbiAgICAgICAgbGV0IG5ldiA9IHRoaXMubWFrZVNvY2tldFJvb21MZWZ0TmFtZShpZCwgcm9vbU5hbWUpXG4gICAgICAgIHJldHVybiBbbmV2LCBuYXJnc11cbiAgICAgIGRlZmF1bHQ6XG4gICAgICAgIHJldHVybiBbZXYsIGFyZ3NdXG4gICAgfVxuICB9XG5cbiAgLy8gVE9ETzogVXNlIGFuIEFQSSBmcm9tIHNvY2tldC5pbyBpZih3aGVuKSBpdCB3aWxsIGJlIGF2YWlsYWJsZS5cbiAgZW1pdCAoZXYsIC4uLmFyZ3MpIHtcbiAgICBsZXQgZGF0YSA9IFsgZXYsIC4uLmFyZ3MgXVxuICAgIGxldCBwYWNrZXQgPSB7IHR5cGU6IChoYXNCaW5hcnkoYXJncykgPyA1IDogMiksIGRhdGEgfVxuICAgIGxldCBvcHRzID0geyByb29tczogWyB0aGlzLmNoYW5uZWwgXSB9XG4gICAgcmV0dXJuIHRoaXMuYWRhcHRlci5icm9hZGNhc3QocGFja2V0LCBvcHRzLCBmYWxzZSlcbiAgfVxuXG4gIG9uUGFja2V0IChwYWNrZXQpIHtcbiAgICBsZXQgW2V2LCAuLi5hcmdzXSA9IHBhY2tldC5kYXRhXG4gICAgaWYgKF8uaW5jbHVkZXModGhpcy5pbnRlbmFsRXZlbnRzLCBldikpIHtcbiAgICAgIGxldCBbbmV2LCBuYXJnc10gPSB0aGlzLm1lcmdlRXZlbnROYW1lKGV2LCBhcmdzKVxuICAgICAgcmV0dXJuIHN1cGVyLmVtaXQobmV2LCAuLi5uYXJncylcbiAgICB9IGVsc2Uge1xuICAgICAgcmV0dXJuIHN1cGVyLmVtaXQoZXYsIC4uLmFyZ3MpXG4gICAgfVxuICB9XG5cbiAgYnJvYWRjYXN0SG9vayAocGFja2V0LCBvcHRzKSB7XG4gICAgbGV0IGlzQnVzQ2Fobm5lbCA9IF8uaW5kZXhPZihvcHRzLnJvb21zLCB0aGlzLmNoYW5uZWwpID49IDBcbiAgICBsZXQgaXNCdXNUeXBlID0gXy5pbmRleE9mKHRoaXMudHlwZXMsIHBhY2tldC50eXBlKSA+PSAwXG4gICAgaWYgKGlzQnVzQ2Fobm5lbCAmJiBpc0J1c1R5cGUpIHtcbiAgICAgIHRoaXMub25QYWNrZXQocGFja2V0KVxuICAgIH1cbiAgfVxuXG4gIC8vIFRPRE86IFVzZSBhbiBBUEkgZnJvbSBzb2NrZXQuaW8gaWYod2hlbikgaXQgd2lsbCBiZSBhdmFpbGFibGUuXG4gIGluamVjdEJ1c0hvb2sgKCkge1xuICAgIGxldCBicm9hZGNhc3RIb29rID0gdGhpcy5icm9hZGNhc3RIb29rLmJpbmQodGhpcylcbiAgICBsZXQgYWRhcHRlciA9IHRoaXMuYWRhcHRlclxuICAgIGxldCBvcmlnID0gdGhpcy5hZGFwdGVyLmJyb2FkY2FzdFxuICAgIGFkYXB0ZXIuYnJvYWRjYXN0ID0gZnVuY3Rpb24gKC4uLmFyZ3MpIHtcbiAgICAgIGJyb2FkY2FzdEhvb2soLi4uYXJncylcbiAgICAgIG9yaWcuYXBwbHkoYWRhcHRlciwgYXJncylcbiAgICB9XG4gIH1cblxufVxuXG5tb2R1bGUuZXhwb3J0cyA9IFNvY2tldElPQ2x1c3RlckJ1c1xuIl19