tweak script

bluesky-social · Nov 8, 2023 · 0bc2542 · 0bc2542
1 parent 1a99165
commit 0bc2542
Showing 1 changed file with 25 additions and 18 deletions.
diff --git a/packages/pds/src/migrate-script/migrate-all.ts b/packages/pds/src/migrate-script/migrate-all.ts
@@ -110,6 +110,12 @@ const migrateRepo = async (
       adminHeaders,
     )
     status.importedRev = importedRev
+    status.phase = TransferPhase.transferredPds
+    await updateStatus(db, status)
+  }
+
+  if (status.phase < TransferPhase.transferredEntryway) {
+    await updatePdsOnEntryway(ctx, pds, status.did)
     status.phase = TransferPhase.transferredEntryway
     await updateStatus(db, status)
   }
@@ -271,30 +277,31 @@ const lockAndTransfer = async (
       { headers: adminHeaders },
     )
 
-    status.phase = TransferPhase.transferredPds
-    await updateStatus(db, status)
-
-    transferDefer.resolve()
-
-    await ctx.db.transaction(async (dbTxn) => {
-      await dbTxn.db
-        .updateTable('user_account')
-        .where('did', '=', status.did)
-        .set({ pdsId: pds.id })
-        .execute()
-      await dbTxn.db
-        .updateTable('repo_root')
-        .where('did', '=', status.did)
-        .set({ did: `migrated-${status.did}` })
-        .execute()
-    })
-
     return importedRev
   } finally {
     transferDefer.resolve()
   }
 }
 
+const updatePdsOnEntryway = async (
+  ctx: AppContext,
+  pds: PdsInfo,
+  did: string,
+) => {
+  await ctx.db.transaction(async (dbTxn) => {
+    await dbTxn.db
+      .updateTable('user_account')
+      .where('did', '=', did)
+      .set({ pdsId: pds.id })
+      .execute()
+    await dbTxn.db
+      .updateTable('repo_root')
+      .where('did', '=', did)
+      .set({ did: `migrated-${did}` })
+      .execute()
+  })
+}
+
 const transferPreferences = async (
   ctx: AppContext,
   pds: PdsInfo,