import { MigrationHelper } from "../migration-helper";
import { Direction, IRREVERSIBLE, Migrator } from "../migrator";

type ExpectedAccountType = { keys?: { legacyEtmKey?: string } };

export class RemoveLegacyEtmKeyMigrator extends Migrator<5, 6> {
  async migrate(helper: MigrationHelper): Promise<void> {
    const accounts = await helper.getAccounts<ExpectedAccountType>();

    async function updateAccount(userId: string, account: ExpectedAccountType) {
      if (account?.keys?.legacyEtmKey) {
        delete account.keys.legacyEtmKey;
        await helper.set(userId, account);
      }
    }

    await Promise.all(accounts.map(({ userId, account }) => updateAccount(userId, account)));
  }

  async rollback(helper: MigrationHelper): Promise<void> {
    throw IRREVERSIBLE;
  }

  // Override is necessary because default implementation assumes `stateVersion` at the root, but for this version
  // it is nested inside a global object.
  override async updateVersion(helper: MigrationHelper, direction: Direction): Promise<void> {
    const endVersion = direction === "up" ? this.toVersion : this.fromVersion;
    helper.currentVersion = endVersion;
    const global: { stateVersion: number } = (await helper.get("global")) || ({} as any);
    await helper.set("global", { ...global, stateVersion: endVersion });
  }
}