Implement Fight button cancelability

Author: Zarel

data/mods/gen1/conditions.ts

@@ -196,19 +196,23 @@ export const Conditions: import('../../../sim/dex-conditions').ModdedConditionDa
 		onRestart() {
 			this.effectState.duration = 2;
 		},
-		// onDisableMove(target) {
-		// 	target.maybeDisabled = true;
-		// },
+		onLockMove() {
+			// exact move doesn't matter, no move is ever actually used
+			return 'struggle';
+		},
+		onDisableMove(target) {
+			target.maybeLocked = true;
+		},
+	},
+	fakepartiallytrapped: {
+		name: 'fakepartiallytrapped',
+		// Wrap ended this turn, but you don't know that
+		// until you try to use an attack
+		duration: 2,
+		onDisableMove(target) {
+			target.maybeLocked = true;
+		},
 	},
-	// fakepartiallytrapped: {
-	// 	name: 'fakepartiallytrapped',
-	// 	// Wrap ended this turn, but you don't know that
-	// 	// until you try to use an attack
-	// 	duration: 2,
-	// 	onDisableMove(target) {
-	// 		target.maybeDisabled = true;
-	// 	},
-	// },
 	partialtrappinglock: {
 		name: 'partialtrappinglock',
 		durationCallback() {
@@ -240,18 +244,21 @@ export const Conditions: import('../../../sim/dex-conditions').ModdedConditionDa
 			this.add('move', pokemon, this.effectState.move, foe, `[from] ${this.effectState.move}`);
 			this.damage(this.effectState.damage, foe, pokemon, move);
 			if (this.effectState.duration === 1) {
-				// if (this.effectState.totalDuration !== 5) {
-				// 	pokemon.addVolatile('fakepartiallytrapped');
-				// 	foe.addVolatile('fakepartiallytrapped');
-				// }
+				if (this.effectState.totalDuration !== 5) {
+					pokemon.addVolatile('fakepartiallytrapped');
+					foe.addVolatile('fakepartiallytrapped');
+				}
 			} else {
 				foe.addVolatile('partiallytrapped', pokemon, move);
 			}
 			return false;
 		},
-		// onDisableMove(pokemon) {
-		// 	pokemon.maybeDisabled = true;
-		// },
+		onLockMove() {
+			return this.effectState.move;
+		},
+		onDisableMove(pokemon) {
+			pokemon.maybeLocked = true;
+		},
 	},
 	mustrecharge: {
 		inherit: true,

sim/pokemon.ts

@@ -127,6 +127,8 @@ export class Pokemon {
 	trapped: boolean | "hidden";
 	maybeTrapped: boolean;
 	maybeDisabled: boolean;
+	/** true = locked,  */
+	maybeLocked: boolean | null;
 
 	illusion: Pokemon | null;
 	transformed: boolean;
@@ -418,6 +420,7 @@ export class Pokemon {
 		this.trapped = false;
 		this.maybeTrapped = false;
 		this.maybeDisabled = false;
+		this.maybeLocked = false;
 
 		this.illusion = null;
 		this.transformed = false;
@@ -1048,7 +1051,7 @@ export class Pokemon {
 	}
 
 	getMoveRequestData() {
-		let lockedMove = this.getLockedMove();
+		let lockedMove = this.maybeLocked ? null : this.getLockedMove();
 
 		// Information should be restricted for the last active Pokémon
 		const isLastActive = this.isLastActive();
@@ -1068,6 +1071,9 @@ export class Pokemon {
 			if (this.maybeDisabled) {
 				data.maybeDisabled = this.maybeDisabled;
 			}
+			if (this.maybeLocked) {
+				data.maybeLocked = this.maybeLocked;
+			}
 			if (canSwitchIn) {
 				if (this.trapped === true) {
 					data.trapped = true;

sim/side.ts

@@ -91,8 +91,9 @@ export interface PokemonSwitchRequestData {
 	terastallized?: string;
 }
 export interface PokemonMoveRequestData {
-	moves: { move: string, id: ID, target?: string, disabled?: string | boolean }[];
+	moves: { move: string, id: ID, target?: string, disabled?: string | boolean, disabledSource?: string }[];
 	maybeDisabled?: boolean;
+	maybeLocked?: boolean;
 	trapped?: boolean;
 	maybeTrapped?: boolean;
 	canMegaEvo?: boolean;
@@ -639,6 +640,18 @@ export class Side {
 				moveid: lockedMoveID,
 			});
 			return true;
+		} else if (moveid === 'fight') {
+			// test fight button
+			if (!pokemon.maybeLocked) {
+				return this.emitChoiceError(`Can't move: ${pokemon.name}'s Fight button is known to be safe`);
+			}
+			pokemon.maybeLocked = false;
+			this.updateRequestForPokemon(pokemon, req => {
+				delete req.maybeLocked;
+				return true;
+			});
+			this.emitChoiceError(`Can't move: ${pokemon.name} is not locked`, true);
+			this.emitRequest(this.activeRequest!);
 		} else if (!moves.length && !zMove) {
 			// Override action and use Struggle if there are no enabled moves with PP
 			// Gen 4 and earlier announce a Pokemon has no moves left before the turn begins, and only to that player's side.
@@ -769,7 +782,7 @@ export class Side {
 		return true;
 	}
 
-	updateRequestForPokemon(pokemon: Pokemon, update: (req: AnyObject) => boolean) {
+	updateRequestForPokemon(pokemon: Pokemon, update: (req: PokemonMoveRequestData) => boolean) {
 		if (!(this.activeRequest as MoveRequest)?.active) {
 			throw new Error(`Can't update a request without active Pokemon`);
 		}