1.0.1

Added Donkeys

src/main/java/com/rsscripting/mobcage/description/descriptor/DonkeyDescriptor.java

@@ -0,0 +1,86 @@
+package com.rsscripting.mobcage.description.descriptor;
+
+import com.rsscripting.mobcage.description.CaptureResult;
+import com.rsscripting.mobcage.description.MobAttribute;
+import com.rsscripting.mobcage.util.MobAgeUtil;
+import com.rsscripting.mobcage.util.MobHealthUtil;
+import com.rsscripting.mobcage.MobCage;
+import com.rsscripting.mobcage.util.MobEquipmentUtil;
+import net.kyori.adventure.text.Component;
+import net.kyori.adventure.text.serializer.plain.PlainTextComponentSerializer;
+import org.bukkit.Location;
+import org.bukkit.entity.Donkey;
+import org.bukkit.entity.EntityType;
+import org.bukkit.entity.LivingEntity;
+import org.bukkit.entity.Player;
+
+import java.util.List;
+
+public class DonkeyDescriptor implements MobDescriptor {
+
+    @Override
+    public boolean supports(LivingEntity entity) {
+        return entity instanceof Donkey;
+    }
+
+    @Override
+    public CaptureResult beforeCapture(LivingEntity entity, Player player) {
+
+        CaptureResult result = MobHealthUtil.validateFullHealth(entity);
+
+        if (!result.isAllowed()) {
+            return result;
+        }
+
+        MobAgeUtil.normalize(entity);
+
+        MobEquipmentUtil.handleEquipmentOnCapture(
+                MobCage.getInstance(),
+                entity,
+                player
+        );
+
+        return CaptureResult.allowed();
+    }
+
+    @Override
+    public String getDisplayName(LivingEntity entity) {
+        return MobAgeUtil.getAgePrefix(entity) + "Donkey";
+    }
+
+    @Override
+    public List<MobAttribute> getAttributes(LivingEntity entity) {
+        return List.of();
+    }
+
+    @Override
+    public void release(Location location, List<Component> lore) {
+
+        PlainTextComponentSerializer plain =
+                PlainTextComponentSerializer.plainText();
+
+        boolean baby = false;
+
+        for (int i = 0; i < lore.size(); i++) {
+
+            String line = plain.serialize(lore.get(i));
+
+            if (line.equals("Contains:") && i + 1 < lore.size()) {
+                baby = plain.serialize(lore.get(i + 1)).equals("Baby Donkey");
+            }
+        }
+
+        Donkey donkey = (Donkey) location.getWorld().spawnEntity(
+                location,
+                EntityType.DONKEY
+        );
+
+        if (baby) {
+            donkey.setBaby();
+        } else {
+            donkey.setAdult();
+        }
+
+        donkey.setPersistent(true);
+    }
+}

src/main/java/com/rsscripting/mobcage/description/descriptor/HorseDescriptor.java

@@ -13,7 +13,6 @@ import org.bukkit.Location;
 import org.bukkit.entity.EntityType;
 import org.bukkit.entity.LivingEntity;
 import org.bukkit.entity.Horse;
-import org.bukkit.entity.EntityType;
 import org.bukkit.entity.Player;
 
 import java.util.List;

src/main/java/com/rsscripting/mobcage/description/descriptor/MobDescriptorRegistry.java

@@ -11,6 +11,7 @@ public final class MobDescriptorRegistry {
             new CaveSpiderDescriptor(),
             new ChickenDescriptor(),
             new CowDescriptor(),
+            new DonkeyDescriptor(),
             new HorseDescriptor(),
             new PigDescriptor(),
             new SheepDescriptor(),

src/main/java/com/rsscripting/mobcage/listener/MobReleaseListener.java

@@ -73,6 +73,9 @@ public class MobReleaseListener implements Listener {
             case "Cow", "Baby Cow" ->
                     new CowDescriptor().release(location, lore);
 
+            case "Donkey", "Baby Donkey" ->
+                    new DonkeyDescriptor().release(location, lore);
+
             case "Horse", "Baby Horse" ->
                     new HorseDescriptor().release(location, lore);
 

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

@@ -1,6 +1,8 @@
 package com.rsscripting.mobcage.util;
 
 import com.rsscripting.mobcage.MobCage;
+import org.bukkit.Material;
+import org.bukkit.entity.ChestedHorse;
 import org.bukkit.entity.LivingEntity;
 import org.bukkit.entity.Player;
 import org.bukkit.inventory.EntityEquipment;
@@ -16,55 +18,109 @@ public class MobEquipmentUtil {
 
         EntityEquipment equipment = entity.getEquipment();
 
-        if (equipment == null) {
-            return;
+        boolean hasEquipment =
+                hasItem(equipment != null ? equipment.getHelmet() : null) ||
+                        hasItem(equipment != null ? equipment.getChestplate() : null) ||
+                        hasItem(equipment != null ? equipment.getLeggings() : null) ||
+                        hasItem(equipment != null ? equipment.getBoots() : null) ||
+                        hasItem(equipment != null ? equipment.getItemInMainHand() : null) ||
+                        hasItem(equipment != null ? equipment.getItemInOffHand() : null);
+
+        boolean hasChestedHorseEquipment = false;
+
+        if (entity instanceof ChestedHorse chestedHorse) {
+
+            if (hasItem(chestedHorse.getInventory().getSaddle())) {
+                hasChestedHorseEquipment = true;
+            }
+
+            if (chestedHorse.isCarryingChest()) {
+                hasChestedHorseEquipment = true;
+
+                for (ItemStack item : chestedHorse.getInventory().getStorageContents()) {
+                    if (hasItem(item)) {
+                        hasChestedHorseEquipment = true;
+                        break;
+                    }
+                }
+            }
         }
 
-        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) {
+        if (!hasEquipment && !hasChestedHorseEquipment) {
             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());
+            if (equipment != null) {
+                dropItem(entity, equipment.getHelmet());
+                dropItem(entity, equipment.getChestplate());
+                dropItem(entity, equipment.getLeggings());
+                dropItem(entity, equipment.getBoots());
+                dropItem(entity, equipment.getItemInMainHand());
+                dropItem(entity, equipment.getItemInOffHand());
+            }
 
+            if (entity instanceof ChestedHorse chestedHorse) {
+
+                if (chestedHorse.isCarryingChest()) {
+
+                    for (ItemStack item : chestedHorse.getInventory().getStorageContents()) {
+                        dropItem(entity, item);
+                    }
+
+                    // The chest itself is also a possession.
+                    dropItem(
+                            entity,
+                            new ItemStack(Material.CHEST)
+                    );
+                }
+            }
+
+            String message =
+                    plugin.getConfig().getString("equipment-drop-message", "");
+
+            if (!message.isBlank()) {
+                player.sendRichMessage(message);
+            }
+
+        } else {
+
+            player.sendRichMessage(
+                    "<yellow>The mob struggled during capture. During the struggle all of the mob's possessions were destroyed.</yellow>"
+            );
         }
 
-        equipment.setHelmet(null);
-        equipment.setChestplate(null);
-        equipment.setLeggings(null);
-        equipment.setBoots(null);
-        equipment.setItemInMainHand(null);
-        equipment.setItemInOffHand(null);
-
-        if (!message.isBlank()) {
-            player.sendRichMessage(message);
+        if (equipment != null) {
+            equipment.setHelmet(null);
+            equipment.setChestplate(null);
+            equipment.setLeggings(null);
+            equipment.setBoots(null);
+            equipment.setItemInMainHand(null);
+            equipment.setItemInOffHand(null);
         }
 
+        if (entity instanceof ChestedHorse chestedHorse) {
+
+            chestedHorse.getInventory().setSaddle(null);
+
+            if (chestedHorse.isCarryingChest()) {
+                chestedHorse.getInventory().clear();
+                chestedHorse.setCarryingChest(false);
+            }
+        }
+    }
+
+    private static boolean hasItem(ItemStack item) {
+        return item != null && !item.getType().isAir();
     }
 
     private static void dropItem(LivingEntity entity, ItemStack item) {
 
-        if (item == null || item.getType().isAir()) {
+        if (!hasItem(item)) {
             return;
         }
 
@@ -73,5 +129,4 @@ public class MobEquipmentUtil {
                 item.clone()
         );
     }
-
 }