package com.turt2live.nodrop;

import org.bukkit.entity.Player;
import org.bukkit.event.EventHandler;
import org.bukkit.event.EventPriority;
import org.bukkit.event.Listener;
import org.bukkit.event.entity.EntityDeathEvent;
import org.bukkit.inventory.ItemStack;
import org.bukkit.plugin.java.JavaPlugin;

import java.util.ArrayList;
import java.util.List;

public class NoDrop extends JavaPlugin implements Listener {

    public void onEnable() {
        getServer().getPluginManager().registerEvents(this, this);
    }

    @EventHandler(priority = EventPriority.HIGHEST, ignoreCancelled = true)
    public void onEntityDeath(EntityDeathEvent event) {
        if (!(event.getEntity() instanceof Player)) {
            List<ItemStack> remove = new ArrayList<ItemStack>();
            for (ItemStack stack : event.getDrops()) {
                if (stack != null) {
                    switch (stack.getType()) {
                        case CHAINMAIL_BOOTS:
                        case CHAINMAIL_CHESTPLATE:
                        case CHAINMAIL_HELMET:
                        case CHAINMAIL_LEGGINGS:
                            remove.add(stack);
                            break;
                        default:
                            continue;
                    }
                }
            }
            event.getDrops().removeAll(remove);
        }
    }

}