fix(addVersionHeader): set on error response as well

Author: coderbyheart

src/addVersionHeader.ts

@@ -1,16 +1,13 @@
-import type { MiddlewareObj } from '@middy/core'
+import type { MiddlewareObj, Request } from '@middy/core'
 import type {
 	APIGatewayProxyEventV2,
 	APIGatewayProxyStructuredResultV2,
 } from 'aws-lambda'
 
 export const addVersionHeader = (
 	version: string,
-): MiddlewareObj<
-	APIGatewayProxyEventV2,
-	APIGatewayProxyStructuredResultV2
-> => ({
-	after: async (req) => {
+): MiddlewareObj<APIGatewayProxyEventV2, APIGatewayProxyStructuredResultV2> => {
+	const setVersionHeader = async (req: Request) => {
 		if (req.response === null) return
 		req.response = {
 			...req.response,
@@ -19,5 +16,10 @@ export const addVersionHeader = (
 				'x-backend-version': version,
 			},
 		}
-	},
-})
+	}
+
+	return {
+		after: setVersionHeader,
+		onError: setVersionHeader,
+	}
+}