chat-service/lib/CommandBinder.js

'use strict';

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

var _slicedToArray3 = _interopRequireDefault(_slicedToArray2);

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

var _toConsumableArray3 = _interopRequireDefault(_toConsumableArray2);

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

var _classCallCheck3 = _interopRequireDefault(_classCallCheck2);

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

var _createClass3 = _interopRequireDefault(_createClass2);

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

var ExecInfo = require('./ExecInfo');
var Promise = require('bluebird');
var _ = require('lodash');

var _require = require('./utils');

var execHook = _require.execHook;
var logError = _require.logError;
var possiblyCallback = _require.possiblyCallback;
var resultsTransform = _require.resultsTransform;

// Implements command functions binding and wrapping.

var CommandBinder = function () {
  function CommandBinder(server, transport, userName) {
    (0, _classCallCheck3.default)(this, CommandBinder);

    this.server = server;
    this.transport = transport;
    this.userName = userName;
  }

  (0, _createClass3.default)(CommandBinder, [{
    key: 'commandWatcher',
    value: function commandWatcher(id, name) {
      var _this = this;

      this.server.runningCommands++;
      return Promise.resolve().disposer(function () {
        _this.server.runningCommands--;
        if (_this.transport.closed && _this.server.runningCommands <= 0) {
          _this.server.emit('commandsFinished');
        }
      });
    }
  }, {
    key: 'makeCommand',
    value: function makeCommand(name, fn) {
      var _this2 = this;

      var validator = this.server.validator;

      var beforeHook = this.server.hooks[name + 'Before'];
      var afterHook = this.server.hooks[name + 'After'];
      return function (args, info) {
        var execInfo = new ExecInfo();
        _.assign(execInfo, { server: _this2.server, userName: _this2.userName });
        _.assign(execInfo, info);
        _.assign(execInfo, validator.splitArguments(name, args));
        return Promise.using(_this2.commandWatcher(info.id, name), function () {
          return validator.checkArguments.apply(validator, [name].concat((0, _toConsumableArray3.default)(execInfo.args))).then(function () {
            if (beforeHook && !execInfo.bypassHooks) {
              return execHook(beforeHook, execInfo);
            } else {
              return Promise.resolve();
            }
          }).then(function (results) {
            if (results && results.length) {
              return results;
            }
            return fn.apply(undefined, (0, _toConsumableArray3.default)(execInfo.args).concat([execInfo])).then(function (result) {
              execInfo.results = [result];
            }, function (error) {
              execInfo.error = error;
            }).then(function () {
              if (afterHook && !execInfo.bypassHooks) {
                return execHook(afterHook, execInfo);
              } else {
                return Promise.resolve();
              }
            }).then(function (results) {
              if (results && results.length) {
                return results;
              } else if (execInfo.error) {
                return Promise.reject(execInfo.error);
              } else {
                return execInfo.results;
              }
            });
          });
        }).catch(function (error) {
          return logError(error);
        });
      };
    }
  }, {
    key: 'bindDisconnect',
    value: function bindDisconnect(id, fn) {
      var _this3 = this;

      var server = this.server;
      var hook = this.server.hooks.onDisconnect;
      this.transport.bindHandler(id, 'disconnect', function () {
        return Promise.using(_this3.commandWatcher(id, 'disconnect'), function () {
          return fn(id).catch(function (error) {
            return logError(error);
          }).catchReturn().then(function () {
            return execHook(hook, server, id);
          }).catch(function (error) {
            return logError(error);
          }).catchReturn();
        });
      });
    }
  }, {
    key: 'bindCommand',
    value: function bindCommand(id, name, fn) {
      var cmd = this.makeCommand(name, fn);
      var useErrorObjects = this.server.useRawErrorObjects;
      var info = { id: id };
      return this.transport.bindHandler(id, name, function () {
        var _possiblyCallback = possiblyCallback(arguments);

        var _possiblyCallback2 = (0, _slicedToArray3.default)(_possiblyCallback, 2);

        var args = _possiblyCallback2[0];
        var cb = _possiblyCallback2[1];

        var ack = resultsTransform(useErrorObjects, cb);
        return cmd(args, info).asCallback(ack, { spread: true });
      });
    }
  }]);
  return CommandBinder;
}();

module.exports = CommandBinder;
//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJzb3VyY2VzIjpbIi4uL3NyYy9Db21tYW5kQmluZGVyLmpzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiJBQUFBOzs7Ozs7Ozs7Ozs7Ozs7Ozs7OztBQUVBLElBQU0sV0FBVyxRQUFRLFlBQVIsQ0FBakI7QUFDQSxJQUFNLFVBQVUsUUFBUSxVQUFSLENBQWhCO0FBQ0EsSUFBTSxJQUFJLFFBQVEsUUFBUixDQUFWOztlQUVRLFFBQVEsU0FBUixDOztJQURBLFEsWUFBQSxRO0lBQVUsUSxZQUFBLFE7SUFBVSxnQixZQUFBLGdCO0lBQWtCLGdCLFlBQUEsZ0I7O0FBRzlDOztJQUNNLGE7QUFFSix5QkFBYSxNQUFiLEVBQXFCLFNBQXJCLEVBQWdDLFFBQWhDLEVBQTBDO0FBQUE7O0FBQ3hDLFNBQUssTUFBTCxHQUFjLE1BQWQ7QUFDQSxTQUFLLFNBQUwsR0FBaUIsU0FBakI7QUFDQSxTQUFLLFFBQUwsR0FBZ0IsUUFBaEI7QUFDRDs7OzttQ0FFZSxFLEVBQUksSSxFQUFNO0FBQUE7O0FBQ3hCLFdBQUssTUFBTCxDQUFZLGVBQVo7QUFDQSxhQUFPLFFBQVEsT0FBUixHQUFrQixRQUFsQixDQUEyQixZQUFNO0FBQ3RDLGNBQUssTUFBTCxDQUFZLGVBQVo7QUFDQSxZQUFJLE1BQUssU0FBTCxDQUFlLE1BQWYsSUFBeUIsTUFBSyxNQUFMLENBQVksZUFBWixJQUErQixDQUE1RCxFQUErRDtBQUM3RCxnQkFBSyxNQUFMLENBQVksSUFBWixDQUFpQixrQkFBakI7QUFDRDtBQUNGLE9BTE0sQ0FBUDtBQU1EOzs7Z0NBRVksSSxFQUFNLEUsRUFBSTtBQUFBOztBQUFBLFVBQ2YsU0FEZSxHQUNELEtBQUssTUFESixDQUNmLFNBRGU7O0FBRXJCLFVBQUksYUFBYSxLQUFLLE1BQUwsQ0FBWSxLQUFaLENBQXFCLElBQXJCLFlBQWpCO0FBQ0EsVUFBSSxZQUFZLEtBQUssTUFBTCxDQUFZLEtBQVosQ0FBcUIsSUFBckIsV0FBaEI7QUFDQSxhQUFPLFVBQUMsSUFBRCxFQUFPLElBQVAsRUFBZ0I7QUFDckIsWUFBSSxXQUFXLElBQUksUUFBSixFQUFmO0FBQ0EsVUFBRSxNQUFGLENBQVMsUUFBVCxFQUFtQixFQUFFLFFBQVEsT0FBSyxNQUFmLEVBQXVCLFVBQVUsT0FBSyxRQUF0QyxFQUFuQjtBQUNBLFVBQUUsTUFBRixDQUFTLFFBQVQsRUFBbUIsSUFBbkI7QUFDQSxVQUFFLE1BQUYsQ0FBUyxRQUFULEVBQW1CLFVBQVUsY0FBVixDQUF5QixJQUF6QixFQUErQixJQUEvQixDQUFuQjtBQUNBLGVBQU8sUUFBUSxLQUFSLENBQ0wsT0FBSyxjQUFMLENBQW9CLEtBQUssRUFBekIsRUFBNkIsSUFBN0IsQ0FESyxFQUVMO0FBQUEsaUJBQU0sVUFBVSxjQUFWLG1CQUF5QixJQUF6QiwwQ0FBa0MsU0FBUyxJQUEzQyxJQUNILElBREcsQ0FDRSxZQUFNO0FBQ1YsZ0JBQUksY0FBYyxDQUFDLFNBQVMsV0FBNUIsRUFBeUM7QUFDdkMscUJBQU8sU0FBUyxVQUFULEVBQXFCLFFBQXJCLENBQVA7QUFDRCxhQUZELE1BRU87QUFDTCxxQkFBTyxRQUFRLE9BQVIsRUFBUDtBQUNEO0FBQUUsV0FORCxFQU9ILElBUEcsQ0FPRSxtQkFBVztBQUNmLGdCQUFJLFdBQVcsUUFBUSxNQUF2QixFQUErQjtBQUFFLHFCQUFPLE9BQVA7QUFBZ0I7QUFDakQsbUJBQU8scURBQU0sU0FBUyxJQUFmLFVBQXFCLFFBQXJCLElBQ0osSUFESSxDQUNDLGtCQUFVO0FBQUUsdUJBQVMsT0FBVCxHQUFtQixDQUFDLE1BQUQsQ0FBbkI7QUFBNkIsYUFEMUMsRUFFQyxpQkFBUztBQUFFLHVCQUFTLEtBQVQsR0FBaUIsS0FBakI7QUFBd0IsYUFGcEMsRUFHSixJQUhJLENBR0MsWUFBTTtBQUNWLGtCQUFJLGFBQWEsQ0FBQyxTQUFTLFdBQTNCLEVBQXdDO0FBQ3RDLHVCQUFPLFNBQVMsU0FBVCxFQUFvQixRQUFwQixDQUFQO0FBQ0QsZUFGRCxNQUVPO0FBQ0wsdUJBQU8sUUFBUSxPQUFSLEVBQVA7QUFDRDtBQUFFLGFBUkEsRUFTSixJQVRJLENBU0MsbUJBQVc7QUFDZixrQkFBSSxXQUFXLFFBQVEsTUFBdkIsRUFBK0I7QUFDN0IsdUJBQU8sT0FBUDtBQUNELGVBRkQsTUFFTyxJQUFJLFNBQVMsS0FBYixFQUFvQjtBQUN6Qix1QkFBTyxRQUFRLE1BQVIsQ0FBZSxTQUFTLEtBQXhCLENBQVA7QUFDRCxlQUZNLE1BRUE7QUFDTCx1QkFBTyxTQUFTLE9BQWhCO0FBQ0Q7QUFDRixhQWpCSSxDQUFQO0FBa0JELFdBM0JHLENBQU47QUFBQSxTQUZLLEVBOEJKLEtBOUJJLENBOEJFO0FBQUEsaUJBQVMsU0FBUyxLQUFULENBQVQ7QUFBQSxTQTlCRixDQUFQO0FBK0JELE9BcENEO0FBcUNEOzs7bUNBRWUsRSxFQUFJLEUsRUFBSTtBQUFBOztBQUN0QixVQUFJLFNBQVMsS0FBSyxNQUFsQjtBQUNBLFVBQUksT0FBTyxLQUFLLE1BQUwsQ0FBWSxLQUFaLENBQWtCLFlBQTdCO0FBQ0EsV0FBSyxTQUFMLENBQWUsV0FBZixDQUEyQixFQUEzQixFQUErQixZQUEvQixFQUE2QyxZQUFNO0FBQ2pELGVBQU8sUUFBUSxLQUFSLENBQ0wsT0FBSyxjQUFMLENBQW9CLEVBQXBCLEVBQXdCLFlBQXhCLENBREssRUFFTDtBQUFBLGlCQUFNLEdBQUcsRUFBSCxFQUNILEtBREcsQ0FDRztBQUFBLG1CQUFTLFNBQVMsS0FBVCxDQUFUO0FBQUEsV0FESCxFQUVILFdBRkcsR0FHSCxJQUhHLENBR0U7QUFBQSxtQkFBTSxTQUFTLElBQVQsRUFBZSxNQUFmLEVBQXVCLEVBQXZCLENBQU47QUFBQSxXQUhGLEVBSUgsS0FKRyxDQUlHO0FBQUEsbUJBQVMsU0FBUyxLQUFULENBQVQ7QUFBQSxXQUpILEVBS0gsV0FMRyxFQUFOO0FBQUEsU0FGSyxDQUFQO0FBUUQsT0FURDtBQVVEOzs7Z0NBRVksRSxFQUFJLEksRUFBTSxFLEVBQUk7QUFDekIsVUFBSSxNQUFNLEtBQUssV0FBTCxDQUFpQixJQUFqQixFQUF1QixFQUF2QixDQUFWO0FBQ0EsVUFBSSxrQkFBa0IsS0FBSyxNQUFMLENBQVksa0JBQWxDO0FBQ0EsVUFBSSxPQUFPLEVBQUMsTUFBRCxFQUFYO0FBQ0EsYUFBTyxLQUFLLFNBQUwsQ0FBZSxXQUFmLENBQTJCLEVBQTNCLEVBQStCLElBQS9CLEVBQXFDLFlBQVk7QUFBQSxnQ0FDckMsaUJBQWlCLFNBQWpCLENBRHFDOztBQUFBOztBQUFBLFlBQ2pELElBRGlEO0FBQUEsWUFDM0MsRUFEMkM7O0FBRXRELFlBQUksTUFBTSxpQkFBaUIsZUFBakIsRUFBa0MsRUFBbEMsQ0FBVjtBQUNBLGVBQU8sSUFBSSxJQUFKLEVBQVUsSUFBVixFQUFnQixVQUFoQixDQUEyQixHQUEzQixFQUFnQyxFQUFFLFFBQVEsSUFBVixFQUFoQyxDQUFQO0FBQ0QsT0FKTSxDQUFQO0FBS0Q7Ozs7O0FBSUgsT0FBTyxPQUFQLEdBQWlCLGFBQWpCIiwiZmlsZSI6IkNvbW1hbmRCaW5kZXIuanMiLCJzb3VyY2VzQ29udGVudCI6WyIndXNlIHN0cmljdCdcblxuY29uc3QgRXhlY0luZm8gPSByZXF1aXJlKCcuL0V4ZWNJbmZvJylcbmNvbnN0IFByb21pc2UgPSByZXF1aXJlKCdibHVlYmlyZCcpXG5jb25zdCBfID0gcmVxdWlyZSgnbG9kYXNoJylcbmNvbnN0IHsgZXhlY0hvb2ssIGxvZ0Vycm9yLCBwb3NzaWJseUNhbGxiYWNrLCByZXN1bHRzVHJhbnNmb3JtIH0gPVxuICAgICAgICByZXF1aXJlKCcuL3V0aWxzJylcblxuLy8gSW1wbGVtZW50cyBjb21tYW5kIGZ1bmN0aW9ucyBiaW5kaW5nIGFuZCB3cmFwcGluZy5cbmNsYXNzIENvbW1hbmRCaW5kZXIge1xuXG4gIGNvbnN0cnVjdG9yIChzZXJ2ZXIsIHRyYW5zcG9ydCwgdXNlck5hbWUpIHtcbiAgICB0aGlzLnNlcnZlciA9IHNlcnZlclxuICAgIHRoaXMudHJhbnNwb3J0ID0gdHJhbnNwb3J0XG4gICAgdGhpcy51c2VyTmFtZSA9IHVzZXJOYW1lXG4gIH1cblxuICBjb21tYW5kV2F0Y2hlciAoaWQsIG5hbWUpIHtcbiAgICB0aGlzLnNlcnZlci5ydW5uaW5nQ29tbWFuZHMrK1xuICAgIHJldHVybiBQcm9taXNlLnJlc29sdmUoKS5kaXNwb3NlcigoKSA9PiB7XG4gICAgICB0aGlzLnNlcnZlci5ydW5uaW5nQ29tbWFuZHMtLVxuICAgICAgaWYgKHRoaXMudHJhbnNwb3J0LmNsb3NlZCAmJiB0aGlzLnNlcnZlci5ydW5uaW5nQ29tbWFuZHMgPD0gMCkge1xuICAgICAgICB0aGlzLnNlcnZlci5lbWl0KCdjb21tYW5kc0ZpbmlzaGVkJylcbiAgICAgIH1cbiAgICB9KVxuICB9XG5cbiAgbWFrZUNvbW1hbmQgKG5hbWUsIGZuKSB7XG4gICAgbGV0IHsgdmFsaWRhdG9yIH0gPSB0aGlzLnNlcnZlclxuICAgIGxldCBiZWZvcmVIb29rID0gdGhpcy5zZXJ2ZXIuaG9va3NbYCR7bmFtZX1CZWZvcmVgXVxuICAgIGxldCBhZnRlckhvb2sgPSB0aGlzLnNlcnZlci5ob29rc1tgJHtuYW1lfUFmdGVyYF1cbiAgICByZXR1cm4gKGFyZ3MsIGluZm8pID0+IHtcbiAgICAgIGxldCBleGVjSW5mbyA9IG5ldyBFeGVjSW5mbygpXG4gICAgICBfLmFzc2lnbihleGVjSW5mbywgeyBzZXJ2ZXI6IHRoaXMuc2VydmVyLCB1c2VyTmFtZTogdGhpcy51c2VyTmFtZSB9KVxuICAgICAgXy5hc3NpZ24oZXhlY0luZm8sIGluZm8pXG4gICAgICBfLmFzc2lnbihleGVjSW5mbywgdmFsaWRhdG9yLnNwbGl0QXJndW1lbnRzKG5hbWUsIGFyZ3MpKVxuICAgICAgcmV0dXJuIFByb21pc2UudXNpbmcoXG4gICAgICAgIHRoaXMuY29tbWFuZFdhdGNoZXIoaW5mby5pZCwgbmFtZSksXG4gICAgICAgICgpID0+IHZhbGlkYXRvci5jaGVja0FyZ3VtZW50cyhuYW1lLCAuLi5leGVjSW5mby5hcmdzKVxuICAgICAgICAgIC50aGVuKCgpID0+IHtcbiAgICAgICAgICAgIGlmIChiZWZvcmVIb29rICYmICFleGVjSW5mby5ieXBhc3NIb29rcykge1xuICAgICAgICAgICAgICByZXR1cm4gZXhlY0hvb2soYmVmb3JlSG9vaywgZXhlY0luZm8pXG4gICAgICAgICAgICB9IGVsc2Uge1xuICAgICAgICAgICAgICByZXR1cm4gUHJvbWlzZS5yZXNvbHZlKClcbiAgICAgICAgICAgIH0gfSlcbiAgICAgICAgICAudGhlbihyZXN1bHRzID0+IHtcbiAgICAgICAgICAgIGlmIChyZXN1bHRzICYmIHJlc3VsdHMubGVuZ3RoKSB7IHJldHVybiByZXN1bHRzIH1cbiAgICAgICAgICAgIHJldHVybiBmbiguLi5leGVjSW5mby5hcmdzLCBleGVjSW5mbylcbiAgICAgICAgICAgICAgLnRoZW4ocmVzdWx0ID0+IHsgZXhlY0luZm8ucmVzdWx0cyA9IFtyZXN1bHRdIH0sXG4gICAgICAgICAgICAgICAgICAgIGVycm9yID0+IHsgZXhlY0luZm8uZXJyb3IgPSBlcnJvciB9KVxuICAgICAgICAgICAgICAudGhlbigoKSA9PiB7XG4gICAgICAgICAgICAgICAgaWYgKGFmdGVySG9vayAmJiAhZXhlY0luZm8uYnlwYXNzSG9va3MpIHtcbiAgICAgICAgICAgICAgICAgIHJldHVybiBleGVjSG9vayhhZnRlckhvb2ssIGV4ZWNJbmZvKVxuICAgICAgICAgICAgICAgIH0gZWxzZSB7XG4gICAgICAgICAgICAgICAgICByZXR1cm4gUHJvbWlzZS5yZXNvbHZlKClcbiAgICAgICAgICAgICAgICB9IH0pXG4gICAgICAgICAgICAgIC50aGVuKHJlc3VsdHMgPT4ge1xuICAgICAgICAgICAgICAgIGlmIChyZXN1bHRzICYmIHJlc3VsdHMubGVuZ3RoKSB7XG4gICAgICAgICAgICAgICAgICByZXR1cm4gcmVzdWx0c1xuICAgICAgICAgICAgICAgIH0gZWxzZSBpZiAoZXhlY0luZm8uZXJyb3IpIHtcbiAgICAgICAgICAgICAgICAgIHJldHVybiBQcm9taXNlLnJlamVjdChleGVjSW5mby5lcnJvcilcbiAgICAgICAgICAgICAgICB9IGVsc2Uge1xuICAgICAgICAgICAgICAgICAgcmV0dXJuIGV4ZWNJbmZvLnJlc3VsdHNcbiAgICAgICAgICAgICAgICB9XG4gICAgICAgICAgICAgIH0pXG4gICAgICAgICAgfSkpXG4gICAgICAgIC5jYXRjaChlcnJvciA9PiBsb2dFcnJvcihlcnJvcikpXG4gICAgfVxuICB9XG5cbiAgYmluZERpc2Nvbm5lY3QgKGlkLCBmbikge1xuICAgIGxldCBzZXJ2ZXIgPSB0aGlzLnNlcnZlclxuICAgIGxldCBob29rID0gdGhpcy5zZXJ2ZXIuaG9va3Mub25EaXNjb25uZWN0XG4gICAgdGhpcy50cmFuc3BvcnQuYmluZEhhbmRsZXIoaWQsICdkaXNjb25uZWN0JywgKCkgPT4ge1xuICAgICAgcmV0dXJuIFByb21pc2UudXNpbmcoXG4gICAgICAgIHRoaXMuY29tbWFuZFdhdGNoZXIoaWQsICdkaXNjb25uZWN0JyksXG4gICAgICAgICgpID0+IGZuKGlkKVxuICAgICAgICAgIC5jYXRjaChlcnJvciA9PiBsb2dFcnJvcihlcnJvcikpXG4gICAgICAgICAgLmNhdGNoUmV0dXJuKClcbiAgICAgICAgICAudGhlbigoKSA9PiBleGVjSG9vayhob29rLCBzZXJ2ZXIsIGlkKSlcbiAgICAgICAgICAuY2F0Y2goZXJyb3IgPT4gbG9nRXJyb3IoZXJyb3IpKVxuICAgICAgICAgIC5jYXRjaFJldHVybigpKVxuICAgIH0pXG4gIH1cblxuICBiaW5kQ29tbWFuZCAoaWQsIG5hbWUsIGZuKSB7XG4gICAgbGV0IGNtZCA9IHRoaXMubWFrZUNvbW1hbmQobmFtZSwgZm4pXG4gICAgbGV0IHVzZUVycm9yT2JqZWN0cyA9IHRoaXMuc2VydmVyLnVzZVJhd0Vycm9yT2JqZWN0c1xuICAgIGxldCBpbmZvID0ge2lkfVxuICAgIHJldHVybiB0aGlzLnRyYW5zcG9ydC5iaW5kSGFuZGxlcihpZCwgbmFtZSwgZnVuY3Rpb24gKCkge1xuICAgICAgbGV0IFthcmdzLCBjYl0gPSBwb3NzaWJseUNhbGxiYWNrKGFyZ3VtZW50cylcbiAgICAgIGxldCBhY2sgPSByZXN1bHRzVHJhbnNmb3JtKHVzZUVycm9yT2JqZWN0cywgY2IpXG4gICAgICByZXR1cm4gY21kKGFyZ3MsIGluZm8pLmFzQ2FsbGJhY2soYWNrLCB7IHNwcmVhZDogdHJ1ZSB9KVxuICAgIH0pXG4gIH1cblxufVxuXG5tb2R1bGUuZXhwb3J0cyA9IENvbW1hbmRCaW5kZXJcbiJdfQ==