UpgradePotion.java

package chbachman.armour.upgrade.upgradeList;

import chbachman.api.upgrade.IUpgrade;
import chbachman.api.upgrade.Upgrade;
import chbachman.api.util.ArmourSlot;
import chbachman.armour.register.Vanilla;
import chbachman.armour.util.EnergyUtil;
import cpw.mods.fml.relauncher.Side;
import cpw.mods.fml.relauncher.SideOnly;
import net.minecraft.entity.player.EntityPlayer;
import net.minecraft.init.Items;
import net.minecraft.item.ItemStack;
import net.minecraft.potion.Potion;
import net.minecraft.potion.PotionEffect;
import net.minecraft.util.IIcon;
import net.minecraft.world.World;

public class UpgradePotion extends Upgrade {

    public final PotionEffect effect;
    public final int energyCost;

    public UpgradePotion(String name, Potion effect, int level, int energyCost) {
        super(name);
        this.effect = new PotionEffect(effect.id, level, 0, true);
        this.energyCost = energyCost;
    }

    public UpgradePotion(String name, Potion effect, int level, int duration, int energyCost) {
        super(name);
        this.effect = new PotionEffect(effect.id, level, duration, true);
        this.energyCost = energyCost;
    }

    @Override
    public int onTick(World world, EntityPlayer player, ItemStack stack, ArmourSlot slot) {

        if (EnergyUtil.isEmpty(stack)) {
            return 0;
        }

        player.addPotionEffect(new PotionEffect(effect));
        return energyCost;
    }

    @Override
    public IUpgrade[] getDependencies() {

        return new IUpgrade[] { Vanilla.basePotion };
    }
    
    @Override
    @SideOnly(Side.CLIENT)
    public IIcon getIcon(){
    	
    	IIcon icon = new ItemStack(Items.apple, 1).getIconIndex();
    	//return null;
    	return icon;
    }

}