diff options
Diffstat (limited to 'src/main/java/de/hysky/skyblocker/skyblock/item')
| -rw-r--r-- | src/main/java/de/hysky/skyblocker/skyblock/item/slottext/adders/EnchantmentLevelAdder.java | 24 |
1 files changed, 17 insertions, 7 deletions
diff --git a/src/main/java/de/hysky/skyblocker/skyblock/item/slottext/adders/EnchantmentLevelAdder.java b/src/main/java/de/hysky/skyblocker/skyblock/item/slottext/adders/EnchantmentLevelAdder.java index babf27d6..44b3c982 100644 --- a/src/main/java/de/hysky/skyblocker/skyblock/item/slottext/adders/EnchantmentLevelAdder.java +++ b/src/main/java/de/hysky/skyblocker/skyblock/item/slottext/adders/EnchantmentLevelAdder.java @@ -21,12 +21,22 @@ public class EnchantmentLevelAdder extends SlotTextAdder { public @Nullable Text getText(Slot slot) { final ItemStack itemStack = slot.getStack(); if (!itemStack.isOf(Items.ENCHANTED_BOOK)) return null; - List<Text> lore = ItemUtils.getLore(itemStack); - if (lore.isEmpty()) return null; - String firstLine = lore.getFirst().getString(); - String romanNumeral = firstLine.substring(firstLine.lastIndexOf(' ') + 1); //+1 because we don't need the space itself - int level = ChocolateFactorySolver.romanToDecimal(romanNumeral); //Temporary line. The method will be moved out later. - if (level == 0) return null; - return Text.literal(String.valueOf(level)).formatted(Formatting.GREEN); + String name = itemStack.getName().getString(); + if (name.equals("Enchanted Book")) { + List<Text> lore = ItemUtils.getLore(itemStack); + if (lore.isEmpty()) return null; + int level = getEnchantLevelFromString(lore.getFirst().getString()); + if (level == 0) return null; + return Text.literal(String.valueOf(level)).formatted(Formatting.GREEN); + } else { //In bazaar, the books have the enchantment name in the name + int level = getEnchantLevelFromString(name); + if (level == 0) return null; + return Text.literal(String.valueOf(level)).formatted(Formatting.GREEN); + } + } + + private static int getEnchantLevelFromString(String str) { + String romanNumeral = str.substring(str.lastIndexOf(' ') + 1); //+1 because we don't need the space itself + return ChocolateFactorySolver.romanToDecimal(romanNumeral); //Temporary line. The method will be moved out later. } } |