fix middleware wrapper

solidjs · Jul 22, 2024 · e93153b · e93153b
1 parent 6aa7d1f
commit e93153b
Showing 1 changed file with 4 additions and 8 deletions.
diff --git 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;
     }
   };
 }