🐛 Fix merge guest user into logged in user (#907)

apps/web/src/utils/auth.ts

@@ -18,9 +18,11 @@ import CredentialsProvider from "next-auth/providers/credentials";
 import EmailProvider from "next-auth/providers/email";
 
 import { LegacyTokenProvider } from "@/utils/auth/legacy-token-provider";
+import { mergeGuestsIntoUser } from "@/utils/auth/merge-user";
 import { emailClient } from "@/utils/emails";
 
-const authOptions = {
+const getAuthOptions = (...args: GetServerSessionParams) =>
+  ({
     adapter: PrismaAdapter(prisma),
     secret: process.env.SECRET_PASSWORD,
     session: {
@@ -115,14 +117,8 @@ const authOptions = {
       error: "/auth/error",
     },
     callbacks: {
-    async redirect({ url, baseUrl }) {
-      // Allows relative callback URLs
-      if (url.startsWith("/")) return `${baseUrl}${url}`;
-      // Allows callback URLs on the same origin
-      else if (new URL(url).origin === baseUrl) return url;
-      return baseUrl;
-    },
-    async signIn({ user }) {
+      async signIn({ user, email }) {
+        if (email?.verificationRequest) {
           if (user.email) {
             const userExists =
               (await prisma.user.count({
@@ -130,6 +126,7 @@ const authOptions = {
                   email: user.email,
                 },
               })) > 0;
+
             if (userExists) {
               if (isEmailBlocked(user.email)) {
                 return false;
@@ -139,6 +136,12 @@ const authOptions = {
               return false;
             }
           }
+        } else {
+          const session = await getServerSession(...args);
+          if (session && session.user.email === null) {
+            await mergeGuestsIntoUser(user.id, [session.user.id]);
+          }
+        }
 
         return true;
       },
@@ -185,18 +188,19 @@ const authOptions = {
         return session;
       },
     },
-} satisfies NextAuthOptions;
+  } satisfies NextAuthOptions);
 
-export async function getServerSession(
-  ...args:
+type GetServerSessionParams =
   | [GetServerSidePropsContext["req"], GetServerSidePropsContext["res"]]
   | [NextApiRequest, NextApiResponse]
-    | []
-) {
-  return getServerSessionWithOptions(...args, authOptions);
+  | [];
+
+export function getServerSession(...args: GetServerSessionParams) {
+  return getServerSessionWithOptions(...args, getAuthOptions(...args));
 }
 
 export async function AuthApiRoute(req: NextApiRequest, res: NextApiResponse) {
+  const authOptions = getAuthOptions(req, res);
   return NextAuth(req, res, authOptions);
 }
 

apps/web/src/utils/auth/merge-user.ts

@@ -0,0 +1,39 @@
+import { prisma } from "@rallly/database";
+
+export const mergeGuestsIntoUser = async (
+  userId: string,
+  guestIds: string[],
+) => {
+  await prisma.poll.updateMany({
+    where: {
+      userId: {
+        in: guestIds,
+      },
+    },
+    data: {
+      userId: userId,
+    },
+  });
+
+  await prisma.participant.updateMany({
+    where: {
+      userId: {
+        in: guestIds,
+      },
+    },
+    data: {
+      userId: userId,
+    },
+  });
+
+  await prisma.comment.updateMany({
+    where: {
+      userId: {
+        in: guestIds,
+      },
+    },
+    data: {
+      userId: userId,
+    },
+  });
+};

packages/backend/trpc/routers/auth.ts

@@ -6,44 +6,6 @@ import { generateOtp } from "../../utils/nanoid";
 import { publicProcedure, router } from "../trpc";
 import { RegistrationTokenPayload } from "../types";
 
-// assigns participants and comments created by guests to a user
-// we could have multiple guests because a login might be triggered from one device
-// and opened in another one.
-const mergeGuestsIntoUser = async (userId: string, guestIds: string[]) => {
-  await prisma.poll.updateMany({
-    where: {
-      userId: {
-        in: guestIds,
-      },
-    },
-    data: {
-      userId: userId,
-    },
-  });
-
-  await prisma.participant.updateMany({
-    where: {
-      userId: {
-        in: guestIds,
-      },
-    },
-    data: {
-      userId: userId,
-    },
-  });
-
-  await prisma.comment.updateMany({
-    where: {
-      userId: {
-        in: guestIds,
-      },
-    },
-    data: {
-      userId: userId,
-    },
-  });
-};
-
 export const auth = router({
   // @deprecated
   requestRegistration: publicProcedure
@@ -107,7 +69,7 @@ export const auth = router({
         locale: z.string().optional(),
       }),
     )
-    .mutation(async ({ input, ctx }) => {
+    .mutation(async ({ input }) => {
       const payload = await decryptToken<RegistrationTokenPayload>(input.token);
 
       if (!payload) {
@@ -129,10 +91,6 @@ export const auth = router({
         },
       });
 
-      if (ctx.user.isGuest) {
-        await mergeGuestsIntoUser(user.id, [ctx.user.id]);
-      }
-
       return { ok: true, user };
     }),
   getUserPermission: publicProcedure