package ru.betterend.mixin.common;

import net.minecraft.sounds.SoundEvent;
import net.minecraft.util.Mth;
import net.minecraft.world.damagesource.DamageSource;
import net.minecraft.world.effect.MobEffect;
import net.minecraft.world.effect.MobEffectInstance;
import net.minecraft.world.effect.MobEffects;
import net.minecraft.world.entity.*;
import net.minecraft.world.entity.ai.attributes.AttributeMap;
import net.minecraft.world.entity.ai.attributes.AttributeModifier;
import net.minecraft.world.entity.ai.attributes.AttributeSupplier;
import net.minecraft.world.entity.ai.attributes.Attributes;
import net.minecraft.world.entity.animal.FlyingAnimal;
import net.minecraft.world.item.ElytraItem;
import net.minecraft.world.item.Item;
import net.minecraft.world.item.ItemStack;
import net.minecraft.world.level.Level;
import net.minecraft.world.phys.Vec3;
import org.spongepowered.asm.mixin.Mixin;
import org.spongepowered.asm.mixin.Shadow;
import org.spongepowered.asm.mixin.injection.At;
import org.spongepowered.asm.mixin.injection.At.Shift;
import org.spongepowered.asm.mixin.injection.Inject;
import org.spongepowered.asm.mixin.injection.ModifyArg;
import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
import org.spongepowered.asm.mixin.injection.callback.CallbackInfoReturnable;
import ru.betterend.interfaces.MobEffectApplier;
import ru.betterend.item.ArmoredElytra;
import ru.betterend.item.CrystaliteArmor;
import ru.betterend.registry.EndAttributes;

import java.util.Collection;

@Mixin(LivingEntity.class)
public abstract class LivingEntityMixin extends Entity {

	public LivingEntityMixin(EntityType<?> entityType, Level level) {
		super(entityType, level);
	}

	@Shadow
	protected int fallFlyTicks;

	@Shadow
	public abstract boolean hasEffect(MobEffect mobEffect);

	@Shadow
	public abstract ItemStack getItemBySlot(EquipmentSlot equipmentSlot);

	@Shadow
	public abstract void calculateEntityAnimation(LivingEntity livingEntity, boolean b);

	@Shadow
	protected abstract SoundEvent getFallDamageSound(int i);

	@Shadow
	public abstract boolean isFallFlying();

	@Shadow
	public abstract AttributeMap getAttributes();

	private Entity lastAttacker;

	@Inject(method = "createLivingAttributes", at = @At("RETURN"), cancellable = true)
	private static void be_addLivingAttributes(CallbackInfoReturnable<AttributeSupplier.Builder> info) {
		EndAttributes.addLivingEntityAttributes(info.getReturnValue());
	}

	@Inject(method = "tickEffects", at = @At("HEAD"))
	protected void be_applyEffects(CallbackInfo info) {
		if (!level.isClientSide()) {
			LivingEntity owner = LivingEntity.class.cast(this);
			if (CrystaliteArmor.hasFullSet(owner)) {
				CrystaliteArmor.applySetEffect(owner);
			}
			getArmorSlots().forEach(itemStack -> {
				if (itemStack.getItem() instanceof MobEffectApplier) {
					((MobEffectApplier) itemStack.getItem()).applyEffect(owner);
				}
			});
		}
	}

	@Inject(method = "canBeAffected", at = @At("HEAD"), cancellable = true)
	public void be_canBeAffected(MobEffectInstance mobEffectInstance, CallbackInfoReturnable<Boolean> info) {
		if (mobEffectInstance.getEffect() == MobEffects.BLINDNESS && getAttributes().getValue(EndAttributes.BLINDNESS_RESISTANCE) > 0.0) {
			info.setReturnValue(false);
		}
	}

	@Inject(method = "hurt", at = @At("HEAD"))
	public void be_hurt(DamageSource source, float amount, CallbackInfoReturnable<Boolean> info) {
		this.lastAttacker = source.getEntity();
	}
	
	@ModifyArg(method = "hurt", at = @At(value = "INVOKE", target = "Lnet/minecraft/world/entity/LivingEntity;knockback(FDD)V"))
	private float be_increaseKnockback(float value, double x, double z) {
		if (lastAttacker != null && lastAttacker instanceof LivingEntity) {
			LivingEntity attacker = (LivingEntity) lastAttacker;
			value += this.be_getKnockback(attacker.getMainHandItem().getItem());
		}
		return value;
	}

	@Inject(method = "updateFallFlying", at = @At("HEAD"), cancellable = true)
	private void be_updateFallFlying(CallbackInfo info) {
		ItemStack itemStack = getItemBySlot(EquipmentSlot.CHEST);
		if (!level.isClientSide && itemStack.getItem() instanceof ArmoredElytra) {
			boolean isFlying = getSharedFlag(7);
			if (isFlying && !onGround && !isPassenger() && !hasEffect(MobEffects.LEVITATION)) {
				if (ElytraItem.isFlyEnabled(itemStack)) {
					if ((fallFlyTicks + 1) % 20 == 0) {
						itemStack.hurtAndBreak(1, LivingEntity.class.cast(this),
								livingEntity -> livingEntity.broadcastBreakEvent(EquipmentSlot.CHEST));
					}
					isFlying = true;
				} else {
					isFlying = false;
				}
			} else {
				isFlying = false;
			}
			setSharedFlag(7, isFlying);
			info.cancel();
		}
	}

	@Inject(method = "travel", at = @At(value = "INVOKE",
			target = "Lnet/minecraft/world/entity/LivingEntity;isFallFlying()Z",
			shift = Shift.AFTER), cancellable = true)
	public void be_travel(Vec3 vec3, CallbackInfo info) {
		ItemStack itemStack = getItemBySlot(EquipmentSlot.CHEST);
		if (isFallFlying() && itemStack.getItem() instanceof ArmoredElytra) {
			Vec3 moveDelta = getDeltaMovement();
			if (moveDelta.y > -0.5D) {
				fallDistance = 1.0F;
			}

			Vec3 lookAngle = getLookAngle();
			double d = 0.08D;
			float rotX = xRot * 0.017453292F;
			double k = Math.sqrt(lookAngle.x * lookAngle.x + lookAngle.z * lookAngle.z);
			double l = Math.sqrt(getHorizontalDistanceSqr(moveDelta));
			double lookLen = lookAngle.length();
			float n = Mth.cos(rotX);
			n = (float) (n * n * Math.min(1.0D, lookLen / 0.4D));
			moveDelta = getDeltaMovement().add(0.0D, d * (-1.0D + (double) n * 0.75D), 0.0D);
			double coef;
			if (moveDelta.y < 0.0D && k > 0.0D) {
				coef = moveDelta.y * -0.1D * (double) n;
				moveDelta = moveDelta.add(lookAngle.x * coef / k, coef, lookAngle.z * coef / k);
			}

			if (rotX < 0.0F && k > 0.0D) {
				coef = l * (double) (-Mth.sin(rotX)) * 0.04D;
				moveDelta = moveDelta.add(-lookAngle.x * coef / k, coef * 3.2D, -lookAngle.z * coef / k);
			}

			if (k > 0.0D) {
				moveDelta = moveDelta.add((lookAngle.x / k * l - moveDelta.x) * 0.1D, 0.0D, (lookAngle.z / k * l - moveDelta.z) * 0.1D);
			}
			moveDelta = moveDelta.multiply(0.9900000095367432D, 0.9800000190734863D, 0.9900000095367432D);
			double movementFactor = ((ArmoredElytra) itemStack.getItem()).getMovementFactor();
			moveDelta = moveDelta.multiply(movementFactor, 1.0D, movementFactor);
			setDeltaMovement(moveDelta);
			move(MoverType.SELF, moveDelta);
			if (!level.isClientSide) {
				if (horizontalCollision) {
					coef = Math.sqrt(getHorizontalDistanceSqr(moveDelta));
					double r = l - coef;
					float dmg = (float) (r * 10.0D - 3.0D);
					if (dmg > 0.0F) {
						playSound(getFallDamageSound((int) dmg), 1.0F, 1.0F);
						hurt(DamageSource.FLY_INTO_WALL, dmg);
					}
				}
				if (onGround) {
					setSharedFlag(7, false);
				}
			}

			calculateEntityAnimation(LivingEntity.class.cast(this), this instanceof FlyingAnimal);
			info.cancel();
		}
	}

	private double be_getKnockback(Item tool) {
		if (tool == null) return 0.0D;
		Collection<AttributeModifier> modifiers = tool.getDefaultAttributeModifiers(EquipmentSlot.MAINHAND).get(Attributes.ATTACK_KNOCKBACK);
		if (modifiers.size() > 0) {
			return modifiers.iterator().next().getAmount();
		}
		return 0.0D;
	}
}