Fix error when categorizing entity not tracked by geary

MineInAbyss · Dec 2, 2023 · c3b007c · c3b007c
1 parent d6923e1
commit c3b007c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mobzy-spawning/src/main/kotlin/com/mineinabyss/mobzy/spawning/vertical/SpawnInfo.kt b/mobzy-spawning/src/main/kotlin/com/mineinabyss/mobzy/spawning/vertical/SpawnInfo.kt
@@ -1,6 +1,7 @@
 package com.mineinabyss.mobzy.spawning.vertical
 
 import com.mineinabyss.geary.papermc.tracking.entities.toGeary
+import com.mineinabyss.geary.papermc.tracking.entities.toGearyOrNull
 import com.mineinabyss.geary.prefabs.PrefabKey
 import com.mineinabyss.geary.prefabs.helpers.prefabs
 import com.mineinabyss.idofront.location.down
@@ -84,7 +85,7 @@ class SpawnInfo(
     companion object {
         //TODO perhaps give normal mobs prefab keys too to make this more type safe
         fun categorizeByType(mobs: Collection<Entity>): Map<PrefabKey?, Int> =
-            mobs.groupingBy { it.toGeary().prefabs.firstOrNull()?.get<PrefabKey>() }.eachCount()
+            mobs.groupingBy { it.toGearyOrNull()?.prefabs?.firstOrNull()?.get<PrefabKey>() }.eachCount()
     }
 }