diff --git a/paper-server/src/main/java/org/bukkit/craftbukkit/entity/CraftLightningStrike.java b/paper-server/src/main/java/org/bukkit/craftbukkit/entity/CraftLightningStrike.java index 0b2ba5c933..6fed8075aa 100644 --- a/paper-server/src/main/java/org/bukkit/craftbukkit/entity/CraftLightningStrike.java +++ b/paper-server/src/main/java/org/bukkit/craftbukkit/entity/CraftLightningStrike.java @@ -66,4 +66,23 @@ public class CraftLightningStrike extends CraftEntity implements LightningStrike return this.spigot; } // Spigot end + + // Paper start + @Override + public int getFlashCount() { + return getHandle().flashes; + } + + @Override + public void setFlashCount(int flashes) { + com.google.common.base.Preconditions.checkArgument(flashes >= 0, "Flashes has to be a positive number!"); + getHandle().flashes = flashes; + } + + @Override + public @org.jetbrains.annotations.Nullable org.bukkit.entity.Entity getCausingEntity() { + final var cause = this.getHandle().getCause(); + return cause == null ? null : cause.getBukkitEntity(); + } + // Paper end }