package com.bewitchment.api.registry.entity;

import com.bewitchment.Util;
import com.bewitchment.api.capability.magicpower.MagicPower;
import net.minecraft.block.state.IBlockState;
import net.minecraft.entity.Entity;
import net.minecraft.entity.EntityLivingBase;
import net.minecraft.entity.MoverType;
import net.minecraft.entity.player.EntityPlayer;
import net.minecraft.inventory.InventoryHelper;
import net.minecraft.item.ItemStack;
import net.minecraft.nbt.NBTTagCompound;
import net.minecraft.util.DamageSource;
import net.minecraft.util.EnumActionResult;
import net.minecraft.util.EnumHand;
import net.minecraft.util.math.BlockPos;
import net.minecraft.util.math.Vec3d;
import net.minecraft.world.World;
import net.minecraftforge.fml.common.ObfuscationReflectionHelper;

@SuppressWarnings({"NullableProblems"})
public abstract class EntityBroom extends Entity {
    private ItemStack item;

    private boolean canFly;

    public EntityBroom(World world) {
        super(world);
        setSize(0.7f, 0.7f);
        setNoGravity(true);
    }

    @Override
    public EnumActionResult applyPlayerInteraction(EntityPlayer player, Vec3d vec, EnumHand hand) {
        if (!player.isRiding() && !player.isSneaking()) {
            player.rotationYaw = rotationYaw;
            player.rotationPitch = rotationPitch;
            player.startRiding(this);
            return EnumActionResult.SUCCESS;
        }
        return super.applyPlayerInteraction(player, vec, hand);
    }

    @Override
    public Entity getControllingPassenger() {
        return getPassengers().isEmpty() ? null : getPassengers().get(0);
    }

    @Override
    public boolean attackEntityFrom(DamageSource source, float amount) {
        if (getPassengers().isEmpty() && source.getImmediateSource() instanceof EntityPlayer) {
            dropBroom();
        }
        return super.attackEntityFrom(source, amount);
    }

    @Override
    public boolean canBeCollidedWith() {
        return !isDead;
    }

    @Override
    public boolean canBePushed() {
        return canBeCollidedWith();
    }

    @Override
    public boolean processInitialInteract(EntityPlayer player, EnumHand hand) {
        if (item == null) {
            item = player.getHeldItem(hand).splitStack(1);
            return true;
        }
        return super.processInitialInteract(player, hand);
    }

    @Override
    public double getMountedYOffset() {
        return 0.4;
    }

    @Override
    public void onUpdate() {
        super.onUpdate();
        float friction = 0.98f;
        if (onGround) friction = 0.4f;
        motionX *= friction;
        motionZ *= friction;
        Entity rider = getControllingPassenger();
        if (rider instanceof EntityPlayer) {
            boolean jump = getJump((EntityPlayer) rider);
            if (rider.ticksExisted % 20 == 0 && (jump || !onGround)) canFly = MagicPower.attemptDrain(null, (EntityPlayer) rider, getMagicCost());
            if (canFly) {
                float speed = getSpeed();
                float maxSpeed = getMaxSpeed();
                if (Math.abs(motionX) < maxSpeed) motionX += rider.motionX * speed;
                if (Math.abs(motionZ) < maxSpeed) motionZ += rider.motionZ * speed;
                rotationYaw = rider.rotationYaw;
                if (jump && motionY < 1) motionY += (0.1f + getThrust());
            }
        }
        if (collidedHorizontally) {
            motionX = 0;
            motionZ = 0;
        }
        motionY -= 0.1f;
        motionY *= 0.75f;
        move(MoverType.SELF, motionX, motionY, motionZ);
    }

    @Override
    protected void updateFallState(double y, boolean onGround, IBlockState state, BlockPos pos) {
    }

    @Override
    protected void entityInit() {
    }

    @Override
    protected void writeEntityToNBT(NBTTagCompound tag) {
        if (item != null) tag.setTag("item", item.serializeNBT());
    }

    @Override
    protected void readEntityFromNBT(NBTTagCompound tag) {
        item = tag.hasKey("item") ? new ItemStack(tag.getCompoundTag("item")) : ItemStack.EMPTY;
    }

    protected abstract float getSpeed();

    protected abstract float getMaxSpeed();

    protected abstract float getThrust();

    protected abstract int getMagicCost();

    public void dismount() {
        dropBroom();
    }

    public void dropBroom() {
        if (!world.isRemote) {
            if (getRidingEntity() != null && this.getControllingPassenger() instanceof EntityPlayer) {
                EntityPlayer player = (EntityPlayer) getControllingPassenger();
                Util.giveItem(player, item.copy());
            }
            else InventoryHelper.spawnItemStack(world, posX, posY, posZ, item.copy());
        }
        setDead();
    }

    private static boolean getJump(EntityLivingBase rider) throws IllegalArgumentException {
        return ObfuscationReflectionHelper.getPrivateValue(EntityLivingBase.class, rider, "isJumping", "field_70703_bu");
    }
}