vite-plugin-kiru 0.31.0 → 0.32.0-preview.1

package/build.dev.ts

@@ -3,13 +3,13 @@ import fs from "node:fs"
 
 await esbuild
   .context({
-    entryPoints: ["src/index.ts"],
+    entryPoints: ["src/index.ts", "src/server.ts"],
     bundle: true,
     platform: "node",
     target: "esnext",
     format: "esm",
-    outfile: "./dist/index.js",
-    external: ["kiru", "vite"],
+    outdir: "./dist",
+    external: ["kiru", "vite", "virtual:kiru:entry-server"],
     write: true,
     plugins: [
       {
@@ -18,6 +18,7 @@ await esbuild
           onEnd(() => {
             console.log("[vite-plugin-kiru]: Build complete!")
             fs.copyFileSync("./src/types.d.ts", "dist/index.d.ts")
+            fs.copyFileSync("./src/types.server.d.ts", "dist/server.d.ts")
           })
         },
       },

package/dist/index.d.ts

@@ -116,6 +116,11 @@ export interface SSGOptions {
    * @default "layout.{tsx,jsx}"
    */
   layout?: string
+  /**
+   * The filename of guard files to search for
+   * @default "guard.{ts,js}"
+   */
+  guard?: string
   /**
    * Enable transitions for all routes and loading states
    * @default false
@@ -133,6 +138,49 @@ export interface SSGOptions {
   build?: SSGBuildOptions
 }
 
+export interface SSROptions {
+  /**
+   * The path to the server entry file
+   * @example "src/server.ts"
+   */
+  runtimeEntry: string
+  /**
+   * The base URL of the app
+   * @default "/"
+   */
+  baseUrl?: string
+  /**
+   * The directory of the app
+   * @default "src/pages"
+   */
+  dir?: string
+  /**
+   * The name of the document component
+   * @default "document.{tsx,jsx}"
+   */
+  document?: string
+  /**
+   * The filename of page components to search for
+   * @default "index.{tsx,jsx}"
+   */
+  page?: string
+  /**
+   * The filename of layout components to search for
+   * @default "layout.{tsx,jsx}"
+   */
+  layout?: string
+  /**
+   * The filename of guard files to search for
+   * @default "guard.{ts,js}"
+   */
+  guard?: string
+  /**
+   * Enable transitions for all routes and loading states
+   * @default false
+   */
+  transition?: boolean
+}
+
 export interface DevtoolsOptions {
   /**
    * Specifies the path to the devtools app displayed via popup
@@ -193,6 +241,21 @@ export interface KiruPluginOptions {
    * ```
    */
   ssg?: SSGOptions | true
+
+  /**
+   * Options for SSR
+   * @example
+   * ```ts
+   * ssr: {
+   *   dir: "./src/app",
+   *   document: "document.{tsx,jsx}",
+   *   page: "index.{tsx,jsx}",
+   *   layout: "layout.{tsx,jsx}",
+   *   transition: true
+   * }
+   * ```
+   */
+  ssr?: SSROptions
 }
 
 export const defaultEsBuildOptions: ESBuildOptions

package/dist/index.js

@@ -27,7 +27,7 @@ var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__ge
 // ../../node_modules/.pnpm/@jridgewell+sourcemap-codec@1.5.5/node_modules/@jridgewell/sourcemap-codec/dist/sourcemap-codec.umd.js
 var require_sourcemap_codec_umd = __commonJS({
   "../../node_modules/.pnpm/@jridgewell+sourcemap-codec@1.5.5/node_modules/@jridgewell/sourcemap-codec/dist/sourcemap-codec.umd.js"(exports, module) {
-    (function(global, factory) {
+    (function(global2, factory) {
       if (typeof exports === "object" && typeof module !== "undefined") {
         factory(module);
         module.exports = def(module);
@@ -39,8 +39,8 @@ var require_sourcemap_codec_umd = __commonJS({
       } else {
         const mod = { exports: {} };
         factory(mod);
-        global = typeof globalThis !== "undefined" ? globalThis : global || self;
-        global.sourcemapCodec = def(mod);
+        global2 = typeof globalThis !== "undefined" ? globalThis : global2 || self;
+        global2.sourcemapCodec = def(mod);
       }
       function def(m) {
         return "default" in m.exports ? m.exports.default : m.exports;
@@ -8808,8 +8808,8 @@ function createLogger(state) {
     console.log(ANSI.cyan("[vite-plugin-kiru]"), ...data);
   };
 }
-function resolveUserDocument(projectRoot, ssgOptions) {
-  const { dir, document } = ssgOptions;
+function resolveUserDocument(projectRoot, options) {
+  const { dir, document } = options;
   const fp = path2.resolve(projectRoot, dir, document).replace(/\\/g, "/");
   const matches = globSync(fp);
   if (!matches.length) {
@@ -8840,57 +8840,80 @@ function shouldTransformFile(id, state) {
 }
 
 // src/virtual-modules.ts
-var VIRTUAL_ROUTES_ID = "virtual:kiru:routes";
+var VIRTUAL_CONFIG_ID = "virtual:kiru:config";
 var VIRTUAL_ENTRY_SERVER_ID = "virtual:kiru:entry-server";
 var VIRTUAL_ENTRY_CLIENT_ID = "virtual:kiru:entry-client";
-async function createVirtualModules(projectRoot, ssgOptions) {
-  const userDoc = resolveUserDocument(projectRoot, ssgOptions);
-  function createRoutesModule() {
-    const { dir, baseUrl, page, layout, transition } = ssgOptions;
+async function createVirtualModules(projectRoot, options, mode) {
+  const userDoc = resolveUserDocument(projectRoot, options);
+  function createConfigModule() {
+    const { dir, baseUrl, page, layout, guard, transition } = options;
     return `
 import { formatViteImportMap, normalizePrefixPath } from "kiru/router/utils"
 
 const dir = normalizePrefixPath("${dir}")
 const baseUrl = normalizePrefixPath("${baseUrl}")
-const pagesMap = import.meta.glob(["/**/${page}"])
-const layoutsMap = import.meta.glob(["/**/${layout}"])
-const pages = formatViteImportMap(pagesMap, dir, baseUrl)
-const layouts = formatViteImportMap(layoutsMap, dir, baseUrl)
+const pages = formatViteImportMap(import.meta.glob(["/**/${page}"]), dir, baseUrl)
+const layouts = formatViteImportMap(import.meta.glob(["/**/${layout}"]), dir, baseUrl)
+const guards = formatViteImportMap(import.meta.glob(["/**/${guard}"]), dir, baseUrl)
 const transition = ${transition}
 
-export { dir, baseUrl, pages, layouts, transition }
+export { dir, baseUrl, pages, layouts, guards, transition }
 `;
   }
   function createEntryServerModule() {
+    const documentModuleId = userDoc.substring(projectRoot.length);
+    if (mode === "ssr") {
+      return `
+import { render as kiruServerRender } from "kiru/router/ssr"
+import Document from "${userDoc}"
+import * as config from "${VIRTUAL_CONFIG_ID}"
+
+export const documentModuleId = "${documentModuleId}"
+
+export async function render(url, ctx) {
+  return kiruServerRender(url, { ...ctx, ...config, Document })
+}
+`;
+    }
     return `
 import {
-  render as kiruServerRender,
-  generateStaticPaths as kiruServerGenerateStaticPaths
-} from "kiru/router/server"
+  render as kiruStaticRender,
+  generateStaticPaths as kiruGenerateStaticPaths
+} from "kiru/router/ssg"
 import Document from "${userDoc}"
-import { pages, layouts } from "${VIRTUAL_ROUTES_ID}"
+import * as config from "${VIRTUAL_CONFIG_ID}"
+
+export const documentModuleId = "${documentModuleId}"
 
 export async function render(url, ctx) {
-  const { registerModule, registerPreloadedPageProps } = ctx
-  return kiruServerRender(url, { registerModule, registerPreloadedPageProps, Document, pages, layouts })
+  return kiruStaticRender(url, { ...ctx, ...config, Document })
 }
 
 export async function generateStaticPaths() {
-  return kiruServerGenerateStaticPaths(pages)
+  return kiruGenerateStaticPaths(config.pages)
 }
 `;
   }
   function createEntryClientModule() {
+    if (mode === "ssr") {
+      return `
+import { initClient } from "kiru/router/client"
+import * as config from "${VIRTUAL_CONFIG_ID}"
+import "${userDoc}"
+
+initClient({ ...config, hydrationMode: "dynamic" })
+`;
+    }
     return `
 import { initClient } from "kiru/router/client"
-import { dir, baseUrl, pages, layouts, transition } from "${VIRTUAL_ROUTES_ID}"
+import * as config from "${VIRTUAL_CONFIG_ID}"
 import "${userDoc}"
 
-initClient({ dir, baseUrl, pages, layouts, transition })
+initClient({ ...config })
 `;
   }
   return {
-    [VIRTUAL_ROUTES_ID]: createRoutesModule,
+    [VIRTUAL_CONFIG_ID]: createConfigModule,
     [VIRTUAL_ENTRY_SERVER_ID]: createEntryServerModule,
     [VIRTUAL_ENTRY_CLIENT_ID]: createEntryClientModule
   };
@@ -8911,11 +8934,21 @@ var defaultSSGOptions = {
   document: "document.{tsx,jsx}",
   page: "index.{tsx,jsx}",
   layout: "layout.{tsx,jsx}",
+  guard: "guard.{ts,js}",
   transition: false,
   build: {
     maxConcurrentRenders: 100
   }
 };
+var defaultSSROptions = {
+  baseUrl: "/",
+  dir: "src/pages",
+  document: "document.{tsx,jsx}",
+  page: "index.{tsx,jsx}",
+  layout: "layout.{tsx,jsx}",
+  guard: "guard.{ts,js}",
+  transition: false
+};
 function createPluginState(opts = {}) {
   let fileLinkFormatter = (path8, line) => `vscode://file/${path8}:${line}`;
   let dtClientPathname = "/__devtools__";
@@ -8936,47 +8969,78 @@ function createPluginState(opts = {}) {
     dtHostScriptPath: "/__devtools_host__.js",
     manifestPath: "vite-manifest.json",
     loggingEnabled: opts.loggingEnabled === true,
-    ssgOptions: null
+    ssgOptions: null,
+    ssrOptions: null
   };
-  const { ssg } = opts;
-  if (!ssg) return state;
-  if (ssg === true) {
+  if (opts.ssg && opts.ssr) {
+    throw new Error(
+      "[vite-plugin-kiru]: Cannot enable both SSG and SSR modes simultaneously"
+    );
+  }
+  const { ssg, ssr } = opts;
+  if (ssg) {
+    if (ssg === true) {
+      return {
+        ...state,
+        ssgOptions: defaultSSGOptions
+      };
+    }
+    if (ssg.baseUrl && !ssg.baseUrl.startsWith("/")) {
+      throw new Error("[vite-plugin-kiru]: ssg.baseUrl must start with '/'");
+    }
+    const {
+      baseUrl,
+      dir,
+      document,
+      page,
+      layout,
+      guard,
+      transition,
+      build: { maxConcurrentRenders }
+    } = defaultSSGOptions;
     return {
       ...state,
-      ssgOptions: defaultSSGOptions
+      ssgOptions: {
+        baseUrl,
+        dir,
+        document,
+        page,
+        layout,
+        guard,
+        transition,
+        sitemap: ssg.sitemap,
+        ...ssg,
+        build: {
+          maxConcurrentRenders: ssg.build?.maxConcurrentRenders ?? maxConcurrentRenders
+        }
+      }
     };
   }
-  if (ssg.baseUrl && !ssg.baseUrl.startsWith("/")) {
-    throw new Error("[vite-plugin-kiru]: ssg.baseUrl must start with '/'");
-  }
-  const {
-    baseUrl,
-    dir,
-    document,
-    page,
-    layout,
-    transition,
-    build: { maxConcurrentRenders }
-  } = defaultSSGOptions;
-  return {
-    ...state,
-    ssgOptions: {
-      ...ssg,
-      baseUrl: ssg.baseUrl ?? baseUrl,
-      dir: ssg.dir ?? dir,
-      document: ssg.document ?? document,
-      page: ssg.page ?? page,
-      layout: ssg.layout ?? layout,
-      transition: ssg.transition ?? transition,
-      sitemap: ssg.sitemap,
-      build: {
-        maxConcurrentRenders: ssg.build?.maxConcurrentRenders ?? maxConcurrentRenders
-      }
+  if (ssr) {
+    if (ssr.baseUrl && !ssr.baseUrl.startsWith("/")) {
+      throw new Error("[vite-plugin-kiru]: ssr.baseUrl must start with '/'");
     }
-  };
+    const { baseUrl, dir, document, page, layout, guard, transition } = defaultSSROptions;
+    return {
+      ...state,
+      ssrOptions: {
+        baseUrl,
+        dir,
+        document,
+        page,
+        layout,
+        guard,
+        transition,
+        ...ssr
+      }
+    };
+  }
+  return state;
 }
 function createViteConfig(config, opts) {
-  if (!opts.ssg) {
+  const hasSSG = !!opts.ssg;
+  const hasSSR = !!opts.ssr;
+  if (!hasSSG && !hasSSR) {
     return {
       ...config,
       esbuild: {
@@ -9046,6 +9110,7 @@ function updatePluginState(state, config, opts) {
     dtHostScriptPath: state.dtHostScriptPath,
     manifestPath: state.manifestPath,
     ssgOptions: state.ssgOptions,
+    ssrOptions: state.ssrOptions,
     staticProps: {}
   };
 }
@@ -11274,7 +11339,7 @@ async function getClientAssets(clientOutDirAbs, manifestPath) {
     const clientManifestPath = path6.resolve(clientOutDirAbs, manifestPath);
     if (fs3.existsSync(clientManifestPath)) {
       const manifest = JSON.parse(fs3.readFileSync(clientManifestPath, "utf-8"));
-      const clientEntryKey = "virtual:kiru:entry-client";
+      const clientEntryKey = VIRTUAL_ENTRY_CLIENT_ID;
       if (manifest[clientEntryKey]?.file) {
         clientEntry = manifest[clientEntryKey].file;
       }
@@ -11300,7 +11365,7 @@ function collectCssForModules(manifest, moduleIds, projectRoot) {
     (it.css || []).forEach((c) => cssFiles.add(c));
     (it.imports || []).forEach((imp) => collectCss(imp));
   };
-  const entryClientKey = "virtual:kiru:entry-client";
+  const entryClientKey = VIRTUAL_ENTRY_CLIENT_ID;
   if (manifest[entryClientKey] && !seen.has(entryClientKey)) {
     collectCss(entryClientKey);
   }
@@ -11363,10 +11428,10 @@ async function renderRoute(state, mod, route, srcFilePath, clientEntry, manifest
   const documentModuleId = documentPath.replace(/\\/g, "/");
   moduleIds.push(documentModuleId);
   const ctx = {
-    registerModule: (moduleId) => {
+    registerModule(moduleId) {
       moduleIds.push(moduleId);
     },
-    registerPreloadedPageProps: (props) => {
+    registerStaticProps(props) {
       ;
       (state.staticProps[srcFilePath] ??= {})[route] = props;
     }
@@ -11519,7 +11584,7 @@ async function appendStaticPropsToClientModules(state, clientOutDirAbs, log) {
       Object.keys(manifest).length,
       "entries"
     );
-    const clientEntryChunk = manifest["virtual:kiru:entry-client"];
+    const clientEntryChunk = manifest[VIRTUAL_ENTRY_CLIENT_ID];
     if (!clientEntryChunk) {
       throw new Error("Client entry chunk not found in manifest");
     }
@@ -11548,6 +11613,36 @@ ${code}`, "utf-8");
 
 // src/index.ts
 import { build } from "vite";
+
+// src/globals.ts
+var $KIRU_HEADLESS_GLOBAL = Symbol.for("kiru.headlessGlobal");
+var global = globalThis[$KIRU_HEADLESS_GLOBAL] ??= {
+  viteDevServer: null,
+  server: null
+};
+var VITE_DEV_SERVER_INSTANCE = {
+  get current() {
+    return global.viteDevServer;
+  },
+  set current(server) {
+    global.viteDevServer = server;
+  }
+};
+var entryServerResolvers = [];
+var KIRU_SERVER_ENTRY = {
+  get current() {
+    return global.serverEntry;
+  },
+  set current(server) {
+    global.serverEntry = server;
+    if (server) {
+      entryServerResolvers.forEach((fn) => fn());
+      entryServerResolvers = [];
+    }
+  }
+};
+
+// src/index.ts
 function kiru(opts = {}) {
   let state;
   let log;
@@ -11563,10 +11658,17 @@ function kiru(opts = {}) {
       const initialState = createPluginState(opts);
       state = updatePluginState(initialState, config, opts);
       log = createLogger(state);
-      if (state.ssgOptions) {
+      if (state.ssrOptions) {
+        virtualModules = await createVirtualModules(
+          state.projectRoot,
+          state.ssrOptions,
+          "ssr"
+        );
+      } else if (state.ssgOptions) {
         virtualModules = await createVirtualModules(
           state.projectRoot,
-          state.ssgOptions
+          state.ssgOptions,
+          "ssg"
         );
       }
     },
@@ -11583,10 +11685,12 @@ function kiru(opts = {}) {
         createPreviewMiddleware(state.projectRoot, state.baseOutDir)
       );
     },
-    configureServer(server) {
+    async configureServer(server) {
+      VITE_DEV_SERVER_INSTANCE.current = server;
       if (state.isProduction || state.isBuild) return;
       const {
         ssgOptions,
+        ssrOptions,
         devtoolsEnabled,
         dtClientPathname,
         dtHostScriptPath,
@@ -11607,26 +11711,40 @@ function kiru(opts = {}) {
             const url = req.originalUrl || req.url || "/";
             const filePath = path7.join(state.baseOutDir, "client", url);
             const extName = path7.extname(filePath);
-            if (extName && extName !== ".html") {
-              return next();
-            }
             const accept = req.headers["accept"] || "";
-            if (typeof accept === "string" && accept.includes("text/html") && !url.startsWith("/node_modules/") && !url.startsWith("/@") && !url.startsWith(dtHostScriptPath) && !url.startsWith(dtClientPathname)) {
-              const { status, html } = await handleSSR(
-                server,
-                url,
-                state.projectRoot,
-                () => resolveUserDocument(projectRoot, ssgOptions)
-              );
-              res.statusCode = status;
-              res.setHeader("Content-Type", "text/html");
-              res.end(html);
-              return;
+            const shouldHandle = typeof accept === "string" && accept.includes("text/html") && !url.startsWith("/node_modules/") && !url.startsWith("/@") && !url.startsWith(dtHostScriptPath) && !url.startsWith(dtClientPathname) && (extName === ".html" || !extName);
+            if (!shouldHandle) {
+              return next();
             }
+            const { status, html } = await handleSSR(
+              server,
+              url,
+              state.projectRoot,
+              () => resolveUserDocument(projectRoot, ssgOptions)
+            );
+            res.statusCode = status;
+            res.setHeader("Content-Type", "text/html");
+            res.end(html);
           } catch (e) {
-            console.error(e);
+            const error = e;
+            console.error(
+              ANSI.red("[SSG] Middleware Error"),
+              `
+${ANSI.yellow("URL:")} ${req.url}`,
+              `
+${ANSI.yellow("Error:")} ${error.message}`
+            );
+            if (error.stack) {
+              console.error(ANSI.black_bright(error.stack));
+            }
+            next(e);
           }
-          next();
+        });
+      } else if (ssrOptions) {
+        queueMicrotask(() => {
+          server.ssrLoadModule(VIRTUAL_ENTRY_SERVER_ID).then((mod) => {
+            KIRU_SERVER_ENTRY.current = mod;
+          });
         });
       }
     },
@@ -11696,15 +11814,14 @@ function kiru(opts = {}) {
       };
     }
   };
-  return [
-    mainPlugin,
-    {
+  const plugins = [mainPlugin];
+  if (opts.ssg) {
+    plugins.push({
       name: "vite-plugin-kiru:ssg",
       apply: "build",
       enforce: "post",
       async closeBundle(error) {
-        if (error || this.environment.config.build.ssr || !state.ssgOptions)
-          return;
+        if (error || this.environment.config.build.ssr) return;
         log(ANSI.cyan("[SSG]"), "Starting SSG build...");
         await build({
           ...inlineConfig,
@@ -11716,8 +11833,35 @@ function kiru(opts = {}) {
         });
         log(ANSI.cyan("[SSG]"), "SSG build complete!");
       }
-    }
-  ];
+    });
+  } else if (opts.ssr) {
+    plugins.push({
+      name: "vite-plugin-kiru:ssr",
+      apply: "build",
+      enforce: "post",
+      async closeBundle(error) {
+        if (error || this.environment.config.build.ssr || !state.ssrOptions)
+          return;
+        log(ANSI.cyan("[SSR]"), "Starting SSR build...");
+        await build({
+          ...inlineConfig,
+          configFile: false,
+          build: {
+            ...inlineConfig?.build,
+            ssr: true,
+            rollupOptions: {
+              input: [
+                path7.resolve(state.projectRoot, state.ssrOptions.runtimeEntry),
+                VIRTUAL_ENTRY_SERVER_ID
+              ]
+            }
+          }
+        });
+        log(ANSI.cyan("[SSR]"), "SSR build complete!");
+      }
+    });
+  }
+  return plugins;
 }
 function onHMR(callback) {
 }

package/dist/server.d.ts

@@ -1,10 +1,8 @@
-import type { Readable } from "node:stream"
-
 export interface SSRHttpResponse {
   html: string
   statusCode: number
   headers: Array<[string, string]>
-  stream: Readable
+  stream: ReadableStream | null
 }
 
 export interface SSRRenderResult {
@@ -13,6 +11,7 @@ export interface SSRRenderResult {
 
 export interface ServerRenderOptions {
   url: string
+  context: Kiru.RequestContext
 }
 
 export declare function renderPage(

package/dist/server.js

@@ -1,9 +1,6 @@
 // src/server.ts
-import path3 from "node:path";
-import fs from "node:fs";
-
-// src/utils.ts
 import path2 from "node:path";
+import fs from "node:fs";
 
 // ../../node_modules/.pnpm/@isaacs+balanced-match@4.0.1/node_modules/@isaacs/balanced-match/dist/esm/index.js
 var balanced = (a, b, str) => {
@@ -1557,7 +1554,7 @@ minimatch.Minimatch = Minimatch;
 minimatch.escape = escape;
 minimatch.unescape = unescape;
 
-// ../../node_modules/.pnpm/glob@11.0.3/node_modules/glob/dist/esm/glob.js
+// ../../node_modules/.pnpm/glob@12.0.0/node_modules/glob/dist/esm/glob.js
 import { fileURLToPath as fileURLToPath2 } from "node:url";
 
 // ../../node_modules/.pnpm/lru-cache@11.2.2/node_modules/lru-cache/dist/esm/index.js
@@ -4111,12 +4108,12 @@ var PathBase = class {
   /**
    * Get the Path object referenced by the string path, resolved from this Path
    */
-  resolve(path4) {
-    if (!path4) {
+  resolve(path3) {
+    if (!path3) {
       return this;
     }
-    const rootPath = this.getRootString(path4);
-    const dir = path4.substring(rootPath.length);
+    const rootPath = this.getRootString(path3);
+    const dir = path3.substring(rootPath.length);
     const dirParts = dir.split(this.splitSep);
     const result = rootPath ? this.getRoot(rootPath).#resolveParts(dirParts) : this.#resolveParts(dirParts);
     return result;
@@ -4868,8 +4865,8 @@ var PathWin32 = class _PathWin32 extends PathBase {
   /**
    * @internal
    */
-  getRootString(path4) {
-    return win32.parse(path4).root;
+  getRootString(path3) {
+    return win32.parse(path3).root;
   }
   /**
    * @internal
@@ -4915,8 +4912,8 @@ var PathPosix = class _PathPosix extends PathBase {
   /**
    * @internal
    */
-  getRootString(path4) {
-    return path4.startsWith("/") ? "/" : "";
+  getRootString(path3) {
+    return path3.startsWith("/") ? "/" : "";
   }
   /**
    * @internal
@@ -5005,11 +5002,11 @@ var PathScurryBase = class {
   /**
    * Get the depth of a provided path, string, or the cwd
    */
-  depth(path4 = this.cwd) {
-    if (typeof path4 === "string") {
-      path4 = this.cwd.resolve(path4);
+  depth(path3 = this.cwd) {
+    if (typeof path3 === "string") {
+      path3 = this.cwd.resolve(path3);
     }
-    return path4.depth();
+    return path3.depth();
   }
   /**
    * Return the cache of child entries.  Exposed so subclasses can create
@@ -5496,9 +5493,9 @@ var PathScurryBase = class {
     process2();
     return results;
   }
-  chdir(path4 = this.cwd) {
+  chdir(path3 = this.cwd) {
     const oldCwd = this.cwd;
-    this.cwd = typeof path4 === "string" ? this.cwd.resolve(path4) : path4;
+    this.cwd = typeof path3 === "string" ? this.cwd.resolve(path3) : path3;
     this.cwd[setAsCwd](oldCwd);
   }
 };
@@ -5572,7 +5569,7 @@ var PathScurryDarwin = class extends PathScurryPosix {
 var Path = process.platform === "win32" ? PathWin32 : PathPosix;
 var PathScurry = process.platform === "win32" ? PathScurryWin32 : process.platform === "darwin" ? PathScurryDarwin : PathScurryPosix;
 
-// ../../node_modules/.pnpm/glob@11.0.3/node_modules/glob/dist/esm/pattern.js
+// ../../node_modules/.pnpm/glob@12.0.0/node_modules/glob/dist/esm/pattern.js
 var isPatternList = (pl) => pl.length >= 1;
 var isGlobList = (gl) => gl.length >= 1;
 var Pattern = class _Pattern {
@@ -5737,7 +5734,7 @@ var Pattern = class _Pattern {
   }
 };
 
-// ../../node_modules/.pnpm/glob@11.0.3/node_modules/glob/dist/esm/ignore.js
+// ../../node_modules/.pnpm/glob@12.0.0/node_modules/glob/dist/esm/ignore.js
 var defaultPlatform2 = typeof process === "object" && process && typeof process.platform === "string" ? process.platform : "linux";
 var Ignore = class {
   relative;
@@ -5824,7 +5821,7 @@ var Ignore = class {
   }
 };
 
-// ../../node_modules/.pnpm/glob@11.0.3/node_modules/glob/dist/esm/processor.js
+// ../../node_modules/.pnpm/glob@12.0.0/node_modules/glob/dist/esm/processor.js
 var HasWalkedCache = class _HasWalkedCache {
   store;
   constructor(store = /* @__PURE__ */ new Map()) {
@@ -5854,8 +5851,8 @@ var MatchRecord = class {
   }
   // match, absolute, ifdir
   entries() {
-    return [...this.store.entries()].map(([path4, n]) => [
-      path4,
+    return [...this.store.entries()].map(([path3, n]) => [
+      path3,
       !!(n & 2),
       !!(n & 1)
     ]);
@@ -6045,7 +6042,7 @@ var Processor = class _Processor {
   }
 };
 
-// ../../node_modules/.pnpm/glob@11.0.3/node_modules/glob/dist/esm/walker.js
+// ../../node_modules/.pnpm/glob@12.0.0/node_modules/glob/dist/esm/walker.js
 var makeIgnore = (ignore, opts) => typeof ignore === "string" ? new Ignore([ignore], opts) : Array.isArray(ignore) ? new Ignore(ignore, opts) : ignore;
 var GlobUtil = class {
   path;
@@ -6060,9 +6057,9 @@ var GlobUtil = class {
   signal;
   maxDepth;
   includeChildMatches;
-  constructor(patterns, path4, opts) {
+  constructor(patterns, path3, opts) {
     this.patterns = patterns;
-    this.path = path4;
+    this.path = path3;
     this.opts = opts;
     this.#sep = !opts.posix && opts.platform === "win32" ? "\\" : "/";
     this.includeChildMatches = opts.includeChildMatches !== false;
@@ -6081,11 +6078,11 @@ var GlobUtil = class {
       });
     }
   }
-  #ignored(path4) {
-    return this.seen.has(path4) || !!this.#ignore?.ignored?.(path4);
+  #ignored(path3) {
+    return this.seen.has(path3) || !!this.#ignore?.ignored?.(path3);
   }
-  #childrenIgnored(path4) {
-    return !!this.#ignore?.childrenIgnored?.(path4);
+  #childrenIgnored(path3) {
+    return !!this.#ignore?.childrenIgnored?.(path3);
   }
   // backpressure mechanism
   pause() {
@@ -6300,8 +6297,8 @@ var GlobUtil = class {
 };
 var GlobWalker = class extends GlobUtil {
   matches = /* @__PURE__ */ new Set();
-  constructor(patterns, path4, opts) {
-    super(patterns, path4, opts);
+  constructor(patterns, path3, opts) {
+    super(patterns, path3, opts);
   }
   matchEmit(e) {
     this.matches.add(e);
@@ -6338,8 +6335,8 @@ var GlobWalker = class extends GlobUtil {
 };
 var GlobStream = class extends GlobUtil {
   results;
-  constructor(patterns, path4, opts) {
-    super(patterns, path4, opts);
+  constructor(patterns, path3, opts) {
+    super(patterns, path3, opts);
     this.results = new Minipass({
       signal: this.signal,
       objectMode: true
@@ -6372,7 +6369,7 @@ var GlobStream = class extends GlobUtil {
   }
 };
 
-// ../../node_modules/.pnpm/glob@11.0.3/node_modules/glob/dist/esm/glob.js
+// ../../node_modules/.pnpm/glob@12.0.0/node_modules/glob/dist/esm/glob.js
 var defaultPlatform3 = typeof process === "object" && process && typeof process.platform === "string" ? process.platform : "linux";
 var Glob = class {
   absolute;
@@ -6572,7 +6569,7 @@ var Glob = class {
   }
 };
 
-// ../../node_modules/.pnpm/glob@11.0.3/node_modules/glob/dist/esm/has-magic.js
+// ../../node_modules/.pnpm/glob@12.0.0/node_modules/glob/dist/esm/has-magic.js
 var hasMagic = (pattern, options = {}) => {
   if (!Array.isArray(pattern)) {
     pattern = [pattern];
@@ -6584,7 +6581,7 @@ var hasMagic = (pattern, options = {}) => {
   return false;
 };
 
-// ../../node_modules/.pnpm/glob@11.0.3/node_modules/glob/dist/esm/index.js
+// ../../node_modules/.pnpm/glob@12.0.0/node_modules/glob/dist/esm/index.js
 function globStreamSync(pattern, options = {}) {
   return new Glob(pattern, options).streamSync();
 }
@@ -6660,47 +6657,66 @@ function tEntries(obj) {
   return Object.entries(obj);
 }
 
-// src/utils.ts
-function resolveUserDocument(projectRoot, options) {
-  const { dir, document } = options;
-  const fp = path2.resolve(projectRoot, dir, document).replace(/\\/g, "/");
-  const matches = globSync(fp);
-  if (!matches.length) {
-    throw new Error(`Document not found at ${fp}`);
-  }
-  return path2.resolve(projectRoot, matches[0]).replace(/\\/g, "/");
-}
-
 // src/virtual-modules.ts
+var VIRTUAL_ENTRY_SERVER_ID = "virtual:kiru:entry-server";
 var VIRTUAL_ENTRY_CLIENT_ID = "virtual:kiru:entry-client";
 
 // src/globals.ts
+var $KIRU_HEADLESS_GLOBAL = Symbol.for("kiru.headlessGlobal");
+var global = globalThis[$KIRU_HEADLESS_GLOBAL] ??= {
+  viteDevServer: null,
+  server: null
+};
 var VITE_DEV_SERVER_INSTANCE = {
   get current() {
-    return (
-      // @ts-ignore
-      globalThis.KIRU_VITE_DEV_SERVER_INSTANCE ?? null
-    );
+    return global.viteDevServer;
   },
   set current(server) {
-    globalThis.KIRU_VITE_DEV_SERVER_INSTANCE = server;
+    global.viteDevServer = server;
   }
 };
+var entryServerResolvers = [];
+var KIRU_SERVER_ENTRY = {
+  get current() {
+    return global.serverEntry;
+  },
+  set current(server) {
+    global.serverEntry = server;
+    if (server) {
+      entryServerResolvers.forEach((fn) => fn());
+      entryServerResolvers = [];
+    }
+  }
+};
+async function awaitServerRendererInitialized_Dev() {
+  if (KIRU_SERVER_ENTRY.current) {
+    return Promise.resolve(KIRU_SERVER_ENTRY.current);
+  }
+  return new Promise((resolve, reject) => {
+    const timeout = setTimeout(() => {
+      entryServerResolvers = entryServerResolvers.filter((r) => r !== resolve);
+      reject(new Error("Failed to acquire server renderer. Seek help!"));
+    }, 1e4);
+    entryServerResolvers.push(() => {
+      clearTimeout(timeout);
+      resolve(KIRU_SERVER_ENTRY.current);
+    });
+  });
+}
 
 // src/server.ts
 async function getClientAssets(clientOutDirAbs, manifestPath) {
   let clientEntry = null;
   let manifest = null;
   try {
-    const clientManifestPath = path3.resolve(clientOutDirAbs, manifestPath);
+    const clientManifestPath = path2.resolve(clientOutDirAbs, manifestPath);
     if (fs.existsSync(clientManifestPath)) {
       const parsedManifest = JSON.parse(
         fs.readFileSync(clientManifestPath, "utf-8")
       );
       manifest = parsedManifest;
-      const clientEntryKey = "virtual:kiru:entry-client";
-      if (parsedManifest[clientEntryKey]?.file) {
-        clientEntry = parsedManifest[clientEntryKey].file;
+      if (parsedManifest[VIRTUAL_ENTRY_CLIENT_ID]?.file) {
+        clientEntry = parsedManifest[VIRTUAL_ENTRY_CLIENT_ID].file;
       }
     }
   } catch {
@@ -6723,7 +6739,7 @@ function collectCssForModules(manifest, moduleIds, projectRoot) {
     (it.css || []).forEach((c) => cssFiles.add(c));
     (it.imports || []).forEach((imp) => collectCss(imp));
   };
-  const entryClientKey = "virtual:kiru:entry-client";
+  const entryClientKey = VIRTUAL_ENTRY_CLIENT_ID;
   if (manifestRef[entryClientKey] && !seen.has(entryClientKey)) {
     collectCss(entryClientKey);
   }
@@ -6749,8 +6765,8 @@ function collectCssForModules(manifest, moduleIds, projectRoot) {
     }
     for (const key in manifestRef) {
       const keyNormalized = key.replace(/\\/g, "/");
-      const moduleBaseName = path3.basename(normalizedId);
-      const keyBaseName = path3.basename(keyNormalized);
+      const moduleBaseName = path2.basename(normalizedId);
+      const keyBaseName = path2.basename(keyNormalized);
       if ((keyNormalized.endsWith(normalizedId) || normalizedId.endsWith(keyNormalized)) && moduleBaseName === keyBaseName) {
         collectCss(key);
         break;
@@ -6763,17 +6779,61 @@ function collectCssForModules(manifest, moduleIds, projectRoot) {
   }
   return "";
 }
+async function resolveServerRenderer() {
+  if (process.env.NODE_ENV !== "production") {
+    return awaitServerRendererInitialized_Dev();
+  }
+  return loadServerRenderer_Production();
+}
+async function loadServerRenderer_Production() {
+  const projectRoot = process.cwd().replace(/\\/g, "/");
+  const serverOutDirAbs = path2.resolve(projectRoot, "dist/server");
+  const manifestPath = path2.resolve(serverOutDirAbs, "vite-manifest.json");
+  if (!fs.existsSync(manifestPath)) {
+    throw new Error(
+      `Server manifest not found at ${manifestPath}. Make sure the SSR build has been completed.`
+    );
+  }
+  const manifest = JSON.parse(
+    fs.readFileSync(manifestPath, "utf-8")
+  );
+  const virtualEntryServerModule = Object.values(manifest).find(
+    (value) => value.src === VIRTUAL_ENTRY_SERVER_ID
+  );
+  if (!virtualEntryServerModule) {
+    throw new Error(
+      "Virtual entry server module not found in manifest. Make sure the SSR build has been completed."
+    );
+  }
+  const entryServerFile = virtualEntryServerModule.file;
+  const entryServerPath = path2.resolve(serverOutDirAbs, entryServerFile);
+  if (!fs.existsSync(entryServerPath)) {
+    throw new Error(
+      `Virtual entry server module file not found at ${entryServerPath}`
+    );
+  }
+  const fileUrl = `file://${entryServerPath.replace(/\\/g, "/")}`;
+  const module = await import(
+    /* @vite-ignore */
+    fileUrl
+  );
+  if (!module.render || !module.documentModuleId) {
+    throw new Error(
+      "Virtual entry server module does not export render and documentModuleId"
+    );
+  }
+  return {
+    render: module.render,
+    documentModuleId: module.documentModuleId
+  };
+}
 async function renderPage(options) {
-  const { render } = await import("virtual:kiru:entry-server");
-  const moduleIds = [];
+  const { render, documentModuleId } = await resolveServerRenderer();
+  const moduleIds = [documentModuleId];
   const projectRoot = process.cwd().replace(/\\/g, "/");
-  const documentModule = resolveUserDocument(projectRoot, {
-    dir: "src/pages",
-    document: "document.tsx"
-  }).substring(projectRoot.length);
-  moduleIds.push(documentModule);
   const { httpResponse } = await render(options.url, {
-    registerModule: (moduleId) => {
+    userContext: options.context,
+    registerModule(moduleId) {
       moduleIds.push(moduleId);
     }
   });
@@ -6785,7 +6845,7 @@ async function renderPage(options) {
   const isDevelopment = process.env.NODE_ENV !== "production";
   let html = httpResponse.html;
   if (isDevelopment) {
-    const scriptTag = `<script type="module" src="/@id/${VIRTUAL_ENTRY_CLIENT_ID}"></script>`;
+    const scriptTag = `<script type="module" src="/@id/${VIRTUAL_ENTRY_CLIENT_ID}" async></script>`;
     html = html.includes("</body>") ? html.replace("</body>", scriptTag + "</body>") : html + scriptTag;
     const importedModules = /* @__PURE__ */ new Set();
     const seen = /* @__PURE__ */ new Set();
@@ -6799,7 +6859,7 @@ async function renderPage(options) {
       }
     };
     for (const id of moduleIds) {
-      const p = path3.join(projectRoot, id).replace(/\\/g, "/");
+      const p = path2.join(projectRoot, id).replace(/\\/g, "/");
       const mod = VITE_DEV_SERVER_INSTANCE.current.moduleGraph.getModuleById(p);
       if (!mod) {
         console.error(`Module not found: ${p}`);
@@ -6819,7 +6879,7 @@ async function renderPage(options) {
       html = html.replace("<head>", "<head>" + stylesheets.join("\n"));
     }
   } else {
-    const clientOutDirAbs = path3.resolve(projectRoot, "dist/client");
+    const clientOutDirAbs = path2.resolve(projectRoot, "dist/client");
     const { clientEntry, manifest } = await getClientAssets(
       clientOutDirAbs,
       "vite-manifest.json"
@@ -6831,10 +6891,15 @@ async function renderPage(options) {
       }
     }
     if (clientEntry) {
-      const scriptTag = `<script type="module" src="/${clientEntry}"></script>`;
+      const scriptTag = `<script type="module" src="/${clientEntry}" async></script>`;
       html = html.includes("</body>") ? html.replace("</body>", scriptTag + "</body>") : html + scriptTag;
     }
   }
+  const contextString = JSON.stringify(options.context);
+  html = html.replace(
+    "</head>",
+    `<script type="application/json" k-request-context>${contextString}</script></head>`
+  );
   return {
     httpResponse: {
       ...httpResponse,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "vite-plugin-kiru",
-  "version": "0.31.0",
+  "version": "0.32.0-preview.1",
   "description": "",
   "main": "dist/index.js",
   "type": "module",
@@ -8,8 +8,18 @@
   "keywords": [],
   "author": "",
   "license": "ISC",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "default": "./dist/index.js"
+    },
+    "./server": {
+      "types": "./dist/server.d.ts",
+      "default": "./dist/server.js"
+    }
+  },
   "peerDependencies": {
-    "kiru": ">=0.53.0"
+    "kiru": ">=0.54.0-preview.0"
   },
   "devDependencies": {
     "@types/node": "^22.17.0",
@@ -17,8 +27,9 @@
     "rollup": "^4.46.2",
     "tsx": "^4.20.3",
     "typescript": "^5.9.2",
+    "kiru-devtools-client": "^0.0.0",
     "kiru-devtools-host": "^1.0.0",
-    "kiru-devtools-client": "^0.0.0"
+    "kiru": "^0.54.0-preview.0"
   },
   "dependencies": {
     "glob": "^12.0.0",