Unicopia/src/main/java/com/minelittlepony/unicopia/item/EnchantedStaffItem.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.entity.player.Pony;
import com.minelittlepony.unicopia.magic.Affinity;
import com.minelittlepony.unicopia.magic.CasterUtils;
import com.minelittlepony.unicopia.magic.Affine;
import com.minelittlepony.unicopia.magic.Caster;
import com.minelittlepony.unicopia.magic.TossedMagicEffect;
import com.minelittlepony.unicopia.util.projectile.TossableItem;

import net.minecraft.block.BlockState;
import net.minecraft.client.item.TooltipContext;
import net.minecraft.entity.EquipmentSlot;
import net.minecraft.entity.LivingEntity;
import net.minecraft.entity.damage.DamageSource;
import net.minecraft.entity.player.PlayerEntity;
import net.minecraft.item.ItemStack;
import net.minecraft.particle.ParticleTypes;
import net.minecraft.sound.SoundCategory;
import net.minecraft.sound.SoundEvents;
import net.minecraft.text.Text;
import net.minecraft.util.Hand;
import net.minecraft.util.TypedActionResult;
import net.minecraft.util.math.BlockPos;
import net.minecraft.util.math.Vec3d;
import net.minecraft.world.World;

public class EnchantedStaffItem extends StaffItem implements Affine, TossableItem {

    @Nonnull
    private final TossedMagicEffect effect;

    public EnchantedStaffItem(Settings settings, @Nonnull TossedMagicEffect effect) {
        super(settings.maxDamage(500));

        this.effect = effect;
    }

    @Override
    public void appendTooltip(ItemStack stack, @Nullable World world, List<Text> tooltip, TooltipContext context) {
        tooltip.add(getAffinity().getName());
    }

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

            player.setCurrentHand(hand);
            return TypedActionResult.consume(itemstack);
        }

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

    @Override
    public void onStoppedUsing(ItemStack itemstack, World world, LivingEntity entity, int timeLeft) {
        if (EquinePredicates.PLAYER_UNICORN.test(entity)) {
            int i = getMaxUseTime(itemstack) - timeLeft;

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

    @Override
    protected boolean castContainedEffect(ItemStack stack, LivingEntity target, LivingEntity attacker) {
        if (attacker.isSneaking()) {
            stack.damage(50, attacker, p -> p.sendEquipmentBreakStatus(EquipmentSlot.MAINHAND));

            CasterUtils.toCaster(attacker).ifPresent(c -> c.subtractEnergyCost(4));
            CasterUtils.toCaster(target).ifPresent(c -> onImpact(
                    c,
                    target.getBlockPos(),
                    target.getEntityWorld().getBlockState(target.getBlockPos())
            ));

            return true;
        }

        return false;
    }

    @Override
    public void usageTick(World world, LivingEntity entity, ItemStack stack, int ticksRemaining) {
        if (entity instanceof LivingEntity) {
            LivingEntity living = entity;

            if (living.getActiveItem().getItem() == this) {
                Vec3d eyes = entity.getCameraPosVec(1);

                float i = getMaxUseTime(stack) - ticksRemaining;

                world.addParticle(i > 150 ? ParticleTypes.LARGE_SMOKE : ParticleTypes.CLOUD, eyes.x, eyes.y, eyes.z,
                        (world.random.nextGaussian() - 0.5) / 10,
                        (world.random.nextGaussian() - 0.5) / 10,
                        (world.random.nextGaussian() - 0.5) / 10
                );
                world.playSound(null, entity.getBlockPos(), SoundEvents.ENTITY_GUARDIAN_ATTACK, SoundCategory.PLAYERS, 1, i / 20);

                if (i > 200) {
                    living.clearActiveItem();
                    living.damage(DamageSource.MAGIC, 1200);
                    CasterUtils.toCaster(entity).ifPresent(c -> onImpact(
                            c,
                            entity.getBlockPos(),
                            entity.getEntityWorld().getBlockState(entity.getBlockPos())
                    ));
                }
            }
        }

    }

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

    @Override
    public void toss(World world, ItemStack stack, PlayerEntity player) {
        Pony iplayer = Pony.of(player);

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

        stack.damage(1, player, p -> p.sendEquipmentBreakStatus(EquipmentSlot.MAINHAND));
    }

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

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

}