diff --git a/src/health/event.rs b/src/health/event.rs new file mode 100644 index 0000000..e69de29 diff --git a/src/health/mod.rs b/src/health/mod.rs index cf0ee62..91c2115 100644 --- a/src/health/mod.rs +++ b/src/health/mod.rs @@ -1,9 +1,73 @@ +use std::ops::Deref; + use bevy::prelude::*; +pub mod event; + +/// Health plugin provides systems to handle health modifications and events pub struct HealthPlugin; impl Plugin for HealthPlugin { fn build(&self, app: &mut App) { - //todo!() + app.add_systems(PostUpdate,cleanup_killed); + } +} + +/// Component that holds health points and max health points of entity +#[derive(Component)] +pub struct Health { + health: usize, + max_health: usize, +} + +impl Default for Health { + fn default() -> Self { + Self { health: 1, max_health: 1 } + } +} + +impl Health { + /// Create new health component with set 'health' and 'max_health'. + pub fn new(health: usize, max_health: usize) -> Self { + Self { + health, + max_health + } + } + /// Create new health component from max health. + pub fn from_max(max_health: usize) -> Self { + Self { + health: max_health, + max_health + } + } + + /// Substract 'damage' amount of health points from component with boundary check. + pub fn damage(&mut self, damage: usize) { + self.health = if damage > self.health {0} else {self.health - damage}; + } + /// Add 'heal' amount of health points to component with max health as a boundary. + pub fn heal_no_overheal(&mut self, heal: usize) { + self.health = if self.health + heal > self.max_health {self.max_health} else {self.health + heal}; + } + /// Add 'heal' amount of health points to component with usize::MAX as a boundary. + pub fn heal_with_overheal(&mut self, heal: usize) { + self.health = if usize::MAX-heal < self.health {usize::MAX} else {self.health + heal}; + } + pub fn get_health(&self) -> usize { + self.health + } +} + +/// Marker component that marks entity to NOT be despawned at the end of frame when health reaches zero +#[derive(Component)] +#[require(Health)] +pub struct Unkillable; + +fn cleanup_killed(mut commands: Commands, query: Query<(Entity, &Health), Without>) { + for (entity, health) in query { + if health.health == 0 { + commands.entity(entity).despawn(); + } } }