import { getPokemonSpecies, getPokemonSpeciesForm } from "#app/data/pokemon-species";
import { globalScene } from "#app/global-scene";
import { DexAttr, type SessionSaveData, type SystemSaveData } from "#app/system/game-data";
import * as Utils from "#app/utils";

export const systemMigrators = [
  /**
   * If a starter is caught, but the only forms registered as caught are not starterSelectable,
   * unlock the default form.
   * @param data {@linkcode SystemSaveData}
   */
  function migrateUnselectableForms(data: SystemSaveData) {
    if (data.starterData && data.dexData) {
      Object.keys(data.starterData).forEach(sd => {
        const caughtAttr = data.dexData[sd]?.caughtAttr;
        const speciesNumber = Number(sd);
        if (!speciesNumber) { // An unknown bug at some point in time caused some accounts to have starter data for pokedex number 0 which crashes
          return;
        }
        const species = getPokemonSpecies(speciesNumber);
        if (caughtAttr && species.forms?.length > 1) {
          const selectableForms = species.forms.filter((form, formIndex) => form.isStarterSelectable && (caughtAttr & globalScene.gameData.getFormAttr(formIndex)));
          if (selectableForms.length === 0) {
            data.dexData[sd].caughtAttr += DexAttr.DEFAULT_FORM;
          }
        }
      });
    }
  },
] as const;

export const settingsMigrators = [] as const;

export const sessionMigrators = [
  function migrateTera(data: SessionSaveData) {
    for (let i = 0; i < data.modifiers.length;) {
      if (data.modifiers[i].className === "TerastallizeModifier") {
        data.party.forEach((p) => {
          if (p.id === data.modifiers[i].args[0]) {
            p.teraType = data.modifiers[i].args[1];
          }
        });
        data.modifiers.splice(i, 1);
      } else {
        i++;
      }
    }

    for (let i = 0; i < data.enemyModifiers.length;) {
      if (data.enemyModifiers[i].className === "TerastallizeModifier") {
        data.enemyParty.forEach((p) => {
          if (p.id === data.enemyModifiers[i].args[0]) {
            p.teraType = data.enemyModifiers[i].args[1];
          }
        });
        data.enemyModifiers.splice(i, 1);
      } else {
        i++;
      }
    }

    data.party.forEach(p => {
      if (Utils.isNullOrUndefined(p.teraType)) {
        p.teraType = getPokemonSpeciesForm(p.species, p.formIndex).type1;
      }
    });

    data.enemyParty.forEach(p => {
      if (Utils.isNullOrUndefined(p.teraType)) {
        p.teraType = getPokemonSpeciesForm(p.species, p.formIndex).type1;
      }
    });
  }
] as const;