make attacker immune to sword lightning bolt

AlexModGuy · Dec 26, 2023 · 51db79b · 51db79b
1 parent 6b732b2
commit 51db79b
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/src/main/java/com/github/alexthe666/iceandfire/event/ServerEvents.java b/src/main/java/com/github/alexthe666/iceandfire/event/ServerEvents.java
@@ -683,8 +683,9 @@ public void onVillagerTrades(VillagerTradesEvent event) {
 
     @SubscribeEvent
     public void onLightningHit(final EntityStruckByLightningEvent event) {
-        // TODO :: Set owner UUID in the tags as well to prevent damage to them?
-        if (event.getLightning().getTags().contains(BOLT_DONT_DESTROY_LOOT) && (event.getEntity() instanceof ItemEntity || event.getEntity() instanceof ExperienceOrb)) {
+        if ((event.getEntity() instanceof ItemEntity || event.getEntity() instanceof ExperienceOrb) && event.getLightning().getTags().contains(BOLT_DONT_DESTROY_LOOT)) {
+            event.setCanceled(true);
+        } else if (event.getLightning().getTags().contains(event.getEntity().getStringUUID())) {
             event.setCanceled(true);
         }
     }

diff --git a/src/main/java/com/github/alexthe666/iceandfire/item/DragonSteelOverrides.java b/src/main/java/com/github/alexthe666/iceandfire/item/DragonSteelOverrides.java
@@ -94,6 +94,7 @@ default void hurtEnemy(T item, ItemStack stack, LivingEntity target, LivingEntit
             if (!attacker.level().isClientSide && flag) {
                 LightningBolt lightningboltentity = EntityType.LIGHTNING_BOLT.create(target.level());
                 lightningboltentity.getTags().add(ServerEvents.BOLT_DONT_DESTROY_LOOT);
+                lightningboltentity.getTags().add(attacker.getStringUUID());
                 lightningboltentity.moveTo(target.position());
                 if (!target.level().isClientSide) {
                     target.level().addFreshEntity(lightningboltentity);

diff --git a/src/main/java/com/github/alexthe666/iceandfire/item/ItemAlchemySword.java b/src/main/java/com/github/alexthe666/iceandfire/item/ItemAlchemySword.java
@@ -54,6 +54,7 @@ public boolean hurtEnemy(@NotNull ItemStack stack, @NotNull LivingEntity target,
             if (!attacker.level().isClientSide && flag) {
                 LightningBolt lightningboltentity = EntityType.LIGHTNING_BOLT.create(target.level());
                 lightningboltentity.getTags().add(ServerEvents.BOLT_DONT_DESTROY_LOOT);
+                lightningboltentity.getTags().add(attacker.getStringUUID());
                 lightningboltentity.moveTo(target.position());
                 if (!target.level().isClientSide) {
                     target.level().addFreshEntity(lightningboltentity);