From 3031cc1245fcf2b88f8b71451f6b50df8fe3adf3 Mon Sep 17 00:00:00 2001
From: Frederico Santos <frederico.f.santos@tecnico.ulisboa.pt>
Date: Sun, 19 May 2024 17:13:33 +0100
Subject: [PATCH] Implement Ball Fetch (#1129)

* Implemented Ball Fetch

* Added once per battle condition

* Ability only functions on player side

* Update ability.ts

---------

Co-authored-by: SamuelHudson <samuel.hudson2017@gmail.com>
Co-authored-by: Benjamin Odom <bennybroseph@gmail.com>
---
 src/battle.ts       |  3 +++
 src/data/ability.ts | 32 ++++++++++++++++++++++++++++++--
 src/phases.ts       |  1 +
 3 files changed, 34 insertions(+), 2 deletions(-)

diff --git a/src/battle.ts b/src/battle.ts
index 580bad9a508..601cbeb9cb6 100644
--- a/src/battle.ts
+++ b/src/battle.ts
@@ -11,6 +11,7 @@ import { BattleSpec } from "./enums/battle-spec";
 import { PlayerGender } from "./system/game-data";
 import { MoneyMultiplierModifier, PokemonHeldItemModifier } from "./modifier/modifier";
 import { MoneyAchv } from "./system/achv";
+import { PokeballType } from "./data/pokeball";
 
 export enum BattleType {
     WILD,
@@ -61,6 +62,7 @@ export default class Battle {
     public battleSeed: string;
     private battleSeedState: string;
     public moneyScattered: number;
+    public lastUsedPokeball: PokeballType;
 
     private rngCounter: integer = 0;
 
@@ -86,6 +88,7 @@ export default class Battle {
         this.battleSeed = Utils.randomString(16, true);
         this.battleSeedState = null;
         this.moneyScattered = 0;
+        this.lastUsedPokeball = null;
     }
 
     private initBattleSpec(): void {
diff --git a/src/data/ability.ts b/src/data/ability.ts
index 519e2411612..b5e4a91a11c 100644
--- a/src/data/ability.ts
+++ b/src/data/ability.ts
@@ -22,6 +22,7 @@ import i18next, { Localizable } from "#app/plugins/i18n.js";
 import { Command } from "../ui/command-ui-handler";
 import Battle from "#app/battle.js";
 import { ability } from "#app/locales/en/ability.js";
+import { PokeballType, getPokeballName } from "./pokeball";
 
 export class Ability implements Localizable {
   public id: Abilities;
@@ -2276,6 +2277,33 @@ export class PostTurnFormChangeAbAttr extends PostTurnAbAttr {
   }
 }
 
+/** 
+ * Grabs the last failed Pokeball used 
+ * @extends PostTurnAbAttr 
+ * @see {@linkcode applyPostTurn} */
+export class FetchBallAbAttr extends PostTurnAbAttr {
+  constructor() {
+    super();
+  }
+  /** 
+   * Adds the last used Pokeball back into the player's inventory 
+   * @param pokemon {@linkcode Pokemon} with this ability 
+   * @param passive N/A 
+   * @param args N/A 
+   * @returns true if player has used a pokeball and this pokemon is owned by the player 
+   */
+  applyPostTurn(pokemon: Pokemon, passive: boolean, args: any[]): boolean {
+    let lastUsed = pokemon.scene.currentBattle.lastUsedPokeball;
+    if(lastUsed != null && pokemon.isPlayer) {
+      pokemon.scene.pokeballCounts[lastUsed]++;
+      pokemon.scene.currentBattle.lastUsedPokeball = null;
+      pokemon.scene.queueMessage(getPokemonMessage(pokemon, ` found a\n${getPokeballName(lastUsed)}!`));
+      return true;
+    }
+    return false;
+  }
+}
+
 export class PostBiomeChangeAbAttr extends AbAttr { }
 
 export class PostBiomeChangeWeatherChangeAbAttr extends PostBiomeChangeAbAttr {
@@ -2644,7 +2672,6 @@ export class SuppressFieldAbilitiesAbAttr extends AbAttr {
   }
 }
 
-
 export class AlwaysHitAbAttr extends AbAttr { }
 
 export class UncopiableAbilityAbAttr extends AbAttr {
@@ -3627,7 +3654,8 @@ export function initAbilities() {
     new Ability(Abilities.LIBERO, 8)
       .unimplemented(),
     new Ability(Abilities.BALL_FETCH, 8)
-      .unimplemented(),
+      .attr(FetchBallAbAttr)
+      .condition(getOncePerBattleCondition(Abilities.BALL_FETCH)),
     new Ability(Abilities.COTTON_DOWN, 8)
       .attr(PostDefendStatChangeAbAttr, (target, user, move) => move.category !== MoveCategory.STATUS, BattleStat.SPD, -1, false, true)
       .bypassFaint(),
diff --git a/src/phases.ts b/src/phases.ts
index 63573b28e06..1d4d0ba5f34 100644
--- a/src/phases.ts
+++ b/src/phases.ts
@@ -4350,6 +4350,7 @@ export class AttemptCapturePhase extends PokemonPhase {
       scale: 1
     });
     
+    this.scene.currentBattle.lastUsedPokeball = this.pokeballType;
     this.removePb();
     this.end();
   }