Fix: Crystalite Elytra should give set bonus

OpenBagTwo · Oct 14, 2024 · 57fc43a · 57fc43a
1 parent 9a2b329
commit 57fc43a
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/src/main/java/org/betterx/betterend/client/BetterEndClient.java b/src/main/java/org/betterx/betterend/client/BetterEndClient.java
@@ -8,6 +8,7 @@
 import org.betterx.betterend.events.ItemTooltipCallback;
 import org.betterx.betterend.interfaces.MultiModelItem;
 import org.betterx.betterend.item.CrystaliteArmor;
+import org.betterx.betterend.item.CrystaliteElytra;
 import org.betterx.betterend.registry.*;
 import org.betterx.betterend.world.generator.GeneratorOptions;
 
@@ -65,7 +66,9 @@ public void onInitializeClient() {
 
     public static void registerTooltips() {
         ItemTooltipCallback.EVENT.register((player, stack, context, lines) -> {
-            if (stack.getItem() instanceof CrystaliteArmor) {
+            if ((stack.getItem() instanceof CrystaliteArmor)
+                || (stack.getItem() instanceof CrystaliteElytra)
+            ) {
                 boolean hasSet = false;
                 if (player != null) {
                     hasSet = CrystaliteArmor.hasFullSet(player);

diff --git a/src/main/java/org/betterx/betterend/item/CrystaliteArmor.java b/src/main/java/org/betterx/betterend/item/CrystaliteArmor.java
@@ -22,7 +22,10 @@ public CrystaliteArmor(Type type, Properties settings) {
 
     public static boolean hasFullSet(LivingEntity owner) {
         for (ItemStack armorStack : owner.getArmorSlots()) {
-            if (!(armorStack.getItem() instanceof CrystaliteArmor)) {
+            if (
+                !(armorStack.getItem() instanceof CrystaliteArmor)
+                    && !(armorStack.getItem() instanceof CrystaliteElytra)
+            ) {
                 return false;
             }
         }