import { MigrateDownArgs, MigrateUpArgs, sql } from '@payloadcms/db-postgres'

const columns = [
  'progression_minigames_aim_trainer_games_played',
  'progression_minigames_aim_trainer_best_score',
  'progression_minigames_aim_trainer_recruit_best_score',
  'progression_minigames_aim_trainer_veteran_best_score',
  'progression_minigames_aim_trainer_elite_best_score',
  'progression_minigames_aim_trainer_recruit_best_accuracy',
  'progression_minigames_aim_trainer_veteran_best_accuracy',
  'progression_minigames_aim_trainer_elite_best_accuracy',
  'progression_minigames_aim_trainer_xp_earned',
] as const

export async function up({ db }: MigrateUpArgs): Promise<void> {
  for (const column of columns) {
    await db.execute(sql.raw(`ALTER TABLE "profiles" ADD COLUMN "${column}" numeric DEFAULT 0;`))
  }
}

export async function down({ db }: MigrateDownArgs): Promise<void> {
  for (const column of [...columns].reverse()) {
    await db.execute(sql.raw(`ALTER TABLE "profiles" DROP COLUMN "${column}";`))
  }
}