diff options
Diffstat (limited to 'packages/integrations/node/src/nodeMiddleware.ts')
-rw-r--r-- | packages/integrations/node/src/nodeMiddleware.ts | 26 |
1 files changed, 18 insertions, 8 deletions
diff --git a/packages/integrations/node/src/nodeMiddleware.ts b/packages/integrations/node/src/nodeMiddleware.ts index 8d31b6806..fa0cdfda8 100644 --- a/packages/integrations/node/src/nodeMiddleware.ts +++ b/packages/integrations/node/src/nodeMiddleware.ts @@ -1,19 +1,29 @@ import type { NodeApp } from 'astro/app/node'; -import type { IncomingMessage, ServerResponse } from 'node:http'; +import type { ServerResponse } from 'node:http'; import type { Readable } from 'stream'; import { createOutgoingHttpHeaders } from './createOutgoingHttpHeaders'; import { responseIterator } from './response-iterator'; -import type { Options } from './types'; +import type { ErrorHandlerParams, Options, RequestHandlerParams } from './types'; // Disable no-unused-vars to avoid breaking signature change // eslint-disable-next-line @typescript-eslint/no-unused-vars export default function (app: NodeApp, _mode: Options['mode']) { - return async function ( - req: IncomingMessage, - res: ServerResponse, - next?: (err?: unknown) => void, - locals?: object - ) { + return async function (...args: RequestHandlerParams | ErrorHandlerParams) { + let error = null; + let [req, res, next, locals] = args as RequestHandlerParams; + + if (args[0] instanceof Error) { + [error, req, res, next, locals] = args as ErrorHandlerParams; + + if (error) { + if (next) { + return next(error); + } else { + throw error; + } + } + } + try { const route = app.match(req); if (route) { |