add legacy user migration flow

.changeset/curvy-starfishes-change.md

@@ -0,0 +1,5 @@
+---
+"@treasure-dev/tdk-core": patch
+---
+
+Added API client helper for legacy user migration

.changeset/ninety-shirts-pay.md

@@ -0,0 +1,5 @@
+---
+"@treasure-dev/tdk-react": minor
+---
+
+Added legacy user migration flow to connect modal

apps/api/prisma/migrations/20241118225941_smart_account_index_address_only/migration.sql

@@ -0,0 +1,2 @@
+-- CreateIndex
+CREATE INDEX "user_smart_account_address_idx" ON "public"."user_smart_account"("address");

apps/api/prisma/migrations/20241119215629_user_profile_delete_cascade/migration.sql

@@ -0,0 +1,5 @@
+-- AddForeignKey
+ALTER TABLE "public"."user_social_account" ADD CONSTRAINT "user_social_account_legacy_user_profile_id_fkey" FOREIGN KEY ("legacy_user_profile_id") REFERENCES "public"."user_profile"("id") ON DELETE CASCADE ON UPDATE CASCADE;
+
+-- AddForeignKey
+ALTER TABLE "public"."user_notification_settings" ADD CONSTRAINT "user_notification_settings_legacy_user_profile_id_fkey" FOREIGN KEY ("legacy_user_profile_id") REFERENCES "public"."user_profile"("id") ON DELETE CASCADE ON UPDATE CASCADE;

apps/api/prisma/schema.prisma

@@ -58,6 +58,7 @@ model UserSmartAccount {
   updatedAt DateTime @updatedAt @map("updated_at")
 
   @@unique([chainId, address])
+  @@index([address])
   @@index([userId])
   @@map("user_smart_account")
   @@schema("public")
@@ -102,8 +103,10 @@ model UserProfile {
   testnetFaucetLastUsedAt DateTime? @map("testnet_faucet_last_used_at")
 
   // Relations
-  userId String? @unique @map("user_id")
-  user   User?   @relation(fields: [userId], references: [id])
+  userId                     String?                    @unique @map("user_id")
+  user                       User?                      @relation(fields: [userId], references: [id])
+  legacySocialAccounts       UserSocialAccount[]
+  legacyNotificationSettings UserNotificationSettings[]
 
   // Computed
   createdAt DateTime @default(now()) @map("created_at")
@@ -138,7 +141,8 @@ model UserSocialAccount {
   isPublic      Boolean       @default(false) @map("is_public")
 
   // Migration
-  legacyUserProfileId String? @map("legacy_user_profile_id")
+  legacyUserProfileId String?      @map("legacy_user_profile_id")
+  legacyUserProfile   UserProfile? @relation(fields: [legacyUserProfileId], references: [id], onDelete: Cascade)
 
   // Relations
   userId String? @map("user_id")
@@ -176,7 +180,8 @@ model UserNotificationSettings {
   isEnabledInApp Boolean          @default(true) @map("is_enabled_in_app")
 
   // Migration
-  legacyUserProfileId String? @map("legacy_user_profile_id")
+  legacyUserProfileId String?      @map("legacy_user_profile_id")
+  legacyUserProfile   UserProfile? @relation(fields: [legacyUserProfileId], references: [id], onDelete: Cascade)
 
   // Relations
   userId String? @map("user_id")

apps/api/src/routes/auth.ts

@@ -28,7 +28,6 @@ import type { TdkApiContext } from "../types";
 import {
   USER_NOTIFICATION_SETTINGS_SELECT_FIELDS,
   USER_PROFILE_SELECT_FIELDS,
-  USER_PUBLIC_PROFILE_SELECT_FIELDS,
   USER_SELECT_FIELDS,
   USER_SMART_ACCOUNT_SELECT_FIELDS,
   USER_SOCIAL_ACCOUNT_SELECT_FIELDS,
@@ -37,6 +36,7 @@ import { throwUnauthorizedError, throwUserNotFoundError } from "../utils/error";
 import { log } from "../utils/log";
 import {
   getThirdwebUser,
+  migrateLegacyUser,
   parseThirdwebUserEmail,
   transformUserProfileResponseFields,
 } from "../utils/user";
@@ -173,7 +173,7 @@ export const authRoutes =
           });
         }
 
-        const [user, profile, legacyUserProfiles] = await Promise.all([
+        const [user, profile, legacyProfiles] = await Promise.all([
           // Fetch user
           db.user.findUnique({
             where: { id: userSmartAccount.userId },
@@ -207,17 +207,12 @@ export const authRoutes =
                   legacyEmail: userSmartAccount.initialEmail,
                   legacyEmailVerifiedAt: { not: null },
                 },
-                select: {
-                  ...USER_PUBLIC_PROFILE_SELECT_FIELDS,
-                  id: true,
-                },
               })
             : userSmartAccount.initialWalletAddress
-              ? db.userProfile.findUnique({
+              ? db.userProfile.findMany({
                   where: {
                     legacyAddress: userSmartAccount.initialWalletAddress,
                   },
-                  select: { ...USER_PUBLIC_PROFILE_SELECT_FIELDS, id: true },
                 })
               : [],
         ]);
@@ -268,20 +263,35 @@ export const authRoutes =
           endTimestamp: session.endTimestamp.toString(),
         }));
 
+        let updatedProfile: typeof profile | undefined;
+
+        if (env.USER_MIGRATION_ENABLED) {
+          // Automatically migrate legacy profile if only one exists
+          if (legacyProfiles.length === 1 && !!legacyProfiles[0]) {
+            const result = await migrateLegacyUser({
+              db,
+              userId: user.id,
+              userProfileId: profile.id,
+              legacyProfile: legacyProfiles[0],
+            });
+            updatedProfile = result.updatedProfile;
+          }
+        }
+
+        const finalProfile = updatedProfile ?? profile;
         reply.send({
           token: authTokenResult.value,
           user: {
             ...user,
-            ...profile,
-            ...transformUserProfileResponseFields(profile),
+            ...finalProfile,
+            ...transformUserProfileResponseFields(finalProfile),
             address,
             sessions,
           },
-          legacyProfiles: legacyUserProfiles
-            ? Array.isArray(legacyUserProfiles)
-              ? legacyUserProfiles
-              : [legacyUserProfiles]
-            : [],
+          legacyProfiles:
+            env.USER_MIGRATION_ENABLED && legacyProfiles.length > 1
+              ? legacyProfiles
+              : [],
         });
       },
     );