@alignable/bifrost-fastify 1.0.0 → 1.0.2

package/dist/index.d.ts

@@ -12,6 +12,7 @@ type RenderedPageContext = Awaited<ReturnType<typeof renderPage<{
 declare module "fastify" {
     interface FastifyRequest {
         bifrostPageId?: string | null;
+        vikePageContext?: Partial<PageContextServer> | null;
         getLayout: GetLayout;
     }
 }

package/dist/index.js

@@ -2,6 +2,7 @@
 import proxy from "@fastify/http-proxy";
 import accepts from "@fastify/accepts";
 import forwarded from "@fastify/forwarded";
+import { PassThrough } from "stream";
 import { renderPage } from "vike/server";
 
 // lib/extractDomElements.ts
@@ -51,11 +52,14 @@ var viteProxyPlugin = async (fastify, { upstream, host, onError, buildPageContex
     if (!httpResponse) {
       return reply.code(404).type("text/html").send("Not Found");
     }
-    const { body, statusCode, headers } = httpResponse;
-    return reply.status(statusCode).headers(Object.fromEntries(headers)).send(body);
+    const { pipe, statusCode, headers } = httpResponse;
+    const stream = new PassThrough();
+    pipe(stream);
+    return reply.status(statusCode).headers(Object.fromEntries(headers)).send(stream);
   }
   await fastify.register(accepts);
   fastify.decorateRequest("bifrostPageId", null);
+  fastify.decorateRequest("vikePageContext", null);
   fastify.decorateRequest("getLayout", null);
   await fastify.register(proxy, {
     upstream: upstream.href,
@@ -64,10 +68,12 @@ var viteProxyPlugin = async (fastify, { upstream, host, onError, buildPageContex
       if ((req.method === "GET" || req.method === "HEAD") && req.accepts().type(["html"]) === "html") {
         const pageContextInit = {
           urlOriginal: req.url,
+          headersOriginal: req.headers,
           ...buildPageContextInit ? await buildPageContextInit(req) : {}
         };
         const pageContext = await renderPage(pageContextInit);
         req.bifrostPageId = pageContext.pageId;
+        req.vikePageContext = pageContext;
         const proxyMode = pageContext.config?.proxyMode;
         switch (proxyMode) {
           case "passthru": {
@@ -127,7 +133,6 @@ var viteProxyPlugin = async (fastify, { upstream, host, onError, buildPageContex
       async onResponse(req, reply, res) {
         if ([301, 302, 303, 307, 308].includes(reply.statusCode)) {
           const location = reply.getHeader("location");
-          console.log(location);
           if (location) {
             const url = new URL(location, host.href);
             if (url.host === upstream.host || url.host === host.host) {
@@ -149,6 +154,7 @@ var viteProxyPlugin = async (fastify, { upstream, host, onError, buildPageContex
         }
         const pageContextInit = {
           urlOriginal: req.url,
+          headersOriginal: req.headers,
           // Critical that we don't set any passToClient values in pageContextInit
           // If we do, Vike re-requests pageContext on client navigation. This breaks wrapped proxy.
           _wrappedServerOnly: {

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../index.ts","../lib/extractDomElements.ts"],"sourcesContent":["// Note that this file isn't processed by Vite, see https://github.com/brillout/vike/issues/562\nimport { FastifyReply, RawServerBase, FastifyPluginAsync } from \"fastify\";\nimport { FastifyRequest, RequestGenericInterface } from \"fastify/types/request\";\nimport proxy from \"@fastify/http-proxy\";\nimport accepts from \"@fastify/accepts\";\nimport forwarded from \"@fastify/forwarded\";\nimport type { GetLayout, WrappedServerOnly } from \"@alignable/bifrost/config\";\nimport { Writable } from \"stream\";\nimport { IncomingMessage } from \"http\";\nimport { renderPage } from \"vike/server\";\nimport { PageContextServer } from \"vike/types\";\nimport { extractDomElements } from \"./lib/extractDomElements\";\n\ntype RenderedPageContext = Awaited<\n  ReturnType<\n    typeof renderPage<\n      {\n        isClientSideNavigation?: boolean;\n      },\n      { urlOriginal: string }\n    >\n  >\n>;\n\ndeclare module \"fastify\" {\n  interface FastifyRequest {\n    bifrostPageId?: string | null;\n    getLayout: GetLayout;\n  }\n}\n\ntype RawRequestExtendedWithProxy = FastifyRequest<\n  RequestGenericInterface,\n  RawServerBase\n>[\"raw\"] & {\n  _bfproxy?: boolean;\n};\n\nfunction streamToString(stream: Writable): Promise<string> {\n  const chunks: Buffer[] = [];\n  return new Promise((resolve, reject) => {\n    stream.on(\"data\", (chunk) => chunks.push(Buffer.from(chunk)));\n    stream.on(\"error\", (err) => reject(err));\n    stream.on(\"end\", () => resolve(Buffer.concat(chunks).toString(\"utf8\")));\n  });\n}\n\ninterface ViteProxyPluginOptions {\n  upstream: URL;\n  host: URL;\n  onError?: (error: any, pageContext: RenderedPageContext) => void;\n  buildPageContextInit?: (\n    req: FastifyRequest\n  ) => Promise<Partial<Omit<PageContextServer, \"headers\">>>;\n}\n/**\n * Fastify plugin that wraps @fasitfy/http-proxy to proxy Rails/Turbolinks server into a vike site.\n */\nexport const viteProxyPlugin: FastifyPluginAsync<\n  ViteProxyPluginOptions\n> = async (fastify, { upstream, host, onError, buildPageContextInit }) => {\n  async function replyWithPage(\n    reply: FastifyReply<RawServerBase>,\n    pageContext: RenderedPageContext\n  ): Promise<FastifyReply> {\n    const { httpResponse } = pageContext;\n\n    if (\n      onError &&\n      httpResponse?.statusCode === 500 &&\n      pageContext.errorWhileRendering\n    ) {\n      onError(pageContext.errorWhileRendering, pageContext);\n    }\n\n    if (!httpResponse) {\n      return reply.code(404).type(\"text/html\").send(\"Not Found\");\n    }\n\n    const { body, statusCode, headers } = httpResponse;\n    return reply\n      .status(statusCode)\n      .headers(Object.fromEntries(headers))\n      .send(body);\n  }\n  await fastify.register(accepts);\n  fastify.decorateRequest(\"bifrostPageId\", null);\n  fastify.decorateRequest(\"getLayout\", null);\n  await fastify.register(proxy, {\n    upstream: upstream.href,\n    websocket: true,\n    async preHandler(req, reply) {\n      if (\n        (req.method === \"GET\" || req.method === \"HEAD\") &&\n        req.accepts().type([\"html\"]) === \"html\"\n      ) {\n        const pageContextInit = {\n          urlOriginal: req.url,\n          ...(buildPageContextInit ? await buildPageContextInit(req) : {}),\n        };\n\n        const pageContext = await renderPage(pageContextInit);\n\n        // this does not handle getting the original pageId when errors are thrown: https://github.com/vikejs/vike/issues/1112\n        req.bifrostPageId = pageContext.pageId;\n\n        const proxyMode = pageContext.config?.proxyMode;\n\n        switch (proxyMode) {\n          case \"passthru\": {\n            req.log.info(`bifrost: passthru proxy to backend`);\n            return;\n          }\n          case \"wrapped\": {\n            req.log.info(`bifrost: proxy route matched, proxying to backend`);\n            if (!!pageContext.isClientSideNavigation) {\n              // This should never happen because wrapped proxy routes have no onBeforeRender. onRenderClient should make a request to the legacy backend.\n              req.log.error(\n                \"Wrapped proxy route is requesting index.pageContext.json. Something is wrong with the client.\"\n              );\n              return reply.redirect(\n                req.url.replace(\"/index.pageContext.json\", \"\")\n              );\n            }\n            if (!pageContext.config?.getLayout) {\n              req.log.error(\n                \"Config missing getLayout on wrapped route! Falling back to passthru proxy\"\n              );\n              return;\n            }\n\n            let proxyHeadersAlreadySet = true;\n            for (const [key, val] of Object.entries(\n              pageContext.config?.proxyHeaders || {}\n            )) {\n              proxyHeadersAlreadySet &&= req.headers[key.toLowerCase()] == val;\n              req.headers[key.toLowerCase()] = val;\n            }\n            // If proxy headers set, this is a client navigation meant to go direct to legacy backend.\n            // Use passthru proxy in this case.\n            // ALB CANNOT be used for this. see `onBeforeRenderClient` for details\n            if (proxyHeadersAlreadySet) return;\n\n            (req.raw as RawRequestExtendedWithProxy)._bfproxy = true;\n            req.getLayout = pageContext.config.getLayout;\n            return;\n          }\n          default:\n            req.log.info(`bifrost: rendering page ${pageContext.pageId}`);\n            return replyWithPage(reply, pageContext);\n        }\n      }\n    },\n    replyOptions: {\n      rewriteRequestHeaders(request, headers) {\n        const fwd = forwarded(request as IncomingMessage).reverse();\n        // fwd.push(request.ip); TODO: not sure if this is needed\n        headers[\"X-Forwarded-For\"] = fwd.join(\", \");\n        headers[\"X-Forwarded-Host\"] = host.host;\n        headers[\"X-Forwarded-Proto\"] = host.protocol;\n\n        if ((request as RawRequestExtendedWithProxy)._bfproxy) {\n          // Proxying and wrapping\n\n          // Delete cache headers\n          delete headers[\"if-modified-since\"];\n          delete headers[\"if-none-match\"];\n          delete headers[\"if-unmodified-since\"];\n          delete headers[\"if-none-match\"];\n          delete headers[\"if-range\"];\n        }\n        return headers;\n      },\n      async onResponse(req, reply: FastifyReply<RawServerBase>, res) {\n        if ([301, 302, 303, 307, 308].includes(reply.statusCode)) {\n          const location = reply.getHeader(\"location\") as string;\n          console.log(location);\n          if (location) {\n            const url = new URL(location, host.href);\n            if (url.host === upstream.host || url.host === host.host) {\n              // rewrite redirect on upstream's host to the proxy host\n              url.host = host.host;\n              url.protocol = host.protocol;\n            }\n            reply.header(\"location\", url);\n            return reply.send(res);\n          }\n        }\n\n        const proxyLayoutInfo = req.getLayout?.(reply.getHeaders());\n        if (!proxyLayoutInfo) {\n          return reply.send(res);\n        }\n\n        const html = await streamToString(res);\n\n        const { bodyAttributes, bodyInnerHtml, headInnerHtml } =\n          extractDomElements(html);\n\n        if (!bodyInnerHtml || !headInnerHtml) {\n          throw new Error(\"Proxy failed\");\n        }\n\n        const pageContextInit = {\n          urlOriginal: req.url,\n          // Critical that we don't set any passToClient values in pageContextInit\n          // If we do, Vike re-requests pageContext on client navigation. This breaks wrapped proxy.\n          _wrappedServerOnly: {\n            bodyAttributes,\n            bodyInnerHtml,\n            headInnerHtml,\n            proxyLayoutInfo,\n          } satisfies WrappedServerOnly,\n        };\n        const pageContext = await renderPage(pageContextInit);\n        return replyWithPage(reply, pageContext);\n      },\n    },\n  });\n};\n","import { Parser } from \"htmlparser2\";\nimport { DomHandler, type Element } from \"domhandler\";\nimport { findOne } from \"domutils\";\nimport render from \"dom-serializer\";\n\nfunction getInnerHtml(element: Element): string {\n  return element.children.map((c) => render(c)).join(\"\");\n}\n\nexport function extractDomElements(html: string): {\n  bodyInnerHtml: string | null;\n  headInnerHtml: string | null;\n  bodyAttributes: Record<string, string>;\n} {\n  let bodyInnerHtml: string | null = null;\n  let headInnerHtml: string | null = null;\n  let bodyAttributes: Record<string, string> = {};\n  const handler = new DomHandler((error, dom) => {\n    if (!error) {\n      const body = findOne((elem) => elem.name === \"body\", dom);\n      const head = findOne((elem) => elem.name === \"head\", dom);\n      if (body && head) {\n        bodyAttributes = body.attribs;\n        bodyInnerHtml = getInnerHtml(body);\n        headInnerHtml = getInnerHtml(head);\n      }\n    }\n  });\n  const parser = new Parser(handler);\n  parser.write(html);\n  parser.end();\n  return { bodyInnerHtml, headInnerHtml, bodyAttributes };\n}\n"],"mappings":";AAGA,OAAO,WAAW;AAClB,OAAO,aAAa;AACpB,OAAO,eAAe;AAItB,SAAS,kBAAkB;;;ACT3B,SAAS,cAAc;AACvB,SAAS,kBAAgC;AACzC,SAAS,eAAe;AACxB,OAAO,YAAY;AAEnB,SAAS,aAAa,SAA0B;AAC9C,SAAO,QAAQ,SAAS,IAAI,CAAC,MAAM,OAAO,CAAC,CAAC,EAAE,KAAK,EAAE;AACvD;AAEO,SAAS,mBAAmB,MAIjC;AACA,MAAI,gBAA+B;AACnC,MAAI,gBAA+B;AACnC,MAAI,iBAAyC,CAAC;AAC9C,QAAM,UAAU,IAAI,WAAW,CAAC,OAAO,QAAQ;AAC7C,QAAI,CAAC,OAAO;AACV,YAAM,OAAO,QAAQ,CAAC,SAAS,KAAK,SAAS,QAAQ,GAAG;AACxD,YAAM,OAAO,QAAQ,CAAC,SAAS,KAAK,SAAS,QAAQ,GAAG;AACxD,UAAI,QAAQ,MAAM;AAChB,yBAAiB,KAAK;AACtB,wBAAgB,aAAa,IAAI;AACjC,wBAAgB,aAAa,IAAI;AAAA,MACnC;AAAA,IACF;AAAA,EACF,CAAC;AACD,QAAM,SAAS,IAAI,OAAO,OAAO;AACjC,SAAO,MAAM,IAAI;AACjB,SAAO,IAAI;AACX,SAAO,EAAE,eAAe,eAAe,eAAe;AACxD;;;ADMA,SAAS,eAAe,QAAmC;AACzD,QAAM,SAAmB,CAAC;AAC1B,SAAO,IAAI,QAAQ,CAAC,SAAS,WAAW;AACtC,WAAO,GAAG,QAAQ,CAAC,UAAU,OAAO,KAAK,OAAO,KAAK,KAAK,CAAC,CAAC;AAC5D,WAAO,GAAG,SAAS,CAAC,QAAQ,OAAO,GAAG,CAAC;AACvC,WAAO,GAAG,OAAO,MAAM,QAAQ,OAAO,OAAO,MAAM,EAAE,SAAS,MAAM,CAAC,CAAC;AAAA,EACxE,CAAC;AACH;AAaO,IAAM,kBAET,OAAO,SAAS,EAAE,UAAU,MAAM,SAAS,qBAAqB,MAAM;AACxE,iBAAe,cACb,OACA,aACuB;AACvB,UAAM,EAAE,aAAa,IAAI;AAEzB,QACE,WACA,cAAc,eAAe,OAC7B,YAAY,qBACZ;AACA,cAAQ,YAAY,qBAAqB,WAAW;AAAA,IACtD;AAEA,QAAI,CAAC,cAAc;AACjB,aAAO,MAAM,KAAK,GAAG,EAAE,KAAK,WAAW,EAAE,KAAK,WAAW;AAAA,IAC3D;AAEA,UAAM,EAAE,MAAM,YAAY,QAAQ,IAAI;AACtC,WAAO,MACJ,OAAO,UAAU,EACjB,QAAQ,OAAO,YAAY,OAAO,CAAC,EACnC,KAAK,IAAI;AAAA,EACd;AACA,QAAM,QAAQ,SAAS,OAAO;AAC9B,UAAQ,gBAAgB,iBAAiB,IAAI;AAC7C,UAAQ,gBAAgB,aAAa,IAAI;AACzC,QAAM,QAAQ,SAAS,OAAO;AAAA,IAC5B,UAAU,SAAS;AAAA,IACnB,WAAW;AAAA,IACX,MAAM,WAAW,KAAK,OAAO;AAC3B,WACG,IAAI,WAAW,SAAS,IAAI,WAAW,WACxC,IAAI,QAAQ,EAAE,KAAK,CAAC,MAAM,CAAC,MAAM,QACjC;AACA,cAAM,kBAAkB;AAAA,UACtB,aAAa,IAAI;AAAA,UACjB,GAAI,uBAAuB,MAAM,qBAAqB,GAAG,IAAI,CAAC;AAAA,QAChE;AAEA,cAAM,cAAc,MAAM,WAAW,eAAe;AAGpD,YAAI,gBAAgB,YAAY;AAEhC,cAAM,YAAY,YAAY,QAAQ;AAEtC,gBAAQ,WAAW;AAAA,UACjB,KAAK,YAAY;AACf,gBAAI,IAAI,KAAK,oCAAoC;AACjD;AAAA,UACF;AAAA,UACA,KAAK,WAAW;AACd,gBAAI,IAAI,KAAK,mDAAmD;AAChE,gBAAI,CAAC,CAAC,YAAY,wBAAwB;AAExC,kBAAI,IAAI;AAAA,gBACN;AAAA,cACF;AACA,qBAAO,MAAM;AAAA,gBACX,IAAI,IAAI,QAAQ,2BAA2B,EAAE;AAAA,cAC/C;AAAA,YACF;AACA,gBAAI,CAAC,YAAY,QAAQ,WAAW;AAClC,kBAAI,IAAI;AAAA,gBACN;AAAA,cACF;AACA;AAAA,YACF;AAEA,gBAAI,yBAAyB;AAC7B,uBAAW,CAAC,KAAK,GAAG,KAAK,OAAO;AAAA,cAC9B,YAAY,QAAQ,gBAAgB,CAAC;AAAA,YACvC,GAAG;AACD,kEAA2B,IAAI,QAAQ,IAAI,YAAY,CAAC,KAAK;AAC7D,kBAAI,QAAQ,IAAI,YAAY,CAAC,IAAI;AAAA,YACnC;AAIA,gBAAI;AAAwB;AAE5B,YAAC,IAAI,IAAoC,WAAW;AACpD,gBAAI,YAAY,YAAY,OAAO;AACnC;AAAA,UACF;AAAA,UACA;AACE,gBAAI,IAAI,KAAK,2BAA2B,YAAY,QAAQ;AAC5D,mBAAO,cAAc,OAAO,WAAW;AAAA,QAC3C;AAAA,MACF;AAAA,IACF;AAAA,IACA,cAAc;AAAA,MACZ,sBAAsB,SAAS,SAAS;AACtC,cAAM,MAAM,UAAU,OAA0B,EAAE,QAAQ;AAE1D,gBAAQ,iBAAiB,IAAI,IAAI,KAAK,IAAI;AAC1C,gBAAQ,kBAAkB,IAAI,KAAK;AACnC,gBAAQ,mBAAmB,IAAI,KAAK;AAEpC,YAAK,QAAwC,UAAU;AAIrD,iBAAO,QAAQ,mBAAmB;AAClC,iBAAO,QAAQ,eAAe;AAC9B,iBAAO,QAAQ,qBAAqB;AACpC,iBAAO,QAAQ,eAAe;AAC9B,iBAAO,QAAQ,UAAU;AAAA,QAC3B;AACA,eAAO;AAAA,MACT;AAAA,MACA,MAAM,WAAW,KAAK,OAAoC,KAAK;AAC7D,YAAI,CAAC,KAAK,KAAK,KAAK,KAAK,GAAG,EAAE,SAAS,MAAM,UAAU,GAAG;AACxD,gBAAM,WAAW,MAAM,UAAU,UAAU;AAC3C,kBAAQ,IAAI,QAAQ;AACpB,cAAI,UAAU;AACZ,kBAAM,MAAM,IAAI,IAAI,UAAU,KAAK,IAAI;AACvC,gBAAI,IAAI,SAAS,SAAS,QAAQ,IAAI,SAAS,KAAK,MAAM;AAExD,kBAAI,OAAO,KAAK;AAChB,kBAAI,WAAW,KAAK;AAAA,YACtB;AACA,kBAAM,OAAO,YAAY,GAAG;AAC5B,mBAAO,MAAM,KAAK,GAAG;AAAA,UACvB;AAAA,QACF;AAEA,cAAM,kBAAkB,IAAI,YAAY,MAAM,WAAW,CAAC;AAC1D,YAAI,CAAC,iBAAiB;AACpB,iBAAO,MAAM,KAAK,GAAG;AAAA,QACvB;AAEA,cAAM,OAAO,MAAM,eAAe,GAAG;AAErC,cAAM,EAAE,gBAAgB,eAAe,cAAc,IACnD,mBAAmB,IAAI;AAEzB,YAAI,CAAC,iBAAiB,CAAC,eAAe;AACpC,gBAAM,IAAI,MAAM,cAAc;AAAA,QAChC;AAEA,cAAM,kBAAkB;AAAA,UACtB,aAAa,IAAI;AAAA;AAAA;AAAA,UAGjB,oBAAoB;AAAA,YAClB;AAAA,YACA;AAAA,YACA;AAAA,YACA;AAAA,UACF;AAAA,QACF;AACA,cAAM,cAAc,MAAM,WAAW,eAAe;AACpD,eAAO,cAAc,OAAO,WAAW;AAAA,MACzC;AAAA,IACF;AAAA,EACF,CAAC;AACH;","names":[]}
+{"version":3,"sources":["../index.ts","../lib/extractDomElements.ts"],"sourcesContent":["// Note that this file isn't processed by Vite, see https://github.com/brillout/vike/issues/562\nimport { FastifyReply, RawServerBase, FastifyPluginAsync } from \"fastify\";\nimport { FastifyRequest, RequestGenericInterface } from \"fastify/types/request\";\nimport proxy from \"@fastify/http-proxy\";\nimport accepts from \"@fastify/accepts\";\nimport forwarded from \"@fastify/forwarded\";\nimport type { GetLayout, WrappedServerOnly } from \"@alignable/bifrost/config\";\nimport { PassThrough, Writable } from \"stream\";\nimport { IncomingMessage } from \"http\";\nimport { renderPage } from \"vike/server\";\nimport { PageContextServer } from \"vike/types\";\nimport { extractDomElements } from \"./lib/extractDomElements\";\n\ntype RenderedPageContext = Awaited<\n  ReturnType<\n    typeof renderPage<\n      {\n        isClientSideNavigation?: boolean;\n      },\n      { urlOriginal: string }\n    >\n  >\n>;\n\ndeclare module \"fastify\" {\n  interface FastifyRequest {\n    bifrostPageId?: string | null;\n    vikePageContext?: Partial<PageContextServer> | null;\n    getLayout: GetLayout;\n  }\n}\n\ntype RawRequestExtendedWithProxy = FastifyRequest<\n  RequestGenericInterface,\n  RawServerBase\n>[\"raw\"] & {\n  _bfproxy?: boolean;\n};\n\nfunction streamToString(stream: Writable): Promise<string> {\n  const chunks: Buffer[] = [];\n  return new Promise((resolve, reject) => {\n    stream.on(\"data\", (chunk) => chunks.push(Buffer.from(chunk)));\n    stream.on(\"error\", (err) => reject(err));\n    stream.on(\"end\", () => resolve(Buffer.concat(chunks).toString(\"utf8\")));\n  });\n}\n\ninterface ViteProxyPluginOptions {\n  upstream: URL;\n  host: URL;\n  onError?: (error: any, pageContext: RenderedPageContext) => void;\n  buildPageContextInit?: (\n    req: FastifyRequest\n  ) => Promise<Partial<Omit<PageContextServer, \"headers\">>>;\n}\n/**\n * Fastify plugin that wraps @fasitfy/http-proxy to proxy Rails/Turbolinks server into a vike site.\n */\nexport const viteProxyPlugin: FastifyPluginAsync<\n  ViteProxyPluginOptions\n> = async (fastify, { upstream, host, onError, buildPageContextInit }) => {\n  async function replyWithPage(\n    reply: FastifyReply<RawServerBase>,\n    pageContext: RenderedPageContext\n  ): Promise<FastifyReply> {\n    const { httpResponse } = pageContext;\n\n    if (\n      onError &&\n      httpResponse?.statusCode === 500 &&\n      pageContext.errorWhileRendering\n    ) {\n      onError(pageContext.errorWhileRendering, pageContext);\n    }\n\n    if (!httpResponse) {\n      return reply.code(404).type(\"text/html\").send(\"Not Found\");\n    }\n\n    const { pipe, statusCode, headers } = httpResponse;\n    const stream = new PassThrough();\n    pipe(stream);\n    return reply\n      .status(statusCode)\n      .headers(Object.fromEntries(headers))\n      .send(stream);\n  }\n  await fastify.register(accepts);\n  fastify.decorateRequest(\"bifrostPageId\", null);\n  fastify.decorateRequest(\"vikePageContext\", null);\n  fastify.decorateRequest(\"getLayout\", null);\n  await fastify.register(proxy, {\n    upstream: upstream.href,\n    websocket: true,\n    async preHandler(req, reply) {\n      if (\n        (req.method === \"GET\" || req.method === \"HEAD\") &&\n        req.accepts().type([\"html\"]) === \"html\"\n      ) {\n        const pageContextInit = {\n          urlOriginal: req.url,\n          headersOriginal: req.headers,\n          ...(buildPageContextInit ? await buildPageContextInit(req) : {}),\n        };\n\n        const pageContext = await renderPage(pageContextInit);\n\n        // this does not handle getting the original pageId when errors are thrown: https://github.com/vikejs/vike/issues/1112\n        req.bifrostPageId = pageContext.pageId;\n        req.vikePageContext = pageContext;\n\n        const proxyMode = pageContext.config?.proxyMode;\n\n        switch (proxyMode) {\n          case \"passthru\": {\n            req.log.info(`bifrost: passthru proxy to backend`);\n            return;\n          }\n          case \"wrapped\": {\n            req.log.info(`bifrost: proxy route matched, proxying to backend`);\n            if (!!pageContext.isClientSideNavigation) {\n              // This should never happen because wrapped proxy routes have no onBeforeRender. onRenderClient should make a request to the legacy backend.\n              req.log.error(\n                \"Wrapped proxy route is requesting index.pageContext.json. Something is wrong with the client.\"\n              );\n              return reply.redirect(\n                req.url.replace(\"/index.pageContext.json\", \"\")\n              );\n            }\n            if (!pageContext.config?.getLayout) {\n              req.log.error(\n                \"Config missing getLayout on wrapped route! Falling back to passthru proxy\"\n              );\n              return;\n            }\n\n            let proxyHeadersAlreadySet = true;\n            for (const [key, val] of Object.entries(\n              pageContext.config?.proxyHeaders || {}\n            )) {\n              proxyHeadersAlreadySet &&= req.headers[key.toLowerCase()] == val;\n              req.headers[key.toLowerCase()] = val;\n            }\n            // If proxy headers set, this is a client navigation meant to go direct to legacy backend.\n            // Use passthru proxy in this case.\n            // ALB CANNOT be used for this. see `onBeforeRenderClient` for details\n            if (proxyHeadersAlreadySet) return;\n\n            (req.raw as RawRequestExtendedWithProxy)._bfproxy = true;\n            req.getLayout = pageContext.config.getLayout;\n            return;\n          }\n          default:\n            req.log.info(`bifrost: rendering page ${pageContext.pageId}`);\n            return replyWithPage(reply, pageContext);\n        }\n      }\n    },\n    replyOptions: {\n      rewriteRequestHeaders(request, headers) {\n        const fwd = forwarded(request as IncomingMessage).reverse();\n        // fwd.push(request.ip); TODO: not sure if this is needed\n        headers[\"X-Forwarded-For\"] = fwd.join(\", \");\n        headers[\"X-Forwarded-Host\"] = host.host;\n        headers[\"X-Forwarded-Proto\"] = host.protocol;\n\n        if ((request as RawRequestExtendedWithProxy)._bfproxy) {\n          // Proxying and wrapping\n\n          // Delete cache headers\n          delete headers[\"if-modified-since\"];\n          delete headers[\"if-none-match\"];\n          delete headers[\"if-unmodified-since\"];\n          delete headers[\"if-none-match\"];\n          delete headers[\"if-range\"];\n        }\n        return headers;\n      },\n      async onResponse(req, reply: FastifyReply<RawServerBase>, res) {\n        if ([301, 302, 303, 307, 308].includes(reply.statusCode)) {\n          const location = reply.getHeader(\"location\") as string;\n          if (location) {\n            const url = new URL(location, host.href);\n            if (url.host === upstream.host || url.host === host.host) {\n              // rewrite redirect on upstream's host to the proxy host\n              url.host = host.host;\n              url.protocol = host.protocol;\n            }\n            reply.header(\"location\", url);\n            return reply.send(res);\n          }\n        }\n\n        const proxyLayoutInfo = req.getLayout?.(reply.getHeaders());\n        if (!proxyLayoutInfo) {\n          return reply.send(res);\n        }\n\n        const html = await streamToString(res);\n\n        const { bodyAttributes, bodyInnerHtml, headInnerHtml } =\n          extractDomElements(html);\n\n        if (!bodyInnerHtml || !headInnerHtml) {\n          throw new Error(\"Proxy failed\");\n        }\n\n        const pageContextInit = {\n          urlOriginal: req.url,\n          headersOriginal: req.headers,\n          // Critical that we don't set any passToClient values in pageContextInit\n          // If we do, Vike re-requests pageContext on client navigation. This breaks wrapped proxy.\n          _wrappedServerOnly: {\n            bodyAttributes,\n            bodyInnerHtml,\n            headInnerHtml,\n            proxyLayoutInfo,\n          } satisfies WrappedServerOnly,\n        };\n        const pageContext = await renderPage(pageContextInit);\n        return replyWithPage(reply, pageContext);\n      },\n    },\n  });\n};\n","import { Parser } from \"htmlparser2\";\nimport { DomHandler, type Element } from \"domhandler\";\nimport { findOne } from \"domutils\";\nimport render from \"dom-serializer\";\n\nfunction getInnerHtml(element: Element): string {\n  return element.children.map((c) => render(c)).join(\"\");\n}\n\nexport function extractDomElements(html: string): {\n  bodyInnerHtml: string | null;\n  headInnerHtml: string | null;\n  bodyAttributes: Record<string, string>;\n} {\n  let bodyInnerHtml: string | null = null;\n  let headInnerHtml: string | null = null;\n  let bodyAttributes: Record<string, string> = {};\n  const handler = new DomHandler((error, dom) => {\n    if (!error) {\n      const body = findOne((elem) => elem.name === \"body\", dom);\n      const head = findOne((elem) => elem.name === \"head\", dom);\n      if (body && head) {\n        bodyAttributes = body.attribs;\n        bodyInnerHtml = getInnerHtml(body);\n        headInnerHtml = getInnerHtml(head);\n      }\n    }\n  });\n  const parser = new Parser(handler);\n  parser.write(html);\n  parser.end();\n  return { bodyInnerHtml, headInnerHtml, bodyAttributes };\n}\n"],"mappings":";AAGA,OAAO,WAAW;AAClB,OAAO,aAAa;AACpB,OAAO,eAAe;AAEtB,SAAS,mBAA6B;AAEtC,SAAS,kBAAkB;;;ACT3B,SAAS,cAAc;AACvB,SAAS,kBAAgC;AACzC,SAAS,eAAe;AACxB,OAAO,YAAY;AAEnB,SAAS,aAAa,SAA0B;AAC9C,SAAO,QAAQ,SAAS,IAAI,CAAC,MAAM,OAAO,CAAC,CAAC,EAAE,KAAK,EAAE;AACvD;AAEO,SAAS,mBAAmB,MAIjC;AACA,MAAI,gBAA+B;AACnC,MAAI,gBAA+B;AACnC,MAAI,iBAAyC,CAAC;AAC9C,QAAM,UAAU,IAAI,WAAW,CAAC,OAAO,QAAQ;AAC7C,QAAI,CAAC,OAAO;AACV,YAAM,OAAO,QAAQ,CAAC,SAAS,KAAK,SAAS,QAAQ,GAAG;AACxD,YAAM,OAAO,QAAQ,CAAC,SAAS,KAAK,SAAS,QAAQ,GAAG;AACxD,UAAI,QAAQ,MAAM;AAChB,yBAAiB,KAAK;AACtB,wBAAgB,aAAa,IAAI;AACjC,wBAAgB,aAAa,IAAI;AAAA,MACnC;AAAA,IACF;AAAA,EACF,CAAC;AACD,QAAM,SAAS,IAAI,OAAO,OAAO;AACjC,SAAO,MAAM,IAAI;AACjB,SAAO,IAAI;AACX,SAAO,EAAE,eAAe,eAAe,eAAe;AACxD;;;ADOA,SAAS,eAAe,QAAmC;AACzD,QAAM,SAAmB,CAAC;AAC1B,SAAO,IAAI,QAAQ,CAAC,SAAS,WAAW;AACtC,WAAO,GAAG,QAAQ,CAAC,UAAU,OAAO,KAAK,OAAO,KAAK,KAAK,CAAC,CAAC;AAC5D,WAAO,GAAG,SAAS,CAAC,QAAQ,OAAO,GAAG,CAAC;AACvC,WAAO,GAAG,OAAO,MAAM,QAAQ,OAAO,OAAO,MAAM,EAAE,SAAS,MAAM,CAAC,CAAC;AAAA,EACxE,CAAC;AACH;AAaO,IAAM,kBAET,OAAO,SAAS,EAAE,UAAU,MAAM,SAAS,qBAAqB,MAAM;AACxE,iBAAe,cACb,OACA,aACuB;AACvB,UAAM,EAAE,aAAa,IAAI;AAEzB,QACE,WACA,cAAc,eAAe,OAC7B,YAAY,qBACZ;AACA,cAAQ,YAAY,qBAAqB,WAAW;AAAA,IACtD;AAEA,QAAI,CAAC,cAAc;AACjB,aAAO,MAAM,KAAK,GAAG,EAAE,KAAK,WAAW,EAAE,KAAK,WAAW;AAAA,IAC3D;AAEA,UAAM,EAAE,MAAM,YAAY,QAAQ,IAAI;AACtC,UAAM,SAAS,IAAI,YAAY;AAC/B,SAAK,MAAM;AACX,WAAO,MACJ,OAAO,UAAU,EACjB,QAAQ,OAAO,YAAY,OAAO,CAAC,EACnC,KAAK,MAAM;AAAA,EAChB;AACA,QAAM,QAAQ,SAAS,OAAO;AAC9B,UAAQ,gBAAgB,iBAAiB,IAAI;AAC7C,UAAQ,gBAAgB,mBAAmB,IAAI;AAC/C,UAAQ,gBAAgB,aAAa,IAAI;AACzC,QAAM,QAAQ,SAAS,OAAO;AAAA,IAC5B,UAAU,SAAS;AAAA,IACnB,WAAW;AAAA,IACX,MAAM,WAAW,KAAK,OAAO;AAC3B,WACG,IAAI,WAAW,SAAS,IAAI,WAAW,WACxC,IAAI,QAAQ,EAAE,KAAK,CAAC,MAAM,CAAC,MAAM,QACjC;AACA,cAAM,kBAAkB;AAAA,UACtB,aAAa,IAAI;AAAA,UACjB,iBAAiB,IAAI;AAAA,UACrB,GAAI,uBAAuB,MAAM,qBAAqB,GAAG,IAAI,CAAC;AAAA,QAChE;AAEA,cAAM,cAAc,MAAM,WAAW,eAAe;AAGpD,YAAI,gBAAgB,YAAY;AAChC,YAAI,kBAAkB;AAEtB,cAAM,YAAY,YAAY,QAAQ;AAEtC,gBAAQ,WAAW;AAAA,UACjB,KAAK,YAAY;AACf,gBAAI,IAAI,KAAK,oCAAoC;AACjD;AAAA,UACF;AAAA,UACA,KAAK,WAAW;AACd,gBAAI,IAAI,KAAK,mDAAmD;AAChE,gBAAI,CAAC,CAAC,YAAY,wBAAwB;AAExC,kBAAI,IAAI;AAAA,gBACN;AAAA,cACF;AACA,qBAAO,MAAM;AAAA,gBACX,IAAI,IAAI,QAAQ,2BAA2B,EAAE;AAAA,cAC/C;AAAA,YACF;AACA,gBAAI,CAAC,YAAY,QAAQ,WAAW;AAClC,kBAAI,IAAI;AAAA,gBACN;AAAA,cACF;AACA;AAAA,YACF;AAEA,gBAAI,yBAAyB;AAC7B,uBAAW,CAAC,KAAK,GAAG,KAAK,OAAO;AAAA,cAC9B,YAAY,QAAQ,gBAAgB,CAAC;AAAA,YACvC,GAAG;AACD,kEAA2B,IAAI,QAAQ,IAAI,YAAY,CAAC,KAAK;AAC7D,kBAAI,QAAQ,IAAI,YAAY,CAAC,IAAI;AAAA,YACnC;AAIA,gBAAI;AAAwB;AAE5B,YAAC,IAAI,IAAoC,WAAW;AACpD,gBAAI,YAAY,YAAY,OAAO;AACnC;AAAA,UACF;AAAA,UACA;AACE,gBAAI,IAAI,KAAK,2BAA2B,YAAY,QAAQ;AAC5D,mBAAO,cAAc,OAAO,WAAW;AAAA,QAC3C;AAAA,MACF;AAAA,IACF;AAAA,IACA,cAAc;AAAA,MACZ,sBAAsB,SAAS,SAAS;AACtC,cAAM,MAAM,UAAU,OAA0B,EAAE,QAAQ;AAE1D,gBAAQ,iBAAiB,IAAI,IAAI,KAAK,IAAI;AAC1C,gBAAQ,kBAAkB,IAAI,KAAK;AACnC,gBAAQ,mBAAmB,IAAI,KAAK;AAEpC,YAAK,QAAwC,UAAU;AAIrD,iBAAO,QAAQ,mBAAmB;AAClC,iBAAO,QAAQ,eAAe;AAC9B,iBAAO,QAAQ,qBAAqB;AACpC,iBAAO,QAAQ,eAAe;AAC9B,iBAAO,QAAQ,UAAU;AAAA,QAC3B;AACA,eAAO;AAAA,MACT;AAAA,MACA,MAAM,WAAW,KAAK,OAAoC,KAAK;AAC7D,YAAI,CAAC,KAAK,KAAK,KAAK,KAAK,GAAG,EAAE,SAAS,MAAM,UAAU,GAAG;AACxD,gBAAM,WAAW,MAAM,UAAU,UAAU;AAC3C,cAAI,UAAU;AACZ,kBAAM,MAAM,IAAI,IAAI,UAAU,KAAK,IAAI;AACvC,gBAAI,IAAI,SAAS,SAAS,QAAQ,IAAI,SAAS,KAAK,MAAM;AAExD,kBAAI,OAAO,KAAK;AAChB,kBAAI,WAAW,KAAK;AAAA,YACtB;AACA,kBAAM,OAAO,YAAY,GAAG;AAC5B,mBAAO,MAAM,KAAK,GAAG;AAAA,UACvB;AAAA,QACF;AAEA,cAAM,kBAAkB,IAAI,YAAY,MAAM,WAAW,CAAC;AAC1D,YAAI,CAAC,iBAAiB;AACpB,iBAAO,MAAM,KAAK,GAAG;AAAA,QACvB;AAEA,cAAM,OAAO,MAAM,eAAe,GAAG;AAErC,cAAM,EAAE,gBAAgB,eAAe,cAAc,IACnD,mBAAmB,IAAI;AAEzB,YAAI,CAAC,iBAAiB,CAAC,eAAe;AACpC,gBAAM,IAAI,MAAM,cAAc;AAAA,QAChC;AAEA,cAAM,kBAAkB;AAAA,UACtB,aAAa,IAAI;AAAA,UACjB,iBAAiB,IAAI;AAAA;AAAA;AAAA,UAGrB,oBAAoB;AAAA,YAClB;AAAA,YACA;AAAA,YACA;AAAA,YACA;AAAA,UACF;AAAA,QACF;AACA,cAAM,cAAc,MAAM,WAAW,eAAe;AACpD,eAAO,cAAc,OAAO,WAAW;AAAA,MACzC;AAAA,IACF;AAAA,EACF,CAAC;AACH;","names":[]}

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@alignable/bifrost-fastify",
   "repository": "https://github.com/Alignable/bifrost.git",
-  "version": "1.0.0",
+  "version": "1.0.2",
   "type": "module",
   "main": "./dist/index.js",
   "types": "./dist/index.d.ts",
@@ -21,15 +21,15 @@
     "vite": "^6.3.5"
   },
   "peerDependencies": {
-    "@alignable/bifrost": "1.0.0",
+    "@alignable/bifrost": "1.0.2",
     "fastify": "^4.9.2",
     "vike": "0.4.247"
   },
   "devDependencies": {
     "@types/connect": "^3.4.35",
     "@types/node": "^18.11.9",
-    "@types/react": "^18.0.8",
-    "@types/react-dom": "^18.0.3",
+    "@types/react": "^19.0.0",
+    "@types/react-dom": "^19.0.0",
     "react": "^19.0.0",
     "react-dom": "^19.0.0",
     "tsup": "^6.7.0",