pokerogue/src/system/pokemon-data.ts

import { BattleType } from "../battle";
import { globalScene } from "#app/global-scene";
import type { Gender } from "../data/gender";
import type { Nature } from "#enums/nature";
import type { PokeballType } from "#enums/pokeball";
import { getPokemonSpecies, getPokemonSpeciesForm } from "../data/pokemon-species";
import { Status } from "../data/status-effect";
import Pokemon, { EnemyPokemon, PokemonMove, PokemonSummonData } from "../field/pokemon";
import { TrainerSlot } from "../data/trainer-config";
import type { Variant } from "#app/data/variant";
import { loadBattlerTag } from "../data/battler-tags";
import type { Biome } from "#enums/biome";
import { Moves } from "#enums/moves";
import type { Species } from "#enums/species";
import { CustomPokemonData } from "#app/data/custom-pokemon-data";
import type { PokemonType } from "#enums/pokemon-type";
import { getSpeciesFormChangeMessage } from "#app/data/pokemon-forms";

export default class PokemonData {
  public id: number;
  public player: boolean;
  public species: Species;
  public nickname: string;
  public formIndex: number;
  public abilityIndex: number;
  public passive: boolean;
  public shiny: boolean;
  public variant: Variant;
  public pokeball: PokeballType;
  public level: number;
  public exp: number;
  public levelExp: number;
  public gender: Gender;
  public hp: number;
  public stats: number[];
  public ivs: number[];
  public nature: Nature;
  public moveset: (PokemonMove | null)[];
  public status: Status | null;
  public friendship: number;
  public metLevel: number;
  public metBiome: Biome | -1; // -1 for starters
  public metSpecies: Species;
  public metWave: number; // 0 for unknown (previous saves), -1 for starters
  public luck: number;
  public pauseEvolutions: boolean;
  public pokerus: boolean;
  public usedTMs: Moves[];
  public evoCounter: number;
  public teraType: PokemonType;
  public isTerastallized: boolean;
  public stellarTypesBoosted: PokemonType[];

  public fusionSpecies: Species;
  public fusionFormIndex: number;
  public fusionAbilityIndex: number;
  public fusionShiny: boolean;
  public fusionVariant: Variant;
  public fusionGender: Gender;
  public fusionLuck: number;
  public fusionTeraType: PokemonType;

  public boss: boolean;
  public bossSegments?: number;

  public summonData: PokemonSummonData;
  public summonDataSpeciesFormIndex: number;

  /** Data that can customize a Pokemon in non-standard ways from its Species */
  public customPokemonData: CustomPokemonData;
  public fusionCustomPokemonData: CustomPokemonData;

  // Deprecated attributes, needed for now to allow SessionData migration (see PR#4619 comments)
  public natureOverride: Nature | -1;
  public mysteryEncounterPokemonData: CustomPokemonData | null;
  public fusionMysteryEncounterPokemonData: CustomPokemonData | null;

  constructor(source: Pokemon | any, forHistory = false) {
    const sourcePokemon = source instanceof Pokemon ? source : null;
    this.id = source.id;
    this.player = sourcePokemon ? sourcePokemon.isPlayer() : source.player;
    this.species = sourcePokemon ? sourcePokemon.species.speciesId : source.species;
    this.nickname = sourcePokemon ? sourcePokemon.nickname : source.nickname;
    this.formIndex = Math.max(Math.min(source.formIndex, getPokemonSpecies(this.species).forms.length - 1), 0);
    this.abilityIndex = source.abilityIndex;
    this.passive = source.passive;
    this.shiny = source.shiny;
    this.variant = source.variant;
    this.pokeball = source.pokeball;
    this.level = source.level;
    this.exp = source.exp;
    if (!forHistory) {
      this.levelExp = source.levelExp;
    }
    this.gender = source.gender;
    if (!forHistory) {
      this.hp = source.hp;
    }
    this.stats = source.stats;
    this.ivs = source.ivs;
    this.nature = source.nature !== undefined ? source.nature : (0 as Nature);
    this.friendship =
      source.friendship !== undefined ? source.friendship : getPokemonSpecies(this.species).baseFriendship;
    this.metLevel = source.metLevel || 5;
    this.metBiome = source.metBiome !== undefined ? source.metBiome : -1;
    this.metSpecies = source.metSpecies;
    this.metWave = source.metWave ?? (this.metBiome === -1 ? -1 : 0);
    this.luck = source.luck !== undefined ? source.luck : source.shiny ? source.variant + 1 : 0;
    if (!forHistory) {
      this.pauseEvolutions = !!source.pauseEvolutions;
      this.evoCounter = source.evoCounter ?? 0;
    }
    this.pokerus = !!source.pokerus;
    this.teraType = source.teraType as PokemonType;
    this.isTerastallized = source.isTerastallized || false;
    this.stellarTypesBoosted = source.stellarTypesBoosted || [];

    this.fusionSpecies = sourcePokemon ? sourcePokemon.fusionSpecies?.speciesId : source.fusionSpecies;
    this.fusionFormIndex = source.fusionFormIndex;
    this.fusionAbilityIndex = source.fusionAbilityIndex;
    this.fusionShiny = source.fusionShiny;
    this.fusionVariant = source.fusionVariant;
    this.fusionGender = source.fusionGender;
    this.fusionLuck =
      source.fusionLuck !== undefined ? source.fusionLuck : source.fusionShiny ? source.fusionVariant + 1 : 0;
    this.fusionCustomPokemonData = new CustomPokemonData(source.fusionCustomPokemonData);
    this.fusionTeraType = (source.fusionTeraType ?? 0) as PokemonType;
    this.usedTMs = source.usedTMs ?? [];

    this.customPokemonData = new CustomPokemonData(source.customPokemonData);

    // Deprecated, but needed for session data migration
    this.natureOverride = source.natureOverride;
    this.mysteryEncounterPokemonData = source.mysteryEncounterPokemonData
      ? new CustomPokemonData(source.mysteryEncounterPokemonData)
      : null;
    this.fusionMysteryEncounterPokemonData = source.fusionMysteryEncounterPokemonData
      ? new CustomPokemonData(source.fusionMysteryEncounterPokemonData)
      : null;

    if (!forHistory) {
      this.boss = (source instanceof EnemyPokemon && !!source.bossSegments) || (!this.player && !!source.boss);
      this.bossSegments = source.bossSegments;
    }

    if (sourcePokemon) {
      this.moveset = sourcePokemon.moveset;
      if (!forHistory) {
        this.status = sourcePokemon.status;
        if (this.player && sourcePokemon.summonData) {
          this.summonData = sourcePokemon.summonData;
          this.summonDataSpeciesFormIndex = this.getSummonDataSpeciesFormIndex();
        }
      }
    } else {
      this.moveset = (source.moveset || [new PokemonMove(Moves.TACKLE), new PokemonMove(Moves.GROWL)])
        .filter(m => m)
        .map((m: any) => new PokemonMove(m.moveId, m.ppUsed, m.ppUp, m.virtual, m.maxPpOverride));
      if (!forHistory) {
        this.status = source.status
          ? new Status(source.status.effect, source.status.toxicTurnCount, source.status.sleepTurnsRemaining)
          : null;
      }

      this.summonData = new PokemonSummonData();
      if (!forHistory && source.summonData) {
        this.summonData.stats = source.summonData.stats;
        this.summonData.statStages = source.summonData.statStages;
        this.summonData.moveQueue = source.summonData.moveQueue;
        this.summonData.abilitySuppressed = source.summonData.abilitySuppressed;
        this.summonData.abilitiesApplied = source.summonData.abilitiesApplied;

        this.summonData.ability = source.summonData.ability;
        this.summonData.moveset = source.summonData.moveset?.map(m => PokemonMove.loadMove(m));
        this.summonData.types = source.summonData.types;
        this.summonData.speciesForm = source.summonData.speciesForm;
        this.summonDataSpeciesFormIndex = source.summonDataSpeciesFormIndex;

        if (source.summonData.tags) {
          this.summonData.tags = source.summonData.tags?.map(t => loadBattlerTag(t));
        } else {
          this.summonData.tags = [];
        }
      }
    }
  }

  toPokemon(battleType?: BattleType, partyMemberIndex = 0, double = false): Pokemon {
    const species = getPokemonSpecies(this.species);
    const ret: Pokemon = this.player
      ? globalScene.addPlayerPokemon(
          species,
          this.level,
          this.abilityIndex,
          this.formIndex,
          this.gender,
          this.shiny,
          this.variant,
          this.ivs,
          this.nature,
          this,
          playerPokemon => {
            if (this.nickname) {
              playerPokemon.nickname = this.nickname;
            }
          },
        )
      : globalScene.addEnemyPokemon(
          species,
          this.level,
          battleType === BattleType.TRAINER
            ? !double || !(partyMemberIndex % 2)
              ? TrainerSlot.TRAINER
              : TrainerSlot.TRAINER_PARTNER
            : TrainerSlot.NONE,
          this.boss,
          false,
          this,
        );
    if (this.summonData) {
      // when loading from saved session, recover summonData.speciesFrom and form index species object
      // used to stay transformed on reload session
      if (this.summonData.speciesForm) {
        this.summonData.speciesForm = getPokemonSpeciesForm(
          this.summonData.speciesForm.speciesId,
          this.summonDataSpeciesFormIndex,
        );
      }
      ret.primeSummonData(this.summonData);
    }
    return ret;
  }

  /**
   * Method to save summon data species form index
   * Necessary in case the pokemon is transformed
   * to reload the correct form
   */
  getSummonDataSpeciesFormIndex(): number {
    if (this.summonData.speciesForm) {
      return this.summonData.speciesForm.formIndex;
    }
    return 0;
  }
}