1 files changed, 25 insertions, 1 deletions

diff --git a/src/main/java/at/hannibal2/skyhanni/utils/SkyBlockItemModifierUtils.kt b/src/main/java/at/hannibal2/skyhanni/utils/SkyBlockItemModifierUtils.kt
index d092774af..0cf1d3b32 100644
--- a/src/main/java/at/hannibal2/skyhanni/utils/SkyBlockItemModifierUtils.kt
+++ b/src/main/java/at/hannibal2/skyhanni/utils/SkyBlockItemModifierUtils.kt
@@ -9,6 +9,7 @@ import at.hannibal2.skyhanni.utils.NEUInternalName.Companion.asInternalName
 import at.hannibal2.skyhanni.utils.StringUtils.matchMatcher
 import com.google.gson.JsonObject
 import net.minecraft.item.ItemStack
+import java.util.*
 
 object SkyBlockItemModifierUtils {
     private val drillPartTypes = listOf("drill_part_upgrade_module", "drill_part_engine", "drill_part_fuel_tank")
@@ -43,7 +44,7 @@ object SkyBlockItemModifierUtils {
         getAttributeInt("upgrade_level") ?: getAttributeInt("dungeon_item_level")
     } else null
 
-    private fun ItemStack.isDungeonItem(): Boolean = getLore().lastOrNull()?.contains("DUNGEON") == true
+    private fun ItemStack.isDungeonItem() = getLore().any { it.contains("DUNGEON ") }
 
     fun ItemStack.getPetExp() = getPetInfo()?.get("exp")?.asDouble
 
@@ -263,4 +264,27 @@ object SkyBlockItemModifierUtils {
             fun getByName(name: String) = entries.firstOrNull { it.name == name }
         }
     }
+
+    enum class GemstoneSlotType(val colorCode: Char) {
+        JADE('a'),
+        AMBER('6'),
+        TOPAZ('e'),
+        SAPPHIRE('b'),
+        AMETHYST('5'),
+        JASPER('d'),
+        RUBY('c'),
+        OPAL('f'),
+        COMBAT('4'),
+        OFFENSIVE('9'),
+        DEFENSIVE('a'),
+        MINING('5'),
+        UNIVERSAL('f')
+        ;
+
+        companion object {
+            fun getColorCode(name: String) = entries.stream().filter {
+                name.uppercase(Locale.ENGLISH).contains(it.name)
+            }.findFirst().get().colorCode
+        }
+    }
 }
\ No newline at end of file