package org.unifiedtech.bukkit;

import org.bukkit.Material;
import org.bukkit.entity.Player;
import org.bukkit.event.Listener;
import org.bukkit.event.entity.EntityDamageEvent;
import org.bukkit.event.entity.EntityDamageEvent.DamageCause;
import org.bukkit.inventory.PlayerInventory;
import org.bukkit.plugin.java.JavaPlugin;

public class MyEventHandler implements Listener {

    private JavaPlugin plugin;

    public MyEventHandler(JavaPlugin plugin) {
        this.plugin = plugin;
    }

    public void onEntityDamage(EntityDamageEvent event) {
        if (!(event.getEntity() instanceof Player)) {
            return;
        }

        if (!(plugin instanceof MyBukkitPlugin)) {
            return;
        }

        Player player = (Player)event.getEntity();
        PlayerInventory inventory = player.getInventory();

        boolean chainArmourSet = (inventory.getHelmet().getType() == Material.CHAINMAIL_HELMET &&
                                    inventory.getChestplate().getType() == Material.CHAINMAIL_CHESTPLATE &&
                                    inventory.getLeggings().getType() == Material.CHAINMAIL_LEGGINGS &&
                                    inventory.getBoots().getType() == Material.CHAINMAIL_BOOTS);

        boolean goldArmourSet = (inventory.getHelmet().getType() == Material.GOLD_HELMET &&
                                    inventory.getChestplate().getType() == Material.GOLD_CHESTPLATE &&
                                    inventory.getLeggings().getType() == Material.GOLD_LEGGINGS &&
                                    inventory.getBoots().getType() == Material.GOLD_BOOTS);

        MyBukkitPlugin bPlugin = (MyBukkitPlugin)plugin;

        if (event.getCause() == DamageCause.FALL) {
            if (chainArmourSet) {
                event.setCancelled(true);
            }
        } else if (event.getCause() == DamageCause.LAVA ||
                    event.getCause() == DamageCause.FIRE ||
                    event.getCause() == DamageCause.FIRE_TICK) {

            if (goldArmourSet) {
                event.setCancelled(true);
            }
        }
    }
}