api: update login endpoint to detect possible user migrations

apps/api/prisma/migrations/20241105213123_legacy_profile_updates/migration.sql

@@ -0,0 +1,5 @@
+-- AlterTable
+ALTER TABLE "public"."user_profile" ADD COLUMN     "legacy_profile_migrated_at" TIMESTAMP(3);
+
+-- CreateIndex
+CREATE INDEX "user_profile_legacy_email_legacy_email_verified_at_idx" ON "public"."user_profile"("legacy_email", "legacy_email_verified_at");

apps/api/prisma/schema.prisma

@@ -93,9 +93,10 @@ model UserProfile {
   showGemsBalance  Boolean @default(true) @map("show_gems_balance")
 
   // Migration
-  legacyAddress         String?   @unique @map("legacy_address") @db.VarChar(42)
-  legacyEmail           String?   @map("legacy_email")
-  legacyEmailVerifiedAt DateTime? @map("legacy_email_verified_at")
+  legacyAddress           String?   @unique @map("legacy_address") @db.VarChar(42)
+  legacyEmail             String?   @map("legacy_email")
+  legacyEmailVerifiedAt   DateTime? @map("legacy_email_verified_at")
+  legacyProfileMigratedAt DateTime? @map("legacy_profile_migrated_at")
 
   // Status
   testnetFaucetLastUsedAt DateTime? @map("testnet_faucet_last_used_at")
@@ -109,6 +110,7 @@ model UserProfile {
   updatedAt DateTime @updatedAt @map("updated_at")
 
   @@index([legacyAddress])
+  @@index([legacyEmail, legacyEmailVerifiedAt])
   @@index([tag])
   @@map("user_profile")
   @@schema("public")

apps/api/src/routes/auth.ts

@@ -27,6 +27,7 @@ import {
 import type { TdkApiContext } from "../types";
 import {
   USER_PROFILE_SELECT_FIELDS,
+  USER_PUBLIC_PROFILE_SELECT_FIELDS,
   USER_SMART_ACCOUNT_INCLUDE_FIELDS,
 } from "../utils/db";
 import { throwUnauthorizedError } from "../utils/error";
@@ -172,15 +173,38 @@ export const authRoutes =
         }
 
         const { user } = userSmartAccount;
-        const profile = await db.userProfile.upsert({
-          where: { userId: user.id },
-          update: {},
-          create: {
-            userId: user.id,
-            email: userSmartAccount.initialEmail,
-          },
-          select: USER_PROFILE_SELECT_FIELDS,
-        });
+        const [profile, legacyUserProfiles] = await Promise.all([
+          // Fetch or create user profile
+          db.userProfile.upsert({
+            where: { userId: user.id },
+            update: {},
+            create: {
+              userId: user.id,
+              email: userSmartAccount.initialEmail,
+            },
+            select: USER_PROFILE_SELECT_FIELDS,
+          }),
+          // Detect any migrations related to this user
+          userSmartAccount.initialEmail
+            ? db.userProfile.findMany({
+                where: {
+                  legacyEmail: userSmartAccount.initialEmail,
+                  legacyEmailVerifiedAt: { not: null },
+                },
+                select: {
+                  ...USER_PUBLIC_PROFILE_SELECT_FIELDS,
+                  id: true,
+                },
+              })
+            : userSmartAccount.initialWalletAddress
+              ? db.userProfile.findUnique({
+                  where: {
+                    legacyAddress: userSmartAccount.initialWalletAddress,
+                  },
+                  select: { ...USER_PUBLIC_PROFILE_SELECT_FIELDS, id: true },
+                })
+              : [],
+        ]);
 
         const [authTokenResult, userSessionsResult] = await Promise.allSettled([
           auth.generateJWT<UserContext>(address, {
@@ -232,6 +256,11 @@ export const authRoutes =
             address,
             sessions,
           },
+          legacyProfiles: legacyUserProfiles
+            ? Array.isArray(legacyUserProfiles)
+              ? legacyUserProfiles
+              : [legacyUserProfiles]
+            : [],
         });
       },
     );

apps/api/src/schema/auth.ts

@@ -3,6 +3,7 @@ import {
   EXAMPLE_WALLET_ADDRESS,
   sessionSchema,
   userProfileSchema,
+  userPublicProfileSchema,
   userSchema,
 } from "./shared";
 
@@ -45,6 +46,14 @@ export const loginReplySchema = Type.Object({
       sessions: Type.Array(sessionSchema),
     }),
   ]),
+  legacyProfiles: Type.Array(
+    Type.Intersect([
+      Type.Object({
+        id: Type.String(),
+      }),
+      userPublicProfileSchema,
+    ]),
+  ),
 });
 
 export const loginCustomBodySchema = Type.Object({