first commit

calcom/packages/trpc/server/createContext.ts

@@ -0,0 +1,104 @@
+/* eslint-disable @typescript-eslint/no-explicit-any */
+import type { GetServerSidePropsContext, NextApiRequest, NextApiResponse } from "next";
+import type { Session } from "next-auth";
+import type { serverSideTranslations } from "next-i18next/serverSideTranslations";
+
+import { getLocale } from "@calcom/features/auth/lib/getLocale";
+import getIP from "@calcom/lib/getIP";
+import prisma, { readonlyPrisma } from "@calcom/prisma";
+import type { SelectedCalendar, User as PrismaUser } from "@calcom/prisma/client";
+
+import type { CreateNextContextOptions } from "@trpc/server/adapters/next";
+
+type CreateContextOptions =
+  | (Omit<CreateNextContextOptions, "info"> & {
+      info?: CreateNextContextOptions["info"];
+    })
+  | GetServerSidePropsContext;
+
+export type CreateInnerContextOptions = {
+  sourceIp?: string;
+  session?: Session | null;
+  locale: string;
+  user?:
+    | Omit<
+        PrismaUser,
+        | "locale"
+        | "twoFactorSecret"
+        | "emailVerified"
+        | "password"
+        | "identityProviderId"
+        | "invitedTo"
+        | "allowDynamicBooking"
+        | "verified"
+      > & {
+        locale: Exclude<PrismaUser["locale"], null>;
+        credentials?: Credential[];
+        selectedCalendars?: Partial<SelectedCalendar>[];
+        rawAvatar?: string;
+      };
+  i18n?: Awaited<ReturnType<typeof serverSideTranslations>>;
+} & Partial<CreateContextOptions>;
+
+export type GetSessionFn =
+  | ((_options: {
+      req: GetServerSidePropsContext["req"] | NextApiRequest;
+      res: GetServerSidePropsContext["res"] | NextApiResponse;
+    }) => Promise<Session | null>)
+  | (() => Promise<Session | null>);
+
+export type InnerContext = CreateInnerContextOptions & {
+  prisma: typeof prisma;
+  insightsDb: typeof readonlyPrisma;
+};
+
+/**
+ * Inner context. Will always be available in your procedures, in contrast to the outer context.
+ *
+ * Also useful for:
+ * - testing, so you don't have to mock Next.js' `req`/`res`
+ * - tRPC's `createServerSideHelpers` where we don't have `req`/`res`
+ *
+ * @see https://trpc.io/docs/context#inner-and-outer-context
+ */
+export async function createContextInner(opts: CreateInnerContextOptions): Promise<InnerContext> {
+  return {
+    prisma,
+    insightsDb: readonlyPrisma,
+    ...opts,
+  };
+}
+
+type Context = InnerContext & {
+  req: CreateContextOptions["req"];
+  res: CreateContextOptions["res"];
+};
+
+/**
+ * Creates context for an incoming request
+ * @link https://trpc.io/docs/context
+ */
+export const createContext = async (
+  { req, res }: CreateContextOptions,
+  sessionGetter?: GetSessionFn
+): Promise<Context> => {
+  const locale = await getLocale(req);
+
+  // This type may not be accurate if this request is coming from SSG init but they both should satisfy the requirements of getIP.
+  // TODO: @sean - figure out a way to make getIP be happy with trpc req. params
+  const sourceIp = getIP(req as NextApiRequest);
+  const session = !!sessionGetter ? await sessionGetter({ req, res }) : null;
+  const contextInner = await createContextInner({ locale, session, sourceIp });
+  return {
+    ...contextInner,
+    req,
+    res,
+  };
+};
+
+export type TRPCContext = Awaited<ReturnType<typeof createContext>>;
+export type TRPCContextInner = Awaited<ReturnType<typeof createContextInner>>;
+export type WithLocale<T extends TRPCContext = any> = T &
+  Required<Pick<CreateInnerContextOptions, "i18n" | "locale">>;
+export type WithSession<T extends TRPCContext = any> = T &
+  Required<Pick<CreateInnerContextOptions, "session">>;