52 lines
1.7 KiB
Plaintext
52 lines
1.7 KiB
Plaintext
Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
|
|
//#region src/compose.ts
|
|
/**
|
|
* Compose middleware functions into a single function based on `koa-compose` package.
|
|
*
|
|
* @template E - The environment type.
|
|
*
|
|
* @param {[[Function, unknown], unknown][] | [[Function]][]} middleware - An array of middleware functions and their corresponding parameters.
|
|
* @param {ErrorHandler<E>} [onError] - An optional error handler function.
|
|
* @param {NotFoundHandler<E>} [onNotFound] - An optional not-found handler function.
|
|
*
|
|
* @returns {(context: Context, next?: Next) => Promise<Context>} - A composed middleware function.
|
|
*/
|
|
const compose = (middleware, onError, onNotFound) => {
|
|
return (context, next) => {
|
|
let index = -1;
|
|
return dispatch(0);
|
|
/**
|
|
* Dispatch the middleware functions.
|
|
*
|
|
* @param {number} i - The current index in the middleware array.
|
|
*
|
|
* @returns {Promise<Context>} - A promise that resolves to the context.
|
|
*/
|
|
async function dispatch(i) {
|
|
if (i <= index) throw new Error("next() called multiple times");
|
|
index = i;
|
|
let res;
|
|
let isError = false;
|
|
let handler;
|
|
if (middleware[i]) {
|
|
handler = middleware[i][0][0];
|
|
context.req.routeIndex = i;
|
|
} else handler = i === middleware.length && next || void 0;
|
|
if (handler) try {
|
|
res = await handler(context, () => dispatch(i + 1));
|
|
} catch (err) {
|
|
if (err instanceof Error && onError) {
|
|
context.error = err;
|
|
res = await onError(err, context);
|
|
isError = true;
|
|
} else throw err;
|
|
}
|
|
else if (context.finalized === false && onNotFound) res = await onNotFound(context);
|
|
if (res && (context.finalized === false || isError)) context.res = res;
|
|
return context;
|
|
}
|
|
};
|
|
};
|
|
//#endregion
|
|
exports.compose = compose;
|