From daa24b1dbfddc4ecb4504700969c58d1cb52a810 Mon Sep 17 00:00:00 2001
From: frutescens <info@laptop>
Date: Fri, 20 Sep 2024 12:42:29 -0700
Subject: [PATCH] Disabling Moves

---
 src/data/battler-tags.ts | 26 ++++++++++++++++++++++++--
 1 file changed, 24 insertions(+), 2 deletions(-)

diff --git a/src/data/battler-tags.ts b/src/data/battler-tags.ts
index 63d6ee3205e..833b5106ad5 100644
--- a/src/data/battler-tags.ts
+++ b/src/data/battler-tags.ts
@@ -2181,9 +2181,9 @@ export class ExposedTag extends BattlerTag {
 /**
  * Describes the behavior of a Heal Block Tag.
  */
-export class HealBlockTag extends BattlerTag {
+export class HealBlockTag extends MoveRestrictionBattlerTag {
   constructor() {
-    super(BattlerTagType.HEAL_BLOCK, BattlerTagLapseType.TURN_END, 5, Moves.HEAL_BLOCK);
+    super(BattlerTagType.HEAL_BLOCK, 5, Moves.HEAL_BLOCK);
   }
 
   override onAdd(pokemon: Pokemon): void {
@@ -2198,6 +2198,28 @@ export class HealBlockTag extends BattlerTag {
     return i18next.t("battle:battlerTagsHealBlock", { pokemonNameWithAffix: getPokemonNameWithAffix(pokemon) });
   }
 
+  override isMoveRestricted(move: Moves): boolean {
+    if (move.hasFlag(MoveFlags.TRIAGE_MOVE) && move.category() === MoveCategory.STATUS) {
+      return true;
+    }
+    return false;
+  }
+
+  override selectionDeniedText(pokemon: Pokemon, move: Moves): string {
+    return i18next.t("battle:moveDisabled", { moveName: allMoves[move].name });
+  }
+
+  override isMoveRestricted(move: Moves): boolean {
+    if (move.hasFlag(MoveFlags.TRIAGE_MOVE) && move.category() === MoveCategory.STATUS) {
+      return true;
+    }
+    return false;
+  }
+
+  override selectionDeniedText(pokemon: Pokemon, move: Moves): string {
+    return i18next.t("battle:moveDisabled", { moveName: allMoves[move].name });
+  }
+
   override onRemove(pokemon: Pokemon): void {
     super.onRemove(pokemon);