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;

// from socket.io-protocol v4

var EVENT = 2;
var BINARY_EVENT = 5;

// Instances communication via a socket.io-adapter implementation.

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 = [EVENT, BINARY_EVENT];
    _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) ? BINARY_EVENT : EVENT, data: data };
      var opts = { rooms: [this.channel] };
      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 = _.includes(opts.rooms, this.channel);
      var isBusType = _.includes(this.types, packet.type);
      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,eyJ2ZXJzaW9uIjozLCJzb3VyY2VzIjpbIi4uL3NyYy9Tb2NrZXRJT0NsdXN0ZXJCdXMuanMiXSwibmFtZXMiOltdLCJtYXBwaW5ncyI6IkFBQUE7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7QUFFQSxJQUFNLFVBQVUsUUFBUSxVQUFSLENBQWhCO0FBQ0EsSUFBTSxJQUFJLFFBQVEsUUFBUixDQUFWO0FBQ0EsSUFBTSxZQUFZLFFBQVEsWUFBUixDQUFsQjs7ZUFDeUIsUUFBUSxRQUFSLEM7O0lBQWpCLFksWUFBQSxZOztBQUVSOztBQUNBLElBQU0sUUFBUSxDQUFkO0FBQ0EsSUFBTSxlQUFlLENBQXJCOztBQUVBOztJQUNNLGtCOzs7QUFFSiw4QkFBYSxNQUFiLEVBQXFCLE9BQXJCLEVBQThCO0FBQUE7O0FBQUE7O0FBRTVCLFVBQUssTUFBTCxHQUFjLE1BQWQ7QUFDQSxVQUFLLE9BQUwsR0FBZSxPQUFmO0FBQ0EsVUFBSyxPQUFMLEdBQWUsYUFBZjtBQUNBLFVBQUssYUFBTCxHQUFxQixDQUFFLGlCQUFGLEVBQ0UsZ0JBREYsRUFFRSx1QkFGRixDQUFyQjtBQUdBLFVBQUssS0FBTCxHQUFhLENBQUUsS0FBRixFQUFTLFlBQVQsQ0FBYjtBQUNBLFVBQUssYUFBTDtBQVQ0QjtBQVU3Qjs7Ozs2QkFFUztBQUFBOztBQUNSLGFBQU8sUUFBUSxZQUFSLENBQXFCLGNBQU07QUFDaEMsZUFBTyxPQUFLLE9BQUwsQ0FBYSxHQUFiLENBQWlCLE9BQUssTUFBTCxDQUFZLFdBQTdCLEVBQTBDLE9BQUssT0FBL0MsRUFBd0QsRUFBeEQsQ0FBUDtBQUNELE9BRk0sQ0FBUDtBQUdEOzs7MkNBRXVCLEUsRUFBSSxRLEVBQVU7QUFDcEMsaUNBQXlCLEVBQXpCLFNBQStCLFFBQS9CO0FBQ0Q7OzttQ0FFZSxFLEVBQUksSSxFQUFNO0FBQ3hCLGNBQVEsRUFBUjtBQUNFLGFBQUssZ0JBQUw7QUFBQSw2Q0FDaUMsSUFEakM7O0FBQUEsY0FDTyxFQURQO0FBQUEsY0FDVyxRQURYOztBQUFBLGNBQ3dCLEtBRHhCOztBQUVFLGNBQUksTUFBTSxLQUFLLHNCQUFMLENBQTRCLEVBQTVCLEVBQWdDLFFBQWhDLENBQVY7QUFDQSxpQkFBTyxDQUFDLEdBQUQsRUFBTSxLQUFOLENBQVA7QUFDRjtBQUNFLGlCQUFPLENBQUMsRUFBRCxFQUFLLElBQUwsQ0FBUDtBQU5KO0FBUUQ7O0FBRUQ7Ozs7eUJBQ00sRSxFQUFhO0FBQUEsd0NBQU4sSUFBTTtBQUFOLFlBQU07QUFBQTs7QUFDakIsVUFBSSxRQUFTLEVBQVQsU0FBZ0IsSUFBaEIsQ0FBSjtBQUNBLFVBQUksU0FBUyxFQUFFLE1BQU8sVUFBVSxJQUFWLElBQWtCLFlBQWxCLEdBQWlDLEtBQTFDLEVBQWtELFVBQWxELEVBQWI7QUFDQSxVQUFJLE9BQU8sRUFBRSxPQUFPLENBQUUsS0FBSyxPQUFQLENBQVQsRUFBWDtBQUNBLFdBQUssT0FBTCxDQUFhLFNBQWIsQ0FBdUIsTUFBdkIsRUFBK0IsSUFBL0IsRUFBcUMsS0FBckM7QUFDRDs7OzZCQUVTLE0sRUFBUTtBQUFBLGdEQUNJLE9BQU8sSUFEWDs7QUFBQSxVQUNYLEVBRFc7O0FBQUEsVUFDSixJQURJOztBQUVoQixVQUFJLEVBQUUsUUFBRixDQUFXLEtBQUssYUFBaEIsRUFBK0IsRUFBL0IsQ0FBSixFQUF3QztBQUFBOztBQUFBLDhCQUNuQixLQUFLLGNBQUwsQ0FBb0IsRUFBcEIsRUFBd0IsSUFBeEIsQ0FEbUI7O0FBQUE7O0FBQUEsWUFDakMsR0FEaUM7QUFBQSxZQUM1QixLQUQ0Qjs7QUFFdEMsZ0pBQWtCLEdBQWxCLDBDQUEwQixLQUExQjtBQUNELE9BSEQsTUFHTztBQUFBOztBQUNMLGdKQUFrQixFQUFsQiwwQ0FBeUIsSUFBekI7QUFDRDtBQUNGOzs7a0NBRWMsTSxFQUFRLEksRUFBTTtBQUMzQixVQUFJLGVBQWUsRUFBRSxRQUFGLENBQVcsS0FBSyxLQUFoQixFQUF1QixLQUFLLE9BQTVCLENBQW5CO0FBQ0EsVUFBSSxZQUFZLEVBQUUsUUFBRixDQUFXLEtBQUssS0FBaEIsRUFBdUIsT0FBTyxJQUE5QixDQUFoQjtBQUNBLFVBQUksZ0JBQWdCLFNBQXBCLEVBQStCO0FBQzdCLGFBQUssUUFBTCxDQUFjLE1BQWQ7QUFDRDtBQUNGOztBQUVEOzs7O29DQUNpQjtBQUNmLFVBQUksZ0JBQWdCLEtBQUssYUFBTCxDQUFtQixJQUFuQixDQUF3QixJQUF4QixDQUFwQjtBQUNBLFVBQUksVUFBVSxLQUFLLE9BQW5CO0FBQ0EsVUFBSSxPQUFPLEtBQUssT0FBTCxDQUFhLFNBQXhCO0FBQ0EsY0FBUSxTQUFSLEdBQW9CLFlBQW1CO0FBQUEsMkNBQU4sSUFBTTtBQUFOLGNBQU07QUFBQTs7QUFDckMsdUNBQWlCLElBQWpCO0FBQ0EsYUFBSyxLQUFMLENBQVcsT0FBWCxFQUFvQixJQUFwQjtBQUNELE9BSEQ7QUFJRDs7O0VBdEU4QixZOztBQTBFakMsT0FBTyxPQUFQLEdBQWlCLGtCQUFqQiIsImZpbGUiOiJTb2NrZXRJT0NsdXN0ZXJCdXMuanMiLCJzb3VyY2VzQ29udGVudCI6WyIndXNlIHN0cmljdCdcblxuY29uc3QgUHJvbWlzZSA9IHJlcXVpcmUoJ2JsdWViaXJkJylcbmNvbnN0IF8gPSByZXF1aXJlKCdsb2Rhc2gnKVxuY29uc3QgaGFzQmluYXJ5ID0gcmVxdWlyZSgnaGFzLWJpbmFyeScpXG5jb25zdCB7IEV2ZW50RW1pdHRlciB9ID0gcmVxdWlyZSgnZXZlbnRzJylcblxuLy8gZnJvbSBzb2NrZXQuaW8tcHJvdG9jb2wgdjRcbmNvbnN0IEVWRU5UID0gMlxuY29uc3QgQklOQVJZX0VWRU5UID0gNVxuXG4vLyBJbnN0YW5jZXMgY29tbXVuaWNhdGlvbiB2aWEgYSBzb2NrZXQuaW8tYWRhcHRlciBpbXBsZW1lbnRhdGlvbi5cbmNsYXNzIFNvY2tldElPQ2x1c3RlckJ1cyBleHRlbmRzIEV2ZW50RW1pdHRlciB7XG5cbiAgY29uc3RydWN0b3IgKHNlcnZlciwgYWRhcHRlcikge1xuICAgIHN1cGVyKClcbiAgICB0aGlzLnNlcnZlciA9IHNlcnZlclxuICAgIHRoaXMuYWRhcHRlciA9IGFkYXB0ZXJcbiAgICB0aGlzLmNoYW5uZWwgPSAnY2x1c3RlcjpidXMnXG4gICAgdGhpcy5pbnRlbmFsRXZlbnRzID0gWyAncm9vbUxlYXZlU29ja2V0JyxcbiAgICAgICAgICAgICAgICAgICAgICAgICAgICdzb2NrZXRSb29tTGVmdCcsXG4gICAgICAgICAgICAgICAgICAgICAgICAgICAnZGlzY29ubmVjdFVzZXJTb2NrZXRzJyBdXG4gICAgdGhpcy50eXBlcyA9IFsgRVZFTlQsIEJJTkFSWV9FVkVOVCBdXG4gICAgdGhpcy5pbmplY3RCdXNIb29rKClcbiAgfVxuXG4gIGxpc3RlbiAoKSB7XG4gICAgcmV0dXJuIFByb21pc2UuZnJvbUNhbGxiYWNrKGNiID0+IHtcbiAgICAgIHJldHVybiB0aGlzLmFkYXB0ZXIuYWRkKHRoaXMuc2VydmVyLmluc3RhbmNlVUlELCB0aGlzLmNoYW5uZWwsIGNiKVxuICAgIH0pXG4gIH1cblxuICBtYWtlU29ja2V0Um9vbUxlZnROYW1lIChpZCwgcm9vbU5hbWUpIHtcbiAgICByZXR1cm4gYHNvY2tldFJvb21MZWZ0OiR7aWR9OiR7cm9vbU5hbWV9YFxuICB9XG5cbiAgbWVyZ2VFdmVudE5hbWUgKGV2LCBhcmdzKSB7XG4gICAgc3dpdGNoIChldikge1xuICAgICAgY2FzZSAnc29ja2V0Um9vbUxlZnQnOlxuICAgICAgICBsZXQgW2lkLCByb29tTmFtZSwgLi4ubmFyZ3NdID0gYXJnc1xuICAgICAgICBsZXQgbmV2ID0gdGhpcy5tYWtlU29ja2V0Um9vbUxlZnROYW1lKGlkLCByb29tTmFtZSlcbiAgICAgICAgcmV0dXJuIFtuZXYsIG5hcmdzXVxuICAgICAgZGVmYXVsdDpcbiAgICAgICAgcmV0dXJuIFtldiwgYXJnc11cbiAgICB9XG4gIH1cblxuICAvLyBUT0RPOiBVc2UgYW4gQVBJIGZyb20gc29ja2V0LmlvIGlmKHdoZW4pIGl0IHdpbGwgYmUgYXZhaWxhYmxlLlxuICBlbWl0IChldiwgLi4uYXJncykge1xuICAgIGxldCBkYXRhID0gWyBldiwgLi4uYXJncyBdXG4gICAgbGV0IHBhY2tldCA9IHsgdHlwZTogKGhhc0JpbmFyeShhcmdzKSA/IEJJTkFSWV9FVkVOVCA6IEVWRU5UKSwgZGF0YSB9XG4gICAgbGV0IG9wdHMgPSB7IHJvb21zOiBbIHRoaXMuY2hhbm5lbCBdIH1cbiAgICB0aGlzLmFkYXB0ZXIuYnJvYWRjYXN0KHBhY2tldCwgb3B0cywgZmFsc2UpXG4gIH1cblxuICBvblBhY2tldCAocGFja2V0KSB7XG4gICAgbGV0IFtldiwgLi4uYXJnc10gPSBwYWNrZXQuZGF0YVxuICAgIGlmIChfLmluY2x1ZGVzKHRoaXMuaW50ZW5hbEV2ZW50cywgZXYpKSB7XG4gICAgICBsZXQgW25ldiwgbmFyZ3NdID0gdGhpcy5tZXJnZUV2ZW50TmFtZShldiwgYXJncylcbiAgICAgIHJldHVybiBzdXBlci5lbWl0KG5ldiwgLi4ubmFyZ3MpXG4gICAgfSBlbHNlIHtcbiAgICAgIHJldHVybiBzdXBlci5lbWl0KGV2LCAuLi5hcmdzKVxuICAgIH1cbiAgfVxuXG4gIGJyb2FkY2FzdEhvb2sgKHBhY2tldCwgb3B0cykge1xuICAgIGxldCBpc0J1c0NhaG5uZWwgPSBfLmluY2x1ZGVzKG9wdHMucm9vbXMsIHRoaXMuY2hhbm5lbClcbiAgICBsZXQgaXNCdXNUeXBlID0gXy5pbmNsdWRlcyh0aGlzLnR5cGVzLCBwYWNrZXQudHlwZSlcbiAgICBpZiAoaXNCdXNDYWhubmVsICYmIGlzQnVzVHlwZSkge1xuICAgICAgdGhpcy5vblBhY2tldChwYWNrZXQpXG4gICAgfVxuICB9XG5cbiAgLy8gVE9ETzogVXNlIGFuIEFQSSBmcm9tIHNvY2tldC5pbyBpZih3aGVuKSBpdCB3aWxsIGJlIGF2YWlsYWJsZS5cbiAgaW5qZWN0QnVzSG9vayAoKSB7XG4gICAgbGV0IGJyb2FkY2FzdEhvb2sgPSB0aGlzLmJyb2FkY2FzdEhvb2suYmluZCh0aGlzKVxuICAgIGxldCBhZGFwdGVyID0gdGhpcy5hZGFwdGVyXG4gICAgbGV0IG9yaWcgPSB0aGlzLmFkYXB0ZXIuYnJvYWRjYXN0XG4gICAgYWRhcHRlci5icm9hZGNhc3QgPSBmdW5jdGlvbiAoLi4uYXJncykge1xuICAgICAgYnJvYWRjYXN0SG9vayguLi5hcmdzKVxuICAgICAgb3JpZy5hcHBseShhZGFwdGVyLCBhcmdzKVxuICAgIH1cbiAgfVxuXG59XG5cbm1vZHVsZS5leHBvcnRzID0gU29ja2V0SU9DbHVzdGVyQnVzXG4iXX0=