weifuwu 0.18.2 → 0.18.3

package/dist/index.js

@@ -306,6 +306,11 @@ var Router = class _Router {
     return this;
   }
   route(method, path2, ...args) {
+    const last = args[args.length - 1];
+    if (last instanceof _Router) {
+      this._mountRouter(path2, last);
+      return this;
+    }
     const handler = args.pop();
     const middlewares = args;
     const segments = this.splitPath(path2);
@@ -5330,9 +5335,19 @@ function streamResponse(reactStream, opts) {
 var als = new AsyncLocalStorage();
 __registerAls(() => als.getStore());
 var isDev = process.env.NODE_ENV !== "production";
+var bundleCache = /* @__PURE__ */ new Map();
 function id2(s) {
   return createHash3("md5").update(s).digest("hex").slice(0, 8);
 }
+function serializeLoaderData(ctx) {
+  const data = {};
+  for (const key of Object.keys(ctx)) {
+    if (!["params", "query", "mountPath", "layoutStack"].includes(key)) {
+      data[key] = ctx[key];
+    }
+  }
+  return data;
+}
 async function buildClientBundle(entryPath, layoutPaths) {
   try {
     const absEntry = resolve4(entryPath);
@@ -5379,21 +5394,17 @@ async function buildClientBundle(entryPath, layoutPaths) {
     return null;
   }
 }
-var bundleRegistry = /* @__PURE__ */ new Map();
-function serializeLoaderData(ctx) {
-  const data = {};
-  for (const key of Object.keys(ctx)) {
-    if (!["params", "query", "mountPath", "layoutStack"].includes(key)) {
-      data[key] = ctx[key];
-    }
-  }
-  return data;
-}
 function ssr(path2) {
   const entryId = id2(resolve4(path2));
   const bundleKey = `/__ssr/${entryId}.js`;
-  let bundleBuilt = false;
-  return async (req, ctx) => {
+  const r = new Router();
+  r.get("/__ssr/:path", (req, ctx) => {
+    const buf = bundleCache.get("/__ssr/" + ctx.params.path);
+    return buf ? new Response(buf, {
+      headers: { "content-type": "application/javascript; charset=utf-8" }
+    }) : new Response("", { status: 404 });
+  });
+  r.get("/", async (req, ctx) => {
     const pageMod = await compileTsx(path2);
     const Component = pageMod.default;
     if (!Component) return new Response("", { status: 500 });
@@ -5441,14 +5452,11 @@ function ssr(path2) {
         }
       }
       let bundle = null;
-      if (!bundleBuilt) {
+      if (!bundleCache.has(bundleKey)) {
         const buf = await buildClientBundle(path2, layoutPaths);
-        if (buf) {
-          bundleRegistry.set(bundleKey, buf);
-          bundleBuilt = true;
-        }
+        if (buf) bundleCache.set(bundleKey, buf);
       }
-      if (bundleRegistry.has(bundleKey)) {
+      if (bundleCache.has(bundleKey)) {
         bundle = { url: bundleKey };
       }
       const { renderToReadableStream } = await import("react-dom/server");
@@ -5461,7 +5469,8 @@ function ssr(path2) {
         loaderData
       });
     });
-  };
+  });
+  return r;
 }
 
 // ssr/layout.ts

package/dist/router.d.ts

@@ -19,16 +19,16 @@ export declare class Router {
     use(router: Router): this;
     use(path: string, router: Router): this;
     use(path: string, mw: Middleware): this;
-    get(path: string, ...args: [...Middleware[], Handler]): this;
-    post(path: string, ...args: [...Middleware[], Handler]): this;
-    put(path: string, ...args: [...Middleware[], Handler]): this;
-    delete(path: string, ...args: [...Middleware[], Handler]): this;
-    patch(path: string, ...args: [...Middleware[], Handler]): this;
-    head(path: string, ...args: [...Middleware[], Handler]): this;
-    options(path: string, ...args: [...Middleware[], Handler]): this;
-    all(path: string, ...args: [...Middleware[], Handler]): this;
+    get(path: string, ...args: [...Middleware[], Handler | Router]): this;
+    post(path: string, ...args: [...Middleware[], Handler | Router]): this;
+    put(path: string, ...args: [...Middleware[], Handler | Router]): this;
+    delete(path: string, ...args: [...Middleware[], Handler | Router]): this;
+    patch(path: string, ...args: [...Middleware[], Handler | Router]): this;
+    head(path: string, ...args: [...Middleware[], Handler | Router]): this;
+    options(path: string, ...args: [...Middleware[], Handler | Router]): this;
+    all(path: string, ...args: [...Middleware[], Handler | Router]): this;
     onError(handler: ErrorHandler): this;
-    route(method: string, path: string, ...args: [...Middleware[], Handler]): this;
+    route(method: string, path: string, ...args: [...Middleware[], Handler | Router]): this;
     ws(path: string, ...args: [...Middleware[], WebSocketHandler]): this;
     handler(): Handler;
     websocketHandler(): WsUpgradeHandler;

package/dist/ssr/index.d.ts

@@ -1,4 +1,4 @@
-export { ssr, ssrBundleHandler } from './ssr.ts';
+export { ssr } from './ssr.ts';
 export { layout } from './layout.ts';
 export { tailwind } from './tailwind.ts';
 export { notFound } from './not-found.ts';

package/dist/ssr/index.js

@@ -231,279 +231,6 @@ function setCtx(value) {
 }
 var TsxContext = createContext(DEFAULT_CTX);
 
-// ssr/ssr.ts
-var als = new AsyncLocalStorage();
-__registerAls(() => als.getStore());
-var isDev = process.env.NODE_ENV !== "production";
-function id2(s) {
-  return createHash2("md5").update(s).digest("hex").slice(0, 8);
-}
-async function buildClientBundle(entryPath, layoutPaths) {
-  try {
-    const absEntry = resolve2(entryPath);
-    const absLayouts = layoutPaths.map((p) => resolve2(p));
-    const layoutImports = absLayouts.map((p) => `import${JSON.stringify(p)};`).join("");
-    const _sc = `(function(){var k='__WEIFUWU_CTX_STORE';var s=typeof globalThis!='undefined'&&globalThis[k];if(!s)return function(){};return function(v){s._ctx={...s._ctx,...v};s._snapshot={params:s._ctx.params,query:s._ctx.query,user:s._ctx.user,parsed:s._ctx.parsed,prefs:s._ctx.prefs,env:s._ctx.env};s._listeners.forEach(function(fn){fn()})}})()`;
-    const code = [
-      layoutImports,
-      `import{hydrateRoot}from'react-dom/client';`,
-      `import{createElement,useState,useEffect}from'react';`,
-      `import{TsxContext}from'weifuwu/react';`,
-      `import P from${JSON.stringify(absEntry)};`,
-      `var setCtx=${_sc};`,
-      `const c=document.getElementById('__weifuwu_root');`,
-      `if(window.__WEIFUWU_PROPS)setCtx({loaderData:window.__WEIFUWU_PROPS});`,
-      `if(!window.__WFW_ROOT){`,
-      `function App(){`,
-      `const[p,setP]=useState({C:P});`,
-      `useEffect(()=>{window.__WFW_SET_PAGE=(C)=>{setCtx({loaderData:window.__WEIFUWU_PROPS});setP({C})}},[]);`,
-      `const ctx=window.__WEIFUWU_CTX||{};`,
-      `return createElement(TsxContext.Provider,{value:ctx},`,
-      `createElement(p.C,null))`,
-      `}`,
-      `window.__WFW_ROOT=hydrateRoot(c,createElement(App));`,
-      `}else{`,
-      `window.__WFW_SET_PAGE?.(P);`,
-      `}`
-    ].join("");
-    const { default: esbuild2 } = await import("esbuild");
-    const result = await esbuild2.build({
-      stdin: { contents: code, loader: "tsx", resolveDir: dirname2(absEntry) },
-      bundle: true,
-      format: "esm",
-      jsx: "automatic",
-      jsxImportSource: "react",
-      banner: { js: "self.process={env:{}};" },
-      loader: { ".node": "empty" },
-      write: false,
-      minify: true
-    });
-    return result.outputFiles[0].contents;
-  } catch (err) {
-    console.error("hydration bundle failed:", err);
-    return null;
-  }
-}
-var bundleRegistry = /* @__PURE__ */ new Map();
-function serializeLoaderData(ctx) {
-  const data = {};
-  for (const key of Object.keys(ctx)) {
-    if (!["params", "query", "mountPath", "layoutStack"].includes(key)) {
-      data[key] = ctx[key];
-    }
-  }
-  return data;
-}
-function ssr(path) {
-  const entryId = id2(resolve2(path));
-  const bundleKey = `/__ssr/${entryId}.js`;
-  let bundleBuilt = false;
-  return async (req, ctx) => {
-    const pageMod = await compileTsx(path);
-    const Component = pageMod.default;
-    if (!Component) return new Response("", { status: 500 });
-    const layouts = ctx.layoutStack || [];
-    const layoutComponents = layouts.map((l) => l.component);
-    const layoutPaths = layouts.map((l) => l.path);
-    const base = (ctx.mountPath || "").replace(/\/$/, "");
-    const loaderData = serializeLoaderData(ctx);
-    const ctxValue = {
-      params: ctx.params,
-      query: ctx.query,
-      user: ctx.user ?? {},
-      parsed: ctx.parsed ?? {},
-      prefs: ctx.prefs ?? {},
-      loaderData,
-      env: ctx.env ?? {}
-    };
-    return als.run(ctxValue, async () => {
-      setCtx(ctxValue);
-      let element = createElement(
-        TsxContext.Provider,
-        { value: ctxValue },
-        createElement(
-          "div",
-          { id: "__weifuwu_root" },
-          createElement(Component, null)
-        )
-      );
-      if (layoutComponents.length === 0) {
-        element = createElement(
-          "html",
-          { lang: "en" },
-          createElement(
-            "head",
-            null,
-            createElement("meta", { charSet: "utf-8" }),
-            createElement("meta", { name: "viewport", content: "width=device-width, initial-scale=1" }),
-            createElement("title", null, "weifuwu")
-          ),
-          createElement("body", null, element)
-        );
-      } else {
-        for (const L of layoutComponents.toReversed()) {
-          element = createElement(L, { children: element });
-        }
-      }
-      let bundle = null;
-      if (!bundleBuilt) {
-        const buf = await buildClientBundle(path, layoutPaths);
-        if (buf) {
-          bundleRegistry.set(bundleKey, buf);
-          bundleBuilt = true;
-        }
-      }
-      if (bundleRegistry.has(bundleKey)) {
-        bundle = { url: bundleKey };
-      }
-      const { renderToReadableStream } = await import("react-dom/server");
-      const stream = await renderToReadableStream(element);
-      return streamResponse(stream, {
-        ctx,
-        base,
-        isDev,
-        bundle,
-        loaderData
-      });
-    });
-  };
-}
-function ssrBundleHandler() {
-  return (req, ctx) => {
-    const url = new URL(req.url);
-    const buf = bundleRegistry.get(url.pathname);
-    return buf ? new Response(buf, {
-      headers: { "content-type": "application/javascript; charset=utf-8" }
-    }) : new Response("", { status: 404 });
-  };
-}
-
-// ssr/layout.ts
-function layout(path) {
-  return async (req, ctx, next) => {
-    const mod = await compileTsx(path);
-    const Component = mod.default;
-    if (!Component) throw new Error(`Layout ${path} has no default export`);
-    ctx.layoutStack = [...ctx.layoutStack || [], { path, component: Component }];
-    return next(req, ctx);
-  };
-}
-
-// ssr/tailwind.ts
-import { existsSync as existsSync2, mkdirSync as mkdirSync2, readFileSync as readFileSync2, writeFileSync } from "node:fs";
-import { relative, resolve as resolve3 } from "node:path";
-var isDev2 = process.env.NODE_ENV !== "production";
-function tailwind(cssPath, scanDir) {
-  let compiledCss = "";
-  let twWatcher = null;
-  return async (req, ctx, next) => {
-    const url = new URL(req.url);
-    if (url.pathname === "/__wfw/style.css") {
-      if (!compiledCss) compiledCss = await compile(cssPath, scanDir);
-      return new Response(compiledCss || "", {
-        headers: { "content-type": "text/css; charset=utf-8" }
-      });
-    }
-    ctx.compiledTailwindCss = compiledCss;
-    if (isDev2 && !twWatcher) {
-      twWatcher = watchFile(cssPath, () => {
-        compiledCss = "";
-      });
-    }
-    return next(req, ctx);
-  };
-}
-async function compile(cssPath, scanDir) {
-  try {
-    const inputFile = resolve3(cssPath);
-    if (!existsSync2(inputFile)) {
-      mkdirSync2(dirname3(inputFile), { recursive: true });
-      writeFileSync(inputFile, '@import "tailwindcss"\n', "utf-8");
-    }
-    const { default: tailwindPlugin } = await import("@tailwindcss/postcss");
-    const { default: postcss } = await import("postcss");
-    let src = readFileSync2(inputFile, "utf-8");
-    const scanSource = scanDir ? relative(dirname3(inputFile), scanDir) || "." : ".";
-    const sourcePath = scanSource === "." ? "./" : `./${scanSource}/`;
-    src = `@source "${sourcePath}";
-${src}`;
-    const result = await postcss([tailwindPlugin()]).process(src, { from: inputFile });
-    return result.css;
-  } catch (err) {
-    console.warn("Tailwind CSS processing failed:", err.message);
-    return "";
-  }
-}
-function dirname3(p) {
-  return p.substring(0, p.lastIndexOf("/")) || "/";
-}
-function watchFile(path, onChange) {
-  let watcher = null;
-  import("chokidar").then((chokidar2) => {
-    watcher = chokidar2.default.watch(resolve3(path), { persistent: false });
-    watcher.on("change", onChange);
-  });
-  return watcher;
-}
-
-// ssr/not-found.ts
-function notFound(path) {
-  return async (req, ctx) => {
-    if (!path) return new Response("Not Found", { status: 404 });
-    const mod = await compileTsx(path);
-    const Component = mod?.default;
-    const body = Component ? "404 - Not Found" : "404 - Not Found";
-    return new Response(body, {
-      status: 404,
-      headers: { "content-type": "text/html; charset=utf-8" }
-    });
-  };
-}
-
-// ssr/error-boundary.ts
-import { createElement as createElement2 } from "react";
-import { TextEncoder as TextEncoder2 } from "node:util";
-function errorBoundary(errorPath) {
-  return async (req, ctx, next) => {
-    try {
-      return await next(req, ctx);
-    } catch (err) {
-      const mod = await compileTsx(errorPath);
-      const ErrorComponent = mod.default;
-      if (!ErrorComponent) throw err;
-      const layouts = (ctx.layoutStack || []).map((l) => l.component);
-      const stream = await import("react-dom/server").then((m) => m.renderToReadableStream(
-        createElement2(ErrorComponent, {
-          error: err instanceof Error ? err : new Error(String(err)),
-          reset: () => {
-          }
-        })
-      ));
-      const reader = stream.getReader();
-      const chunks = [];
-      while (true) {
-        const { done, value } = await reader.read();
-        if (done) break;
-        chunks.push(value);
-      }
-      const encoder = new TextEncoder2();
-      const body = chunks.reduce((acc, c) => {
-        const merged = new Uint8Array(acc.length + c.length);
-        merged.set(acc);
-        merged.set(c, acc.length);
-        return merged;
-      }, new Uint8Array(0));
-      return new Response(body, {
-        status: 500,
-        headers: { "content-type": "text/html; charset=utf-8" }
-      });
-    }
-  };
-}
-
-// ssr/live.ts
-import chokidar from "chokidar";
-
 // router.ts
 import { WebSocketServer } from "ws";
 var createTrieNode = () => ({
@@ -632,6 +359,11 @@ var Router = class _Router {
     return this;
   }
   route(method, path, ...args) {
+    const last = args[args.length - 1];
+    if (last instanceof _Router) {
+      this._mountRouter(path, last);
+      return this;
+    }
     const handler = args.pop();
     const middlewares = args;
     const segments = this.splitPath(path);
@@ -779,117 +511,383 @@ var Router = class _Router {
           wildcardIdx = i;
         }
       }
-      const segment = segments[i];
-      if (!segment) break;
-      const next = matchTrieNode(node, segment, params);
-      if (!next) {
-        if (wildcardHandler) {
-          params["*"] = segments.slice(wildcardIdx).join("/");
-          return { handler: wildcardHandler, middlewares: wildcardMws, pathMws, params };
-        }
-        return null;
+      const segment = segments[i];
+      if (!segment) break;
+      const next = matchTrieNode(node, segment, params);
+      if (!next) {
+        if (wildcardHandler) {
+          params["*"] = segments.slice(wildcardIdx).join("/");
+          return { handler: wildcardHandler, middlewares: wildcardMws, pathMws, params };
+        }
+        return null;
+      }
+      node = next;
+    }
+    pathMws.push(...node.pathMws);
+    const handler = node.handlers.get(method) || node.handlers.get("*");
+    if (handler) {
+      if (node.wildcard) params["*"] = segments.slice(segments.length).join("/");
+      return {
+        handler,
+        middlewares: node.middlewares.get(method) || node.middlewares.get("*") || [],
+        pathMws,
+        params
+      };
+    }
+    if (wildcardHandler) {
+      params["*"] = segments.slice(wildcardIdx).join("/");
+      return { handler: wildcardHandler, middlewares: wildcardMws, pathMws, params };
+    }
+    return null;
+  }
+  matchWsTrie(root, segments) {
+    let node = root;
+    const params = {};
+    for (const segment of segments) {
+      const next = matchWsNode(node, segment, params);
+      if (!next) return null;
+      node = next;
+    }
+    return node.handler ? { handler: node.handler, middlewares: node.middlewares, params } : null;
+  }
+  async handle(req, ctx, segments, query) {
+    const match = this.matchTrie(req.method, segments);
+    if (match?.handler) {
+      const { handler, middlewares: routeMws, pathMws, params } = match;
+      const allMws = this.globalMws.length + pathMws.length + routeMws.length === 0 ? [] : [...this.globalMws, ...pathMws, ...routeMws];
+      const ctxWithMatch = { ...ctx, params: { ...ctx.params, ...params } };
+      try {
+        return await this.runChain(allMws, handler, req, ctxWithMatch);
+      } catch (e) {
+        const err = e instanceof Error ? e : new Error(String(e));
+        return this.errorHandler ? this.errorHandler(err, req, ctxWithMatch) : new Response("Internal Server Error", { status: 500 });
+      }
+    }
+    if (this.globalMws.length > 0) {
+      try {
+        const delegate = () => new Response("Not Found", { status: 404 });
+        return await this.runChain(this.globalMws, delegate, req, ctx);
+      } catch (e) {
+        const err = e instanceof Error ? e : new Error(String(e));
+        return this.errorHandler ? this.errorHandler(err, req, ctx) : new Response("Internal Server Error", { status: 500 });
+      }
+    }
+    return new Response("Not Found", { status: 404 });
+  }
+  async runChain(middlewares, finalHandler, req, ctx) {
+    let index = 0;
+    const dispatch = async (req2, ctx2) => {
+      if (index < middlewares.length) {
+        const mw = middlewares[index++];
+        return mw ? await mw(req2, ctx2, dispatch) : new Response("Middleware error", { status: 500 });
+      }
+      return await finalHandler(req2, ctx2);
+    };
+    return dispatch(req, ctx);
+  }
+};
+function upgradeSocket(wss, req, socket, head, handler, ctx) {
+  wss.handleUpgrade(req, socket, head, (ws) => {
+    if (handler.open) {
+      handler.open(ws, ctx);
+    }
+    ws.on("message", (data) => {
+      handler.message?.(ws, ctx, data);
+    });
+    ws.on("close", () => {
+      handler.close?.(ws, ctx);
+    });
+    ws.on("error", (err) => {
+      handler.error?.(ws, ctx, err);
+    });
+  });
+}
+function sendHttpResponseOnSocket(socket, response) {
+  const statusLine = `HTTP/1.1 ${response.status} ${response.statusText}`;
+  const headerLines = [statusLine];
+  response.headers.forEach((value, key) => {
+    headerLines.push(`${key}: ${value}`);
+  });
+  headerLines.push("Connection: close");
+  headerLines.push("");
+  const headerStr = headerLines.join("\r\n");
+  response.arrayBuffer().then((buf) => {
+    const body = Buffer.from(buf);
+    socket.write(headerStr + "\r\n" + body.toString());
+    socket.end();
+  }).catch(() => {
+    socket.write(headerStr + "\r\n");
+    socket.end();
+  });
+}
+
+// ssr/ssr.ts
+var als = new AsyncLocalStorage();
+__registerAls(() => als.getStore());
+var isDev = process.env.NODE_ENV !== "production";
+var bundleCache = /* @__PURE__ */ new Map();
+function id2(s) {
+  return createHash2("md5").update(s).digest("hex").slice(0, 8);
+}
+function serializeLoaderData(ctx) {
+  const data = {};
+  for (const key of Object.keys(ctx)) {
+    if (!["params", "query", "mountPath", "layoutStack"].includes(key)) {
+      data[key] = ctx[key];
+    }
+  }
+  return data;
+}
+async function buildClientBundle(entryPath, layoutPaths) {
+  try {
+    const absEntry = resolve2(entryPath);
+    const absLayouts = layoutPaths.map((p) => resolve2(p));
+    const layoutImports = absLayouts.map((p) => `import${JSON.stringify(p)};`).join("");
+    const _sc = `(function(){var k='__WEIFUWU_CTX_STORE';var s=typeof globalThis!='undefined'&&globalThis[k];if(!s)return function(){};return function(v){s._ctx={...s._ctx,...v};s._snapshot={params:s._ctx.params,query:s._ctx.query,user:s._ctx.user,parsed:s._ctx.parsed,prefs:s._ctx.prefs,env:s._ctx.env};s._listeners.forEach(function(fn){fn()})}})()`;
+    const code = [
+      layoutImports,
+      `import{hydrateRoot}from'react-dom/client';`,
+      `import{createElement,useState,useEffect}from'react';`,
+      `import{TsxContext}from'weifuwu/react';`,
+      `import P from${JSON.stringify(absEntry)};`,
+      `var setCtx=${_sc};`,
+      `const c=document.getElementById('__weifuwu_root');`,
+      `if(window.__WEIFUWU_PROPS)setCtx({loaderData:window.__WEIFUWU_PROPS});`,
+      `if(!window.__WFW_ROOT){`,
+      `function App(){`,
+      `const[p,setP]=useState({C:P});`,
+      `useEffect(()=>{window.__WFW_SET_PAGE=(C)=>{setCtx({loaderData:window.__WEIFUWU_PROPS});setP({C})}},[]);`,
+      `const ctx=window.__WEIFUWU_CTX||{};`,
+      `return createElement(TsxContext.Provider,{value:ctx},`,
+      `createElement(p.C,null))`,
+      `}`,
+      `window.__WFW_ROOT=hydrateRoot(c,createElement(App));`,
+      `}else{`,
+      `window.__WFW_SET_PAGE?.(P);`,
+      `}`
+    ].join("");
+    const { default: esbuild2 } = await import("esbuild");
+    const result = await esbuild2.build({
+      stdin: { contents: code, loader: "tsx", resolveDir: dirname2(absEntry) },
+      bundle: true,
+      format: "esm",
+      jsx: "automatic",
+      jsxImportSource: "react",
+      banner: { js: "self.process={env:{}};" },
+      loader: { ".node": "empty" },
+      write: false,
+      minify: true
+    });
+    return result.outputFiles[0].contents;
+  } catch (err) {
+    console.error("hydration bundle failed:", err);
+    return null;
+  }
+}
+function ssr(path) {
+  const entryId = id2(resolve2(path));
+  const bundleKey = `/__ssr/${entryId}.js`;
+  const r = new Router();
+  r.get("/__ssr/:path", (req, ctx) => {
+    const buf = bundleCache.get("/__ssr/" + ctx.params.path);
+    return buf ? new Response(buf, {
+      headers: { "content-type": "application/javascript; charset=utf-8" }
+    }) : new Response("", { status: 404 });
+  });
+  r.get("/", async (req, ctx) => {
+    const pageMod = await compileTsx(path);
+    const Component = pageMod.default;
+    if (!Component) return new Response("", { status: 500 });
+    const layouts = ctx.layoutStack || [];
+    const layoutComponents = layouts.map((l) => l.component);
+    const layoutPaths = layouts.map((l) => l.path);
+    const base = (ctx.mountPath || "").replace(/\/$/, "");
+    const loaderData = serializeLoaderData(ctx);
+    const ctxValue = {
+      params: ctx.params,
+      query: ctx.query,
+      user: ctx.user ?? {},
+      parsed: ctx.parsed ?? {},
+      prefs: ctx.prefs ?? {},
+      loaderData,
+      env: ctx.env ?? {}
+    };
+    return als.run(ctxValue, async () => {
+      setCtx(ctxValue);
+      let element = createElement(
+        TsxContext.Provider,
+        { value: ctxValue },
+        createElement(
+          "div",
+          { id: "__weifuwu_root" },
+          createElement(Component, null)
+        )
+      );
+      if (layoutComponents.length === 0) {
+        element = createElement(
+          "html",
+          { lang: "en" },
+          createElement(
+            "head",
+            null,
+            createElement("meta", { charSet: "utf-8" }),
+            createElement("meta", { name: "viewport", content: "width=device-width, initial-scale=1" }),
+            createElement("title", null, "weifuwu")
+          ),
+          createElement("body", null, element)
+        );
+      } else {
+        for (const L of layoutComponents.toReversed()) {
+          element = createElement(L, { children: element });
+        }
+      }
+      let bundle = null;
+      if (!bundleCache.has(bundleKey)) {
+        const buf = await buildClientBundle(path, layoutPaths);
+        if (buf) bundleCache.set(bundleKey, buf);
       }
-      node = next;
-    }
-    pathMws.push(...node.pathMws);
-    const handler = node.handlers.get(method) || node.handlers.get("*");
-    if (handler) {
-      if (node.wildcard) params["*"] = segments.slice(segments.length).join("/");
-      return {
-        handler,
-        middlewares: node.middlewares.get(method) || node.middlewares.get("*") || [],
-        pathMws,
-        params
-      };
-    }
-    if (wildcardHandler) {
-      params["*"] = segments.slice(wildcardIdx).join("/");
-      return { handler: wildcardHandler, middlewares: wildcardMws, pathMws, params };
-    }
-    return null;
-  }
-  matchWsTrie(root, segments) {
-    let node = root;
-    const params = {};
-    for (const segment of segments) {
-      const next = matchWsNode(node, segment, params);
-      if (!next) return null;
-      node = next;
-    }
-    return node.handler ? { handler: node.handler, middlewares: node.middlewares, params } : null;
-  }
-  async handle(req, ctx, segments, query) {
-    const match = this.matchTrie(req.method, segments);
-    if (match?.handler) {
-      const { handler, middlewares: routeMws, pathMws, params } = match;
-      const allMws = this.globalMws.length + pathMws.length + routeMws.length === 0 ? [] : [...this.globalMws, ...pathMws, ...routeMws];
-      const ctxWithMatch = { ...ctx, params: { ...ctx.params, ...params } };
-      try {
-        return await this.runChain(allMws, handler, req, ctxWithMatch);
-      } catch (e) {
-        const err = e instanceof Error ? e : new Error(String(e));
-        return this.errorHandler ? this.errorHandler(err, req, ctxWithMatch) : new Response("Internal Server Error", { status: 500 });
+      if (bundleCache.has(bundleKey)) {
+        bundle = { url: bundleKey };
       }
+      const { renderToReadableStream } = await import("react-dom/server");
+      const stream = await renderToReadableStream(element);
+      return streamResponse(stream, {
+        ctx,
+        base,
+        isDev,
+        bundle,
+        loaderData
+      });
+    });
+  });
+  return r;
+}
+
+// ssr/layout.ts
+function layout(path) {
+  return async (req, ctx, next) => {
+    const mod = await compileTsx(path);
+    const Component = mod.default;
+    if (!Component) throw new Error(`Layout ${path} has no default export`);
+    ctx.layoutStack = [...ctx.layoutStack || [], { path, component: Component }];
+    return next(req, ctx);
+  };
+}
+
+// ssr/tailwind.ts
+import { existsSync as existsSync2, mkdirSync as mkdirSync2, readFileSync as readFileSync2, writeFileSync } from "node:fs";
+import { relative, resolve as resolve3 } from "node:path";
+var isDev2 = process.env.NODE_ENV !== "production";
+function tailwind(cssPath, scanDir) {
+  let compiledCss = "";
+  let twWatcher = null;
+  return async (req, ctx, next) => {
+    const url = new URL(req.url);
+    if (url.pathname === "/__wfw/style.css") {
+      if (!compiledCss) compiledCss = await compile(cssPath, scanDir);
+      return new Response(compiledCss || "", {
+        headers: { "content-type": "text/css; charset=utf-8" }
+      });
     }
-    if (this.globalMws.length > 0) {
-      try {
-        const delegate = () => new Response("Not Found", { status: 404 });
-        return await this.runChain(this.globalMws, delegate, req, ctx);
-      } catch (e) {
-        const err = e instanceof Error ? e : new Error(String(e));
-        return this.errorHandler ? this.errorHandler(err, req, ctx) : new Response("Internal Server Error", { status: 500 });
-      }
+    ctx.compiledTailwindCss = compiledCss;
+    if (isDev2 && !twWatcher) {
+      twWatcher = watchFile(cssPath, () => {
+        compiledCss = "";
+      });
     }
-    return new Response("Not Found", { status: 404 });
-  }
-  async runChain(middlewares, finalHandler, req, ctx) {
-    let index = 0;
-    const dispatch = async (req2, ctx2) => {
-      if (index < middlewares.length) {
-        const mw = middlewares[index++];
-        return mw ? await mw(req2, ctx2, dispatch) : new Response("Middleware error", { status: 500 });
-      }
-      return await finalHandler(req2, ctx2);
-    };
-    return dispatch(req, ctx);
-  }
-};
-function upgradeSocket(wss, req, socket, head, handler, ctx) {
-  wss.handleUpgrade(req, socket, head, (ws) => {
-    if (handler.open) {
-      handler.open(ws, ctx);
+    return next(req, ctx);
+  };
+}
+async function compile(cssPath, scanDir) {
+  try {
+    const inputFile = resolve3(cssPath);
+    if (!existsSync2(inputFile)) {
+      mkdirSync2(dirname3(inputFile), { recursive: true });
+      writeFileSync(inputFile, '@import "tailwindcss"\n', "utf-8");
     }
-    ws.on("message", (data) => {
-      handler.message?.(ws, ctx, data);
-    });
-    ws.on("close", () => {
-      handler.close?.(ws, ctx);
-    });
-    ws.on("error", (err) => {
-      handler.error?.(ws, ctx, err);
-    });
-  });
+    const { default: tailwindPlugin } = await import("@tailwindcss/postcss");
+    const { default: postcss } = await import("postcss");
+    let src = readFileSync2(inputFile, "utf-8");
+    const scanSource = scanDir ? relative(dirname3(inputFile), scanDir) || "." : ".";
+    const sourcePath = scanSource === "." ? "./" : `./${scanSource}/`;
+    src = `@source "${sourcePath}";
+${src}`;
+    const result = await postcss([tailwindPlugin()]).process(src, { from: inputFile });
+    return result.css;
+  } catch (err) {
+    console.warn("Tailwind CSS processing failed:", err.message);
+    return "";
+  }
 }
-function sendHttpResponseOnSocket(socket, response) {
-  const statusLine = `HTTP/1.1 ${response.status} ${response.statusText}`;
-  const headerLines = [statusLine];
-  response.headers.forEach((value, key) => {
-    headerLines.push(`${key}: ${value}`);
-  });
-  headerLines.push("Connection: close");
-  headerLines.push("");
-  const headerStr = headerLines.join("\r\n");
-  response.arrayBuffer().then((buf) => {
-    const body = Buffer.from(buf);
-    socket.write(headerStr + "\r\n" + body.toString());
-    socket.end();
-  }).catch(() => {
-    socket.write(headerStr + "\r\n");
-    socket.end();
+function dirname3(p) {
+  return p.substring(0, p.lastIndexOf("/")) || "/";
+}
+function watchFile(path, onChange) {
+  let watcher = null;
+  import("chokidar").then((chokidar2) => {
+    watcher = chokidar2.default.watch(resolve3(path), { persistent: false });
+    watcher.on("change", onChange);
   });
+  return watcher;
+}
+
+// ssr/not-found.ts
+function notFound(path) {
+  return async (req, ctx) => {
+    if (!path) return new Response("Not Found", { status: 404 });
+    const mod = await compileTsx(path);
+    const Component = mod?.default;
+    const body = Component ? "404 - Not Found" : "404 - Not Found";
+    return new Response(body, {
+      status: 404,
+      headers: { "content-type": "text/html; charset=utf-8" }
+    });
+  };
+}
+
+// ssr/error-boundary.ts
+import { createElement as createElement2 } from "react";
+import { TextEncoder as TextEncoder2 } from "node:util";
+function errorBoundary(errorPath) {
+  return async (req, ctx, next) => {
+    try {
+      return await next(req, ctx);
+    } catch (err) {
+      const mod = await compileTsx(errorPath);
+      const ErrorComponent = mod.default;
+      if (!ErrorComponent) throw err;
+      const layouts = (ctx.layoutStack || []).map((l) => l.component);
+      const stream = await import("react-dom/server").then((m) => m.renderToReadableStream(
+        createElement2(ErrorComponent, {
+          error: err instanceof Error ? err : new Error(String(err)),
+          reset: () => {
+          }
+        })
+      ));
+      const reader = stream.getReader();
+      const chunks = [];
+      while (true) {
+        const { done, value } = await reader.read();
+        if (done) break;
+        chunks.push(value);
+      }
+      const encoder = new TextEncoder2();
+      const body = chunks.reduce((acc, c) => {
+        const merged = new Uint8Array(acc.length + c.length);
+        merged.set(acc);
+        merged.set(c, acc.length);
+        return merged;
+      }, new Uint8Array(0));
+      return new Response(body, {
+        status: 500,
+        headers: { "content-type": "text/html; charset=utf-8" }
+      });
+    }
+  };
 }
 
 // ssr/live.ts
+import chokidar from "chokidar";
 var clients = /* @__PURE__ */ new Set();
 function broadcastReload() {
   for (const ws of clients) {
@@ -931,6 +929,5 @@ export {
   liveReload,
   notFound,
   ssr,
-  ssrBundleHandler,
   tailwind
 };

package/dist/ssr/ssr.d.ts

@@ -1,3 +1,2 @@
-import type { Handler } from '../types.ts';
-export declare function ssr(path: string): Handler;
-export declare function ssrBundleHandler(): Handler;
+import { Router } from '../router.ts';
+export declare function ssr(path: string): Router;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "weifuwu",
-  "version": "0.18.2",
+  "version": "0.18.3",
   "description": "Web-standard HTTP framework for Node.js — (req, ctx) => Response",
   "main": "./dist/index.js",
   "types": "./dist/index.d.ts",