pokerogue/src/data/status-effect.ts

import * as Utils from "../utils";

export enum StatusEffect {
  NONE,
  POISON,
  TOXIC,
  PARALYSIS,
  SLEEP,
  FREEZE,
  BURN,
  FAINT
}

export class Status {
  public effect: StatusEffect;
  public turnCount: integer;
  public cureTurn: integer;

  constructor(effect: StatusEffect, turnCount: integer = 0, cureTurn?: integer) {
    this.effect = effect;
    this.turnCount = turnCount === undefined ? 0 : turnCount;
    this.cureTurn = cureTurn;
  }

  incrementTurn(): void {
    this.turnCount++;
  }

  isPostTurn(): boolean {
    return this.effect === StatusEffect.POISON || this.effect === StatusEffect.TOXIC || this.effect === StatusEffect.BURN;
  }
}

export function getStatusEffectObtainText(statusEffect: StatusEffect, sourceText?: string): string {
  const sourceClause = sourceText ? ` ${statusEffect !== StatusEffect.SLEEP ? "by" : "from"} ${sourceText}` : "";
  switch (statusEffect) {
  case StatusEffect.POISON:
    return `\nwas poisoned${sourceClause}!`;
  case StatusEffect.TOXIC:
    return `\nwas badly poisoned${sourceClause}!`;
  case StatusEffect.PARALYSIS:
    return ` was paralyzed${sourceClause}!\nIt may be unable to move!`;
  case StatusEffect.SLEEP:
    return `\nfell asleep${sourceClause}!`;
  case StatusEffect.FREEZE:
    return `\nwas frozen solid${sourceClause}!`;
  case StatusEffect.BURN:
    return `\nwas burned${sourceClause}!`;
  }

  return "";
}

export function getStatusEffectActivationText(statusEffect: StatusEffect): string {
  switch (statusEffect) {
  case StatusEffect.POISON:
  case StatusEffect.TOXIC:
    return " is hurt\nby poison!";
  case StatusEffect.PARALYSIS:
    return " is paralyzed!\nIt can't move!";
  case StatusEffect.SLEEP:
    return " is fast asleep.";
  case StatusEffect.FREEZE:
    return " is\nfrozen solid!";
  case StatusEffect.BURN:
    return " is hurt\nby its burn!";
  }

  return "";
}

export function getStatusEffectOverlapText(statusEffect: StatusEffect): string {
  switch (statusEffect) {
  case StatusEffect.POISON:
  case StatusEffect.TOXIC:
    return " is\nalready poisoned!";
  case StatusEffect.PARALYSIS:
    return " is\nalready paralyzed!";
  case StatusEffect.SLEEP:
    return " is\nalready asleep!";
  case StatusEffect.FREEZE:
    return " is\nalready frozen!";
  case StatusEffect.BURN:
    return " is\nalready burned!";
  }

  return "";
}

export function getStatusEffectHealText(statusEffect: StatusEffect): string {
  switch (statusEffect) {
  case StatusEffect.POISON:
  case StatusEffect.TOXIC:
    return " was\ncured of its poison!";
  case StatusEffect.PARALYSIS:
    return " was\nhealed of paralysis!";
  case StatusEffect.SLEEP:
    return " woke up!";
  case StatusEffect.FREEZE:
    return " was\ndefrosted!";
  case StatusEffect.BURN:
    return " was\nhealed of its burn!";
  }

  return "";
}

export function getStatusEffectDescriptor(statusEffect: StatusEffect): string {
  switch (statusEffect) {
  case StatusEffect.POISON:
  case StatusEffect.TOXIC:
    return "poisoning";
  case StatusEffect.PARALYSIS:
    return "paralysis";
  case StatusEffect.SLEEP:
    return "sleep";
  case StatusEffect.FREEZE:
    return "freezing";
  case StatusEffect.BURN:
    return "burn";
  }
}

export function getStatusEffectCatchRateMultiplier(statusEffect: StatusEffect): number {
  switch (statusEffect) {
  case StatusEffect.POISON:
  case StatusEffect.TOXIC:
  case StatusEffect.PARALYSIS:
  case StatusEffect.BURN:
    return 1.5;
  case StatusEffect.SLEEP:
  case StatusEffect.FREEZE:
    return 2.5;
  }

  return 1;
}

/**
* Returns a random non-volatile StatusEffect
*/
export function generateRandomStatusEffect(): StatusEffect {
  return Utils.randIntRange(1, 6);
}

/**
* Returns a random non-volatile StatusEffect between the two provided
* @param statusEffectA The first StatusEffect
* @param statusEffectA The second StatusEffect
*/
export function getRandomStatusEffect(statusEffectA: StatusEffect, statusEffectB: StatusEffect): StatusEffect {
  if (statusEffectA === StatusEffect.NONE || statusEffectA === StatusEffect.FAINT) {
    return statusEffectB;
  }
  if (statusEffectB === StatusEffect.NONE || statusEffectB === StatusEffect.FAINT) {
    return statusEffectA;
  }

  return Utils.randIntRange(0, 2) ? statusEffectA : statusEffectB;
}

/**
* Returns a random non-volatile StatusEffect between the two provided
* @param statusA The first Status
* @param statusB The second Status
*/
export function getRandomStatus(statusA: Status, statusB: Status): Status {  
  if (statusA === undefined || statusA.effect === StatusEffect.NONE || statusA.effect === StatusEffect.FAINT) {
    return statusB;
  }
  if (statusB === undefined || statusB.effect === StatusEffect.NONE || statusB.effect === StatusEffect.FAINT) {
    return statusA;
  }
  

  return Utils.randIntRange(0, 2) ? statusA : statusB;
}