118 lines
2.8 KiB
TypeScript
118 lines
2.8 KiB
TypeScript
import type { CreateNextContextOptions } from "@trpc/server/adapters/next";
|
|
import type { NextApiRequest } from "next";
|
|
import type { OpenApiMeta } from "trpc-to-openapi";
|
|
import { initTRPC, TRPCError } from "@trpc/server";
|
|
import superjson from "superjson";
|
|
import { ZodError } from "zod";
|
|
|
|
import type { dbClient } from "@kan/db/client";
|
|
import { initAuth } from "@kan/auth";
|
|
import { createDrizzleClient } from "@kan/db/client";
|
|
|
|
export interface User {
|
|
id: string;
|
|
name: string;
|
|
email: string;
|
|
emailVerified: boolean;
|
|
createdAt: Date;
|
|
updatedAt: Date;
|
|
image?: string | null | undefined;
|
|
stripeCustomerId?: string | null | undefined;
|
|
}
|
|
|
|
interface CreateContextOptions {
|
|
user: User | null | undefined;
|
|
db: dbClient;
|
|
}
|
|
|
|
export const createInnerTRPCContext = (opts: CreateContextOptions) => {
|
|
return {
|
|
user: opts.user,
|
|
db: opts.db,
|
|
};
|
|
};
|
|
|
|
export const createTRPCContext = async ({ req }: CreateNextContextOptions) => {
|
|
const db = createDrizzleClient();
|
|
const auth = initAuth(db);
|
|
|
|
const session = await auth.api.getSession({
|
|
// @ts-expect-error
|
|
headers: new Headers(req.headers),
|
|
});
|
|
|
|
return createInnerTRPCContext({ db, user: session?.user });
|
|
};
|
|
|
|
export const createNextApiContext = async (req: NextApiRequest) => {
|
|
const db = createDrizzleClient();
|
|
const auth = initAuth(db);
|
|
|
|
const session = await auth.api.getSession({
|
|
// @ts-expect-error
|
|
headers: new Headers(req.headers),
|
|
});
|
|
|
|
return createInnerTRPCContext({ db, user: session?.user });
|
|
};
|
|
|
|
export const createRESTContext = async ({ req }: CreateNextContextOptions) => {
|
|
const db = createDrizzleClient();
|
|
const auth = initAuth(db);
|
|
|
|
let session;
|
|
try {
|
|
session = await auth.api.getSession({
|
|
// @ts-expect-error
|
|
headers: new Headers(req.headers),
|
|
});
|
|
} catch (error) {
|
|
console.error("Error getting session, ", error);
|
|
throw error;
|
|
}
|
|
|
|
return createInnerTRPCContext({ db, user: session?.user });
|
|
};
|
|
|
|
const t = initTRPC
|
|
.context<typeof createTRPCContext>()
|
|
.meta<OpenApiMeta>()
|
|
.create({
|
|
transformer: superjson,
|
|
errorFormatter({ shape, error }) {
|
|
return {
|
|
...shape,
|
|
data: {
|
|
...shape.data,
|
|
zodError:
|
|
error.cause instanceof ZodError ? error.cause.flatten() : null,
|
|
},
|
|
};
|
|
},
|
|
});
|
|
|
|
export const createTRPCRouter = t.router;
|
|
|
|
export const createCallerFactory = t.createCallerFactory;
|
|
|
|
export const publicProcedure = t.procedure.meta({
|
|
openapi: { method: "GET", path: "/public" },
|
|
});
|
|
|
|
const enforceUserIsAuthed = t.middleware(async ({ ctx, next }) => {
|
|
if (!ctx.user) {
|
|
throw new TRPCError({ code: "UNAUTHORIZED" });
|
|
}
|
|
|
|
return next({
|
|
ctx,
|
|
});
|
|
});
|
|
|
|
export const protectedProcedure = t.procedure.use(enforceUserIsAuthed).meta({
|
|
openapi: {
|
|
method: "GET",
|
|
path: "/protected",
|
|
},
|
|
});
|