Show is mob can be captured in soul vial

1 files changed, 20 insertions, 0 deletions

diff --git a/src/main/java/kubatech/api/helpers/EnderIOHelper.java b/src/main/java/kubatech/api/helpers/EnderIOHelper.java
new file mode 100644
index 0000000000..b6c01c9741
--- /dev/null
+++ b/src/main/java/kubatech/api/helpers/EnderIOHelper.java
@@ -0,0 +1,20 @@
+package kubatech.api.helpers;
+
+import crazypants.enderio.EnderIO;
+import kubatech.api.LoaderReference;
+import net.minecraft.entity.Entity;
+import net.minecraft.entity.EntityList;
+import net.minecraft.entity.boss.IBossDisplayData;
+
+public class EnderIOHelper {
+    public static boolean canEntityBeCapturedWithSoulVial(Entity entity, String entityID) {
+        if (!LoaderReference.EnderIO) return true;
+        if (ReflectionHelper.<Boolean>callMethod(EnderIO.itemSoulVessel, "isBlackListed", false, entityID))
+            return false;
+        return crazypants.enderio.config.Config.soulVesselCapturesBosses || !(entity instanceof IBossDisplayData);
+    }
+
+    public static boolean canEntityBeCapturedWithSoulVial(Entity entity) {
+        return canEntityBeCapturedWithSoulVial(entity, EntityList.getEntityString(entity));
+    }
+}