Fix AuraMobsAPI#getMobLevel error

Archy-X · Jun 29, 2024 · 5f3a203 · 5f3a203
1 parent a4ae5ad
commit 5f3a203
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/src/main/java/dev/aurelium/auramobs/api/AuraMobsAPI.java b/src/main/java/dev/aurelium/auramobs/api/AuraMobsAPI.java
@@ -33,11 +33,18 @@ public static int getMobLevel(Entity e) {
         if (!plugin.isAuraMob(m)) {
             return 1;
         }
-        String persistent = m.getPersistentDataContainer().get(plugin.getMobKey(), PersistentDataType.STRING);
-        if (persistent == null || persistent.isEmpty()) {
+        Integer persistent = m.getPersistentDataContainer().get(plugin.getMobKey(), PersistentDataType.INTEGER);
+        if (persistent == null) {
             return 1;
         }
-        return Integer.parseInt(persistent);
+        return persistent;
+    }
+
+    public static boolean isAuraMob(Entity entity) {
+       if (!(entity instanceof LivingEntity m)) {
+           return false;
+       }
+        return plugin.isAuraMob(m);
     }
 
     /**