@alignable/bifrost-fastify 1.0.10 → 1.0.11

package/dist/index.js

@@ -37,6 +37,7 @@ function extractDomElements(html) {
 // index.ts
 import { Http2ServerRequest } from "http2";
 import { text } from "stream/consumers";
+import { parse as parseContentType } from "fast-content-type-parse";
 var viteProxyPlugin = async (fastify, opts) => {
   const { upstream, host, onError, buildPageContextInit } = opts;
   async function replyWithPage(reply, pageContext) {
@@ -147,10 +148,14 @@ var viteProxyPlugin = async (fastify, opts) => {
         if (!proxyLayoutInfo) {
           return reply.send("stream" in res ? res.stream : res);
         }
+        const contentType = reply.getHeader("content-type");
+        if (!contentType || parseContentType(contentType).type !== "text/html") {
+          return reply.send("stream" in res ? res.stream : res);
+        }
         const html = await text(res.stream);
         const { bodyAttributes, bodyInnerHtml, headInnerHtml } = extractDomElements(html);
         if (!bodyInnerHtml || !headInnerHtml) {
-          throw new Error("Proxy failed");
+          return reply.send(html);
         }
         const pageContextInit = {
           urlOriginal: req.url,

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 {\n  FastifyReply,\n  RawServerBase,\n  FastifyPluginAsync,\n  RouteGenericInterface,\n} from \"fastify\";\nimport { FastifyRequest, RequestGenericInterface } from \"fastify/types/request\";\nimport proxy, { type FastifyHttpProxyOptions } 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 { renderPage } from \"vike/server\";\nimport { PageContextServer } from \"vike/types\";\nimport { extractDomElements } from \"./lib/extractDomElements\";\nimport { Http2ServerRequest } from \"http2\";\nimport { IncomingMessage } from \"http\";\nimport { text } from \"node:stream/consumers\";\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 | null;\n    customPageContextInit: Partial<Omit<PageContextServer, \"headers\">> | null;\n  }\n}\n\ntype RawRequestExtendedWithProxy = FastifyRequest<\n  RequestGenericInterface,\n  RawServerBase\n>[\"raw\"] & {\n  _bfproxy?: boolean;\n};\n\ninterface ViteProxyPluginOptions\n  extends Omit<\n    FastifyHttpProxyOptions,\n    \"upstream\" | \"preHandler\" | \"replyOptions\"\n  > {\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, opts) => {\n  const { upstream, host, onError, buildPageContextInit } = opts;\n  async function replyWithPage(\n    reply: FastifyReply<RouteGenericInterface, 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  fastify.decorateRequest(\"customPageContextInit\", null);\n  await fastify.register(proxy, {\n    ...opts,\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        req.customPageContextInit = buildPageContextInit\n          ? await buildPageContextInit(req)\n          : {};\n\n        const pageContextInit = {\n          urlOriginal: req.url,\n          headersOriginal: req.headers,\n          ...req.customPageContextInit,\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        if (!(request.raw instanceof Http2ServerRequest)) {\n          const fwd = forwarded(request.raw).reverse();\n          headers[\"X-Forwarded-For\"] = fwd.join(\", \");\n          headers[\"X-Forwarded-Host\"] = host.host;\n          headers[\"X-Forwarded-Proto\"] = host.protocol;\n        }\n\n        if ((request.raw 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, 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(\"stream\" in res ? res.stream : res);\n          }\n        }\n\n        const proxyLayoutInfo = req.getLayout?.(reply.getHeaders());\n        if (!proxyLayoutInfo) {\n          return reply.send(\"stream\" in res ? res.stream : res);\n        }\n\n        const html = await text(res.stream);\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          ...req.customPageContextInit,\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":";AAQA,OAAO,WAA6C;AACpD,OAAO,aAAa;AACpB,OAAO,eAAe;AAEtB,SAAS,mBAA6B;AACtC,SAAS,kBAAkB;;;ACb3B,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;;;ADhBA,SAAS,0BAA0B;AAEnC,SAAS,YAAY;AA4Cd,IAAM,kBAET,OAAO,SAAS,SAAS;AAC3B,QAAM,EAAE,UAAU,MAAM,SAAS,qBAAqB,IAAI;AAC1D,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,UAAQ,gBAAgB,yBAAyB,IAAI;AACrD,QAAM,QAAQ,SAAS,OAAO;AAAA,IAC5B,GAAG;AAAA,IACH,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,YAAI,wBAAwB,uBACxB,MAAM,qBAAqB,GAAG,IAC9B,CAAC;AAEL,cAAM,kBAAkB;AAAA,UACtB,aAAa,IAAI;AAAA,UACjB,iBAAiB,IAAI;AAAA,UACrB,GAAG,IAAI;AAAA,QACT;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,YAAI,EAAE,QAAQ,eAAe,qBAAqB;AAChD,gBAAM,MAAM,UAAU,QAAQ,GAAG,EAAE,QAAQ;AAC3C,kBAAQ,iBAAiB,IAAI,IAAI,KAAK,IAAI;AAC1C,kBAAQ,kBAAkB,IAAI,KAAK;AACnC,kBAAQ,mBAAmB,IAAI,KAAK;AAAA,QACtC;AAEA,YAAK,QAAQ,IAAoC,UAAU;AAIzD,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,OAAO,KAAK;AAChC,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,YAAY,MAAM,IAAI,SAAS,GAAG;AAAA,UACtD;AAAA,QACF;AAEA,cAAM,kBAAkB,IAAI,YAAY,MAAM,WAAW,CAAC;AAC1D,YAAI,CAAC,iBAAiB;AACpB,iBAAO,MAAM,KAAK,YAAY,MAAM,IAAI,SAAS,GAAG;AAAA,QACtD;AAEA,cAAM,OAAO,MAAM,KAAK,IAAI,MAAM;AAElC,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,UACA,GAAG,IAAI;AAAA,QACT;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 {\n  FastifyReply,\n  RawServerBase,\n  FastifyPluginAsync,\n  RouteGenericInterface,\n} from \"fastify\";\nimport { FastifyRequest, RequestGenericInterface } from \"fastify/types/request\";\nimport proxy, { type FastifyHttpProxyOptions } 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 { renderPage } from \"vike/server\";\nimport { PageContextServer } from \"vike/types\";\nimport { extractDomElements } from \"./lib/extractDomElements\";\nimport { Http2ServerRequest } from \"http2\";\nimport { IncomingMessage } from \"http\";\nimport { text } from \"node:stream/consumers\";\nimport { parse as parseContentType } from 'fast-content-type-parse';\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 | null;\n    customPageContextInit: Partial<Omit<PageContextServer, \"headers\">> | null;\n  }\n}\n\ntype RawRequestExtendedWithProxy = FastifyRequest<\n  RequestGenericInterface,\n  RawServerBase\n>[\"raw\"] & {\n  _bfproxy?: boolean;\n};\n\ninterface ViteProxyPluginOptions\n  extends Omit<\n    FastifyHttpProxyOptions,\n    \"upstream\" | \"preHandler\" | \"replyOptions\"\n  > {\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, opts) => {\n  const { upstream, host, onError, buildPageContextInit } = opts;\n  async function replyWithPage(\n    reply: FastifyReply<RouteGenericInterface, 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  fastify.decorateRequest(\"customPageContextInit\", null);\n  await fastify.register(proxy, {\n    ...opts,\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        req.customPageContextInit = buildPageContextInit\n          ? await buildPageContextInit(req)\n          : {};\n\n        const pageContextInit = {\n          urlOriginal: req.url,\n          headersOriginal: req.headers,\n          ...req.customPageContextInit,\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        if (!(request.raw instanceof Http2ServerRequest)) {\n          const fwd = forwarded(request.raw).reverse();\n          headers[\"X-Forwarded-For\"] = fwd.join(\", \");\n          headers[\"X-Forwarded-Host\"] = host.host;\n          headers[\"X-Forwarded-Proto\"] = host.protocol;\n        }\n\n        if ((request.raw 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, 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(\"stream\" in res ? res.stream : res);\n          }\n        }\n\n        const proxyLayoutInfo = req.getLayout?.(reply.getHeaders());\n        if (!proxyLayoutInfo) {\n          return reply.send(\"stream\" in res ? res.stream : res);\n        }\n\n        const contentType = reply.getHeader(\"content-type\") as string | undefined;\n        \n        if (!contentType || parseContentType(contentType).type !== 'text/html') {\n          return reply.send(\"stream\" in res ? res.stream : res);\n        }\n\n        const html = await text(res.stream);\n\n        const { bodyAttributes, bodyInnerHtml, headInnerHtml } =\n          extractDomElements(html);\n\n        if (!bodyInnerHtml || !headInnerHtml) {\n          return reply.send(html);\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          ...req.customPageContextInit,\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":";AAQA,OAAO,WAA6C;AACpD,OAAO,aAAa;AACpB,OAAO,eAAe;AAEtB,SAAS,mBAA6B;AACtC,SAAS,kBAAkB;;;ACb3B,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;;;ADhBA,SAAS,0BAA0B;AAEnC,SAAS,YAAY;AACrB,SAAS,SAAS,wBAAwB;AA4CnC,IAAM,kBAET,OAAO,SAAS,SAAS;AAC3B,QAAM,EAAE,UAAU,MAAM,SAAS,qBAAqB,IAAI;AAC1D,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,UAAQ,gBAAgB,yBAAyB,IAAI;AACrD,QAAM,QAAQ,SAAS,OAAO;AAAA,IAC5B,GAAG;AAAA,IACH,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,YAAI,wBAAwB,uBACxB,MAAM,qBAAqB,GAAG,IAC9B,CAAC;AAEL,cAAM,kBAAkB;AAAA,UACtB,aAAa,IAAI;AAAA,UACjB,iBAAiB,IAAI;AAAA,UACrB,GAAG,IAAI;AAAA,QACT;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,YAAI,EAAE,QAAQ,eAAe,qBAAqB;AAChD,gBAAM,MAAM,UAAU,QAAQ,GAAG,EAAE,QAAQ;AAC3C,kBAAQ,iBAAiB,IAAI,IAAI,KAAK,IAAI;AAC1C,kBAAQ,kBAAkB,IAAI,KAAK;AACnC,kBAAQ,mBAAmB,IAAI,KAAK;AAAA,QACtC;AAEA,YAAK,QAAQ,IAAoC,UAAU;AAIzD,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,OAAO,KAAK;AAChC,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,YAAY,MAAM,IAAI,SAAS,GAAG;AAAA,UACtD;AAAA,QACF;AAEA,cAAM,kBAAkB,IAAI,YAAY,MAAM,WAAW,CAAC;AAC1D,YAAI,CAAC,iBAAiB;AACpB,iBAAO,MAAM,KAAK,YAAY,MAAM,IAAI,SAAS,GAAG;AAAA,QACtD;AAEA,cAAM,cAAc,MAAM,UAAU,cAAc;AAElD,YAAI,CAAC,eAAe,iBAAiB,WAAW,EAAE,SAAS,aAAa;AACtE,iBAAO,MAAM,KAAK,YAAY,MAAM,IAAI,SAAS,GAAG;AAAA,QACtD;AAEA,cAAM,OAAO,MAAM,KAAK,IAAI,MAAM;AAElC,cAAM,EAAE,gBAAgB,eAAe,cAAc,IACnD,mBAAmB,IAAI;AAEzB,YAAI,CAAC,iBAAiB,CAAC,eAAe;AACpC,iBAAO,MAAM,KAAK,IAAI;AAAA,QACxB;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,UACA,GAAG,IAAI;AAAA,QACT;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.10",
+  "version": "1.0.11",
   "type": "module",
   "main": "./dist/index.js",
   "types": "./dist/index.d.ts",
@@ -13,13 +13,14 @@
     "dom-serializer": "^2.0.0",
     "domhandler": "^5.0.3",
     "domutils": "^3.2.2",
+    "fast-content-type-parse": "^3.0.0",
     "htmlparser2": "^10.0.0"
   },
   "peerDependencies": {
     "@alignable/bifrost": "1.0.10",
     "fastify": "^5.0.0",
-    "vite": ">=6",
-    "vike": "0.4.247"
+    "vike": "0.4.247",
+    "vite": ">=6"
   },
   "devDependencies": {
     "@types/connect": "^3.4.35",