refactor middleware

.changeset/new-shirts-bake.md

@@ -0,0 +1,5 @@
+---
+"@solidjs/start": minor
+---
+
+refactor middleware. mainly fix onResponse

packages/start/src/middleware/index.ts

@@ -1,53 +1,40 @@
 // @refresh skip
 
-import type { H3Event, Middleware } from "h3";
+import { H3Event, type Middleware, onResponse } from "h3";
 import { getFetchEvent } from "../server/fetchEvent.ts";
 import type { FetchEvent } from "../server/types.ts";
 
-/** Function responsible for receiving an observable [operation]{@link Operation} and returning a [result]{@link OperationResult}. */
+export type MiddlewareFn = (event: FetchEvent) => unknown;
 
-export type MiddlewareFn = (event: FetchEvent) => Promise<unknown> | unknown;
-/** This composes an array of Exchanges into a single ExchangeIO function */
-
-export type RequestMiddleware = (
-  event: FetchEvent,
-) => Response | Promise<Response> | void | Promise<void> | Promise<void | Response>;
-
-// copy-pasted from h3/dist/index.d.ts
-type EventHandlerResponse<T = any> = T | Promise<T>;
-type ResponseMiddlewareResponseParam = { body?: Awaited<EventHandlerResponse> };
+// `unknown` because h3 allows any response type like string, json, etc. not just Response
+export type RequestMiddleware = (event: FetchEvent) => unknown;
 
 export type ResponseMiddleware = (
   event: FetchEvent,
-  response: ResponseMiddlewareResponseParam,
-) => Response | Promise<Response> | void | Promise<void>;
+  response: Response,
+) => ReturnType<Parameters<typeof onResponse>[0]>;
 
-function wrapRequestMiddleware(onRequest: RequestMiddleware) {
+function wrapRequestMiddleware(onRequestFn: RequestMiddleware) {
   return async (h3Event: H3Event) => {
+    // h3 onRequest doesn't allow returning a response, but we will for backwards compatibility with start v1
     const fetchEvent = getFetchEvent(h3Event);
-    const response = await onRequest(fetchEvent);
+    const response = await onRequestFn(fetchEvent);
     if (response) return response;
   };
 }
 
 function wrapResponseMiddleware(onBeforeResponse: ResponseMiddleware): Middleware {
-  return async (h3Event, next) => {
-    const resp = await next();
-
+  return onResponse(async (response, h3Event) => {
     const fetchEvent = getFetchEvent(h3Event);
-    const mwResponse = await onBeforeResponse(fetchEvent, {
-      body: (resp as any)?.body,
-    });
+    const mwResponse = await onBeforeResponse(fetchEvent, response.clone());
     if (mwResponse) return mwResponse;
-  };
+  });
 }
 
 export function createMiddleware(
   args:
     | {
-        /** @deprecated Use H3 `Middleware` */
         onRequest?: RequestMiddleware | RequestMiddleware[] | undefined;
-        /** @deprecated Use H3 `Middleware` */
         onBeforeResponse?: ResponseMiddleware | ResponseMiddleware[] | undefined;
       }
     | Middleware[],