97 lines
3.5 KiB
TypeScript
97 lines
3.5 KiB
TypeScript
import { z } from "zod";
|
|
|
|
import * as trpcNext from "@calcom/trpc/server/adapters/next";
|
|
import { createContext as createTrpcContext } from "@calcom/trpc/server/createContext";
|
|
|
|
import type { AnyRouter } from "@trpc/server";
|
|
|
|
/**
|
|
* Creates an API handler executed by Next.js.
|
|
*/
|
|
export function createNextApiHandler(router: AnyRouter, isPublic = false, namespace = "") {
|
|
return trpcNext.createNextApiHandler({
|
|
router,
|
|
/**
|
|
* @link https://trpc.io/docs/context
|
|
*/
|
|
createContext: ({ req, res }) => {
|
|
return createTrpcContext({ req, res });
|
|
},
|
|
/**
|
|
* @link https://trpc.io/docs/error-handling
|
|
*/
|
|
onError({ error }) {
|
|
if (error.code === "INTERNAL_SERVER_ERROR") {
|
|
// send to bug reporting
|
|
console.error("Something went wrong", error);
|
|
}
|
|
},
|
|
/**
|
|
* Enable query batching
|
|
*/
|
|
batching: {
|
|
enabled: true,
|
|
},
|
|
/**
|
|
* @link https://trpc.io/docs/caching#api-response-caching
|
|
*/
|
|
responseMeta({ ctx, paths, type, errors }) {
|
|
const allOk = errors.length === 0;
|
|
const isQuery = type === "query";
|
|
const noHeaders = {};
|
|
|
|
// We cannot set headers on SSG queries
|
|
if (!ctx?.res) return noHeaders;
|
|
|
|
const defaultHeaders: Record<"headers", Record<string, string>> = {
|
|
headers: {},
|
|
};
|
|
|
|
const timezone = z.string().safeParse(ctx.req?.headers["x-vercel-ip-timezone"]);
|
|
if (timezone.success) defaultHeaders.headers["x-cal-timezone"] = timezone.data;
|
|
|
|
// We need all these conditions to be true to set cache headers
|
|
if (!(isPublic && allOk && isQuery)) return defaultHeaders;
|
|
|
|
// No cache by default
|
|
defaultHeaders.headers["cache-control"] = `no-cache`;
|
|
|
|
if (isPublic && paths) {
|
|
const ONE_DAY_IN_SECONDS = 60 * 60 * 24;
|
|
const FIVE_MINUTES_IN_SECONDS = 5 * 60;
|
|
const ONE_YEAR_IN_SECONDS = 31536000;
|
|
|
|
const cacheRules = {
|
|
session: "no-cache",
|
|
|
|
i18n: `max-age=${ONE_YEAR_IN_SECONDS}`,
|
|
|
|
// FIXME: Using `max-age=1, stale-while-revalidate=60` fails some booking tests.
|
|
"slots.getSchedule": `no-cache`,
|
|
|
|
// Timezones are hardly updated. No need to burden the servers with requests for this by keeping low max-age.
|
|
// Keep it cached for a day and then give it 60 seconds more at most to be updated.
|
|
cityTimezones: `max-age=${ONE_DAY_IN_SECONDS}, stale-while-revalidate=60`,
|
|
|
|
// Feature Flags change but it might be okay to have a 5 minute cache to avoid burdening the servers with requests for this.
|
|
// Note that feature flags can be used to quickly kill a feature if it's not working as expected. So, we have to keep fresh time lesser than the deployment time atleast
|
|
"features.map": `max-age=${FIVE_MINUTES_IN_SECONDS}, stale-while-revalidate=60`, // "map" - Feature Flag Map
|
|
} as const;
|
|
|
|
const prependNamespace = (key: string) =>
|
|
(namespace ? `${namespace}.${key}` : key) as keyof typeof cacheRules;
|
|
const matchedPath = paths.find((v) => prependNamespace(v) in cacheRules);
|
|
if (matchedPath) {
|
|
const cacheRule = cacheRules[prependNamespace(matchedPath)];
|
|
|
|
// We must set cdn-cache-control as well to ensure that Vercel doesn't strip stale-while-revalidate
|
|
// https://vercel.com/docs/concepts/edge-network/caching#:~:text=If%20you%20set,in%20the%20response.
|
|
defaultHeaders.headers["cache-control"] = defaultHeaders.headers["cdn-cache-control"] = cacheRule;
|
|
}
|
|
}
|
|
|
|
return defaultHeaders;
|
|
},
|
|
});
|
|
}
|