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 implementation 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).then(function () {
            return execHook(hook, server, id);
          }).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,eyJ2ZXJzaW9uIjozLCJzb3VyY2VzIjpbIi4uL3NyYy9Db21tYW5kQmluZGVyLmpzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiJBQUFBOzs7Ozs7Ozs7Ozs7Ozs7Ozs7OztBQUVBLElBQU0sV0FBVyxRQUFRLFlBQVIsQ0FBakI7QUFDQSxJQUFNLFVBQVUsUUFBUSxVQUFSLENBQWhCO0FBQ0EsSUFBTSxJQUFJLFFBQVEsUUFBUixDQUFWOztlQUVRLFFBQVEsU0FBUixDOztJQURBLFEsWUFBQSxRO0lBQVUsUSxZQUFBLFE7SUFBVSxnQixZQUFBLGdCO0lBQWtCLGdCLFlBQUEsZ0I7O0FBRzlDOztJQUNNLGE7QUFFSix5QkFBYSxNQUFiLEVBQXFCLFNBQXJCLEVBQWdDLFFBQWhDLEVBQTBDO0FBQUE7O0FBQ3hDLFNBQUssTUFBTCxHQUFjLE1BQWQ7QUFDQSxTQUFLLFNBQUwsR0FBaUIsU0FBakI7QUFDQSxTQUFLLFFBQUwsR0FBZ0IsUUFBaEI7QUFDRDs7OzttQ0FFZSxFLEVBQUksSSxFQUFNO0FBQUE7O0FBQ3hCLFdBQUssTUFBTCxDQUFZLGVBQVo7QUFDQSxhQUFPLFFBQVEsT0FBUixHQUFrQixRQUFsQixDQUEyQixZQUFNO0FBQ3RDLGNBQUssTUFBTCxDQUFZLGVBQVo7QUFDQSxZQUFJLE1BQUssU0FBTCxDQUFlLE1BQWYsSUFBeUIsTUFBSyxNQUFMLENBQVksZUFBWixJQUErQixDQUE1RCxFQUErRDtBQUM3RCxnQkFBSyxNQUFMLENBQVksSUFBWixDQUFpQixrQkFBakI7QUFDRDtBQUNGLE9BTE0sQ0FBUDtBQU1EOzs7Z0NBRVksSSxFQUFNLEUsRUFBSTtBQUFBOztBQUFBLFVBQ2YsU0FEZSxHQUNELEtBQUssTUFESixDQUNmLFNBRGU7O0FBRXJCLFVBQUksYUFBYSxLQUFLLE1BQUwsQ0FBWSxLQUFaLENBQXFCLElBQXJCLFlBQWpCO0FBQ0EsVUFBSSxZQUFZLEtBQUssTUFBTCxDQUFZLEtBQVosQ0FBcUIsSUFBckIsV0FBaEI7QUFDQSxhQUFPLFVBQUMsSUFBRCxFQUFPLElBQVAsRUFBZ0I7QUFDckIsWUFBSSxXQUFXLElBQUksUUFBSixFQUFmO0FBQ0EsVUFBRSxNQUFGLENBQVMsUUFBVCxFQUFtQixFQUFFLFFBQVEsT0FBSyxNQUFmLEVBQXVCLFVBQVUsT0FBSyxRQUF0QyxFQUFuQjtBQUNBLFVBQUUsTUFBRixDQUFTLFFBQVQsRUFBbUIsSUFBbkI7QUFDQSxVQUFFLE1BQUYsQ0FBUyxRQUFULEVBQW1CLFVBQVUsY0FBVixDQUF5QixJQUF6QixFQUErQixJQUEvQixDQUFuQjtBQUNBLGVBQU8sUUFBUSxLQUFSLENBQ0wsT0FBSyxjQUFMLENBQW9CLEtBQUssRUFBekIsRUFBNkIsSUFBN0IsQ0FESyxFQUVMO0FBQUEsaUJBQU0sVUFBVSxjQUFWLG1CQUF5QixJQUF6QiwwQ0FBa0MsU0FBUyxJQUEzQyxJQUNILElBREcsQ0FDRSxZQUFNO0FBQ1YsZ0JBQUksY0FBYyxDQUFDLFNBQVMsV0FBNUIsRUFBeUM7QUFDdkMscUJBQU8sU0FBUyxVQUFULEVBQXFCLFFBQXJCLENBQVA7QUFDRCxhQUZELE1BRU87QUFDTCxxQkFBTyxRQUFRLE9BQVIsRUFBUDtBQUNEO0FBQUUsV0FORCxFQU9ILElBUEcsQ0FPRSxtQkFBVztBQUNmLGdCQUFJLFdBQVcsUUFBUSxNQUF2QixFQUErQjtBQUFFLHFCQUFPLE9BQVA7QUFBZ0I7QUFDakQsbUJBQU8scURBQU0sU0FBUyxJQUFmLFVBQXFCLFFBQXJCLElBQ0osSUFESSxDQUNDLGtCQUFVO0FBQUUsdUJBQVMsT0FBVCxHQUFtQixDQUFDLE1BQUQsQ0FBbkI7QUFBNkIsYUFEMUMsRUFFQyxpQkFBUztBQUFFLHVCQUFTLEtBQVQsR0FBaUIsS0FBakI7QUFBd0IsYUFGcEMsRUFHSixJQUhJLENBR0MsWUFBTTtBQUNWLGtCQUFJLGFBQWEsQ0FBQyxTQUFTLFdBQTNCLEVBQXdDO0FBQ3RDLHVCQUFPLFNBQVMsU0FBVCxFQUFvQixRQUFwQixDQUFQO0FBQ0QsZUFGRCxNQUVPO0FBQ0wsdUJBQU8sUUFBUSxPQUFSLEVBQVA7QUFDRDtBQUFFLGFBUkEsRUFTSixJQVRJLENBU0MsbUJBQVc7QUFDZixrQkFBSSxXQUFXLFFBQVEsTUFBdkIsRUFBK0I7QUFDN0IsdUJBQU8sT0FBUDtBQUNELGVBRkQsTUFFTyxJQUFJLFNBQVMsS0FBYixFQUFvQjtBQUN6Qix1QkFBTyxRQUFRLE1BQVIsQ0FBZSxTQUFTLEtBQXhCLENBQVA7QUFDRCxlQUZNLE1BRUE7QUFDTCx1QkFBTyxTQUFTLE9BQWhCO0FBQ0Q7QUFDRixhQWpCSSxDQUFQO0FBa0JELFdBM0JHLENBQU47QUFBQSxTQUZLLEVBOEJKLEtBOUJJLENBOEJFO0FBQUEsaUJBQVMsU0FBUyxLQUFULENBQVQ7QUFBQSxTQTlCRixDQUFQO0FBK0JELE9BcENEO0FBcUNEOzs7bUNBRWUsRSxFQUFJLEUsRUFBSTtBQUFBOztBQUN0QixVQUFJLFNBQVMsS0FBSyxNQUFsQjtBQUNBLFVBQUksT0FBTyxLQUFLLE1BQUwsQ0FBWSxLQUFaLENBQWtCLFlBQTdCO0FBQ0EsV0FBSyxTQUFMLENBQWUsV0FBZixDQUEyQixFQUEzQixFQUErQixZQUEvQixFQUE2QyxZQUFNO0FBQ2pELGVBQU8sUUFBUSxLQUFSLENBQ0wsT0FBSyxjQUFMLENBQW9CLEVBQXBCLEVBQXdCLFlBQXhCLENBREssRUFFTDtBQUFBLGlCQUFNLEdBQUcsRUFBSCxFQUFPLElBQVAsQ0FBWTtBQUFBLG1CQUFNLFNBQVMsSUFBVCxFQUFlLE1BQWYsRUFBdUIsRUFBdkIsQ0FBTjtBQUFBLFdBQVosRUFBOEMsV0FBOUMsRUFBTjtBQUFBLFNBRkssQ0FBUDtBQUdELE9BSkQ7QUFLRDs7O2dDQUVZLEUsRUFBSSxJLEVBQU0sRSxFQUFJO0FBQ3pCLFVBQUksTUFBTSxLQUFLLFdBQUwsQ0FBaUIsSUFBakIsRUFBdUIsRUFBdkIsQ0FBVjtBQUNBLFVBQUksa0JBQWtCLEtBQUssTUFBTCxDQUFZLGtCQUFsQztBQUNBLFVBQUksT0FBTyxFQUFDLE1BQUQsRUFBWDtBQUNBLGFBQU8sS0FBSyxTQUFMLENBQWUsV0FBZixDQUEyQixFQUEzQixFQUErQixJQUEvQixFQUFxQyxZQUFZO0FBQUEsZ0NBQ3JDLGlCQUFpQixTQUFqQixDQURxQzs7QUFBQTs7QUFBQSxZQUNqRCxJQURpRDtBQUFBLFlBQzNDLEVBRDJDOztBQUV0RCxZQUFJLE1BQU0saUJBQWlCLGVBQWpCLEVBQWtDLEVBQWxDLENBQVY7QUFDQSxlQUFPLElBQUksSUFBSixFQUFVLElBQVYsRUFBZ0IsVUFBaEIsQ0FBMkIsR0FBM0IsRUFBZ0MsRUFBRSxRQUFRLElBQVYsRUFBaEMsQ0FBUDtBQUNELE9BSk0sQ0FBUDtBQUtEOzs7OztBQUlILE9BQU8sT0FBUCxHQUFpQixhQUFqQiIsImZpbGUiOiJDb21tYW5kQmluZGVyLmpzIiwic291cmNlc0NvbnRlbnQiOlsiJ3VzZSBzdHJpY3QnXG5cbmNvbnN0IEV4ZWNJbmZvID0gcmVxdWlyZSgnLi9FeGVjSW5mbycpXG5jb25zdCBQcm9taXNlID0gcmVxdWlyZSgnYmx1ZWJpcmQnKVxuY29uc3QgXyA9IHJlcXVpcmUoJ2xvZGFzaCcpXG5jb25zdCB7IGV4ZWNIb29rLCBsb2dFcnJvciwgcG9zc2libHlDYWxsYmFjaywgcmVzdWx0c1RyYW5zZm9ybSB9ID1cbiAgICAgICAgcmVxdWlyZSgnLi91dGlscycpXG5cbi8vIEltcGxlbWVudHMgY29tbWFuZCBpbXBsZW1lbnRhdGlvbiBmdW5jdGlvbnMgYmluZGluZyBhbmQgd3JhcHBpbmcuXG5jbGFzcyBDb21tYW5kQmluZGVyIHtcblxuICBjb25zdHJ1Y3RvciAoc2VydmVyLCB0cmFuc3BvcnQsIHVzZXJOYW1lKSB7XG4gICAgdGhpcy5zZXJ2ZXIgPSBzZXJ2ZXJcbiAgICB0aGlzLnRyYW5zcG9ydCA9IHRyYW5zcG9ydFxuICAgIHRoaXMudXNlck5hbWUgPSB1c2VyTmFtZVxuICB9XG5cbiAgY29tbWFuZFdhdGNoZXIgKGlkLCBuYW1lKSB7XG4gICAgdGhpcy5zZXJ2ZXIucnVubmluZ0NvbW1hbmRzKytcbiAgICByZXR1cm4gUHJvbWlzZS5yZXNvbHZlKCkuZGlzcG9zZXIoKCkgPT4ge1xuICAgICAgdGhpcy5zZXJ2ZXIucnVubmluZ0NvbW1hbmRzLS1cbiAgICAgIGlmICh0aGlzLnRyYW5zcG9ydC5jbG9zZWQgJiYgdGhpcy5zZXJ2ZXIucnVubmluZ0NvbW1hbmRzIDw9IDApIHtcbiAgICAgICAgdGhpcy5zZXJ2ZXIuZW1pdCgnY29tbWFuZHNGaW5pc2hlZCcpXG4gICAgICB9XG4gICAgfSlcbiAgfVxuXG4gIG1ha2VDb21tYW5kIChuYW1lLCBmbikge1xuICAgIGxldCB7IHZhbGlkYXRvciB9ID0gdGhpcy5zZXJ2ZXJcbiAgICBsZXQgYmVmb3JlSG9vayA9IHRoaXMuc2VydmVyLmhvb2tzW2Ake25hbWV9QmVmb3JlYF1cbiAgICBsZXQgYWZ0ZXJIb29rID0gdGhpcy5zZXJ2ZXIuaG9va3NbYCR7bmFtZX1BZnRlcmBdXG4gICAgcmV0dXJuIChhcmdzLCBpbmZvKSA9PiB7XG4gICAgICBsZXQgZXhlY0luZm8gPSBuZXcgRXhlY0luZm8oKVxuICAgICAgXy5hc3NpZ24oZXhlY0luZm8sIHsgc2VydmVyOiB0aGlzLnNlcnZlciwgdXNlck5hbWU6IHRoaXMudXNlck5hbWUgfSlcbiAgICAgIF8uYXNzaWduKGV4ZWNJbmZvLCBpbmZvKVxuICAgICAgXy5hc3NpZ24oZXhlY0luZm8sIHZhbGlkYXRvci5zcGxpdEFyZ3VtZW50cyhuYW1lLCBhcmdzKSlcbiAgICAgIHJldHVybiBQcm9taXNlLnVzaW5nKFxuICAgICAgICB0aGlzLmNvbW1hbmRXYXRjaGVyKGluZm8uaWQsIG5hbWUpLFxuICAgICAgICAoKSA9PiB2YWxpZGF0b3IuY2hlY2tBcmd1bWVudHMobmFtZSwgLi4uZXhlY0luZm8uYXJncylcbiAgICAgICAgICAudGhlbigoKSA9PiB7XG4gICAgICAgICAgICBpZiAoYmVmb3JlSG9vayAmJiAhZXhlY0luZm8uYnlwYXNzSG9va3MpIHtcbiAgICAgICAgICAgICAgcmV0dXJuIGV4ZWNIb29rKGJlZm9yZUhvb2ssIGV4ZWNJbmZvKVxuICAgICAgICAgICAgfSBlbHNlIHtcbiAgICAgICAgICAgICAgcmV0dXJuIFByb21pc2UucmVzb2x2ZSgpXG4gICAgICAgICAgICB9IH0pXG4gICAgICAgICAgLnRoZW4ocmVzdWx0cyA9PiB7XG4gICAgICAgICAgICBpZiAocmVzdWx0cyAmJiByZXN1bHRzLmxlbmd0aCkgeyByZXR1cm4gcmVzdWx0cyB9XG4gICAgICAgICAgICByZXR1cm4gZm4oLi4uZXhlY0luZm8uYXJncywgZXhlY0luZm8pXG4gICAgICAgICAgICAgIC50aGVuKHJlc3VsdCA9PiB7IGV4ZWNJbmZvLnJlc3VsdHMgPSBbcmVzdWx0XSB9LFxuICAgICAgICAgICAgICAgICAgICBlcnJvciA9PiB7IGV4ZWNJbmZvLmVycm9yID0gZXJyb3IgfSlcbiAgICAgICAgICAgICAgLnRoZW4oKCkgPT4ge1xuICAgICAgICAgICAgICAgIGlmIChhZnRlckhvb2sgJiYgIWV4ZWNJbmZvLmJ5cGFzc0hvb2tzKSB7XG4gICAgICAgICAgICAgICAgICByZXR1cm4gZXhlY0hvb2soYWZ0ZXJIb29rLCBleGVjSW5mbylcbiAgICAgICAgICAgICAgICB9IGVsc2Uge1xuICAgICAgICAgICAgICAgICAgcmV0dXJuIFByb21pc2UucmVzb2x2ZSgpXG4gICAgICAgICAgICAgICAgfSB9KVxuICAgICAgICAgICAgICAudGhlbihyZXN1bHRzID0+IHtcbiAgICAgICAgICAgICAgICBpZiAocmVzdWx0cyAmJiByZXN1bHRzLmxlbmd0aCkge1xuICAgICAgICAgICAgICAgICAgcmV0dXJuIHJlc3VsdHNcbiAgICAgICAgICAgICAgICB9IGVsc2UgaWYgKGV4ZWNJbmZvLmVycm9yKSB7XG4gICAgICAgICAgICAgICAgICByZXR1cm4gUHJvbWlzZS5yZWplY3QoZXhlY0luZm8uZXJyb3IpXG4gICAgICAgICAgICAgICAgfSBlbHNlIHtcbiAgICAgICAgICAgICAgICAgIHJldHVybiBleGVjSW5mby5yZXN1bHRzXG4gICAgICAgICAgICAgICAgfVxuICAgICAgICAgICAgICB9KVxuICAgICAgICAgIH0pKVxuICAgICAgICAuY2F0Y2goZXJyb3IgPT4gbG9nRXJyb3IoZXJyb3IpKVxuICAgIH1cbiAgfVxuXG4gIGJpbmREaXNjb25uZWN0IChpZCwgZm4pIHtcbiAgICBsZXQgc2VydmVyID0gdGhpcy5zZXJ2ZXJcbiAgICBsZXQgaG9vayA9IHRoaXMuc2VydmVyLmhvb2tzLm9uRGlzY29ubmVjdFxuICAgIHRoaXMudHJhbnNwb3J0LmJpbmRIYW5kbGVyKGlkLCAnZGlzY29ubmVjdCcsICgpID0+IHtcbiAgICAgIHJldHVybiBQcm9taXNlLnVzaW5nKFxuICAgICAgICB0aGlzLmNvbW1hbmRXYXRjaGVyKGlkLCAnZGlzY29ubmVjdCcpLFxuICAgICAgICAoKSA9PiBmbihpZCkudGhlbigoKSA9PiBleGVjSG9vayhob29rLCBzZXJ2ZXIsIGlkKSkuY2F0Y2hSZXR1cm4oKSlcbiAgICB9KVxuICB9XG5cbiAgYmluZENvbW1hbmQgKGlkLCBuYW1lLCBmbikge1xuICAgIGxldCBjbWQgPSB0aGlzLm1ha2VDb21tYW5kKG5hbWUsIGZuKVxuICAgIGxldCB1c2VFcnJvck9iamVjdHMgPSB0aGlzLnNlcnZlci51c2VSYXdFcnJvck9iamVjdHNcbiAgICBsZXQgaW5mbyA9IHtpZH1cbiAgICByZXR1cm4gdGhpcy50cmFuc3BvcnQuYmluZEhhbmRsZXIoaWQsIG5hbWUsIGZ1bmN0aW9uICgpIHtcbiAgICAgIGxldCBbYXJncywgY2JdID0gcG9zc2libHlDYWxsYmFjayhhcmd1bWVudHMpXG4gICAgICBsZXQgYWNrID0gcmVzdWx0c1RyYW5zZm9ybSh1c2VFcnJvck9iamVjdHMsIGNiKVxuICAgICAgcmV0dXJuIGNtZChhcmdzLCBpbmZvKS5hc0NhbGxiYWNrKGFjaywgeyBzcHJlYWQ6IHRydWUgfSlcbiAgICB9KVxuICB9XG5cbn1cblxubW9kdWxlLmV4cG9ydHMgPSBDb21tYW5kQmluZGVyXG4iXX0=