ItemRebreatherUpgrade.java

package eiteam.esteemedinnovation.armor.exosuit.steam.upgrades;

import eiteam.esteemedinnovation.api.ChargableUtility;
import eiteam.esteemedinnovation.api.exosuit.ExosuitSlot;
import eiteam.esteemedinnovation.commons.Config;
import net.minecraft.entity.player.EntityPlayer;
import net.minecraft.inventory.EntityEquipmentSlot;
import net.minecraft.item.ItemStack;
import net.minecraft.util.DamageSource;
import net.minecraftforge.event.entity.living.LivingAttackEvent;

import static eiteam.esteemedinnovation.armor.ArmorModule.resource;

public class ItemRebreatherUpgrade extends ItemSteamExosuitUpgrade {
    public ItemRebreatherUpgrade() {
        super(ExosuitSlot.HEAD_GOGGLES, resource("rebreatherUpgrade"), null, 1);
    }

    @Override
    public void onPlayerAttacked(LivingAttackEvent event, EntityPlayer victim, ItemStack armorStack, EntityEquipmentSlot slot) {
        if (event.getSource() == DamageSource.drown && ChargableUtility.hasPower(victim, Config.rebreatherConsumption)) {
            ChargableUtility.drainSteam(victim.getItemStackFromSlot(EntityEquipmentSlot.CHEST), Config.rebreatherConsumption, victim);
            event.setCanceled(true);
        }
    }
}