pokerogue/src/data/terrain.ts

import Pokemon from "../field/pokemon";
import Move from "./move";
import { Type } from "./type";
import * as Utils from "../utils";
import { IncrementMovePriorityAbAttr, applyAbAttrs } from "./ability";
import { ProtectAttr } from "./move";
import { BattlerIndex } from "#app/battle.js";

export enum TerrainType {
  NONE,
  MISTY,
  ELECTRIC,
  GRASSY,
  PSYCHIC
}

export class Terrain {
  public terrainType: TerrainType;
  public turnsLeft: integer;

  constructor(terrainType: TerrainType, turnsLeft?: integer) {
    this.terrainType = terrainType;
    this.turnsLeft = turnsLeft || 0;
  }

  lapse(): boolean {
    if (this.turnsLeft) {
      return !!--this.turnsLeft;
    }

    return true;
  }

  getAttackTypeMultiplier(attackType: Type): number {
    switch (this.terrainType) {
    case TerrainType.ELECTRIC:
      if (attackType === Type.ELECTRIC) {
        return 1.3;
      }
      break;
    case TerrainType.GRASSY:
      if (attackType === Type.GRASS) {
        return 1.3;
      }
      break;
    case TerrainType.PSYCHIC:
      if (attackType === Type.PSYCHIC) {
        return 1.3;
      }
      break;
    }

    return 1;
  }

  isMoveTerrainCancelled(user: Pokemon, targets: BattlerIndex[], move: Move): boolean {
    switch (this.terrainType) {
    case TerrainType.PSYCHIC:
      if (!move.hasAttr(ProtectAttr)) {
        const priority = new Utils.IntegerHolder(move.priority);
        applyAbAttrs(IncrementMovePriorityAbAttr, user, null, move, priority);
        return priority.value > 0 && user.getOpponents().filter(o => targets.includes(o.getBattlerIndex())).length > 0;
      }
    }

    return false;
  }
}

export function getTerrainColor(terrainType: TerrainType): [ integer, integer, integer ] {
  switch (terrainType) {
  case TerrainType.MISTY:
    return [ 232, 136, 200 ];
  case TerrainType.ELECTRIC:
    return [ 248, 248, 120 ];
  case TerrainType.GRASSY:
    return [ 120, 200, 80 ];
  case TerrainType.PSYCHIC:
    return [ 160, 64, 160 ];
  }

  return [ 0, 0, 0 ];
}