ItemArmorPaper.class #2

package com.arctic.paperArmor.items;

import com.arctic.paperArmor.init.ModItems;
import net.minecraft.entity.player.EntityPlayer;
import net.minecraft.item.ItemStack;
import net.minecraft.nbt.NBTTagCompound;
import net.minecraft.util.IChatComponent;
import net.minecraft.world.World;

public class ItemArmorPaper extends ItemArmorPA {
    public ItemArmorPaper(int armorType) {
        super("paper", paper, armorType);
    }

    private int water = 20;
    private int fire = 10;

    @Override
    public void onArmorTick(World world, EntityPlayer player, ItemStack stack) {
        super.onArmorTick(world, player, stack);
        if (!world.isRemote) {
            if (stack.hasTagCompound()) {
                if (player.isWet()){
                    if (stack.getTagCompound().hasKey("WetTick")){
                        wetTick(world, player, stack);
                    } else {
                        stack.getTagCompound().setInteger("WetTick", water);
                    }
                } else if (player.isBurning()){
                    if (stack.getTagCompound().hasKey("FireTick")){
                        fireTick(world, player, stack);
                    } else {
                        stack.getTagCompound().setInteger("FireTick", fire);
                    }
                }
            } else {
                stack.setTagCompound(new NBTTagCompound());
            }
        }
    }

    private void wetTick(World world, EntityPlayer player, ItemStack stack){
        int wetTick = stack.getTagCompound().getInteger("WetTick");
        if (wetTick == 0) {
            if (stack.attemptDamageItem(1, world.rand)) {
                if (stack.getItem() == ModItems.paperBoots) {
                    playWetBreak(player, stack, "water");
                    player.setCurrentItemOrArmor(1, null);
                } else if (stack.getItem() == ModItems.paperLeggings) {
                    playWetBreak(player, stack, "water");
                    player.setCurrentItemOrArmor(2, null);
                } else if (stack.getItem() == ModItems.paperChestPlate) {
                    playWetBreak(player, stack, "water");
                    player.setCurrentItemOrArmor(3, null);
                } else if (stack.getItem() == ModItems.paperHelmet) {
                    playWetBreak(player, stack, "water");
                    player.setCurrentItemOrArmor(4, null);
                }
            }
            wetTick = water;
        } else wetTick--;
        stack.getTagCompound().setInteger("WetTick", wetTick);
    }

    private void fireTick(World world, EntityPlayer player, ItemStack stack){
        int fireTick = stack.getTagCompound().getInteger("FireTick");
        if (fireTick == 0) {
            if (stack.attemptDamageItem(1, world.rand)) {
                if (stack.getItem() == ModItems.paperBoots) {
                    playWetBreak(player, stack, "fire");
                    player.setCurrentItemOrArmor(1, null);
                } else if (stack.getItem() == ModItems.paperLeggings) {
                    playWetBreak(player, stack, "fire");
                    player.setCurrentItemOrArmor(2, null);
                } else if (stack.getItem() == ModItems.paperChestPlate) {
                    playWetBreak(player, stack, "fire");
                    player.setCurrentItemOrArmor(3, null);
                } else if (stack.getItem() == ModItems.paperHelmet) {
                    playWetBreak(player, stack, "fire");
                    player.setCurrentItemOrArmor(4, null);
                }
            }
            fireTick = fire;
        } else fireTick--;
        stack.getTagCompound().setInteger("WetTick", fireTick);
    }

    private void playWetBreak(EntityPlayer player, ItemStack stack, String src){
        IChatComponent msg = stack.func_151000_E();
        msg.appendText(" It seams paper and "+src+" do not mix.");
        player.addChatComponentMessage(msg);
    }
}