82 lines
2.0 KiB
Rust
82 lines
2.0 KiB
Rust
use crate::common_usings::*;
|
|
use pkmn_lib_interface::{pkmn_err, PkmnErr};
|
|
use spin::RwLock;
|
|
|
|
script!(
|
|
DoublePowerUserDamagedByTargetInTurnData,
|
|
"double_power_user_damaged_by_target_in_turn_data",
|
|
hits: RwLock<Vec<Pokemon>>
|
|
);
|
|
|
|
impl Script for DoublePowerUserDamagedByTargetInTurnData {
|
|
fn new() -> Self {
|
|
Self {
|
|
hits: Default::default(),
|
|
}
|
|
}
|
|
|
|
fn get_name(&self) -> &'static str {
|
|
Self::get_const_name()
|
|
}
|
|
|
|
fn get_capabilities(&self) -> &[ScriptCapabilities] {
|
|
&[
|
|
ScriptCapabilities::OnEndTurn,
|
|
ScriptCapabilities::OnIncomingHit,
|
|
]
|
|
}
|
|
|
|
fn on_incoming_hit(
|
|
&self,
|
|
executing_move: ExecutingMove,
|
|
target: Pokemon,
|
|
hit: u8,
|
|
) -> PkmnResult<()> {
|
|
// Only register hits that dealt damage
|
|
if executing_move
|
|
.get_hit_data(&target, hit)
|
|
.or(Err(pkmn_err!("Couldnt resolve hit")))?
|
|
.damage()?
|
|
== 0
|
|
{
|
|
return Ok(());
|
|
}
|
|
let user = executing_move.user();
|
|
let mut write_lock = self.hits.write();
|
|
// Check if the pokemon has already hit the pokemon. Bail out if it has.
|
|
for hit_pokemon in write_lock.iter() {
|
|
if hit_pokemon.reference() == user.reference() {
|
|
return Ok(());
|
|
}
|
|
}
|
|
write_lock.push(user);
|
|
Ok(())
|
|
}
|
|
|
|
fn on_end_turn(&self) -> PkmnResult<()> {
|
|
let pokemon = self.get_owner().unwrap().as_pokemon();
|
|
pokemon.remove_volatile(self)?;
|
|
Ok(())
|
|
}
|
|
|
|
fn as_any(&self) -> &dyn Any {
|
|
self
|
|
}
|
|
}
|
|
|
|
impl DoublePowerUserDamagedByTargetInTurnData {
|
|
pub fn create() -> Box<dyn Script> {
|
|
Box::new(Self::new())
|
|
}
|
|
|
|
pub fn has_been_hit_by_pokemon(&self, pokemon: &Pokemon) -> bool {
|
|
let read_lock = self.hits.read();
|
|
for hit_pokemon in read_lock.iter() {
|
|
if hit_pokemon.reference() == pokemon.reference() {
|
|
return true;
|
|
}
|
|
}
|
|
false
|
|
}
|
|
}
|