84 lines
1.7 KiB
Rust
84 lines
1.7 KiB
Rust
use crate::{
|
|
cash::CashResource, control::ControlState, hitpoints::Hitpoints, player::Player,
|
|
sounds::PlaySound,
|
|
};
|
|
use bevy::prelude::*;
|
|
use lightyear::prelude::input::native::ActionState;
|
|
|
|
pub fn plugin(app: &mut App) {
|
|
app.add_systems(FixedUpdate, on_heal_trigger);
|
|
}
|
|
|
|
#[derive(Debug, PartialEq, Eq)]
|
|
struct HealAction {
|
|
cost: i32,
|
|
damage_healed: u32,
|
|
}
|
|
|
|
fn on_heal_trigger(
|
|
mut cmds: Commands,
|
|
mut cash: ResMut<CashResource>,
|
|
mut query: Query<(&mut Hitpoints, &ActionState<ControlState>), With<Player>>,
|
|
) {
|
|
for (mut hp, controls) in query.iter_mut() {
|
|
if !controls.cash_heal {
|
|
continue;
|
|
}
|
|
|
|
if hp.max() || cash.cash == 0 {
|
|
return;
|
|
}
|
|
|
|
let action = heal(cash.cash, hp.get().1 - hp.get().0);
|
|
|
|
hp.heal(action.damage_healed);
|
|
|
|
cash.cash = cash.cash.saturating_sub(action.cost);
|
|
|
|
//TODO: trigger ui cost animation
|
|
cmds.trigger(PlaySound::CashHeal);
|
|
}
|
|
}
|
|
|
|
fn heal(cash: i32, damage: u32) -> HealAction {
|
|
let cost = (damage as f32 / 10. * 25.) as i32;
|
|
|
|
if cash >= cost {
|
|
HealAction {
|
|
cost,
|
|
damage_healed: damage,
|
|
}
|
|
} else {
|
|
let damage_healed = (cash as f32 * 10. / 25.) as u32;
|
|
|
|
HealAction {
|
|
cost: cash,
|
|
damage_healed,
|
|
}
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod test {
|
|
use super::*;
|
|
|
|
#[test]
|
|
fn test_heal() {
|
|
assert_eq!(
|
|
heal(100, 10),
|
|
HealAction {
|
|
cost: 25,
|
|
damage_healed: 10
|
|
}
|
|
);
|
|
|
|
assert_eq!(
|
|
heal(100, 90),
|
|
HealAction {
|
|
cost: 100,
|
|
damage_healed: 40
|
|
}
|
|
);
|
|
}
|
|
}
|