From 473aa47edbb063aaebb26f32b0879a459084c0e6 Mon Sep 17 00:00:00 2001
From: sandtechnology <20417547+sandtechnology@users.noreply.github.com>
Date: Sat, 5 Aug 2023 13:09:25 +0800
Subject: [PATCH] Fix #583, Update dragonPhase dynamically  (#869)

* #583 Update dragonPhase dynamically

* Remove author annotation

* Fix build

* Formatting
---
 .../mixin/bukkit/CraftEnderDragonMixin.java   | 42 +++++++++++++++++++
 .../resources/mixins.arclight.bukkit.json     |  1 +
 2 files changed, 43 insertions(+)
 create mode 100644 arclight-common/src/main/java/io/izzel/arclight/common/mixin/bukkit/CraftEnderDragonMixin.java

diff --git a/arclight-common/src/main/java/io/izzel/arclight/common/mixin/bukkit/CraftEnderDragonMixin.java b/arclight-common/src/main/java/io/izzel/arclight/common/mixin/bukkit/CraftEnderDragonMixin.java
new file mode 100644
index 00000000..7863480d
--- /dev/null
+++ b/arclight-common/src/main/java/io/izzel/arclight/common/mixin/bukkit/CraftEnderDragonMixin.java
@@ -0,0 +1,42 @@
+package io.izzel.arclight.common.mixin.bukkit;
+
+import io.izzel.arclight.api.EnumHelper;
+import io.izzel.arclight.common.mod.ArclightMod;
+import net.minecraft.world.entity.boss.enderdragon.phases.EnderDragonPhase;
+import org.bukkit.craftbukkit.v.entity.CraftEnderDragon;
+import org.bukkit.entity.EnderDragon;
+import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.injection.At;
+import org.spongepowered.asm.mixin.injection.Inject;
+import org.spongepowered.asm.mixin.injection.callback.CallbackInfoReturnable;
+
+import java.util.ArrayList;
+import java.util.List;
+
+@Mixin(value = CraftEnderDragon.class, remap = false)
+public class CraftEnderDragonMixin {
+
+    @Inject(method = "getPhase", at = @At("HEAD"))
+    public void arclight$getPhase(CallbackInfoReturnable<EnderDragon.Phase> cir) {
+        checkAndUpdateDragonPhase();
+    }
+
+    @Inject(method = "getBukkitPhase", at = @At("HEAD"))
+    private static void arclight$getBukkitPhase(EnderDragonPhase phase, CallbackInfoReturnable<EnderDragon.Phase> cir) {
+        checkAndUpdateDragonPhase();
+    }
+
+    private static void checkAndUpdateDragonPhase() {
+        var forgeCount = EnderDragonPhase.getCount();
+        if (EnderDragon.Phase.values().length != forgeCount) {
+            var newTypes = new ArrayList<EnderDragon.Phase>();
+            for (var id = EnderDragon.Phase.values().length; id < forgeCount; id++) {
+                var name = "MOD_PHASE_" + id;
+                var newPhase = EnumHelper.makeEnum(EnderDragon.Phase.class, name, id, List.of(), List.of());
+                newTypes.add(newPhase);
+                ArclightMod.LOGGER.debug("Registered {} as ender dragon phase {}", name, newPhase);
+            }
+            EnumHelper.addEnums(EnderDragon.Phase.class, newTypes);
+        }
+    }
+}
diff --git a/arclight-common/src/main/resources/mixins.arclight.bukkit.json b/arclight-common/src/main/resources/mixins.arclight.bukkit.json
index 7418822f..c1024af7 100644
--- a/arclight-common/src/main/resources/mixins.arclight.bukkit.json
+++ b/arclight-common/src/main/resources/mixins.arclight.bukkit.json
@@ -18,6 +18,7 @@
     "CraftBlockStateMixin",
     "CraftConsoleCommandSenderMixin",
     "CraftCreativeCategoryMixin",
+    "CraftEnderDragonMixin",
     "CraftEntityMixin",
     "CraftEventFactoryMixin",
     "CraftHumanEntityMixin",