mirror of
https://github.com/freeCodeCamp/freeCodeCamp.git
synced 2026-05-25 10:01:30 -04:00
48 lines
1.1 KiB
JavaScript
48 lines
1.1 KiB
JavaScript
const Layer = require('express/lib/router/layer');
|
|
const Router = require('express/lib/router');
|
|
|
|
const last = (arr = []) => arr[arr.length - 1];
|
|
const noop = Function.prototype;
|
|
|
|
function copyFnProps(oldFn, newFn) {
|
|
Object.keys(oldFn).forEach((key) => {
|
|
newFn[key] = oldFn[key];
|
|
});
|
|
return newFn;
|
|
}
|
|
|
|
function wrap(fn) {
|
|
const newFn = function newFn(...args) {
|
|
const ret = fn.apply(this, args);
|
|
const next = (args.length === 5 ? args[2] : last(args)) || noop;
|
|
if (ret && ret.catch) ret.catch(err => next(err));
|
|
return ret;
|
|
};
|
|
Object.defineProperty(newFn, 'length', {
|
|
value: fn.length,
|
|
writable: false,
|
|
});
|
|
return copyFnProps(fn, newFn);
|
|
}
|
|
|
|
function patchRouterParam() {
|
|
const originalParam = Router.prototype.constructor.param;
|
|
Router.prototype.constructor.param = function param(name, fn) {
|
|
fn = wrap(fn);
|
|
return originalParam.call(this, name, fn);
|
|
};
|
|
}
|
|
|
|
Object.defineProperty(Layer.prototype, 'handle', {
|
|
enumerable: true,
|
|
get() {
|
|
return this.__handle;
|
|
},
|
|
set(fn) {
|
|
fn = wrap(fn);
|
|
this.__handle = fn;
|
|
},
|
|
});
|
|
|
|
patchRouterParam();
|