|  | 
|  | 1 | +package adhdmc.simplebucketmobs.listener; | 
|  | 2 | + | 
|  | 3 | +import adhdmc.simplebucketmobs.SimpleBucketMobs; | 
|  | 4 | +import net.kyori.adventure.text.minimessage.MiniMessage; | 
|  | 5 | +import org.bukkit.Location; | 
|  | 6 | +import org.bukkit.Material; | 
|  | 7 | +import org.bukkit.NamespacedKey; | 
|  | 8 | +import org.bukkit.entity.Entity; | 
|  | 9 | +import org.bukkit.entity.EntityType; | 
|  | 10 | +import org.bukkit.entity.LivingEntity; | 
|  | 11 | +import org.bukkit.event.EventHandler; | 
|  | 12 | +import org.bukkit.event.EventPriority; | 
|  | 13 | +import org.bukkit.event.Listener; | 
|  | 14 | +import org.bukkit.event.entity.CreatureSpawnEvent; | 
|  | 15 | +import org.bukkit.event.player.PlayerInteractEntityEvent; | 
|  | 16 | +import org.bukkit.event.player.PlayerInteractEvent; | 
|  | 17 | +import org.bukkit.inventory.EquipmentSlot; | 
|  | 18 | +import org.bukkit.inventory.ItemStack; | 
|  | 19 | +import org.bukkit.inventory.meta.ItemMeta; | 
|  | 20 | +import org.bukkit.persistence.PersistentDataContainer; | 
|  | 21 | +import org.bukkit.persistence.PersistentDataType; | 
|  | 22 | + | 
|  | 23 | +import java.io.IOException; | 
|  | 24 | + | 
|  | 25 | +public class BucketMob implements Listener { | 
|  | 26 | + | 
|  | 27 | +    NamespacedKey mobNBTKey = new NamespacedKey(SimpleBucketMobs.getPlugin(), "mob_nbt"); | 
|  | 28 | +    NamespacedKey mobTypeKey = new NamespacedKey(SimpleBucketMobs.getPlugin(), "mob_type"); | 
|  | 29 | + | 
|  | 30 | +    @EventHandler(priority = EventPriority.NORMAL, ignoreCancelled = true) | 
|  | 31 | +    public void bucketMob(PlayerInteractEntityEvent event) { | 
|  | 32 | +        if (event.getHand() != EquipmentSlot.HAND) return; | 
|  | 33 | +        if (!(event.getRightClicked() instanceof LivingEntity)) return; | 
|  | 34 | +        ItemStack bucket = event.getPlayer().getEquipment().getItemInMainHand(); | 
|  | 35 | +        if (bucket.getType() != Material.BUCKET) return; | 
|  | 36 | +        if (bucket.getItemMeta().getPersistentDataContainer().has(mobTypeKey)) return; | 
|  | 37 | +        ItemStack mobBucket = new ItemStack(Material.BUCKET); | 
|  | 38 | +        byte[] serializedMob; | 
|  | 39 | +        try { | 
|  | 40 | +            serializedMob = event.getRightClicked().getPersistentDataContainer().serializeToBytes(); | 
|  | 41 | +        } catch (IOException e) { | 
|  | 42 | +            event.getPlayer().sendRichMessage("<red>Failed to bucket mob (Serialization)."); | 
|  | 43 | +            return; | 
|  | 44 | +        } | 
|  | 45 | +        ItemMeta meta = mobBucket.getItemMeta(); | 
|  | 46 | +        PersistentDataContainer bucketPDC = meta.getPersistentDataContainer(); | 
|  | 47 | +        bucketPDC.set(mobNBTKey, PersistentDataType.BYTE_ARRAY, serializedMob); | 
|  | 48 | +        bucketPDC.set(mobTypeKey, PersistentDataType.STRING, event.getRightClicked().getType().toString()); | 
|  | 49 | +        meta.displayName(MiniMessage.miniMessage().deserialize("<aqua>Mob Bucket")); | 
|  | 50 | +        mobBucket.setItemMeta(meta); | 
|  | 51 | +        bucket.subtract(); | 
|  | 52 | +        event.getPlayer().getInventory().addItem(mobBucket); | 
|  | 53 | +        event.getRightClicked().remove(); | 
|  | 54 | +    } | 
|  | 55 | + | 
|  | 56 | +    @EventHandler(priority = EventPriority.NORMAL, ignoreCancelled = true) | 
|  | 57 | +    public void unbucketMob(PlayerInteractEvent event) { | 
|  | 58 | +        if (event.getHand() != EquipmentSlot.HAND) return; | 
|  | 59 | +        Location interactLoc = event.getInteractionPoint(); | 
|  | 60 | +        if (interactLoc == null) return; | 
|  | 61 | +        ItemStack bucket = event.getPlayer().getEquipment().getItemInMainHand(); | 
|  | 62 | +        if (bucket.getType() != Material.BUCKET) return; | 
|  | 63 | +        if (!bucket.getItemMeta().getPersistentDataContainer().has(mobNBTKey)) return; | 
|  | 64 | +        String mobTypeString = bucket.getItemMeta().getPersistentDataContainer().get(mobTypeKey, PersistentDataType.STRING); | 
|  | 65 | +        byte[] serializedMob = bucket.getItemMeta().getPersistentDataContainer().get(mobNBTKey, PersistentDataType.BYTE_ARRAY); | 
|  | 66 | +        EntityType mobType; | 
|  | 67 | +        try { mobType = EntityType.valueOf(mobTypeString); } | 
|  | 68 | +        catch (IllegalArgumentException e) { | 
|  | 69 | +            event.getPlayer().sendRichMessage("<red>Failed to unbucket mob (IllegalArgumentException)."); | 
|  | 70 | +            return; | 
|  | 71 | +        } | 
|  | 72 | +        Entity entity = interactLoc.getWorld().spawnEntity(interactLoc, mobType, CreatureSpawnEvent.SpawnReason.CUSTOM); | 
|  | 73 | +        try { entity.getPersistentDataContainer().readFromBytes(serializedMob); } | 
|  | 74 | +        catch (IOException e) { | 
|  | 75 | +            event.getPlayer().sendRichMessage("<red>Failed to unbucket mob (Deserialization)."); | 
|  | 76 | +            return; | 
|  | 77 | +        } | 
|  | 78 | +        bucket.subtract(); | 
|  | 79 | +        event.getPlayer().getInventory().addItem(new ItemStack(Material.BUCKET)); | 
|  | 80 | +    } | 
|  | 81 | + | 
|  | 82 | +} | 
0 commit comments