From e93153b0b3221eee37ee5e891220f1a8763e06df Mon Sep 17 00:00:00 2001
From: Davide <43080019+edivados@users.noreply.github.com>
Date: Thu, 18 Jul 2024 21:36:58 +0200
Subject: [PATCH] fix middleware wrapper

---
 packages/start/src/middleware/index.tsx | 12 ++++--------
 1 file changed, 4 insertions(+), 8 deletions(-)

diff --git a/packages/start/src/middleware/index.tsx b/packages/start/src/middleware/index.tsx
index 6e8f83774..4e524c7ae 100644
--- a/packages/start/src/middleware/index.tsx
+++ b/packages/start/src/middleware/index.tsx
@@ -27,10 +27,8 @@ function wrapRequestMiddleware(onRequest: RequestMiddleware) {
   return async (h3Event: HTTPEvent) => {
     const fetchEvent = getFetchEvent(h3Event);
     const response = await onRequest(fetchEvent);
-    if (!response) {
-      return;
-    } else {
-      sendWebResponse(h3Event, response);
+    if (response) {
+      await sendWebResponse(h3Event, response);
     }
   };
 }
@@ -39,10 +37,8 @@ function wrapResponseMiddleware(onBeforeResponse: ResponseMiddleware) {
   return async (h3Event: HTTPEvent, response: ResponseMiddlewareResponseParam) => {
     const fetchEvent = getFetchEvent(h3Event);
     const mwResponse = await onBeforeResponse(fetchEvent, response);
-    if (!mwResponse) {
-      return;
-    } else {
-      sendWebResponse(h3Event, mwResponse);
+    if (mwResponse) {
+      response.body = mwResponse;
     }
   };
 }