@chidchanun/bcp 0.1.8 → 0.1.10

package/packages/server/src/navigation-response.ts

@@ -0,0 +1,284 @@
+import * as http from "node:http";
+
+import {
+    appendVary,
+    compressDynamicContent,
+    selectContentEncoding,
+} from "./compression.js";
+
+export interface NavigationRedirectPayload {
+    redirect: {
+        location: string;
+        status: number;
+    };
+}
+
+export function createNavigationJsonResponse(
+    value: unknown,
+    init: ResponseInit = {}
+): Response {
+    const headers =
+        new Headers(
+            init.headers
+        );
+
+    if (
+        !headers.has(
+            "Cache-Control"
+        )
+    ) {
+        headers.set(
+            "Cache-Control",
+            "no-store"
+        );
+    }
+
+    if (
+        !headers.has(
+            "X-Content-Type-Options"
+        )
+    ) {
+        headers.set(
+            "X-Content-Type-Options",
+            "nosniff"
+        );
+    }
+
+    return Response.json(
+        value,
+        {
+            ...init,
+            headers,
+        }
+    );
+}
+
+export function createNavigationRedirectResponse(
+    response: Response
+): Response | null {
+    if (
+        !isRedirectStatus(
+            response.status
+        )
+    ) {
+        return null;
+    }
+
+    const location =
+        response.headers.get(
+            "location"
+        );
+
+    if (!location) {
+        return null;
+    }
+
+    const headers =
+        new Headers();
+
+    copySetCookieHeaders(
+        response.headers,
+        headers
+    );
+
+    return createNavigationJsonResponse(
+        {
+            redirect: {
+                location,
+                status:
+                    response.status,
+            },
+        } satisfies NavigationRedirectPayload,
+        {
+            status:
+                200,
+            headers,
+        }
+    );
+}
+
+export async function sendNavigationWebResponse(
+    req: http.IncomingMessage,
+    res: http.ServerResponse,
+    response: Response
+): Promise<void> {
+    res.statusCode =
+        response.status;
+
+    response.headers.forEach(
+        (
+            value,
+            key
+        ) => {
+            const normalized =
+                key.toLowerCase();
+
+            if (
+                normalized ===
+                    "set-cookie" ||
+                normalized ===
+                    "content-length" ||
+                normalized ===
+                    "content-encoding"
+            ) {
+                return;
+            }
+
+            res.setHeader(
+                key,
+                value
+            );
+        }
+    );
+
+    const cookies =
+        getSetCookieHeaders(
+            response.headers
+        );
+
+    if (
+        cookies.length > 0
+    ) {
+        res.setHeader(
+            "Set-Cookie",
+            cookies
+        );
+    }
+
+    if (
+        response.status === 204 ||
+        response.status === 304
+    ) {
+        res.end();
+        return;
+    }
+
+    const rawBody =
+        Buffer.from(
+            await response.arrayBuffer()
+        );
+
+    const contentType =
+        response.headers.get(
+            "content-type"
+        ) ??
+        "application/octet-stream";
+
+    const existingEncoding =
+        response.headers.get(
+            "content-encoding"
+        );
+
+    const encoding =
+        existingEncoding
+            ? null
+            : selectContentEncoding(
+                req,
+                contentType,
+                rawBody.length
+            );
+
+    const body =
+        existingEncoding
+            ? rawBody
+            : compressDynamicContent(
+                rawBody,
+                encoding
+            );
+
+    if (existingEncoding) {
+        res.setHeader(
+            "Content-Encoding",
+            existingEncoding
+        );
+    } else if (encoding) {
+        res.setHeader(
+            "Content-Encoding",
+            encoding
+        );
+        res.setHeader(
+            "Vary",
+            appendVary(
+                res.getHeader(
+                    "Vary"
+                )?.toString(),
+                "Accept-Encoding"
+            )
+        );
+    }
+
+    res.setHeader(
+        "Content-Length",
+        body.length
+    );
+
+    if (
+        (
+            req.method ??
+            "GET"
+        ).toUpperCase() ===
+        "HEAD"
+    ) {
+        res.end();
+        return;
+    }
+
+    res.end(
+        body
+    );
+}
+
+export function copySetCookieHeaders(
+    source: Headers,
+    target: Headers
+): void {
+    for (
+        const cookie
+        of getSetCookieHeaders(
+            source
+        )
+    ) {
+        target.append(
+            "Set-Cookie",
+            cookie
+        );
+    }
+}
+
+function getSetCookieHeaders(
+    headers: Headers
+): string[] {
+    const extended =
+        headers as Headers & {
+            getSetCookie?: () =>
+                string[];
+        };
+
+    const values =
+        extended.getSetCookie?.();
+
+    if (values) {
+        return values;
+    }
+
+    const fallback =
+        headers.get(
+            "set-cookie"
+        );
+
+    return fallback
+        ? [fallback]
+        : [];
+}
+
+function isRedirectStatus(
+    status: number
+): boolean {
+    return (
+        status === 301 ||
+        status === 302 ||
+        status === 303 ||
+        status === 307 ||
+        status === 308
+    );
+}