Update HMR. Make it true HMR. Add URL to page server props

2026-03-20 11:19:22 +01:00
parent 7804a34951
commit 52dde6c0ab
52 changed files with 1548 additions and 708 deletions
@@ -0,0 +1,50 @@
+import type { Server } from "bun";
+import type { BunextServerRouteConfig, BunxRouteParams } from "../../types";
+import grabRouteParams from "../../utils/grab-route-params";
+import grabConstants from "../../utils/grab-constants";
+import grabRouter from "../../utils/grab-router";
+
+type Params = {
+    req: Request;
+    server: Server;
+};
+
+export default async function ({ req, server }: Params): Promise<Response> {
+    const referer_url = new URL(req.headers.get("referer") || "");
+    const match = global.ROUTER.match(referer_url.pathname);
+
+    const target_map = match?.filePath
+        ? global.BUNDLER_CTX_MAP?.find((m) => m.local_path == match.filePath)
+        : undefined;
+
+    let controller: ReadableStreamDefaultController<string>;
+    const stream = new ReadableStream<string>({
+        start(c) {
+            controller = c;
+            global.HMR_CONTROLLERS.push({
+                controller: c,
+                page_url: referer_url.href,
+                target_map,
+            });
+        },
+        cancel() {
+            const targetControllerIndex = global.HMR_CONTROLLERS.findIndex(
+                (c) => c.controller == controller,
+            );
+
+            if (
+                typeof targetControllerIndex == "number" &&
+                targetControllerIndex >= 0
+            ) {
+                global.HMR_CONTROLLERS.splice(targetControllerIndex, 1);
+            }
+        },
+    });
+
+    return new Response(stream, {
+        headers: {
+            "Content-Type": "text/event-stream",
+            Connection: "keep-alive",
+        },
+    });
+}