windmill/functions/api/[[path]].ts

//this is purely for cloudflare pages redirection purposes

export async function onRequest(context) {
  // Contents of context object
  const {
    request, // same as existing Worker API
    env, // same as existing Worker API
    params, // if filename includes [id] or [[path]]
    waitUntil, // same as ctx.waitUntil in existing Worker API
    next, // used for middleware or to fetch assets
    data, // arbitrary space for passing data between middlewares
  } = context;

  try {
    const url = new URL(request.url);

    // Check for remote_url cookie
    const cookies = request.headers.get("cookie") || "";
    const remoteUrlMatch = cookies.match(/remote_url=([^;]+)/);
    const targetHostname = remoteUrlMatch
      ? decodeURIComponent(remoteUrlMatch[1])
      : "app.windmill.dev";

    url.hostname = targetHostname;
    const res = await fetch(url.toString(), {
      method: request.method,
      body: request.body,
      headers: request.headers,
      redirect: "manual",
    });
    const newResponse = new Response(res.body, res);

    // Extract domain from target hostname for cookie domain replacement
    const targetDomain = targetHostname.split(".").slice(-2).join(".");

    newResponse.headers.set(
      "set-cookie",
      "token" +
        (newResponse.headers
          .get("set-cookie")
          ?.replace(`Domain=${targetDomain};`, "")
          ?.split("token")
          ?.pop() ?? "")
    );

    newResponse.headers.set("Cross-Origin-Resource-Policy", "cross-origin");
    return newResponse;
  } catch (e) {
    return new Response(e.message, { status: 500 });
  }
}