98 lines
2.7 KiB
TypeScript
98 lines
2.7 KiB
TypeScript
import { createCookieSessionStorage, redirect } from "remix";
|
|
import invariant from "tiny-invariant";
|
|
|
|
import { discordIdentify, User, SessionInformation } from "~/models/user.server";
|
|
import { getUserByDiscordId } from "~/models/user.server";
|
|
|
|
invariant(process.env.SESSION_SECRET, "SESSION_SECRET must be set");
|
|
|
|
export const sessionStorage = createCookieSessionStorage({
|
|
cookie: {
|
|
name: "__session",
|
|
httpOnly: true,
|
|
maxAge: 0,
|
|
path: "/",
|
|
sameSite: "lax",
|
|
secrets: [process.env.SESSION_SECRET],
|
|
secure: process.env.NODE_ENV === "production",
|
|
},
|
|
});
|
|
|
|
const USER_SESSION_KEY = "userId";
|
|
const BEARER_TOKEN_KEY = "bearerToken";
|
|
const REFRESH_TOKEN_KEY = "refreshToken";
|
|
|
|
export async function getSession(request: Request) {
|
|
const cookie = request.headers.get("Cookie");
|
|
return sessionStorage.getSession(cookie);
|
|
}
|
|
|
|
export async function getUserId(request: Request): Promise<string | undefined> {
|
|
const session = await getSession(request);
|
|
const userId = session.get(USER_SESSION_KEY);
|
|
return userId;
|
|
}
|
|
|
|
export async function getUser(request: Request): Promise<null | User> {
|
|
const userId = await getUserId(request);
|
|
if (userId === undefined) return null;
|
|
|
|
const user = await getUserByDiscordId(userId);
|
|
if (user) return user;
|
|
|
|
throw await logout(request);
|
|
}
|
|
|
|
export async function requireUserId(
|
|
request: Request,
|
|
redirectTo: string = new URL(request.url).pathname
|
|
): Promise<string> {
|
|
const userId = await getUserId(request);
|
|
if (!userId) {
|
|
const searchParams = new URLSearchParams([["redirectTo", redirectTo]]);
|
|
throw redirect(`/loginstart?${searchParams}`);
|
|
}
|
|
return userId;
|
|
}
|
|
|
|
export async function requireUser(request: Request) {
|
|
const session = await getSession(request);
|
|
const user_id = session.get(USER_SESSION_KEY);
|
|
const client_id = process.env.DISCORD_CLIENT_ID || "";
|
|
const redirect_uri = process.env.DISCORD_REDIRECT_URI || "";
|
|
if (!user_id) {
|
|
return redirect(`https://discord.com/api/oauth2/authorize?client_id=${client_id}` +
|
|
`&response_type=code&redirect_uri=${encodeURIComponent(redirect_uri)}&scope=identify`);
|
|
}
|
|
}
|
|
|
|
export async function createUserSession({
|
|
request,
|
|
discord_id,
|
|
redirectTo,
|
|
}: {
|
|
request: Request;
|
|
discord_id: string;
|
|
redirectTo: string;
|
|
}) {
|
|
const session = await getSession(request);
|
|
session.set(USER_SESSION_KEY, discord_id);
|
|
console.log(discord_id);
|
|
return redirect(redirectTo, {
|
|
headers: {
|
|
"Set-Cookie": await sessionStorage.commitSession(session, {
|
|
maxAge: 60 * 60 * 24 * 7 // 7 days
|
|
}),
|
|
},
|
|
});
|
|
}
|
|
|
|
export async function logout(request: Request) {
|
|
const session = await getSession(request);
|
|
return redirect("/", {
|
|
headers: {
|
|
"Set-Cookie": await sessionStorage.destroySession(session),
|
|
},
|
|
});
|
|
}
|