From 79484f85e754e8579c5aca505ad985aee72f8bb4 Mon Sep 17 00:00:00 2001
From: Paul Beslin <paul.beslin@gmail.com>
Date: Wed, 1 May 2024 12:03:11 +0200
Subject: [PATCH] Implement move Punishment

---
 src/data/move.ts | 25 +++++++++++++++++++++++--
 1 file changed, 23 insertions(+), 2 deletions(-)

diff --git a/src/data/move.ts b/src/data/move.ts
index 60e07b8f5..f5ff10edb 100644
--- a/src/data/move.ts
+++ b/src/data/move.ts
@@ -1986,9 +1986,13 @@ export class HitCountPowerAttr extends VariablePowerAttr {
   }
 }
 
+const countPositiveStats = (pokemon: Pokemon): number => {
+  return pokemon.summonData.battleStats.reduce((total, stat) => stat > 0 && stat ? total + stat : total, 0);
+}
+
 export class StatChangeCountPowerAttr extends VariablePowerAttr {
   apply(user: Pokemon, target: Pokemon, move: Move, args: any[]): boolean {
-    const positiveStats: number = user.summonData.battleStats.reduce((total, stat) => stat > 0 && stat ? total + stat : total, 0);
+    const positiveStats: number = countPositiveStats(user);
 
     (args[0] as Utils.NumberHolder).value += positiveStats * 20;
 
@@ -1996,6 +2000,22 @@ export class StatChangeCountPowerAttr extends VariablePowerAttr {
   }
 }
 
+export class PunishmentPowerAttr extends VariablePowerAttr {
+  PUNISHMENT_MIN_BASE_POWER = 60;
+  PUNISHMENT_MAX_BASE_POWER = 200;
+
+  apply(user: Pokemon, target: Pokemon, move: Move, args: any[]): boolean {
+    const positiveStats: number = countPositiveStats(target);
+
+    (args[0] as Utils.NumberHolder).value = Math.min(
+      this.PUNISHMENT_MAX_BASE_POWER,
+      this.PUNISHMENT_MIN_BASE_POWER + positiveStats * 20
+    );
+
+    return true;
+  }
+}
+
 export class PresentPowerAttr extends VariablePowerAttr {
   apply(user: Pokemon, target: Pokemon, move: Move, args: any[]): boolean {
 
@@ -4821,7 +4841,8 @@ export function initMoves() {
     new StatusMove(Moves.GUARD_SWAP, Type.PSYCHIC, -1, 10, -1, 0, 4)
       .unimplemented(),
     new AttackMove(Moves.PUNISHMENT, Type.DARK, MoveCategory.PHYSICAL, -1, 100, 5, -1, 0, 4)
-      .unimplemented(),
+      .attr(PunishmentPowerAttr)
+      .makesContact(),
     new AttackMove(Moves.LAST_RESORT, Type.NORMAL, MoveCategory.PHYSICAL, 140, 100, 5, -1, 0, 4)
       .condition((user, target, move) => {
         const uniqueUsedMoveIds = new Set<Moves>();