Fix EnchantmentLevelAdder crash in bazaar

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.
 	}
 }