codeclub/auth.ts at main · kletu-codeclub/codeclub · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
import NextAuth from "next-auth";
import { authConfig } from "@/lib/auth.config";
import { PrismaAdapter } from "@auth/prisma-adapter";
import db from "@/lib/db";

export const { handlers, signIn, signOut, auth } = NextAuth({
 ...authConfig,
 adapter: PrismaAdapter(db),
 session: {
  strategy: "jwt",
 },
 pages: {
  signIn: "/auth",
 },
 callbacks: {
  async jwt({ token, user }) {
   if (user) {
    // When user first signs in, get their data from the database
    // try {
    //   const dbUser = await db.user.findUnique({
    //     where: { email: user.email! },
    //     select: { id: true, cfHandle: true, email: true, name: true }
    //   });

    //   if (dbUser) {
    //     token.id = dbUser.id;
    //     token.cfHandle = dbUser.cfHandle;
    //   } else {
    //     // User doesn't exist in our database yet
    //     token.id = user.id;
    //     token.cfHandle = null;
    //   }
    // } catch (error) {
    //   console.error("Error fetching user from database:", error);
    //   token.id = user.id;
    //   token.cfHandle = null;
    // }
    token.id = user.id;
    token.cfHandle = user.cfHandle || null;
   }

   return token;
  },

  async session({ session, token }) {
   if (token) {
    // set the token data to session
    session.user.id = String(token.id);
    session.user.cfHandle = (token.cfHandle as string) || null;
   }

   if (token.sub) {
      try {
        const dbUser = await db.user.findUnique({
          where: { id: token.sub },
          select: { id: true, cfHandle: true, email: true, name: true }
        });

        if (dbUser) {
          token.id = dbUser.id;
          token.cfHandle = dbUser.cfHandle;
        }
      } catch (error) {
        console.error("Error fetching user from database:", error);
      }
    }

   return session;
  },

  redirect({ url, baseUrl }) {
   let callbackUrl = null;

   try {
    callbackUrl = new URL(url, baseUrl).searchParams.get("callbackUrl");
   } catch (e) {
    callbackUrl = null;
   }

   if (callbackUrl) {
    return callbackUrl;
   }
   return baseUrl;
  },
 },
});