@chidchanun/bcp 0.1.10 → 0.1.12

package/packages/server/src/form-action-transport.ts

@@ -0,0 +1,271 @@
+import {
+    copySetCookieHeaders,
+} from "./navigation-response.js";
+
+import {
+    applyResponseCookies,
+} from "./request-context.js";
+
+import type {
+    PageActionExecution,
+} from "./form-action.js";
+
+export interface CompletedPageActionData {
+    kind: "data";
+    data: unknown;
+    setCookies: string[];
+}
+
+export interface CompletedPageActionResponse {
+    kind: "response";
+    response: Response;
+}
+
+export type CompletedPageAction =
+    | CompletedPageActionData
+    | CompletedPageActionResponse;
+
+export interface FormActionDataPayload {
+    kind: "data";
+    data: unknown;
+}
+
+export interface FormActionRedirectPayload {
+    kind: "redirect";
+    redirect: {
+        location: string;
+        status: number;
+    };
+}
+
+export interface FormActionErrorPayload {
+    kind: "error";
+    error: {
+        message: string;
+    };
+}
+
+export function completePageActionExecution(
+    execution: PageActionExecution
+): CompletedPageAction {
+    if (
+        execution.response
+    ) {
+        return {
+            kind:
+                "response",
+            response:
+                applyResponseCookies(
+                    execution.response
+                ),
+        };
+    }
+
+    const carrier =
+        applyResponseCookies(
+            new Response(
+                null,
+                {
+                    status: 204,
+                }
+            )
+        );
+
+    return {
+        kind:
+            "data",
+        data:
+            execution.data,
+        setCookies:
+            carrier.headers
+                .getSetCookie(),
+    };
+}
+
+export function createFormActionTransportResponse(
+    result: CompletedPageAction,
+    target: URL,
+    enhanced: boolean
+): Response {
+    if (
+        result.kind ===
+        "response"
+    ) {
+        if (enhanced) {
+            const redirect =
+                createRedirectPayloadResponse(
+                    result.response
+                );
+
+            if (redirect) {
+                return redirect;
+            }
+        }
+
+        return result.response;
+    }
+
+    if (enhanced) {
+        const headers =
+            new Headers();
+
+        for (
+            const cookie
+            of result.setCookies
+        ) {
+            headers.append(
+                "Set-Cookie",
+                cookie
+            );
+        }
+
+        return createFormActionJsonResponse(
+            {
+                kind:
+                    "data",
+                data:
+                    result.data,
+            } satisfies FormActionDataPayload,
+            {
+                status: 200,
+                headers,
+            }
+        );
+    }
+
+    const headers =
+        new Headers({
+            Location:
+                `${target.pathname}${target.search}${target.hash}`,
+        });
+
+    for (
+        const cookie
+        of result.setCookies
+    ) {
+        headers.append(
+            "Set-Cookie",
+            cookie
+        );
+    }
+
+    return new Response(
+        null,
+        {
+            status: 303,
+            headers,
+        }
+    );
+}
+
+export function createFormActionErrorResponse(
+    message: string,
+    status = 500
+): Response {
+    return createFormActionJsonResponse(
+        {
+            kind:
+                "error",
+            error: {
+                message,
+            },
+        } satisfies FormActionErrorPayload,
+        {
+            status,
+        }
+    );
+}
+
+export function createFormActionJsonResponse(
+    value: unknown,
+    init: ResponseInit = {}
+): Response {
+    const headers =
+        new Headers(
+            init.headers
+        );
+
+    headers.set(
+        "Cache-Control",
+        "no-store"
+    );
+    headers.set(
+        "X-Content-Type-Options",
+        "nosniff"
+    );
+
+    return Response.json(
+        value,
+        {
+            ...init,
+            headers,
+        }
+    );
+}
+
+export function isEnhancedFormActionRequest(
+    headers: Headers
+): boolean {
+    return (
+        headers.get(
+            "x-bcp-action"
+        ) === "1"
+    );
+}
+
+function createRedirectPayloadResponse(
+    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 createFormActionJsonResponse(
+        {
+            kind:
+                "redirect",
+            redirect: {
+                location,
+                status:
+                    response.status,
+            },
+        } satisfies FormActionRedirectPayload,
+        {
+            status: 200,
+            headers,
+        }
+    );
+}
+
+function isRedirectStatus(
+    status: number
+): boolean {
+    return (
+        status === 301 ||
+        status === 302 ||
+        status === 303 ||
+        status === 307 ||
+        status === 308
+    );
+}