Add platform specific EnderDragonPart access

Author: 2No2Name

common/src/main/java/net/caffeinemc/mods/lithium/common/services/PlatformEntityAccess.java

@@ -0,0 +1,15 @@
+package net.caffeinemc.mods.lithium.common.services;
+
+import net.minecraft.world.entity.Entity;
+import net.minecraft.world.level.Level;
+import net.minecraft.world.phys.AABB;
+
+import java.util.ArrayList;
+import java.util.function.Predicate;
+
+public interface PlatformEntityAccess {
+    PlatformEntityAccess INSTANCE = Services.load(PlatformEntityAccess.class);
+
+    void addEnderDragonParts(Level level, Entity excludedEntity, AABB box, Predicate<? super Entity> entityFilter, ArrayList<Entity> entities);
+
+}

fabric/src/main/java/net/caffeinemc/mods/lithium/fabric/FabricEntityAccess.java

@@ -0,0 +1,27 @@
+package net.caffeinemc.mods.lithium.fabric;
+
+import net.caffeinemc.mods.lithium.common.services.PlatformEntityAccess;
+import net.minecraft.world.entity.Entity;
+import net.minecraft.world.entity.boss.EnderDragonPart;
+import net.minecraft.world.level.Level;
+import net.minecraft.world.phys.AABB;
+
+import java.util.ArrayList;
+import java.util.function.Predicate;
+
+public class FabricEntityAccess implements PlatformEntityAccess {
+
+
+    @Override
+    public void addEnderDragonParts(Level level, Entity excludedEntity, AABB box, Predicate<? super Entity> entityFilter, ArrayList<Entity> entities) {
+        for (EnderDragonPart enderDragonPart : level.dragonParts()) {
+            if (enderDragonPart != excludedEntity
+                    && enderDragonPart.parentMob != excludedEntity
+                    && entityFilter.test(enderDragonPart) &&
+                    box.intersects(enderDragonPart.getBoundingBox())
+            ) {
+                entities.add(enderDragonPart);
+            }
+        }
+    }
+}

fabric/src/main/resources/META-INF/services/net.caffeinemc.mods.lithium.common.services.PlatformEntityAccess

@@ -0,0 +1 @@
+net.caffeinemc.mods.lithium.fabric.FabricEntityAccess

neoforge/src/main/java/net/caffeinemc/mods/lithium/neoforge/NeoForgeEntityAccess.java

@@ -0,0 +1,27 @@
+package net.caffeinemc.mods.lithium.neoforge;
+
+import net.caffeinemc.mods.lithium.common.services.PlatformEntityAccess;
+import net.minecraft.world.entity.Entity;
+import net.minecraft.world.level.Level;
+import net.minecraft.world.phys.AABB;
+import net.neoforged.neoforge.entity.PartEntity;
+
+import java.util.ArrayList;
+import java.util.function.Predicate;
+
+public class NeoForgeEntityAccess implements PlatformEntityAccess {
+
+    @Override
+    public void addEnderDragonParts(Level level, Entity excludedEntity, AABB box, Predicate<? super Entity> entityFilter, ArrayList<Entity> entities) {
+        //  [NeoForge Copy] Support for dragon parts and neoforge part entities
+        for (PartEntity<?> partEntity : level.dragonParts()) {
+            if (partEntity != excludedEntity
+                    && partEntity.getParent() != excludedEntity
+                    && entityFilter.test(partEntity)
+                    && box.intersects(partEntity.getBoundingBox())
+            ) {
+                entities.add(partEntity);
+            }
+        }
+    }
+}

neoforge/src/main/resources/META-INF/services/net.caffeinemc.mods.lithium.common.services.PlatformEntityAccess

@@ -0,0 +1 @@
+net.caffeinemc.mods.lithium.neoforge.NeoForgeEntityAccess