package net.citizensnpcs.editor;

import org.bukkit.Material;
import org.bukkit.entity.Pig;
import org.bukkit.entity.Player;
import org.bukkit.inventory.ItemStack;

import net.citizensnpcs.api.npc.NPC;
import net.citizensnpcs.api.util.Messaging;
import net.citizensnpcs.trait.Saddle;
import net.citizensnpcs.util.Messages;

public class PigEquipper implements Equipper {
    @Override
    public void equip(Player equipper, NPC toEquip) {
        ItemStack hand = equipper.getInventory().getItemInHand();
        Pig pig = (Pig) toEquip.getEntity();
        if (hand.getType() == Material.SADDLE) {
            if (!pig.hasSaddle()) {
                toEquip.getTrait(Saddle.class).toggle();
                hand.setAmount(0);
                Messaging.sendTr(equipper, Messages.SADDLED_SET, toEquip.getName());
            }
        } else if (pig.hasSaddle()) {
            equipper.getWorld().dropItemNaturally(pig.getLocation(), new ItemStack(Material.SADDLE, 1));
            toEquip.getTrait(Saddle.class).toggle();
            Messaging.sendTr(equipper, Messages.SADDLED_STOPPED, toEquip.getName());
        }
        equipper.getInventory().setItemInHand(hand);
    }
}