drop off magical damage to armor with distance

1 files changed, 5 insertions, 2 deletions

diff --git a/src/main/java/net/cflems/mc/penguin_dust/listener/DamageListener.java b/src/main/java/net/cflems/mc/penguin_dust/listener/DamageListener.java
index c1a48f8..f7f2e9f 100644
--- a/src/main/java/net/cflems/mc/penguin_dust/listener/DamageListener.java
+++ b/src/main/java/net/cflems/mc/penguin_dust/listener/DamageListener.java
@@ -56,13 +56,16 @@ public final class DamageListener implements Listener {
         }
 
         Entity damagee = event.getEntity();
+        double distance = damagee.getLocation().distance(damager.getLocation());
+
         if (damagee instanceof LivingEntity livingDamagee) {
             EntityEquipment damageeEquipment = livingDamagee.getEquipment();
             if (damageeEquipment != null)
-                spellStrength = consumeReinforcement(damageeEquipment, spellStrength);
+                spellStrength =
+                        consumeReinforcement(
+                                damageeEquipment, Math.max(spellStrength - (int) distance, 0));
         }
 
-        double distance = damagee.getLocation().distance(damager.getLocation());
         double damage = causingSpell.calculateDamage(spellStrength, distance);
         event.setDamage(damage);
     }