From 23ab346234063e405bf39cfee1a967b0e1d61d80 Mon Sep 17 00:00:00 2001
From: Ben Hollis <ben@benhollis.net>
Date: Wed, 6 May 2026 00:41:50 -0700
Subject: [PATCH] Switch a bit more default to stately

---
 api/db/migration-state-queries.ts    | 5 +++--
 api/stately/init/stately-backfill.ts | 9 +++++++--
 2 files changed, 10 insertions(+), 4 deletions(-)

diff --git a/api/db/migration-state-queries.ts b/api/db/migration-state-queries.ts
index 16bee48..3e554ca 100644
--- a/api/db/migration-state-queries.ts
+++ b/api/db/migration-state-queries.ts
@@ -40,12 +40,13 @@ export async function backfillMigrationState(
   client: ClientBase,
   platformMembershipId: string,
   bungieMembershipId: number | undefined,
+  state: MigrationState = MigrationState.Postgres,
 ): Promise<MigrationState> {
   const result = await client.query<{ state: MigrationState }>({
     name: 'backfill_migration_state',
     text: `insert into migration_state (platform_membership_id, membership_id, state) VALUES ($1, $2, $3)
 on conflict (platform_membership_id) do update set state = migration_state.state returning state`,
-    values: [platformMembershipId, bungieMembershipId, MigrationState.Postgres],
+    values: [platformMembershipId, bungieMembershipId, state],
   });
 
   return result.rows[0].state;
@@ -116,7 +117,7 @@ export async function getMigrationState(
     return {
       bungieMembershipId: undefined,
       platformMembershipId,
-      state: MigrationState.Stately,
+      state: MigrationState.Postgres,
       lastStateChangeAt: 0,
       attemptCount: 0,
     };
diff --git a/api/stately/init/stately-backfill.ts b/api/stately/init/stately-backfill.ts
index 85d5b44..410ce66 100644
--- a/api/stately/init/stately-backfill.ts
+++ b/api/stately/init/stately-backfill.ts
@@ -3,7 +3,7 @@ import fs from 'node:fs/promises';
 import { DatabaseError } from 'pg-protocol';
 import { closeDbPool, transaction } from '../../db/index.js';
 import { addLoadoutShareIgnoring } from '../../db/loadout-share-queries.js';
-import { backfillMigrationState } from '../../db/migration-state-queries.js';
+import { backfillMigrationState, MigrationState } from '../../db/migration-state-queries.js';
 import { getSettings, replaceSettings } from '../../db/settings-queries.js';
 import { Loadout } from '../../shapes/loadouts.js';
 import { defaultSettings } from '../../shapes/settings.js';
@@ -312,7 +312,12 @@ async function runSegment(segmentIndex: number, totalSegments: number, workerCou
     await withRetry(`${logPrefix} Backfill migration states`, async () => {
       await transaction(async (pgClient) => {
         for (const profileId of items) {
-          await backfillMigrationState(pgClient, profileId.toString(), undefined);
+          await backfillMigrationState(
+            pgClient,
+            profileId.toString(),
+            undefined,
+            MigrationState.Stately,
+          );
         }
       });
     });