rwsdk/dist/runtime/server.js

Build fast, server-driven webapps on Cloudflare with SSR, RSC, and realtime

import { requestInfo } from "./requestInfo/worker";
let globalWrap;
/**
 * Register a wrapper that runs around every server action and query handler.
 *
 * Call this once in your worker entry point. The wrapper receives the main
 * handler function, its arguments, and the type ("action" or "query").
 * Interruptors run *outside* the wrapper.
 *
 * Only one wrapper is active at a time; the most recent call wins.
 *
 * @example
 * ```ts
 * import { registerServerFunctionWrap } from "rwsdk/worker";
 * import * as Sentry from "@sentry/cloudflare";
 *
 * registerServerFunctionWrap((fn, args, type) =>
 *   Sentry.startSpan(
 *     { name: fn.name, op: `function.rsc_${type}` },
 *     () => fn(...args)
 *   )
 * );
 * ```
 */
export function registerServerFunctionWrap(wrap) {
    globalWrap = wrap;
}
/**
 * @internal Reset the global wrap — only for tests.
 */
export function __resetServerFunctionWrap() {
    globalWrap = undefined;
}
function createServerFunction(fns, mainFn, options) {
    const wrapped = async (...args) => {
        const { request, ctx } = requestInfo;
        // Execute interruptors
        for (const fn of fns) {
            const result = await fn({ request, ctx, args });
            if (result instanceof Response) {
                // We return the Response so it can be handled by the action handler
                // and serialized into the RSC stream via normalizeActionResult.
                return result;
            }
        }
        if (globalWrap) {
            return globalWrap(mainFn, args, options?.__type ?? "action");
        }
        return mainFn(...args);
    };
    wrapped.method = options?.method ?? "POST";
    return wrapped;
}
/**
 * Wrap a function to be used as a server query.
 *
 * - **Method**: Defaults to `GET`. can be changed via `options`.
 * - **Behavior**: When called from the client, it returns data-only and does **not** rehydrate or re-render the React page.
 * - **Location**: Must be defined in a file with `"use server"`. We recommend `queries.ts` colocated with components.
 * - **Middleware**: You can pass an array of functions as the first argument to act as interruptors (e.g. for auth).
 *
 * @example
 * ```ts
 * // getters.ts
 * "use server"
 *
 * export const getUser = serverQuery(async (id: string) => {
 *   return db.user.findUnique({ where: { id } })
 * })
 * ```
 */
export function serverQuery(fnsOrFn, options) {
    let fns = [];
    let mainFn;
    if (Array.isArray(fnsOrFn)) {
        fns = fnsOrFn.slice(0, -1);
        mainFn = fnsOrFn[fnsOrFn.length - 1];
    }
    else {
        mainFn = fnsOrFn;
    }
    const method = options?.method ?? "GET"; // Default to GET for query
    const wrapped = createServerFunction(fns, mainFn, { ...options, method, __type: "query" });
    wrapped.method = method;
    return wrapped;
}
/**
 * Wrap a function to be used as a server action.
 *
 * - **Method**: Defaults to `POST`. can be changed via `options`.
 * - **Behavior**: When called from the client, it **will** rehydrate and re-render the React page with the new server state.
 * - **Location**: Must be defined in a file with `"use server"`. We recommend `actions.ts` colocated with components.
 * - **Middleware**: You can pass an array of functions as the first argument to act as interruptors (e.g. for auth).
 *
 * @example
 * ```ts
 * // actions.ts
 * "use server"
 *
 * export const updateUser = serverAction(async (id: string, data: any) => {
 *   return db.user.update({ where: { id }, data })
 * })
 * ```
 */
export function serverAction(fnsOrFn, options) {
    let fns = [];
    let mainFn;
    if (Array.isArray(fnsOrFn)) {
        fns = fnsOrFn.slice(0, -1);
        mainFn = fnsOrFn[fnsOrFn.length - 1];
    }
    else {
        mainFn = fnsOrFn;
    }
    const method = options?.method ?? "POST"; // Default to POST for action
    const wrapped = createServerFunction(fns, mainFn, { ...options, method, __type: "action" });
    wrapped.method = method;
    return wrapped;
}