done ui

vite-plugin-ssr-middleware.ts

@@ -0,0 +1,118 @@
+import type { Connect, Plugin } from "vite";
+import { name as packageName } from "./package.json";
+import { createMiddleware } from "@hattip/adapter-node";
+import { pathToFileURL } from "url";
+
+
+export function vitePluginSsrMiddleware({
+  entry,
+  preview,
+  mode = "ssrLoadModule",
+}: {
+  entry: string;
+  preview?: string;
+  mode?: "ssrLoadModule" | "ModuleRunner" | "ModuleRunner-HMR";
+}): Plugin {
+  return {
+    name: packageName,
+
+    apply(config, env) {
+      // skip client build
+      return Boolean(env.command === "serve" || config.build?.ssr);
+    },
+
+    config(config, env) {
+      if (env.command === "serve") {
+        return {
+          // disable builtin HTML middleware, which would rewrite `req.url` to "/index.html"
+          appType: "custom",
+        };
+      }
+      if (env.command === "build" && config.build?.ssr) {
+        return {
+          build: {
+            rollupOptions: {
+              input: {
+                index: entry,
+              },
+            },
+          },
+        };
+      }
+      return;
+    },
+
+    async configureServer(server) {
+      let loadModule = server.ssrLoadModule;
+      if (mode === "ModuleRunner" || mode === "ModuleRunner-HMR") {
+        const { createServerModuleRunner } = await import("vite");
+        const runner = createServerModuleRunner(server.environments.ssr, {
+          hmr: mode === "ModuleRunner-HMR" ? undefined : false,
+        });
+        loadModule = (id: string) => runner.import(id);
+      }
+// const mod = await loadModule(entry);
+      const handler: Connect.NextHandleFunction = async (req, res, next) => {
+        console.log("vite-plugin-ssr-middleware handling request:", req.method, req.url);
+        // expose ViteDevServer via request
+        Object.defineProperty(req, "viteDevServer", { value: server });
+
+        try {
+          const mod = await loadModule(entry);
+        // console.log("preview module loaded:", mod);
+          await createMiddleware((ctx) => mod["default"].fetch(ctx.request))(req, res, next);
+        //   await mod["default"](req, res, next);
+        } catch (e) {
+          next(e);
+        }
+      };
+      return () => server.middlewares.use(handler);
+    },
+
+    async configurePreviewServer(server) {
+      if (preview) {
+        const mod = await import( pathToFileURL(preview).href);
+        return () => server.middlewares.use(createMiddleware((ctx) => mod["default"].fetch(ctx.request)));
+      }
+      return;
+    },
+  };
+}
+
+// minimal logger inspired by
+// https://github.com/koajs/logger
+// https://github.com/honojs/hono/blob/25beca878f2662fedd84ed3fbf80c6a515609cea/src/middleware/logger/index.ts
+
+export function vitePluginLogger(): Plugin {
+  return {
+    name: vitePluginLogger.name,
+    configureServer(server) {
+      return () => server.middlewares.use(loggerMiddleware());
+    },
+    configurePreviewServer(server) {
+      return () => server.middlewares.use(loggerMiddleware());
+    },
+  };
+}
+
+function loggerMiddleware(): Connect.NextHandleFunction {
+  return (req, res, next) => {
+    const url = new URL(req.originalUrl!, "https://test.local");
+    console.log("  -->", req.method, url.pathname);
+    const startTime = Date.now();
+    res.once("close", () => {
+      console.log(
+        "  <--",
+        req.method,
+        url.pathname,
+        res.statusCode,
+        formatDuration(Date.now() - startTime),
+      );
+    });
+    next();
+  };
+}
+
+function formatDuration(ms: number) {
+  return ms < 1000 ? `${Math.floor(ms)}ms` : `${(ms / 1000).toFixed(1)}s`;
+}