package net.minecraft.world.entity.monster.illager;

import java.util.Objects;
import net.minecraft.core.particles.ParticleTypes;
import net.minecraft.server.level.ServerLevel;
import net.minecraft.sounds.SoundEvent;
import net.minecraft.sounds.SoundEvents;
import net.minecraft.world.Difficulty;
import net.minecraft.world.DifficultyInstance;
import net.minecraft.world.damagesource.DamageSource;
import net.minecraft.world.effect.MobEffectInstance;
import net.minecraft.world.effect.MobEffects;
import net.minecraft.world.entity.EntitySpawnReason;
import net.minecraft.world.entity.EntityType;
import net.minecraft.world.entity.EquipmentSlot;
import net.minecraft.world.entity.LivingEntity;
import net.minecraft.world.entity.Mob;
import net.minecraft.world.entity.SpawnGroupData;
import net.minecraft.world.entity.ai.attributes.AttributeSupplier;
import net.minecraft.world.entity.ai.attributes.Attributes;
import net.minecraft.world.entity.ai.goal.AvoidEntityGoal;
import net.minecraft.world.entity.ai.goal.FloatGoal;
import net.minecraft.world.entity.ai.goal.LookAtPlayerGoal;
import net.minecraft.world.entity.ai.goal.RandomStrollGoal;
import net.minecraft.world.entity.ai.goal.RangedBowAttackGoal;
import net.minecraft.world.entity.ai.goal.target.HurtByTargetGoal;
import net.minecraft.world.entity.ai.goal.target.NearestAttackableTargetGoal;
import net.minecraft.world.entity.animal.golem.IronGolem;
import net.minecraft.world.entity.monster.Monster;
import net.minecraft.world.entity.monster.RangedAttackMob;
import net.minecraft.world.entity.monster.creaking.Creaking;
import net.minecraft.world.entity.npc.villager.AbstractVillager;
import net.minecraft.world.entity.player.Player;
import net.minecraft.world.entity.projectile.Projectile;
import net.minecraft.world.entity.projectile.ProjectileUtil;
import net.minecraft.world.entity.projectile.arrow.AbstractArrow;
import net.minecraft.world.entity.raid.Raider;
import net.minecraft.world.item.ItemStack;
import net.minecraft.world.item.Items;
import net.minecraft.world.level.Level;
import net.minecraft.world.level.ServerLevelAccessor;
import net.minecraft.world.phys.Vec3;
import org.jspecify.annotations.Nullable;

public class Illusioner extends SpellcasterIllager implements RangedAttackMob {
	private static final int NUM_ILLUSIONS = 4;
	private static final int ILLUSION_TRANSITION_TICKS = 3;
	public static final int ILLUSION_SPREAD = 3;
	private int clientSideIllusionTicks;
	private final Vec3[][] clientSideIllusionOffsets;

	public Illusioner(final EntityType<? extends Illusioner> type, final Level level) {
		super(type, level);
		this.xpReward = 5;
		this.clientSideIllusionOffsets = new Vec3[2][4];

		for (int i = 0; i < 4; i++) {
			this.clientSideIllusionOffsets[0][i] = Vec3.ZERO;
			this.clientSideIllusionOffsets[1][i] = Vec3.ZERO;
		}
	}

	@Override
	protected void registerGoals() {
		super.registerGoals();
		this.goalSelector.addGoal(0, new FloatGoal(this));
		this.goalSelector.addGoal(1, new SpellcasterIllager.SpellcasterCastingSpellGoal());
		this.goalSelector.addGoal(3, new AvoidEntityGoal(this, Creaking.class, 8.0F, 1.0, 1.2));
		this.goalSelector.addGoal(4, new Illusioner.IllusionerMirrorSpellGoal());
		this.goalSelector.addGoal(5, new Illusioner.IllusionerBlindnessSpellGoal());
		this.goalSelector.addGoal(6, new RangedBowAttackGoal<>(this, 0.5, 20, 15.0F));
		this.goalSelector.addGoal(8, new RandomStrollGoal(this, 0.6));
		this.goalSelector.addGoal(9, new LookAtPlayerGoal(this, Player.class, 3.0F, 1.0F));
		this.goalSelector.addGoal(10, new LookAtPlayerGoal(this, Mob.class, 8.0F));
		this.targetSelector.addGoal(1, new HurtByTargetGoal(this, Raider.class).setAlertOthers());
		this.targetSelector.addGoal(2, new NearestAttackableTargetGoal(this, Player.class, true).setUnseenMemoryTicks(300));
		this.targetSelector.addGoal(3, new NearestAttackableTargetGoal(this, AbstractVillager.class, false).setUnseenMemoryTicks(300));
		this.targetSelector.addGoal(3, new NearestAttackableTargetGoal(this, IronGolem.class, false).setUnseenMemoryTicks(300));
	}

	public static AttributeSupplier.Builder createAttributes() {
		return Monster.createMonsterAttributes().add(Attributes.MOVEMENT_SPEED, 0.5).add(Attributes.FOLLOW_RANGE, 18.0).add(Attributes.MAX_HEALTH, 32.0);
	}

	@Override
	public SpawnGroupData finalizeSpawn(
		final ServerLevelAccessor level, final DifficultyInstance difficulty, final EntitySpawnReason spawnReason, @Nullable final SpawnGroupData groupData
	) {
		this.setItemSlot(EquipmentSlot.MAINHAND, new ItemStack(Items.BOW));
		return super.finalizeSpawn(level, difficulty, spawnReason, groupData);
	}

	@Override
	public void aiStep() {
		super.aiStep();
		if (this.level().isClientSide() && this.isInvisible()) {
			this.clientSideIllusionTicks--;
			if (this.clientSideIllusionTicks < 0) {
				this.clientSideIllusionTicks = 0;
			}

			if (this.hurtTime == 1 || this.tickCount % 1200 == 0) {
				this.clientSideIllusionTicks = 3;
				float minSpread = -6.0F;
				int spreadSpan = 13;

				for (int i = 0; i < 4; i++) {
					this.clientSideIllusionOffsets[0][i] = this.clientSideIllusionOffsets[1][i];
					this.clientSideIllusionOffsets[1][i] = new Vec3(
						(-6.0F + this.random.nextInt(13)) * 0.5, Math.max(0, this.random.nextInt(6) - 4), (-6.0F + this.random.nextInt(13)) * 0.5
					);
				}

				for (int i = 0; i < 16; i++) {
					this.level().addParticle(ParticleTypes.CLOUD, this.getRandomX(0.5), this.getRandomY(), this.getZ(0.5), 0.0, 0.0, 0.0);
				}

				this.level().playLocalSound(this.getX(), this.getY(), this.getZ(), SoundEvents.ILLUSIONER_MIRROR_MOVE, this.getSoundSource(), 1.0F, 1.0F, false);
			} else if (this.hurtTime == this.hurtDuration - 1) {
				this.clientSideIllusionTicks = 3;

				for (int i = 0; i < 4; i++) {
					this.clientSideIllusionOffsets[0][i] = this.clientSideIllusionOffsets[1][i];
					this.clientSideIllusionOffsets[1][i] = new Vec3(0.0, 0.0, 0.0);
				}
			}
		}
	}

	@Override
	public SoundEvent getCelebrateSound() {
		return SoundEvents.ILLUSIONER_AMBIENT;
	}

	public Vec3[] getIllusionOffsets(final float a) {
		if (this.clientSideIllusionTicks <= 0) {
			return this.clientSideIllusionOffsets[1];
		} else {
			double scale = (this.clientSideIllusionTicks - a) / 3.0F;
			scale = Math.pow(scale, 0.25);
			Vec3[] offsets = new Vec3[4];

			for (int i = 0; i < 4; i++) {
				offsets[i] = this.clientSideIllusionOffsets[1][i].scale(1.0 - scale).add(this.clientSideIllusionOffsets[0][i].scale(scale));
			}

			return offsets;
		}
	}

	@Override
	protected SoundEvent getAmbientSound() {
		return SoundEvents.ILLUSIONER_AMBIENT;
	}

	@Override
	protected SoundEvent getDeathSound() {
		return SoundEvents.ILLUSIONER_DEATH;
	}

	@Override
	protected SoundEvent getHurtSound(final DamageSource source) {
		return SoundEvents.ILLUSIONER_HURT;
	}

	@Override
	protected SoundEvent getCastingSoundEvent() {
		return SoundEvents.ILLUSIONER_CAST_SPELL;
	}

	@Override
	public void applyRaidBuffs(final ServerLevel level, final int wave, final boolean isCaptain) {
	}

	@Override
	public void performRangedAttack(final LivingEntity target, final float power) {
		ItemStack bowItem = this.getItemInHand(ProjectileUtil.getWeaponHoldingHand(this, Items.BOW));
		ItemStack projectile = this.getProjectile(bowItem);
		AbstractArrow arrow = ProjectileUtil.getMobArrow(this, projectile, power, bowItem);
		double xd = target.getX() - this.getX();
		double yd = target.getY(0.3333333333333333) - arrow.getY();
		double zd = target.getZ() - this.getZ();
		double distanceToTarget = Math.sqrt(xd * xd + zd * zd);
		if (this.level() instanceof ServerLevel serverLevel) {
			Projectile.spawnProjectileUsingShoot(
				arrow, serverLevel, projectile, xd, yd + distanceToTarget * 0.2F, zd, 1.6F, 14 - serverLevel.getDifficulty().getId() * 4
			);
		}

		this.playSound(SoundEvents.SKELETON_SHOOT, 1.0F, 1.0F / (this.getRandom().nextFloat() * 0.4F + 0.8F));
	}

	@Override
	public AbstractIllager.IllagerArmPose getArmPose() {
		if (this.isCastingSpell()) {
			return AbstractIllager.IllagerArmPose.SPELLCASTING;
		} else {
			return this.isAggressive() ? AbstractIllager.IllagerArmPose.BOW_AND_ARROW : AbstractIllager.IllagerArmPose.CROSSED;
		}
	}

	private class IllusionerBlindnessSpellGoal extends SpellcasterIllager.SpellcasterUseSpellGoal {
		private int lastTargetId;

		private IllusionerBlindnessSpellGoal() {
			Objects.requireNonNull(Illusioner.this);
			super();
		}

		@Override
		public boolean canUse() {
			if (!super.canUse()) {
				return false;
			} else if (Illusioner.this.getTarget() == null) {
				return false;
			} else {
				return Illusioner.this.getTarget().getId() == this.lastTargetId
					? false
					: getServerLevel(Illusioner.this).getCurrentDifficultyAt(Illusioner.this.blockPosition()).isHarderThan(Difficulty.NORMAL.ordinal());
			}
		}

		@Override
		public void start() {
			super.start();
			LivingEntity target = Illusioner.this.getTarget();
			if (target != null) {
				this.lastTargetId = target.getId();
			}
		}

		@Override
		protected int getCastingTime() {
			return 20;
		}

		@Override
		protected int getCastingInterval() {
			return 180;
		}

		@Override
		protected void performSpellCasting() {
			Illusioner.this.getTarget().addEffect(new MobEffectInstance(MobEffects.BLINDNESS, 400), Illusioner.this);
		}

		@Override
		protected SoundEvent getSpellPrepareSound() {
			return SoundEvents.ILLUSIONER_PREPARE_BLINDNESS;
		}

		@Override
		protected SpellcasterIllager.IllagerSpell getSpell() {
			return SpellcasterIllager.IllagerSpell.BLINDNESS;
		}
	}

	private class IllusionerMirrorSpellGoal extends SpellcasterIllager.SpellcasterUseSpellGoal {
		private IllusionerMirrorSpellGoal() {
			Objects.requireNonNull(Illusioner.this);
			super();
		}

		@Override
		public boolean canUse() {
			return !super.canUse() ? false : !Illusioner.this.hasEffect(MobEffects.INVISIBILITY);
		}

		@Override
		protected int getCastingTime() {
			return 20;
		}

		@Override
		protected int getCastingInterval() {
			return 340;
		}

		@Override
		protected void performSpellCasting() {
			Illusioner.this.addEffect(new MobEffectInstance(MobEffects.INVISIBILITY, 1200));
		}

		@Nullable
		@Override
		protected SoundEvent getSpellPrepareSound() {
			return SoundEvents.ILLUSIONER_PREPARE_MIRROR;
		}

		@Override
		protected SpellcasterIllager.IllagerSpell getSpell() {
			return SpellcasterIllager.IllagerSpell.DISAPPEAR;
		}
	}
}