From 182934bb13aab9384b21855cfcde6e46019a6cba Mon Sep 17 00:00:00 2001
From: Jbleezy <thejbirdjustin@yahoo.com>
Date: Wed, 29 Mar 2023 23:03:00 -0700
Subject: [PATCH] Lightning Staff: upgraded damages Panzersoldat

---
 README.md                                     |   1 +
 .../zm/replaced/_zm_weap_staff_lightning.gsc  | 120 ++++++++++++++++++
 scripts/zm/zm_tomb/zm_tomb_reimagined.gsc     |   2 +-
 3 files changed, 122 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 9a1201ca..3af9bb01 100644
--- a/README.md
+++ b/README.md
@@ -235,6 +235,7 @@
 * Get full ammo when initially picked up after upgrading (normally missing one clip)
 * Max ammo no longer fills clip ammo
 * Upgraded: kill on any round
+* Lightning upgraded: damages Panzersoldat
 
 ## Wallbuys
 * Purchasing ammo refills clip ammo
diff --git a/scripts/zm/replaced/_zm_weap_staff_lightning.gsc b/scripts/zm/replaced/_zm_weap_staff_lightning.gsc
index 184230a1..90c18759 100644
--- a/scripts/zm/replaced/_zm_weap_staff_lightning.gsc
+++ b/scripts/zm/replaced/_zm_weap_staff_lightning.gsc
@@ -10,6 +10,86 @@
 #include maps\mp\animscripts\shared;
 #include maps\mp\zombies\_zm_weap_staff_lightning;
 
+staff_lightning_ball_kill_zombies( e_attacker )
+{
+    self endon( "death" );
+    self endon( "stop_killing" );
+
+    while ( true )
+    {
+        a_zombies = staff_lightning_get_valid_targets( e_attacker, self.origin );
+
+        if ( isdefined( a_zombies ) )
+        {
+            foreach ( zombie in a_zombies )
+            {
+                if ( staff_lightning_is_target_valid( zombie ) )
+                {
+                    e_attacker thread staff_lightning_arc_fx( self, zombie );
+                    wait 0.2;
+                }
+            }
+        }
+
+        wait 0.05;
+    }
+}
+
+staff_lightning_get_valid_targets( player, v_source )
+{
+    player endon( "disconnect" );
+    a_enemies = [];
+    a_zombies = getaiarray( level.zombie_team );
+    a_zombies = get_array_of_closest( v_source, a_zombies, undefined, undefined, self.n_range );
+
+    if ( isdefined( a_zombies ) )
+    {
+        foreach ( ai_zombie in a_zombies )
+        {
+            if ( staff_lightning_is_target_valid( ai_zombie ) )
+                a_enemies[a_enemies.size] = ai_zombie;
+        }
+    }
+
+    return a_enemies;
+}
+
+staff_lightning_is_target_valid( ai_zombie )
+{
+    if ( !isdefined( ai_zombie ) )
+        return false;
+
+    if ( is_true( ai_zombie.is_being_zapped ) )
+        return false;
+
+    return true;
+}
+
+staff_lightning_arc_fx( e_source, ai_zombie )
+{
+    self endon( "disconnect" );
+
+    if ( !isdefined( ai_zombie ) )
+        return;
+
+    if ( !bullet_trace_throttled( e_source.origin, ai_zombie.origin + vectorscale( ( 0, 0, 1 ), 20.0 ), ai_zombie ) )
+    {
+        return;
+    }
+
+    if ( isdefined( e_source ) && isdefined( ai_zombie ) && isalive( ai_zombie ) )
+    {
+        if ( is_true( ai_zombie.is_mechz ) )
+        {
+            level thread staff_lightning_ball_damage_over_time_mechz( e_source, ai_zombie, self );
+        }
+        else
+        {
+            level thread staff_lightning_ball_damage_over_time( e_source, ai_zombie, self );
+        }
+    }
+}
+
 staff_lightning_ball_damage_over_time( e_source, e_target, e_attacker )
 {
     e_attacker endon( "disconnect" );
@@ -56,6 +136,46 @@ staff_lightning_ball_damage_over_time( e_source, e_target, e_attacker )
     }
 }
 
+staff_lightning_ball_damage_over_time_mechz( e_source, e_target, e_attacker )
+{
+    e_attacker endon( "disconnect" );
+    n_range_sq = e_source.n_range * e_source.n_range;
+    e_target.is_being_zapped = 1;
+    wait 0.5;
+
+    if ( isdefined( e_source ) )
+    {
+        if ( !isdefined( e_source.n_damage_per_sec ) )
+            e_source.n_damage_per_sec = get_lightning_ball_damage_per_sec( e_attacker.chargeshotlevel );
+
+        n_damage_per_pulse = e_source.n_damage_per_sec * 1.0;
+    }
+
+    while ( isdefined( e_source ) && isalive( e_target ) )
+    {
+        wait 1.0;
+
+        if ( !isdefined( e_source ) || !isalive( e_target ) )
+            break;
+
+        n_dist_sq = distancesquared( e_source.origin, e_target.origin );
+
+        if ( n_dist_sq > n_range_sq )
+            break;
+
+        if ( isalive( e_target ) && isdefined( e_source ) )
+        {
+            e_target do_damage_network_safe( e_attacker, e_source.n_damage_per_sec, e_source.str_weapon, "MOD_RIFLE_BULLET" );
+            break;
+        }
+    }
+
+    if ( isdefined( e_target ) )
+    {
+        e_target.is_being_zapped = 0;
+    }
+}
+
 get_lightning_ball_damage_per_sec( n_charge )
 {
     return 2500;
diff --git a/scripts/zm/zm_tomb/zm_tomb_reimagined.gsc b/scripts/zm/zm_tomb/zm_tomb_reimagined.gsc
index a2b00b27..979e2d2e 100644
--- a/scripts/zm/zm_tomb/zm_tomb_reimagined.gsc
+++ b/scripts/zm/zm_tomb/zm_tomb_reimagined.gsc
@@ -50,7 +50,7 @@ main()
     replaceFunc(maps\mp\zombies\_zm_weap_riotshield_tomb::init, scripts\zm\replaced\_zm_weap_riotshield_tomb::init);
 	replaceFunc(maps\mp\zombies\_zm_weap_riotshield_tomb::player_damage_shield, scripts\zm\replaced\_zm_weap_riotshield_tomb::player_damage_shield);
     replaceFunc(maps\mp\zombies\_zm_weap_staff_fire::flame_damage_fx, scripts\zm\replaced\_zm_weap_staff_fire::flame_damage_fx);
-    replaceFunc(maps\mp\zombies\_zm_weap_staff_lightning::staff_lightning_ball_damage_over_time, scripts\zm\replaced\_zm_weap_staff_lightning::staff_lightning_ball_damage_over_time);
+    replaceFunc(maps\mp\zombies\_zm_weap_staff_lightning::staff_lightning_ball_kill_zombies, scripts\zm\replaced\_zm_weap_staff_lightning::staff_lightning_ball_kill_zombies);
 }
 
 init()