pokerogue/src/items/held-items/base-stat-flat.ts

import type Pokemon from "#app/field/pokemon";
import type { HeldItemId } from "#enums/held-item-id";
import { HeldItem, ITEM_EFFECT } from "../held-item";
import { getStatKey, type Stat } from "#enums/stat";
import i18next from "i18next";

export interface BASE_STAT_FLAT_PARAMS {
  /** The pokemon with the item */
  pokemon: Pokemon;
  /** The amount of exp to gain */
  baseStats: number[];
}

export interface BASE_STAT_FLAT_DATA {
  statModifier: number;
  stats: Stat[];
}

/**
 * Currently used by Old Gateau item
 */
export class BaseStatFlatHeldItem extends HeldItem {
  public effects: ITEM_EFFECT[] = [ITEM_EFFECT.BASE_STAT_FLAT];
  public isTransferable = false;

  constructor(type: HeldItemId, maxStackCount = 1) {
    super(type, maxStackCount);
  }

  get description(): string {
    return i18next.t("modifierType:ModifierType.PokemonBaseStatFlatModifierType.description", {
      stats: this.stats.map(stat => i18next.t(getStatKey(stat))).join("/"),
      statValue: this.statModifier,
    });
  }

  /**
   * Checks if the {@linkcode PokemonBaseStatFlatModifier} should be applied to the {@linkcode Pokemon}.
   * @param pokemon The {@linkcode Pokemon} that holds the item
   * @param baseStats The base stats of the {@linkcode Pokemon}
   * @returns `true` if the {@linkcode PokemonBaseStatFlatModifier} should be applied
   */
  //  override shouldApply(pokemon?: Pokemon, baseStats?: number[]): boolean {
  //    return super.shouldApply(pokemon, baseStats) && Array.isArray(baseStats);
  //  }

  /**
   * Applies the {@linkcode PokemonBaseStatFlatModifier}
   * @param _pokemon The {@linkcode Pokemon} that holds the item
   * @param baseStats The base stats of the {@linkcode Pokemon}
   * @returns always `true`
   */
  apply(params: BASE_STAT_FLAT_PARAMS): boolean {
    const pokemon = params.pokemon;
    const itemData = pokemon.heldItemManager.heldItems[this.type]?.data as BASE_STAT_FLAT_DATA;
    if (!itemData) {
      return false;
    }
    const statModifier = itemData.statModifier;
    const stats = itemData.stats;
    const baseStats = params.baseStats;
    // Modifies the passed in baseStats[] array by a flat value, only if the stat is specified in this.stats
    baseStats.forEach((v, i) => {
      if (stats.includes(i)) {
        const newVal = Math.floor(v + statModifier);
        baseStats[i] = Math.min(Math.max(newVal, 1), 999999);
      }
    });

    return true;
  }
}