Fix dead coral updater (#10)

CloudburstMC · Sep 15, 2024 · 9a63bf9 · 9a63bf9
1 parent 18c65f9
commit 9a63bf9
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/main/java/org/cloudburstmc/blockstateupdater/BlockStateUpdater_1_20_0.java b/src/main/java/org/cloudburstmc/blockstateupdater/BlockStateUpdater_1_20_0.java
@@ -44,6 +44,12 @@ public void registerUpdaters(CompoundTagUpdaterContext ctx) {
         this.addCoralUpdater(ctx,  "yellow", "minecraft:horn_coral");
         this.addCoralUpdater(ctx,  "purple", "minecraft:bubble_coral");
 
+        this.addDeadCoralUpdater(ctx,  "red", "minecraft:dead_fire_coral");
+        this.addDeadCoralUpdater(ctx,  "pink", "minecraft:dead_brain_coral");
+        this.addDeadCoralUpdater(ctx,  "blue", "minecraft:dead_tube_coral");
+        this.addDeadCoralUpdater(ctx,  "yellow", "minecraft:dead_horn_coral");
+        this.addDeadCoralUpdater(ctx,  "purple", "minecraft:dead_bubble_coral");
+
         ctx.addUpdater(1, 20, 0)
                 .match("name", "minecraft:calibrated_sculk_sensor")
                 .visit("states")
@@ -114,7 +120,6 @@ private void addCauldronUpdater(CompoundTagUpdaterContext ctx, String type) {
     }
 
     private void addCoralUpdater(CompoundTagUpdaterContext context, String type, String newIdentifier) {
-        // Two updates to match final version
         context.addUpdater(1, 20, 0)
                 .match("name", "minecraft:coral")
                 .visit("states")
@@ -123,7 +128,9 @@ private void addCoralUpdater(CompoundTagUpdaterContext context, String type, Str
                 .edit("coral_color", helper -> helper.getRootTag().put("name", newIdentifier))
                 .remove("coral_color")
                 .remove("dead_bit");
+    }
 
+    private void addDeadCoralUpdater(CompoundTagUpdaterContext context, String type, String newIdentifier) {
         context.addUpdater(1, 20, 0)
                 .match("name", "minecraft:coral")
                 .visit("states")