From 4dd6c7900ca40db1b2cebed9bd02a9eb00874d8d Mon Sep 17 00:00:00 2001
From: DixCouleur <DixCouleur@qq.com>
Date: Wed, 2 Aug 2023 06:10:29 +0800
Subject: [PATCH] Fix "res.writeHead is not a function" in Express/node
 middleware (#7708)

* fix: res.writeHead is not a function

* fix: handler params type added

* fix: handler function params error

* Update packages/integrations/node/src/nodeMiddleware.ts

---------

Co-authored-by: Nate Moore <natemoo-re@users.noreply.github.com>
---
 .changeset/olive-knives-rush.md               |  5 ++++
 .../integrations/node/src/nodeMiddleware.ts   | 26 +++++++++++++------
 packages/integrations/node/src/types.ts       | 11 ++++++++
 3 files changed, 34 insertions(+), 8 deletions(-)
 create mode 100644 .changeset/olive-knives-rush.md

diff --git a/.changeset/olive-knives-rush.md b/.changeset/olive-knives-rush.md
new file mode 100644
index 000000000000..af8022ba06c7
--- /dev/null
+++ b/.changeset/olive-knives-rush.md
@@ -0,0 +1,5 @@
+---
+'@astrojs/node': patch
+---
+
+fix issuse #7590 "res.writeHead is not a function" in Express/Node middleware
diff --git a/packages/integrations/node/src/nodeMiddleware.ts b/packages/integrations/node/src/nodeMiddleware.ts
index 8d31b68068b4..fa0cdfda8435 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) {
diff --git a/packages/integrations/node/src/types.ts b/packages/integrations/node/src/types.ts
index 2a9ecc2a1ade..b0bd4a6819c3 100644
--- a/packages/integrations/node/src/types.ts
+++ b/packages/integrations/node/src/types.ts
@@ -1,3 +1,5 @@
+import { IncomingMessage, ServerResponse } from 'node:http';
+
 export interface UserOptions {
 	/**
 	 * Specifies the mode that the adapter builds to.
@@ -14,3 +16,12 @@ export interface Options extends UserOptions {
 	server: string;
 	client: string;
 }
+
+export type RequestHandlerParams = [
+	req: IncomingMessage,
+	res: ServerResponse,
+	next?: (err?: unknown) => void,
+	locals?: object
+];
+
+export type ErrorHandlerParams = [unknown, ...RequestHandlerParams];