perf : not ask me i copy from another projeft
This commit is contained in:
@@ -4,6 +4,8 @@ import { DrizzleAdapter } from "@auth/drizzle-adapter";
|
||||
import { db } from "@/db";
|
||||
import { accounts, sessions, users, verificationTokens } from "@/db/schema";
|
||||
|
||||
const discordIdCache = new Map<string, string>();
|
||||
|
||||
export const authOptions: NextAuthOptions = {
|
||||
adapter: DrizzleAdapter(db, {
|
||||
usersTable: users,
|
||||
@@ -21,14 +23,21 @@ export const authOptions: NextAuthOptions = {
|
||||
async session({ session, user }) {
|
||||
if (session.user) {
|
||||
(session.user as any).id = user.id;
|
||||
// Attach discordId from accounts table
|
||||
const acct = await db.query.accounts.findFirst({
|
||||
where: (a, { eq, and }) =>
|
||||
and(eq(a.userId, user.id), eq(a.provider, "discord")),
|
||||
});
|
||||
if (acct) {
|
||||
(session.user as typeof session.user & { discordId: string }).discordId =
|
||||
acct.providerAccountId;
|
||||
|
||||
let discordId = discordIdCache.get(user.id);
|
||||
if (!discordId) {
|
||||
const acct = await db.query.accounts.findFirst({
|
||||
where: (a, { eq, and }) =>
|
||||
and(eq(a.userId, user.id), eq(a.provider, "discord")),
|
||||
columns: { providerAccountId: true },
|
||||
});
|
||||
if (acct) {
|
||||
discordId = acct.providerAccountId;
|
||||
discordIdCache.set(user.id, discordId);
|
||||
}
|
||||
}
|
||||
if (discordId) {
|
||||
(session.user as typeof session.user & { discordId: string }).discordId = discordId;
|
||||
}
|
||||
}
|
||||
return session;
|
||||
|
||||
Reference in New Issue
Block a user