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

package com.minelittlepony.unicopia.item;

import java.util.Random;

import com.minelittlepony.util.MagicalDamageSource;
import com.minelittlepony.util.vector.VecHelper;

import net.minecraft.block.BlockPlanks;
import net.minecraft.creativetab.CreativeTabs;
import net.minecraft.entity.Entity;
import net.minecraft.entity.effect.EntityLightningBolt;
import net.minecraft.entity.monster.EntityCreeper;
import net.minecraft.entity.passive.EntityPig;
import net.minecraft.entity.passive.EntityVillager;
import net.minecraft.entity.player.EntityPlayer;
import net.minecraft.item.EnumRarity;
import net.minecraft.item.ItemFood;
import net.minecraft.item.ItemStack;
import net.minecraft.util.ActionResult;
import net.minecraft.util.EnumActionResult;
import net.minecraft.util.EnumHand;
import net.minecraft.util.NonNullList;
import net.minecraft.util.math.RayTraceResult;
import net.minecraft.world.World;

public class ItemApple extends ItemFood {

	private int[] typeRarities = new int[0];
	private String[] subTypes = new String[0];
	private String[] variants = subTypes;

	public int getRandomAppleMetadata(Random rand, Object variant) {
		int[] rarity = typeRarities;
		int result = 0;
		for (int i = 0; i < rarity.length && i < subTypes.length; i++) {
			if (rand.nextInt(rarity[i]) == 0) {
				result++;
			}
		}
		if (variant == BlockPlanks.EnumType.JUNGLE) {
			result = result == 0 ? 1 : result == 1 ? 0 : result;
		}
		if (variant == BlockPlanks.EnumType.SPRUCE) {
			result = result == 0 ? 3 : result == 3 ? 0 : result;
		}
		if (variant == BlockPlanks.EnumType.BIRCH) {
			result = result == 0 ? 2 : result == 2 ? 0 : result;
		}
		if (variant == BlockPlanks.EnumType.DARK_OAK) {
			result = result == 1 ? getZapAppleMetadata() : result == getZapAppleMetadata() ? 1 : result;
		}
		return result;
	}

	public ItemApple() {
		super(4, 3, false);
		setHasSubtypes(true);
		setMaxDamage(0);
		setTranslationKey("apple");
	}

	@Override
	public ActionResult<ItemStack> onItemRightClick(World world, EntityPlayer player, EnumHand hand) {
		RayTraceResult mop = VecHelper.getObjectMouseOver(player, 5, 0);

		if (mop != null && mop.typeOfHit == RayTraceResult.Type.ENTITY) {
			ItemStack stack = player.getHeldItem(hand);

			if (canFeedTo(stack, mop.entityHit)) {
				return onFedTo(stack, player, mop.entityHit);
			}
		}

		return super.onItemRightClick(world, player, hand);
	}

	@Override
	protected void onFoodEaten(ItemStack stack, World w, EntityPlayer player) {
		super.onFoodEaten(stack, w, player);

		if (isZapApple(stack)) {
			player.attackEntityFrom(MagicalDamageSource.create("zap"), 120);

			w.addWeatherEffect(new EntityLightningBolt(w, player.posX, player.posY, player.posZ, false));
		}
	}

	public boolean canFeedTo(ItemStack stack, Entity e) {
		return isZapApple(stack) && (e instanceof EntityVillager || e instanceof EntityCreeper || e instanceof EntityPig);
	}

	public boolean isZapApple(ItemStack stack) {
		int meta = stack.getMetadata();
		return meta == getZapAppleMetadata() || meta >= subTypes.length;
	}

	public ActionResult<ItemStack> onFedTo(ItemStack stack, EntityPlayer player, Entity e) {
		e.onStruckByLightning(new EntityLightningBolt(e.world, e.posX, e.posY, e.posZ, false));

		if (!player.capabilities.isCreativeMode) {
		    stack.shrink(1);
		}

		return new ActionResult<ItemStack>(EnumActionResult.SUCCESS, stack);
	}

	public int getZapAppleMetadata() {
		return 4;
	}

	public ItemApple setSubTypes(String... types) {
		subTypes = types;
		variants = new String[subTypes.length * 2];

		setTranslationKey(variants[0] = types[0]);

		for (int i = 1; i < variants.length; i++) {
			variants[i] = variants[0] + (i % subTypes.length != 0 ? "_" + subTypes[i % subTypes.length] : "");
		}
		return this;
	}

	public ItemApple setTypeRarities(int ... rarity) {
		typeRarities = rarity;
		return this;
	}

	public String[] getVariants() {
		return variants;
	}

	@Override
	public void getSubItems(CreativeTabs tab, NonNullList<ItemStack> items) {
	    if (isInCreativeTab(tab)) {
    		for (int i = 0; i < subTypes.length; i++) {
    			items.add(new ItemStack(this, 1, i));
    		}
	    }
	}

	@Override
	public EnumRarity getRarity(ItemStack stack) {
		int meta = stack.getMetadata();

		if (meta == getZapAppleMetadata()) {
		    return EnumRarity.EPIC;
		}

        if (meta >= subTypes.length) {
            return EnumRarity.RARE;
        }

        return EnumRarity.COMMON;
    }

	@Override
	public String getTranslationKey(ItemStack stack) {
	    int meta = Math.max(0, stack.getMetadata() % subTypes.length);

		return super.getTranslationKey(stack) + (meta > 0 ? "." + subTypes[meta] : "");
	}

	@Override
	public int getItemBurnTime(ItemStack stack) {
		return stack.getMetadata() == 2 ? 150 : 0;
	}
}