diff --git a/arclight-common/src/main/java/io/izzel/arclight/common/mixin/core/entity/effect/LightningBoltEntityMixin.java b/arclight-common/src/main/java/io/izzel/arclight/common/mixin/core/entity/effect/LightningBoltEntityMixin.java index 44118334..8acd916c 100644 --- a/arclight-common/src/main/java/io/izzel/arclight/common/mixin/core/entity/effect/LightningBoltEntityMixin.java +++ b/arclight-common/src/main/java/io/izzel/arclight/common/mixin/core/entity/effect/LightningBoltEntityMixin.java @@ -3,6 +3,7 @@ package io.izzel.arclight.common.mixin.core.entity.effect; import io.izzel.arclight.common.mixin.core.entity.EntityMixin; import io.izzel.arclight.common.mod.util.ArclightCaptures; import net.minecraft.block.BlockState; +import net.minecraft.entity.Entity; import net.minecraft.entity.effect.LightningBoltEntity; import net.minecraft.util.math.BlockPos; import net.minecraft.world.World; @@ -30,6 +31,11 @@ public abstract class LightningBoltEntityMixin extends EntityMixin { this.isSilent = false; } + @Inject(method = "tick", at = @At(value = "INVOKE", target = "Lnet/minecraft/entity/Entity;onStruckByLightning(Lnet/minecraft/entity/effect/LightningBoltEntity;)V")) + private void arclight$captureEntity(CallbackInfo ci) { + ArclightCaptures.captureDamageEventEntity((Entity) (Object) this); + } + @Inject(method = "tick", at = @At(value = "INVOKE", shift = At.Shift.AFTER, target = "Lnet/minecraft/entity/Entity;onStruckByLightning(Lnet/minecraft/entity/effect/LightningBoltEntity;)V")) private void arclight$resetEntity(CallbackInfo ci) { ArclightCaptures.captureDamageEventEntity(null);