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

package com.minelittlepony.unicopia.item;

import java.util.List;

import javax.annotation.Nonnull;
import javax.annotation.Nullable;

import com.minelittlepony.unicopia.EquinePredicates;
import com.minelittlepony.unicopia.SpeciesList;
import com.minelittlepony.unicopia.entity.capabilities.IPlayer;
import com.minelittlepony.unicopia.projectile.ITossableItem;
import com.minelittlepony.unicopia.spell.CasterUtils;
import com.minelittlepony.unicopia.spell.IAligned;
import com.minelittlepony.unicopia.spell.ICaster;
import com.minelittlepony.unicopia.spell.ITossedEffect;
import com.minelittlepony.unicopia.spell.SpellAffinity;
import com.minelittlepony.util.lang.ClientLocale;

import net.minecraft.block.state.BlockState;
import net.minecraft.client.util.ITooltipFlag;
import net.minecraft.entity.Entity;
import net.minecraft.entity.LivingEntity;
import net.minecraft.entity.player.PlayerEntity;
import net.minecraft.init.SoundEvents;
import net.minecraft.item.ItemStack;
import net.minecraft.util.ActionResult;
import net.minecraft.util.DamageSource;
import net.minecraft.util.EnumActionResult;
import net.minecraft.util.EnumHand;
import net.minecraft.util.EnumParticleTypes;
import net.minecraft.util.SoundCategory;
import net.minecraft.util.math.BlockPos;
import net.minecraft.util.math.Vec3d;
import net.minecraft.world.World;

public class ItemMagicStaff extends ItemStaff implements IAligned, ITossableItem {

    @Nonnull
    private final ITossedEffect effect;

    public ItemMagicStaff(String domain, String name, @Nonnull ITossedEffect effect) {
        super(domain, name);

        this.effect = effect;
        setMaxDamage(500);
    }

    @Override
    public void addInformation(ItemStack stack, @Nullable World worldIn, List<String> tooltip, ITooltipFlag flagIn) {
        SpellAffinity affinity = getAffinity();
        tooltip.add(affinity.getColourCode() + ClientLocale.format(affinity.getUnlocalizedName()));
    }

    @Override
    public TypedActionResult<ItemStack> onItemRightClick(World world, PlayerEntity player, EnumHand hand) {
        if (EquinePredicates.MAGI.test(player) && hand == EnumHand.MAIN_HAND) {
            ItemStack itemstack =  player.getStackInHand(hand);

            player.setActiveHand(hand);

            return new TypedActionResult<ItemStack>(EnumActionResult.SUCCESS, itemstack);
        }

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

    @Override
    public void onPlayerStoppedUsing(ItemStack itemstack, World world, LivingEntity entity, int timeLeft) {
        if (EquinePredicates.MAGI.test(entity) && entity instanceof PlayerEntity) {

            int i = getMaxItemUseDuration(itemstack) - timeLeft;

            if (i > 10) {
                if (canBeThrown(itemstack)) {
                    toss(world, itemstack, (PlayerEntity)entity);
                }
            }
        }
    }

    @Override
    protected boolean castContainedEffect(ItemStack stack, LivingEntity target, LivingEntity attacker) {
        if (attacker.isSneaking()) {
            stack.damageItem(50, attacker);

            CasterUtils.toCaster(attacker).ifPresent(c -> c.subtractEnergyCost(4));

            onImpact(
                    CasterUtils.near(target),
                    target.getPosition(),
                    target.getEntityWorld().getBlockState(target.getPosition())
            );

            return true;
        }

        return false;
    }

    public void onUpdate(ItemStack stack, World world, Entity entity, int itemSlot, boolean isSelected) {

        if (entity instanceof LivingEntity) {
            LivingEntity living = (LivingEntity)entity;

            if (living.getActiveItemStack().getItem() == this) {
                Vec3d eyes = entity.getPositionEyes(1);

                float i = getMaxItemUseDuration(stack) - living.getItemInUseCount();

                world.spawnParticle(i > 150 ? EnumParticleTypes.SMOKE_LARGE : EnumParticleTypes.CLOUD, eyes.x, eyes.y, eyes.z,
                        (world.rand.nextGaussian() - 0.5) / 10,
                        (world.rand.nextGaussian() - 0.5) / 10,
                        (world.rand.nextGaussian() - 0.5) / 10
                );
                world.playSound(null, entity.getPosition(), SoundEvents.ENTITY_GUARDIAN_ATTACK, SoundCategory.PLAYERS, 1, i / 20);

                if (i > 200) {
                    living.resetActiveHand();
                    living.attackEntityFrom(DamageSource.MAGIC, 1200);
                    onImpact(
                            CasterUtils.toCaster(entity).orElseGet(() -> CasterUtils.near(entity)),
                            entity.getPosition(),
                            entity.getEntityWorld().getBlockState(entity.getPosition())
                    );
                }
            }
        }

    }

    @Override
    public int getMaxItemUseDuration(ItemStack stack) {
        return 72000;
    }

    @Override
    public void toss(World world, ItemStack stack, PlayerEntity player) {
        IPlayer iplayer = SpeciesList.instance().getPlayer(player);

        iplayer.subtractEnergyCost(4);
        effect.toss(iplayer);

        stack.damageItem(1, player);
    }

    @Override
    public void onImpact(ICaster<?> caster, BlockPos pos, BlockState state) {
        effect.onImpact(caster, pos, state);
    }

    @Override
    public SpellAffinity getAffinity() {
        return effect.getAffinity();
    }

}