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

package com.minelittlepony.unicopia.item;


import java.util.List;

import com.google.common.collect.Lists;
import com.minelittlepony.unicopia.UItems;
import com.minelittlepony.unicopia.item.consumables.Toxicity;
import com.minelittlepony.util.MagicalDamageSource;
import com.minelittlepony.util.VecHelper;

import net.minecraft.entity.Entity;
import net.minecraft.entity.LightningEntity;
import net.minecraft.entity.LivingEntity;
import net.minecraft.entity.mob.CreeperEntity;
import net.minecraft.entity.passive.PigEntity;
import net.minecraft.entity.passive.VillagerEntity;
import net.minecraft.entity.player.PlayerEntity;
import net.minecraft.item.FoodComponent;
import net.minecraft.item.Item;
import net.minecraft.item.ItemGroup;
import net.minecraft.item.ItemStack;
import net.minecraft.item.Items;
import net.minecraft.server.world.ServerWorld;
import net.minecraft.util.ActionResult;
import net.minecraft.util.DefaultedList;
import net.minecraft.util.Hand;
import net.minecraft.util.Identifier;
import net.minecraft.util.Rarity;
import net.minecraft.util.TypedActionResult;
import net.minecraft.util.hit.EntityHitResult;
import net.minecraft.util.hit.HitResult;
import net.minecraft.util.registry.Registry;
import net.minecraft.world.World;

public class ZapAppleItem extends AppleItem {

    private static final List<Item> ALIASABLE_ITEMS = Lists.newArrayList(
            UItems.red_apple,
            UItems.green_apple,
            UItems.sweet_apple,
            UItems.sour_apple,
            UItems.rotten_apple,
            UItems.cooked_zap_apple
    );

    public ZapAppleItem() {
        super(new FoodComponent.Builder()
                .hunger(4)
                .saturationModifier(0.3F)
                .alwaysEdible().build());
    }

    @Override
    public TypedActionResult<ItemStack> use(World world, PlayerEntity player, Hand hand) {
        HitResult mop = VecHelper.getObjectMouseOver(player, 5, 0);

        if (mop != null && mop.getType() == HitResult.Type.ENTITY) {
            ItemStack stack = player.getStackInHand(hand);

            EntityHitResult ehr = ((EntityHitResult)mop);

            if (canFeedTo(stack, ehr.getEntity())) {
                return onFedTo(stack, player, ehr.getEntity());
            }
        }

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

    @Override
    public ItemStack finishUsing(ItemStack stack, World w, LivingEntity player) {
        stack = super.finishUsing(stack, w, player);

        player.damage(MagicalDamageSource.ZAP_APPLE, 120);

        if (w instanceof ServerWorld) {
            ((ServerWorld) w).addLightning(new LightningEntity(w, player.x, player.y, player.z, false));
        }

        return stack;
    }

    public boolean canFeedTo(ItemStack stack, Entity e) {
        return e instanceof VillagerEntity
                || e instanceof CreeperEntity
                || e instanceof PigEntity;
    }

    public TypedActionResult<ItemStack> onFedTo(ItemStack stack, PlayerEntity player, Entity e) {
        e.onStruckByLightning(new LightningEntity(e.world, e.x, e.y, e.z, false));

        if (!player.abilities.creativeMode) {
            stack.decrement(1);
        }

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

    @Override
    public void appendStacks(ItemGroup tab, DefaultedList<ItemStack> items) {
        super.appendStacks(tab, items);
        if (isIn(tab)) {
            ALIASABLE_ITEMS.forEach(item -> items.add(new ItemStack(item)));
        }
    }

    public Item getAppearance(ItemStack stack) {
        if (stack.hasTag() && stack.getTag().containsKey("appearance")) {
            return Registry.ITEM.get(new Identifier(stack.getTag().getString("appearance")));
        }

        return Items.AIR;
    }

    @Override
    public String getTranslationKey(ItemStack stack) {
        Item appearance = getAppearance(stack);
        return appearance == Items.AIR ? super.getTranslationKey() : appearance.getTranslationKey(stack);
    }

    @Override
    public Toxicity getToxicityLevel(ItemStack stack) {
        return getAppearance(stack) == Items.AIR ? Toxicity.SEVERE : Toxicity.SAFE;
    }

    @Override
    public Rarity getRarity(ItemStack stack) {
        if (getAppearance(stack) == Items.AIR) {
            return Rarity.EPIC;
        }

        return Rarity.RARE;
    }
}