From c18117104bf8a288a3545f665fa54e56b62a5bd7 Mon Sep 17 00:00:00 2001
From: Jake Potrebic <jake.m.potrebic@gmail.com>
Date: Mon, 21 Aug 2023 21:08:43 -0700
Subject: [PATCH] More DragonBattle API (#5655)

---
 patches/api/More-DragonBattle-API.patch    | 55 +++++++++++++
 patches/server/More-DragonBattle-API.patch | 91 ++++++++++++++++++++++
 2 files changed, 146 insertions(+)
 create mode 100644 patches/api/More-DragonBattle-API.patch
 create mode 100644 patches/server/More-DragonBattle-API.patch

diff --git a/patches/api/More-DragonBattle-API.patch b/patches/api/More-DragonBattle-API.patch
new file mode 100644
index 0000000000..614be1c70c
--- /dev/null
+++ b/patches/api/More-DragonBattle-API.patch
@@ -0,0 +1,55 @@
+From 0000000000000000000000000000000000000000 Mon Sep 17 00:00:00 2001
+From: Jake Potrebic <jake.m.potrebic@gmail.com>
+Date: Sun, 18 Dec 2022 13:40:17 -0800
+Subject: [PATCH] More DragonBattle API
+
+
+diff --git a/src/main/java/org/bukkit/boss/DragonBattle.java b/src/main/java/org/bukkit/boss/DragonBattle.java
+index 0000000000000000000000000000000000000000..0000000000000000000000000000000000000000 100644
+--- a/src/main/java/org/bukkit/boss/DragonBattle.java
++++ b/src/main/java/org/bukkit/boss/DragonBattle.java
+@@ -0,0 +0,0 @@ public interface DragonBattle {
+          */
+         NONE;
+     }
++    // Paper start
++    /**
++     * Gets the number of gateways tracked by this DragonBattle.
++     * This starts out at 0 and will increase to 20, once for each
++     * kill of the {@link EnderDragon}.
++     *
++     * @return the number of gateways around the end island tracked by this
++     */
++    int getGatewayCount();
++
++    /**
++     * Tries to spawn a new end gateway using default game mechanics.
++     *
++     * @return true if successful, false if there is already the maximum.
++     */
++    boolean spawnNewGateway();
++
++    /**
++     * Spawns a new end gateway at the specified position. This will
++     * spawn regardless of the number of gateways already present.
++     *
++     * @param position position for the new gateway
++     */
++    void spawnNewGateway(@NotNull io.papermc.paper.math.Position position);
++
++    /**
++     * Gets the {@link org.bukkit.entity.EnderCrystal}s being used to respawn the dragon. If no respawn
++     * is ongoing, the list will be empty.
++     *
++     * @return the respawn crystals
++     */
++    java.util.@NotNull @org.jetbrains.annotations.Unmodifiable List<org.bukkit.entity.EnderCrystal> getRespawnCrystals();
++
++    /**
++     * Gets the {@link org.bukkit.entity.EnderCrystal}s on top of the pillars that heal the dragon.
++     *
++     * @return the healing crystals
++     */
++    java.util.@NotNull @org.jetbrains.annotations.Unmodifiable List<org.bukkit.entity.EnderCrystal> getHealingCrystals();
++    // Paper end
+ }
diff --git a/patches/server/More-DragonBattle-API.patch b/patches/server/More-DragonBattle-API.patch
new file mode 100644
index 0000000000..4879c74868
--- /dev/null
+++ b/patches/server/More-DragonBattle-API.patch
@@ -0,0 +1,91 @@
+From 0000000000000000000000000000000000000000 Mon Sep 17 00:00:00 2001
+From: Jake Potrebic <jake.m.potrebic@gmail.com>
+Date: Sun, 18 Dec 2022 13:40:05 -0800
+Subject: [PATCH] More DragonBattle API
+
+== AT ==
+public net.minecraft.world.level.dimension.end.EndDragonFight GATEWAY_COUNT
+public net.minecraft.world.level.dimension.end.EndDragonFight gateways
+public net.minecraft.world.level.dimension.end.EndDragonFight respawnCrystals
+public net.minecraft.world.level.dimension.end.EndDragonFight spawnNewGateway(Lnet/minecraft/core/BlockPos;)V
+
+diff --git a/src/main/java/net/minecraft/world/level/dimension/end/EndDragonFight.java b/src/main/java/net/minecraft/world/level/dimension/end/EndDragonFight.java
+index 0000000000000000000000000000000000000000..0000000000000000000000000000000000000000 100644
+--- a/src/main/java/net/minecraft/world/level/dimension/end/EndDragonFight.java
++++ b/src/main/java/net/minecraft/world/level/dimension/end/EndDragonFight.java
+@@ -0,0 +0,0 @@ public class EndDragonFight {
+         this.gateways.clear();
+     }
+ 
++    // Paper start
++    public boolean spawnNewGatewayIfPossible() {
++        if (!this.gateways.isEmpty()) {
++            this.spawnNewGateway();
++            return true;
++        }
++        return false;
++    }
++
++    public List<EndCrystal> getSpikeCrystals() {
++        final List<EndCrystal> endCrystals = new java.util.ArrayList<>();
++        for (final SpikeFeature.EndSpike spike : SpikeFeature.getSpikesForLevel(this.level)) {
++            endCrystals.addAll(this.level.getEntitiesOfClass(EndCrystal.class, spike.getTopBoundingBox()));
++        }
++        return endCrystals;
++    }
++    // Paper end
++
+     private void spawnNewGateway() {
+         if (!this.gateways.isEmpty()) {
+             int i = (Integer) this.gateways.remove(this.gateways.size() - 1);
+diff --git a/src/main/java/org/bukkit/craftbukkit/boss/CraftDragonBattle.java b/src/main/java/org/bukkit/craftbukkit/boss/CraftDragonBattle.java
+index 0000000000000000000000000000000000000000..0000000000000000000000000000000000000000 100644
+--- a/src/main/java/org/bukkit/craftbukkit/boss/CraftDragonBattle.java
++++ b/src/main/java/org/bukkit/craftbukkit/boss/CraftDragonBattle.java
+@@ -0,0 +0,0 @@ public class CraftDragonBattle implements DragonBattle {
+     private DragonRespawnAnimation toNMSRespawnPhase(RespawnPhase phase) {
+         return (phase != RespawnPhase.NONE) ? DragonRespawnAnimation.values()[phase.ordinal()] : null;
+     }
++    // Paper start
++    @Override
++    public int getGatewayCount() {
++        return EndDragonFight.GATEWAY_COUNT - this.handle.gateways.size();
++    }
++
++    @Override
++    public boolean spawnNewGateway() {
++        return this.handle.spawnNewGatewayIfPossible();
++    }
++
++    @Override
++    public void spawnNewGateway(final io.papermc.paper.math.Position position) {
++        this.handle.spawnNewGateway(io.papermc.paper.util.MCUtil.toBlockPos(position));
++    }
++
++    @Override
++    public java.util.List<org.bukkit.entity.EnderCrystal> getRespawnCrystals() {
++        if (this.handle.respawnCrystals == null) {
++            return java.util.Collections.emptyList();
++        }
++
++        final java.util.List<org.bukkit.entity.EnderCrystal> enderCrystals = new java.util.ArrayList<>();
++        for (final net.minecraft.world.entity.boss.enderdragon.EndCrystal endCrystal : this.handle.respawnCrystals) {
++            if (!endCrystal.isRemoved() && endCrystal.isAlive() && endCrystal.valid) {
++                enderCrystals.add(((org.bukkit.entity.EnderCrystal) endCrystal.getBukkitEntity()));
++            }
++        }
++        return java.util.Collections.unmodifiableList(enderCrystals);
++    }
++
++    @Override
++    public java.util.List<org.bukkit.entity.EnderCrystal> getHealingCrystals() {
++        final java.util.List<org.bukkit.entity.EnderCrystal> enderCrystals = new java.util.ArrayList<>();
++        for (final net.minecraft.world.entity.boss.enderdragon.EndCrystal endCrystal : this.handle.getSpikeCrystals()) {
++            if (!endCrystal.isRemoved() && endCrystal.isAlive() && endCrystal.valid) {
++                enderCrystals.add(((org.bukkit.entity.EnderCrystal) endCrystal.getBukkitEntity()));
++            }
++        }
++        return java.util.Collections.unmodifiableList(enderCrystals);
++    }
++    // Paper end
+ }