1.0.0

Production Release

src/main/java/com/rsscripting/mobcage/util/MobEquipmentUtil.java

@@ -0,0 +1,77 @@
+package com.rsscripting.mobcage.util;
+
+import com.rsscripting.mobcage.MobCage;
+import org.bukkit.entity.LivingEntity;
+import org.bukkit.entity.Player;
+import org.bukkit.inventory.EntityEquipment;
+import org.bukkit.inventory.ItemStack;
+
+public class MobEquipmentUtil {
+
+    public static void handleEquipmentOnCapture(
+            MobCage plugin,
+            LivingEntity entity,
+            Player player
+    ) {
+
+        EntityEquipment equipment = entity.getEquipment();
+
+        if (equipment == null) {
+            return;
+        }
+
+        boolean hasEquipment =
+                (equipment.getHelmet() != null && !equipment.getHelmet().getType().isAir()) ||
+                        (equipment.getChestplate() != null && !equipment.getChestplate().getType().isAir()) ||
+                        (equipment.getLeggings() != null && !equipment.getLeggings().getType().isAir()) ||
+                        (equipment.getBoots() != null && !equipment.getBoots().getType().isAir()) ||
+                        (equipment.getItemInMainHand() != null && !equipment.getItemInMainHand().getType().isAir()) ||
+                        (equipment.getItemInOffHand() != null && !equipment.getItemInOffHand().getType().isAir());
+
+        if (!hasEquipment) {
+            return;
+        }
+
+        boolean dropEquipment =
+                plugin.getConfig().getBoolean("drop-equip-on-capture");
+
+        String message =
+                plugin.getConfig().getString("equipment-drop-message", "");
+
+        if (dropEquipment) {
+
+            dropItem(entity, equipment.getHelmet());
+            dropItem(entity, equipment.getChestplate());
+            dropItem(entity, equipment.getLeggings());
+            dropItem(entity, equipment.getBoots());
+            dropItem(entity, equipment.getItemInMainHand());
+            dropItem(entity, equipment.getItemInOffHand());
+
+        }
+
+        equipment.setHelmet(null);
+        equipment.setChestplate(null);
+        equipment.setLeggings(null);
+        equipment.setBoots(null);
+        equipment.setItemInMainHand(null);
+        equipment.setItemInOffHand(null);
+
+        if (!message.isBlank()) {
+            player.sendRichMessage(message);
+        }
+
+    }
+
+    private static void dropItem(LivingEntity entity, ItemStack item) {
+
+        if (item == null || item.getType().isAir()) {
+            return;
+        }
+
+        entity.getWorld().dropItemNaturally(
+                entity.getLocation(),
+                item.clone()
+        );
+    }
+
+}