Added dungeon potion level to item number (#334)

Co-authored-by: superhize <superhize@gmail.com>

1 files changed, 16 insertions, 0 deletions

diff --git a/src/main/java/at/hannibal2/skyhanni/features/inventory/ItemDisplayOverlayFeatures.kt b/src/main/java/at/hannibal2/skyhanni/features/inventory/ItemDisplayOverlayFeatures.kt
index 2622c8f14..6fdb3800d 100644
--- a/src/main/java/at/hannibal2/skyhanni/features/inventory/ItemDisplayOverlayFeatures.kt
+++ b/src/main/java/at/hannibal2/skyhanni/features/inventory/ItemDisplayOverlayFeatures.kt
@@ -14,6 +14,7 @@ import at.hannibal2.skyhanni.utils.NumberUtil.romanToDecimal
 import at.hannibal2.skyhanni.utils.NumberUtil.romanToDecimalIfNeeded
 import at.hannibal2.skyhanni.utils.StringUtils.matchMatcher
 import at.hannibal2.skyhanni.utils.StringUtils.matchRegex
+import at.hannibal2.skyhanni.utils.StringUtils.removeColor
 import net.minecraft.item.ItemStack
 import net.minecraftforge.fml.common.eventhandler.SubscribeEvent
 
@@ -176,6 +177,21 @@ class ItemDisplayOverlayFeatures {
             }
         }
 
+        if (SkyHanniMod.feature.inventory.itemNumberAsStackSize.contains(13)) {
+            if (item.getInternalName() == "POTION") {
+                item.name?.let {
+                    "Dungeon (?<level>.*) Potion".toPattern().matchMatcher(it.removeColor()) {
+                        return when (val level = group("level").romanToDecimal()) {
+                            in 1..2 -> "§f$level"
+                            in 3..4 -> "§a$level"
+                            in 5..6 -> "§9$level"
+                            else -> "§5$level"
+                        }
+                    }
+                }
+            }
+        }
+
         return ""
     }