load-context.ts

import { PrismaD1 } from "@prisma/adapter-d1";
import { PrismaClient } from "@prisma/client";
import {
	createCookie,
	createWorkersKVSessionStorage,
	type SessionStorage,
} from "@remix-run/cloudflare";
import { Authenticator } from "remix-auth";
import { DiscordStrategy } from "remix-auth-discord";
import type { PlatformProxy } from "wrangler";

export type Cloudflare = Omit<PlatformProxy<Env>, "dispose">;

declare module "@remix-run/cloudflare" {
	interface AppLoadContext {
		cloudflare: Cloudflare;
		prisma: PrismaClient;
		sessionStorage: SessionStorage;
		authenticator: Authenticator<SessionData>;
	}
}

interface SessionData {
	userId: string;
}

const createAuthenticator = (cloudflare: Cloudflare, prisma: PrismaClient) => {
	const sessionStorage = createWorkersKVSessionStorage<SessionData>({
		cookie: createCookie("session", {
			sameSite: "lax",
			secrets: [cloudflare.env.SESSION_COOKIE_SECRET],
			secure: cloudflare.env.CF_ENV !== "local",
			httpOnly: true,
			maxAge: 60 * 60 * 24 * 7,
		}),
		kv: cloudflare.env.KV_SESSION_STORAGE,
	});

	const authenticator = new Authenticator<SessionData>(sessionStorage);

	authenticator.use(
		new DiscordStrategy(
			{
				clientID: cloudflare.env.DISCORD_CLIENT_ID,
				clientSecret: cloudflare.env.DISCORD_CLIENT_SECRET,
				callbackURL: "/sign-in/callback/discord",
				scope: ["identify"],
				prompt: "none",
			},
			async (params) => {
				const user = await prisma.users.findUnique({
					select: { id: true },
					where: { discordId: params.profile.id },
				});
				if (!user) {
					throw new Error("user not found");
				}
				return { userId: user.id };
			},
		),
		"discord",
	);

	authenticator.use(
		new DiscordStrategy(
			{
				clientID: cloudflare.env.DISCORD_CLIENT_ID,
				clientSecret: cloudflare.env.DISCORD_CLIENT_SECRET,
				callbackURL: "/register/callback/discord",
				scope: ["identify"],
				prompt: "consent",
			},
			async (params) => {
				const user = await prisma.users.upsert({
					create: {
						discordId: params.profile.id,
						displayName: params.profile.displayName,
					},
					update: {},
					where: { discordId: params.profile.id },
					select: { id: true },
				});
				return { userId: user.id };
			},
		),
		"discord-registration",
	);

	return { sessionStorage, authenticator };
};

export const getLoadContext = ({
	context: { cloudflare },
}: {
	context: { cloudflare: Cloudflare };
}) => {
	const prisma = new PrismaClient({ adapter: new PrismaD1(cloudflare.env.DB) });
	const { authenticator, sessionStorage } = createAuthenticator(
		cloudflare,
		prisma,
	);
	return {
		cloudflare,
		prisma,
		sessionStorage,
		authenticator,
	};
};