Add tipped arrow support (#80)

Patbox · Jul 24, 2023 · 663d94c · 663d94c
1 parent 5e7e55c
commit 663d94c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/polymer-core/src/main/java/eu/pb4/polymer/core/api/item/PolymerItemUtils.java b/polymer-core/src/main/java/eu/pb4/polymer/core/api/item/PolymerItemUtils.java
@@ -218,7 +218,7 @@ private static boolean shouldPolymerConvert(ItemStack itemStack, @Nullable Serve
                         return true;
                     }
                 }
-            } else if (itemStack.getItem() instanceof PotionItem) {
+            } else if (itemStack.getItem() instanceof PotionItem || itemStack.getItem() instanceof TippedArrowItem) {
                 if (PotionUtil.getPotion(itemStack) instanceof PolymerObject) {
                     return true;
                 }
@@ -382,7 +382,7 @@ public static ItemStack createItemStack(ItemStack itemStack, TooltipContext tool
                 out.getNbt().put(ItemStack.ENCHANTMENTS_KEY, list);
             }
 
-            if (itemStack.getItem() instanceof PotionItem) {
+            if (itemStack.getItem() instanceof PotionItem || itemStack.getItem() instanceof TippedArrowItem) {
                 out.getOrCreateNbt().putInt("CustomPotionColor", PotionUtil.getColor(itemStack));
             }