123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512 |
- 'use strict';
- function Kareem() {
- this._pres = new Map();
- this._posts = new Map();
- }
- Kareem.prototype.execPre = function(name, context, args, callback) {
- if (arguments.length === 3) {
- callback = args;
- args = [];
- }
- var pres = get(this._pres, name, []);
- var numPres = pres.length;
- var numAsyncPres = pres.numAsync || 0;
- var currentPre = 0;
- var asyncPresLeft = numAsyncPres;
- var done = false;
- var $args = args;
- if (!numPres) {
- return process.nextTick(function() {
- callback(null);
- });
- }
- var next = function() {
- if (currentPre >= numPres) {
- return;
- }
- var pre = pres[currentPre];
- if (pre.isAsync) {
- var args = [
- decorateNextFn(_next),
- decorateNextFn(function(error) {
- if (error) {
- if (done) {
- return;
- }
- done = true;
- return callback(error);
- }
- if (--asyncPresLeft === 0 && currentPre >= numPres) {
- return callback(null);
- }
- })
- ];
- callMiddlewareFunction(pre.fn, context, args, args[0]);
- } else if (pre.fn.length > 0) {
- var args = [decorateNextFn(_next)];
- var _args = arguments.length >= 2 ? arguments : [null].concat($args);
- for (var i = 1; i < _args.length; ++i) {
- args.push(_args[i]);
- }
- callMiddlewareFunction(pre.fn, context, args, args[0]);
- } else {
- let maybePromise = null;
- try {
- maybePromise = pre.fn.call(context);
- } catch (err) {
- if (err != null) {
- return callback(err);
- }
- }
- if (isPromise(maybePromise)) {
- maybePromise.then(() => _next(), err => _next(err));
- } else {
- if (++currentPre >= numPres) {
- if (asyncPresLeft > 0) {
- // Leave parallel hooks to run
- return;
- } else {
- return process.nextTick(function() {
- callback(null);
- });
- }
- }
- next();
- }
- }
- };
- next.apply(null, [null].concat(args));
- function _next(error) {
- if (error) {
- if (done) {
- return;
- }
- done = true;
- return callback(error);
- }
- if (++currentPre >= numPres) {
- if (asyncPresLeft > 0) {
- // Leave parallel hooks to run
- return;
- } else {
- return callback(null);
- }
- }
- next.apply(context, arguments);
- }
- };
- Kareem.prototype.execPreSync = function(name, context, args) {
- var pres = get(this._pres, name, []);
- var numPres = pres.length;
- for (var i = 0; i < numPres; ++i) {
- pres[i].fn.apply(context, args || []);
- }
- };
- Kareem.prototype.execPost = function(name, context, args, options, callback) {
- if (arguments.length < 5) {
- callback = options;
- options = null;
- }
- var posts = get(this._posts, name, []);
- var numPosts = posts.length;
- var currentPost = 0;
- var firstError = null;
- if (options && options.error) {
- firstError = options.error;
- }
- if (!numPosts) {
- return process.nextTick(function() {
- callback.apply(null, [firstError].concat(args));
- });
- }
- var next = function() {
- var post = posts[currentPost].fn;
- var numArgs = 0;
- var argLength = args.length;
- var newArgs = [];
- for (var i = 0; i < argLength; ++i) {
- numArgs += args[i] && args[i]._kareemIgnore ? 0 : 1;
- if (!args[i] || !args[i]._kareemIgnore) {
- newArgs.push(args[i]);
- }
- }
- if (firstError) {
- if (post.length === numArgs + 2) {
- var _cb = decorateNextFn(function(error) {
- if (error) {
- firstError = error;
- }
- if (++currentPost >= numPosts) {
- return callback.call(null, firstError);
- }
- next();
- });
- callMiddlewareFunction(post, context,
- [firstError].concat(newArgs).concat([_cb]), _cb);
- } else {
- if (++currentPost >= numPosts) {
- return callback.call(null, firstError);
- }
- next();
- }
- } else {
- const _cb = decorateNextFn(function(error) {
- if (error) {
- firstError = error;
- return next();
- }
- if (++currentPost >= numPosts) {
- return callback.apply(null, [null].concat(args));
- }
- next();
- });
- if (post.length === numArgs + 2) {
- // Skip error handlers if no error
- if (++currentPost >= numPosts) {
- return callback.apply(null, [null].concat(args));
- }
- return next();
- }
- if (post.length === numArgs + 1) {
- callMiddlewareFunction(post, context, newArgs.concat([_cb]), _cb);
- } else {
- let error;
- let maybePromise;
- try {
- maybePromise = post.apply(context, newArgs);
- } catch (err) {
- error = err;
- firstError = err;
- }
- if (isPromise(maybePromise)) {
- return maybePromise.then(() => _cb(), err => _cb(err));
- }
- if (++currentPost >= numPosts) {
- return callback.apply(null, [error].concat(args));
- }
- next(error);
- }
- }
- };
- next();
- };
- Kareem.prototype.execPostSync = function(name, context, args) {
- const posts = get(this._posts, name, []);
- const numPosts = posts.length;
- for (let i = 0; i < numPosts; ++i) {
- posts[i].fn.apply(context, args || []);
- }
- };
- Kareem.prototype.createWrapperSync = function(name, fn) {
- var kareem = this;
- return function syncWrapper() {
- kareem.execPreSync(name, this, arguments);
- var toReturn = fn.apply(this, arguments);
- kareem.execPostSync(name, this, [toReturn]);
- return toReturn;
- };
- }
- function _handleWrapError(instance, error, name, context, args, options, callback) {
- if (options.useErrorHandlers) {
- var _options = { error: error };
- return instance.execPost(name, context, args, _options, function(error) {
- return typeof callback === 'function' && callback(error);
- });
- } else {
- return typeof callback === 'function' ?
- callback(error) :
- undefined;
- }
- }
- Kareem.prototype.wrap = function(name, fn, context, args, options) {
- const lastArg = (args.length > 0 ? args[args.length - 1] : null);
- const argsWithoutCb = typeof lastArg === 'function' ?
- args.slice(0, args.length - 1) :
- args;
- const _this = this;
- options = options || {};
- const checkForPromise = options.checkForPromise;
- this.execPre(name, context, args, function(error) {
- if (error) {
- const numCallbackParams = options.numCallbackParams || 0;
- const errorArgs = options.contextParameter ? [context] : [];
- for (var i = errorArgs.length; i < numCallbackParams; ++i) {
- errorArgs.push(null);
- }
- return _handleWrapError(_this, error, name, context, errorArgs,
- options, lastArg);
- }
- const end = (typeof lastArg === 'function' ? args.length - 1 : args.length);
- const numParameters = fn.length;
- const ret = fn.apply(context, args.slice(0, end).concat(_cb));
- if (checkForPromise) {
- if (ret != null && typeof ret.then === 'function') {
- // Thenable, use it
- return ret.then(
- res => _cb(null, res),
- err => _cb(err)
- );
- }
- // If `fn()` doesn't have a callback argument and doesn't return a
- // promise, assume it is sync
- if (numParameters < end + 1) {
- return _cb(null, ret);
- }
- }
- function _cb() {
- const args = arguments;
- const argsWithoutError = Array.prototype.slice.call(arguments, 1);
- if (options.nullResultByDefault && argsWithoutError.length === 0) {
- argsWithoutError.push(null);
- }
- if (arguments[0]) {
- // Assume error
- return _handleWrapError(_this, arguments[0], name, context,
- argsWithoutError, options, lastArg);
- } else {
- _this.execPost(name, context, argsWithoutError, function() {
- if (arguments[0]) {
- return typeof lastArg === 'function' ?
- lastArg(arguments[0]) :
- undefined;
- }
- return typeof lastArg === 'function' ?
- lastArg.apply(context, arguments) :
- undefined;
- });
- }
- }
- });
- };
- Kareem.prototype.filter = function(fn) {
- const clone = this.clone();
- const pres = Array.from(clone._pres.keys());
- for (const name of pres) {
- const hooks = this._pres.get(name).
- map(h => Object.assign({}, h, { name: name })).
- filter(fn);
- if (hooks.length === 0) {
- clone._pres.delete(name);
- continue;
- }
- hooks.numAsync = hooks.filter(h => h.isAsync).length;
- clone._pres.set(name, hooks);
- }
- const posts = Array.from(clone._posts.keys());
- for (const name of posts) {
- const hooks = this._posts.get(name).
- map(h => Object.assign({}, h, { name: name })).
- filter(fn);
- if (hooks.length === 0) {
- clone._posts.delete(name);
- continue;
- }
- clone._posts.set(name, hooks);
- }
- return clone;
- };
- Kareem.prototype.hasHooks = function(name) {
- return this._pres.has(name) || this._posts.has(name);
- };
- Kareem.prototype.createWrapper = function(name, fn, context, options) {
- var _this = this;
- if (!this.hasHooks(name)) {
- // Fast path: if there's no hooks for this function, just return the
- // function wrapped in a nextTick()
- return function() {
- process.nextTick(() => fn.apply(this, arguments));
- };
- }
- return function() {
- var _context = context || this;
- var args = Array.prototype.slice.call(arguments);
- _this.wrap(name, fn, _context, args, options);
- };
- };
- Kareem.prototype.pre = function(name, isAsync, fn, error, unshift) {
- let options = {};
- if (typeof isAsync === 'object' && isAsync != null) {
- options = isAsync;
- isAsync = options.isAsync;
- } else if (typeof arguments[1] !== 'boolean') {
- error = fn;
- fn = isAsync;
- isAsync = false;
- }
- const pres = get(this._pres, name, []);
- this._pres.set(name, pres);
- if (isAsync) {
- pres.numAsync = pres.numAsync || 0;
- ++pres.numAsync;
- }
- if (typeof fn !== 'function') {
- throw new Error('pre() requires a function, got "' + typeof fn + '"');
- }
- if (unshift) {
- pres.unshift(Object.assign({}, options, { fn: fn, isAsync: isAsync }));
- } else {
- pres.push(Object.assign({}, options, { fn: fn, isAsync: isAsync }));
- }
- return this;
- };
- Kareem.prototype.post = function(name, options, fn, unshift) {
- const hooks = get(this._posts, name, []);
- if (typeof options === 'function') {
- unshift = !!fn;
- fn = options;
- options = {};
- }
- if (typeof fn !== 'function') {
- throw new Error('post() requires a function, got "' + typeof fn + '"');
- }
- if (unshift) {
- hooks.unshift(Object.assign({}, options, { fn: fn }));
- } else {
- hooks.push(Object.assign({}, options, { fn: fn }));
- }
- this._posts.set(name, hooks);
- return this;
- };
- Kareem.prototype.clone = function() {
- const n = new Kareem();
- for (let key of this._pres.keys()) {
- const clone = this._pres.get(key).slice();
- clone.numAsync = this._pres.get(key).numAsync;
- n._pres.set(key, clone);
- }
- for (let key of this._posts.keys()) {
- n._posts.set(key, this._posts.get(key).slice());
- }
- return n;
- };
- Kareem.prototype.merge = function(other, clone) {
- clone = arguments.length === 1 ? true : clone;
- var ret = clone ? this.clone() : this;
- for (let key of other._pres.keys()) {
- const sourcePres = get(ret._pres, key, []);
- const deduplicated = other._pres.get(key).
- // Deduplicate based on `fn`
- filter(p => sourcePres.map(_p => _p.fn).indexOf(p.fn) === -1);
- const combined = sourcePres.concat(deduplicated);
- combined.numAsync = sourcePres.numAsync || 0;
- combined.numAsync += deduplicated.filter(p => p.isAsync).length;
- ret._pres.set(key, combined);
- }
- for (let key of other._posts.keys()) {
- const sourcePosts = get(ret._posts, key, []);
- const deduplicated = other._posts.get(key).
- filter(p => sourcePosts.indexOf(p) === -1);
- ret._posts.set(key, sourcePosts.concat(deduplicated));
- }
- return ret;
- };
- function get(map, key, def) {
- if (map.has(key)) {
- return map.get(key);
- }
- return def;
- }
- function callMiddlewareFunction(fn, context, args, next) {
- let maybePromise;
- try {
- maybePromise = fn.apply(context, args);
- } catch (error) {
- return next(error);
- }
- if (isPromise(maybePromise)) {
- maybePromise.then(() => next(), err => next(err));
- }
- }
- function isPromise(v) {
- return v != null && typeof v.then === 'function';
- }
- function decorateNextFn(fn) {
- var called = false;
- var _this = this;
- return function() {
- // Ensure this function can only be called once
- if (called) {
- return;
- }
- called = true;
- // Make sure to clear the stack so try/catch doesn't catch errors
- // in subsequent middleware
- return process.nextTick(() => fn.apply(_this, arguments));
- };
- }
- module.exports = Kareem;
|