Gen7ScriptsRs/gen_7_scripts/src/moves/change_all_target_stats.rs

59 lines
1.7 KiB
Rust

use crate::script;
use core::any::Any;
use core::sync::atomic::{AtomicI8, Ordering};
use pkmn_lib_interface::app_interface::list::ImmutableList;
use pkmn_lib_interface::app_interface::{
DynamicLibrary, EffectParameter, ExecutingMove, Pokemon, Statistic,
};
use pkmn_lib_interface::handling::{Script, ScriptCapabilities};
script!(
ChangeAllTargetStats,
"change_all_target_stats",
amount: AtomicI8
);
impl Script for ChangeAllTargetStats {
fn new() -> Self {
Self {
amount: Default::default(),
}
}
fn get_name(&self) -> &'static str {
Self::get_const_name()
}
fn get_capabilities(&self) -> &[ScriptCapabilities] {
&[
ScriptCapabilities::Initialize,
ScriptCapabilities::OnSecondaryEffect,
]
}
fn on_initialize(
&self,
_library: &DynamicLibrary,
parameters: Option<ImmutableList<EffectParameter>>,
) {
self.amount.store(
parameters.unwrap().get(0).unwrap().as_int() as i8,
Ordering::SeqCst,
);
}
fn on_secondary_effect(&self, mv: ExecutingMove, target: Pokemon, _hit: u8) {
let user = mv.user();
let amount = self.amount.load(Ordering::SeqCst);
target.change_stat_boost(Statistic::Attack, amount, user == target);
target.change_stat_boost(Statistic::Defense, amount, user == target);
target.change_stat_boost(Statistic::SpecialAttack, amount, user == target);
target.change_stat_boost(Statistic::SpecialDefense, amount, user == target);
target.change_stat_boost(Statistic::Speed, amount, user == target);
}
fn as_any(&self) -> &dyn Any {
self
}
}