Unicopia/src/main/java/com/minelittlepony/unicopia/item/SugaryItem.java

package com.minelittlepony.unicopia.item;

import com.minelittlepony.unicopia.entity.player.Pony;

import net.minecraft.entity.LivingEntity;
import net.minecraft.entity.player.PlayerEntity;
import net.minecraft.item.Item;
import net.minecraft.item.ItemStack;
import net.minecraft.world.World;

public class SugaryItem extends Item {

    private final int sugarAmount;

    public SugaryItem(Settings settings, int sugar) {
        super(settings);
        sugarAmount = sugar;
    }

    @Override
    public ItemStack finishUsing(ItemStack stack, World world, LivingEntity entity) {
        if (sugarAmount != 0 && entity instanceof PlayerEntity) {
            Pony.of((PlayerEntity)entity).addEnergy(sugarAmount);
        }

        return super.finishUsing(stack, world, entity);
    }
}