From 07fd9ac0234306648664bc935ecf2f3cf911dd59 Mon Sep 17 00:00:00 2001
From: Sam <samuelkeiffer@gmail.com>
Date: Sun, 6 Sep 2020 16:01:56 -0500
Subject: [PATCH] Re-added necessary functionality to beam (heal, lifesteal,
 energy regen).

---
 common/src/comp/beam.rs         |  1 +
 common/src/states/basic_beam.rs |  4 ++-
 common/src/sys/beam.rs          | 51 +++++++++++++++++++++++++++++----
 3 files changed, 50 insertions(+), 6 deletions(-)

diff --git a/common/src/comp/beam.rs b/common/src/comp/beam.rs
index 9eeaa3a83e..cba2b8e04e 100644
--- a/common/src/comp/beam.rs
+++ b/common/src/comp/beam.rs
@@ -11,6 +11,7 @@ pub struct Properties {
     pub damage: u32,
     pub heal: u32,
     pub lifesteal_eff: f32,
+    pub energy_regen: u32,
     pub duration: Duration,
     pub owner: Option<Uid>,
 }
diff --git a/common/src/states/basic_beam.rs b/common/src/states/basic_beam.rs
index 915a7a6d53..eff07c5868 100644
--- a/common/src/states/basic_beam.rs
+++ b/common/src/states/basic_beam.rs
@@ -1,5 +1,5 @@
 use crate::{
-    comp::{beam, CharacterState, EnergySource, Ori, Pos, StateUpdate},
+    comp::{beam, CharacterState, Ori, Pos, StateUpdate},
     event::ServerEvent,
     states::utils::*,
     sys::character_behavior::*,
@@ -69,6 +69,7 @@ impl CharacterBehavior for Data {
         } else if data.inputs.primary.is_pressed() && !self.exhausted {
             let damage = (self.base_dps as f32 / self.tick_rate) as u32;
             let heal = (self.base_hps as f32 / self.tick_rate) as u32;
+            let energy = (self.energy_regen as f32 / self.tick_rate) as u32;
             let speed = self.range / self.beam_duration.as_secs_f32();
             let properties = beam::Properties {
                 angle: self.max_angle.to_radians(),
@@ -76,6 +77,7 @@ impl CharacterBehavior for Data {
                 damage,
                 heal,
                 lifesteal_eff: self.lifesteal_eff,
+                energy_regen: energy,
                 duration: self.beam_duration,
                 owner: Some(*data.uid),
             };
diff --git a/common/src/sys/beam.rs b/common/src/sys/beam.rs
index b35b07ee8d..90310bdad9 100644
--- a/common/src/sys/beam.rs
+++ b/common/src/sys/beam.rs
@@ -1,7 +1,7 @@
 use crate::{
     comp::{
-        group, Beam, Body, CharacterState, Damage, DamageSource, HealthChange, HealthSource, Last,
-        Loadout, Ori, Pos, Scale, Stats,
+        group, Beam, Body, CharacterState, Damage, DamageSource, Energy, EnergySource,
+        HealthChange, HealthSource, Last, Loadout, Ori, Pos, Scale, Stats,
     },
     event::{EventBus, ServerEvent},
     state::{DeltaTime, Time},
@@ -33,6 +33,7 @@ impl<'a> System<'a> for Sys {
         ReadStorage<'a, Loadout>,
         ReadStorage<'a, group::Group>,
         ReadStorage<'a, CharacterState>,
+        WriteStorage<'a, Energy>,
         WriteStorage<'a, Beam>,
     );
 
@@ -54,6 +55,7 @@ impl<'a> System<'a> for Sys {
             loadouts,
             groups,
             character_states,
+            mut energies,
             mut beams,
         ): Self::SystemData,
     ) {
@@ -145,16 +147,37 @@ impl<'a> System<'a> for Sys {
                         .map(|group_a| Some(group_a) == groups.get(b))
                         .unwrap_or(Some(*uid_b) == beam.owner);
 
+                    // If owner, shouldn't heal or damage
+                    if Some(*uid_b) == beam.owner {
+                        continue;
+                    }
+                    // Don't heal if outside group
                     // Don't damage in the same group
-                    if same_group {
+                    let (mut is_heal, mut is_damage) = (false, false);
+                    if !same_group && (beam.damage > 0) {
+                        is_damage = true;
+                    }
+                    if same_group && (beam.heal > 0) {
+                        is_heal = true;
+                    }
+                    if !is_heal && !is_damage {
                         continue;
                     }
 
                     // Weapon gives base damage
-                    let source = DamageSource::Energy;
+                    let source = if is_heal {
+                        DamageSource::Healing
+                    } else {
+                        DamageSource::Energy
+                    };
+                    let healthchange = if is_heal {
+                        beam.heal as f32
+                    } else {
+                        -(beam.damage as f32)
+                    };
 
                     let mut damage = Damage {
-                        healthchange: -(beam.damage as f32),
+                        healthchange,
                         source,
                     };
 
@@ -176,6 +199,24 @@ impl<'a> System<'a> for Sys {
                                 },
                             },
                         });
+                        if is_damage && beam.lifesteal_eff > 0.0 {
+                            server_emitter.emit(ServerEvent::Damage {
+                                uid: beam.owner.unwrap_or(*uid),
+                                change: HealthChange {
+                                    amount: (-damage.healthchange * beam.lifesteal_eff) as i32,
+                                    cause: HealthSource::Attack {
+                                        by: beam.owner.unwrap_or(*uid),
+                                    },
+                                },
+                            });
+                        }
+                        if let Some(energy_mut) = beam
+                            .owner
+                            .and_then(|o| uid_allocator.retrieve_entity_internal(o.into()))
+                            .and_then(|o| energies.get_mut(o))
+                        {
+                            energy_mut.change_by(beam.energy_regen as i32, EnergySource::HitEnemy);
+                        }
                     }
                 }
             }