@chidchanun/bcp 0.1.9 → 0.1.11

package/packages/server/src/form-action-dev-proxy.ts

@@ -0,0 +1,601 @@
+import * as http from "node:http";
+import path from "node:path";
+
+import {
+    runWithRequestCacheContext,
+} from "../../cache/src/index.js";
+
+import {
+    matchRoute,
+    scanRoutes,
+} from "../../router/src/index.js";
+
+import {
+    executePageAction,
+    normalizePageActionMethod,
+} from "./form-action.js";
+
+import {
+    completePageActionExecution,
+    createFormActionErrorResponse,
+    createFormActionTransportResponse,
+    isEnhancedFormActionRequest,
+} from "./form-action-transport.js";
+
+import {
+    sendNavigationWebResponse,
+} from "./navigation-response.js";
+
+import {
+    runWithRequestContext,
+} from "./request-context.js";
+
+export interface FormActionDevProxyOptions {
+    port: number;
+    hostname: string;
+    upstreamPort: number;
+    upstreamHostname?: string;
+    rootDirectory: string;
+}
+
+export function createFormActionDevProxy(
+    options: FormActionDevProxyOptions
+) {
+    const upstreamHostname =
+        options.upstreamHostname ??
+        "127.0.0.1";
+    const rootDirectory =
+        path.resolve(
+            options.rootDirectory
+        );
+    const appDirectory =
+        path.join(
+            rootDirectory,
+            "app"
+        );
+    let server:
+        http.Server | null =
+        null;
+
+    async function start(): Promise<void> {
+        if (server) {
+            return;
+        }
+
+        server =
+            http.createServer(
+                async (
+                    req,
+                    res
+                ) => {
+                    try {
+                        const requestUrl =
+                            new URL(
+                                req.url ?? "/",
+                                `http://${req.headers.host ?? `${options.hostname}:${options.port}`}`
+                            );
+
+                        if (
+                            requestUrl.pathname !==
+                            "/_bcp/action"
+                        ) {
+                            proxyRequest(
+                                req,
+                                res,
+                                upstreamHostname,
+                                options.upstreamPort
+                            );
+                            return;
+                        }
+
+                        await handleActionRequest(
+                            req,
+                            res,
+                            requestUrl,
+                            appDirectory
+                        );
+                    } catch (error) {
+                        console.error(
+                            "[BCP Action Error]",
+                            error
+                        );
+
+                        if (
+                            res.headersSent
+                        ) {
+                            res.destroy();
+                            return;
+                        }
+
+                        await sendNavigationWebResponse(
+                            req,
+                            res,
+                            createFormActionErrorResponse(
+                                "Internal Server Error",
+                                500
+                            )
+                        );
+                    }
+                }
+            );
+
+        server.keepAliveTimeout =
+            65_000;
+        server.headersTimeout =
+            66_000;
+
+        await listen(
+            server,
+            options.port,
+            options.hostname
+        );
+    }
+
+    async function stop(): Promise<void> {
+        if (!server) {
+            return;
+        }
+
+        const current =
+            server;
+        server =
+            null;
+
+        await closeServer(
+            current
+        );
+    }
+
+    return {
+        start,
+        stop,
+    };
+}
+
+async function handleActionRequest(
+    req: http.IncomingMessage,
+    res: http.ServerResponse,
+    requestUrl: URL,
+    appDirectory: string
+): Promise<void> {
+    if (
+        (req.method ?? "GET")
+            .toUpperCase() !==
+        "POST"
+    ) {
+        await sendNavigationWebResponse(
+            req,
+            res,
+            createFormActionErrorResponse(
+                "Method Not Allowed",
+                405
+            )
+        );
+        return;
+    }
+
+    const actionName =
+        requestUrl.searchParams.get(
+            "name"
+        );
+
+    if (!actionName) {
+        await sendNavigationWebResponse(
+            req,
+            res,
+            createFormActionErrorResponse(
+                "Missing form action name.",
+                400
+            )
+        );
+        return;
+    }
+
+    const target =
+        resolveActionTarget(
+            req,
+            requestUrl
+        );
+
+    if (!target) {
+        await sendNavigationWebResponse(
+            req,
+            res,
+            createFormActionErrorResponse(
+                "Missing or invalid form action target.",
+                400
+            )
+        );
+        return;
+    }
+
+    const method =
+        normalizePageActionMethod(
+            requestUrl.searchParams.get(
+                "method"
+            )
+        );
+    const routes =
+        scanRoutes(
+            appDirectory
+        );
+    const match =
+        matchRoute(
+            routes,
+            target.pathname
+        );
+
+    if (!match) {
+        await sendNavigationWebResponse(
+            req,
+            res,
+            createFormActionErrorResponse(
+                "Form action route not found.",
+                404
+            )
+        );
+        return;
+    }
+
+    const formData =
+        await readFormData(
+            req,
+            requestUrl
+        );
+    const actionRequest =
+        new Request(
+            target,
+            {
+                method,
+                headers:
+                    collectHeaders(
+                        req
+                    ),
+            }
+        );
+
+    const completed =
+        await runWithRequestContext(
+            actionRequest,
+            () =>
+                runWithRequestCacheContext(
+                    target.pathname,
+                    async () => {
+                        const execution =
+                            await executePageAction(
+                                match.route.filePath,
+                                actionName,
+                                method,
+                                formData,
+                                match.params,
+                                target,
+                                Date.now()
+                            );
+
+                        return completePageActionExecution(
+                            execution
+                        );
+                    }
+                ),
+            {
+                remoteAddress:
+                    req.socket
+                        .remoteAddress ??
+                    null,
+            }
+        );
+    const response =
+        createFormActionTransportResponse(
+            completed,
+            target,
+            isEnhancedFormActionRequest(
+                collectHeaders(
+                    req
+                )
+            )
+        );
+
+    await sendNavigationWebResponse(
+        req,
+        res,
+        response
+    );
+}
+
+function resolveActionTarget(
+    req: http.IncomingMessage,
+    requestUrl: URL
+): URL | null {
+    const candidates = [
+        requestUrl.searchParams.get(
+            "url"
+        ),
+        headerValue(
+            req.headers[
+                "x-bcp-action-target"
+            ]
+        ),
+        headerValue(
+            req.headers.referer
+        ),
+    ];
+
+    for (
+        const candidate
+        of candidates
+    ) {
+        if (!candidate) {
+            continue;
+        }
+
+        try {
+            const target =
+                new URL(
+                    candidate,
+                    requestUrl.origin
+                );
+
+            if (
+                target.origin ===
+                requestUrl.origin &&
+                target.pathname !==
+                "/_bcp/action"
+            ) {
+                return target;
+            }
+        } catch {
+            // Try the next target source.
+        }
+    }
+
+    return null;
+}
+
+async function readFormData(
+    req: http.IncomingMessage,
+    requestUrl: URL
+): Promise<FormData> {
+    const chunks:
+        Buffer[] = [];
+
+    for await (
+        const chunk
+        of req
+    ) {
+        chunks.push(
+            Buffer.isBuffer(
+                chunk
+            )
+                ? chunk
+                : Buffer.from(
+                    chunk
+                )
+        );
+    }
+
+    const body =
+        Buffer.concat(
+            chunks
+        );
+    const request =
+        new Request(
+            requestUrl,
+            {
+                method: "POST",
+                headers:
+                    collectHeaders(
+                        req
+                    ),
+                body:
+                    body.length > 0
+                        ? body
+                        : undefined,
+            }
+        );
+
+    return request.formData();
+}
+
+function collectHeaders(
+    req: http.IncomingMessage
+): Headers {
+    const headers =
+        new Headers();
+
+    for (
+        const [
+            name,
+            value,
+        ]
+        of Object.entries(
+            req.headers
+        )
+    ) {
+        if (
+            value === undefined
+        ) {
+            continue;
+        }
+
+        if (
+            Array.isArray(
+                value
+            )
+        ) {
+            for (
+                const item
+                of value
+            ) {
+                headers.append(
+                    name,
+                    item
+                );
+            }
+        } else {
+            headers.set(
+                name,
+                value
+            );
+        }
+    }
+
+    return headers;
+}
+
+function headerValue(
+    value: string | string[] | undefined
+): string | null {
+    if (
+        Array.isArray(
+            value
+        )
+    ) {
+        return value[0] ??
+            null;
+    }
+
+    return value ??
+        null;
+}
+
+function proxyRequest(
+    req: http.IncomingMessage,
+    res: http.ServerResponse,
+    upstreamHostname: string,
+    upstreamPort: number
+): void {
+    const upstream =
+        http.request({
+            hostname:
+                upstreamHostname,
+            port:
+                upstreamPort,
+            method:
+                req.method ??
+                "GET",
+            path:
+                req.url ??
+                "/",
+            headers:
+                req.headers,
+        });
+
+    upstream.on(
+        "response",
+        (upstreamResponse) => {
+            res.writeHead(
+                upstreamResponse.statusCode ??
+                502,
+                upstreamResponse.headers
+            );
+            upstreamResponse.pipe(
+                res
+            );
+        }
+    );
+
+    upstream.on(
+        "error",
+        (error) => {
+            if (
+                res.headersSent
+            ) {
+                res.destroy(
+                    error
+                );
+                return;
+            }
+
+            res.writeHead(
+                502,
+                {
+                    "Content-Type":
+                        "text/plain; charset=utf-8",
+                    "Cache-Control":
+                        "no-store",
+                }
+            );
+            res.end(
+                "Bad Gateway"
+            );
+        }
+    );
+
+    req.pipe(
+        upstream
+    );
+}
+
+function listen(
+    server: http.Server,
+    port: number,
+    hostname: string
+): Promise<void> {
+    return new Promise(
+        (
+            resolve,
+            reject
+        ) => {
+            const onError =
+                (error: Error) => {
+                    server.off(
+                        "listening",
+                        onListening
+                    );
+                    reject(
+                        error
+                    );
+                };
+            const onListening =
+                () => {
+                    server.off(
+                        "error",
+                        onError
+                    );
+                    resolve();
+                };
+
+            server.once(
+                "error",
+                onError
+            );
+            server.once(
+                "listening",
+                onListening
+            );
+            server.listen(
+                port,
+                hostname
+            );
+        }
+    );
+}
+
+function closeServer(
+    server: http.Server
+): Promise<void> {
+    if (
+        !server.listening
+    ) {
+        return Promise.resolve();
+    }
+
+    return new Promise(
+        (
+            resolve,
+            reject
+        ) => {
+            server.close(
+                (error) => {
+                    if (error) {
+                        reject(
+                            error
+                        );
+                        return;
+                    }
+                    resolve();
+                }
+            );
+        }
+    );
+}