package net.minecraft.world.entity.monster;

import java.util.Comparator;
import java.util.EnumSet;
import java.util.List;
import net.minecraft.core.BlockPos;
import net.minecraft.core.particles.ParticleTypes;
import net.minecraft.nbt.CompoundTag;
import net.minecraft.network.syncher.EntityDataAccessor;
import net.minecraft.network.syncher.EntityDataSerializers;
import net.minecraft.network.syncher.SynchedEntityData;
import net.minecraft.network.syncher.SynchedEntityData.Builder;
import net.minecraft.server.level.ServerLevel;
import net.minecraft.sounds.SoundEvent;
import net.minecraft.sounds.SoundEvents;
import net.minecraft.sounds.SoundSource;
import net.minecraft.util.Mth;
import net.minecraft.world.DifficultyInstance;
import net.minecraft.world.damagesource.DamageSource;
import net.minecraft.world.entity.Entity;
import net.minecraft.world.entity.EntityDimensions;
import net.minecraft.world.entity.EntitySelector;
import net.minecraft.world.entity.EntitySpawnReason;
import net.minecraft.world.entity.EntityType;
import net.minecraft.world.entity.FlyingMob;
import net.minecraft.world.entity.LivingEntity;
import net.minecraft.world.entity.Mob;
import net.minecraft.world.entity.Pose;
import net.minecraft.world.entity.SpawnGroupData;
import net.minecraft.world.entity.ai.attributes.Attributes;
import net.minecraft.world.entity.ai.control.BodyRotationControl;
import net.minecraft.world.entity.ai.control.LookControl;
import net.minecraft.world.entity.ai.control.MoveControl;
import net.minecraft.world.entity.ai.goal.Goal;
import net.minecraft.world.entity.ai.goal.Goal.Flag;
import net.minecraft.world.entity.ai.targeting.TargetingConditions;
import net.minecraft.world.entity.animal.Cat;
import net.minecraft.world.entity.player.Player;
import net.minecraft.world.level.Level;
import net.minecraft.world.level.ServerLevelAccessor;
import net.minecraft.world.level.levelgen.Heightmap.Types;
import net.minecraft.world.phys.Vec3;
import org.jetbrains.annotations.Nullable;

public class Phantom extends FlyingMob implements Enemy {
	public static final float FLAP_DEGREES_PER_TICK = 7.448451F;
	public static final int TICKS_PER_FLAP = Mth.ceil(24.166098F);
	private static final EntityDataAccessor<Integer> ID_SIZE = SynchedEntityData.defineId(Phantom.class, EntityDataSerializers.INT);
	Vec3 moveTargetPoint = Vec3.ZERO;
	BlockPos anchorPoint = BlockPos.ZERO;
	Phantom.AttackPhase attackPhase = Phantom.AttackPhase.CIRCLE;

	public Phantom(EntityType<? extends Phantom> entityType, Level level) {
		super(entityType, level);
		this.xpReward = 5;
		this.moveControl = new Phantom.PhantomMoveControl(this);
		this.lookControl = new Phantom.PhantomLookControl(this);
	}

	@Override
	public boolean isFlapping() {
		return (this.getUniqueFlapTickOffset() + this.tickCount) % TICKS_PER_FLAP == 0;
	}

	@Override
	protected BodyRotationControl createBodyControl() {
		return new Phantom.PhantomBodyRotationControl(this);
	}

	@Override
	protected void registerGoals() {
		this.goalSelector.addGoal(1, new Phantom.PhantomAttackStrategyGoal());
		this.goalSelector.addGoal(2, new Phantom.PhantomSweepAttackGoal());
		this.goalSelector.addGoal(3, new Phantom.PhantomCircleAroundAnchorGoal());
		this.targetSelector.addGoal(1, new Phantom.PhantomAttackPlayerTargetGoal());
	}

	@Override
	protected void defineSynchedData(Builder builder) {
		super.defineSynchedData(builder);
		builder.define(ID_SIZE, 0);
	}

	public void setPhantomSize(int phantomSize) {
		this.entityData.set(ID_SIZE, Mth.clamp(phantomSize, 0, 64));
	}

	private void updatePhantomSizeInfo() {
		this.refreshDimensions();
		this.getAttribute(Attributes.ATTACK_DAMAGE).setBaseValue(6 + this.getPhantomSize());
	}

	public int getPhantomSize() {
		return this.entityData.get(ID_SIZE);
	}

	@Override
	public void onSyncedDataUpdated(EntityDataAccessor<?> dataAccessor) {
		if (ID_SIZE.equals(dataAccessor)) {
			this.updatePhantomSizeInfo();
		}

		super.onSyncedDataUpdated(dataAccessor);
	}

	public int getUniqueFlapTickOffset() {
		return this.getId() * 3;
	}

	@Override
	protected boolean shouldDespawnInPeaceful() {
		return true;
	}

	@Override
	public void tick() {
		super.tick();
		if (this.level().isClientSide) {
			float f = Mth.cos((this.getUniqueFlapTickOffset() + this.tickCount) * 7.448451F * (float) (Math.PI / 180.0) + (float) Math.PI);
			float g = Mth.cos((this.getUniqueFlapTickOffset() + this.tickCount + 1) * 7.448451F * (float) (Math.PI / 180.0) + (float) Math.PI);
			if (f > 0.0F && g <= 0.0F) {
				this.level()
					.playLocalSound(
						this.getX(),
						this.getY(),
						this.getZ(),
						SoundEvents.PHANTOM_FLAP,
						this.getSoundSource(),
						0.95F + this.random.nextFloat() * 0.05F,
						0.95F + this.random.nextFloat() * 0.05F,
						false
					);
			}

			float h = this.getBbWidth() * 1.48F;
			float i = Mth.cos(this.getYRot() * (float) (Math.PI / 180.0)) * h;
			float j = Mth.sin(this.getYRot() * (float) (Math.PI / 180.0)) * h;
			float k = (0.3F + f * 0.45F) * this.getBbHeight() * 2.5F;
			this.level().addParticle(ParticleTypes.MYCELIUM, this.getX() + i, this.getY() + k, this.getZ() + j, 0.0, 0.0, 0.0);
			this.level().addParticle(ParticleTypes.MYCELIUM, this.getX() - i, this.getY() + k, this.getZ() - j, 0.0, 0.0, 0.0);
		}
	}

	@Override
	public void aiStep() {
		if (this.isAlive() && this.isSunBurnTick()) {
			this.igniteForSeconds(8.0F);
		}

		super.aiStep();
	}

	@Override
	public SpawnGroupData finalizeSpawn(
		ServerLevelAccessor level, DifficultyInstance difficulty, EntitySpawnReason spawnReason, @Nullable SpawnGroupData spawnGroupData
	) {
		this.anchorPoint = this.blockPosition().above(5);
		this.setPhantomSize(0);
		return super.finalizeSpawn(level, difficulty, spawnReason, spawnGroupData);
	}

	@Override
	public void readAdditionalSaveData(CompoundTag tag) {
		super.readAdditionalSaveData(tag);
		if (tag.contains("AX")) {
			this.anchorPoint = new BlockPos(tag.getInt("AX"), tag.getInt("AY"), tag.getInt("AZ"));
		}

		this.setPhantomSize(tag.getInt("Size"));
	}

	@Override
	public void addAdditionalSaveData(CompoundTag tag) {
		super.addAdditionalSaveData(tag);
		tag.putInt("AX", this.anchorPoint.getX());
		tag.putInt("AY", this.anchorPoint.getY());
		tag.putInt("AZ", this.anchorPoint.getZ());
		tag.putInt("Size", this.getPhantomSize());
	}

	@Override
	public boolean shouldRenderAtSqrDistance(double distance) {
		return true;
	}

	@Override
	public SoundSource getSoundSource() {
		return SoundSource.HOSTILE;
	}

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

	@Override
	protected SoundEvent getHurtSound(DamageSource damageSource) {
		return SoundEvents.PHANTOM_HURT;
	}

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

	@Override
	protected float getSoundVolume() {
		return 1.0F;
	}

	@Override
	public boolean canAttackType(EntityType<?> entityType) {
		return true;
	}

	@Override
	public EntityDimensions getDefaultDimensions(Pose pose) {
		int i = this.getPhantomSize();
		EntityDimensions entityDimensions = super.getDefaultDimensions(pose);
		return entityDimensions.scale(1.0F + 0.15F * i);
	}

	boolean canAttack(ServerLevel level, LivingEntity entity, TargetingConditions targetingConditions) {
		return targetingConditions.test(level, this, entity);
	}

	static enum AttackPhase {
		CIRCLE,
		SWOOP;
	}

	class PhantomAttackPlayerTargetGoal extends Goal {
		private final TargetingConditions attackTargeting = TargetingConditions.forCombat().range(64.0);
		private int nextScanTick = reducedTickDelay(20);

		@Override
		public boolean canUse() {
			if (this.nextScanTick > 0) {
				this.nextScanTick--;
				return false;
			} else {
				this.nextScanTick = reducedTickDelay(60);
				ServerLevel serverLevel = getServerLevel(Phantom.this.level());
				List<Player> list = serverLevel.getNearbyPlayers(this.attackTargeting, Phantom.this, Phantom.this.getBoundingBox().inflate(16.0, 64.0, 16.0));
				if (!list.isEmpty()) {
					list.sort(Comparator.comparing(Entity::getY).reversed());

					for (Player player : list) {
						if (Phantom.this.canAttack(serverLevel, player, TargetingConditions.DEFAULT)) {
							Phantom.this.setTarget(player);
							return true;
						}
					}
				}

				return false;
			}
		}

		@Override
		public boolean canContinueToUse() {
			LivingEntity livingEntity = Phantom.this.getTarget();
			return livingEntity != null ? Phantom.this.canAttack(getServerLevel(Phantom.this.level()), livingEntity, TargetingConditions.DEFAULT) : false;
		}
	}

	class PhantomAttackStrategyGoal extends Goal {
		private int nextSweepTick;

		@Override
		public boolean canUse() {
			LivingEntity livingEntity = Phantom.this.getTarget();
			return livingEntity != null ? Phantom.this.canAttack(getServerLevel(Phantom.this.level()), livingEntity, TargetingConditions.DEFAULT) : false;
		}

		@Override
		public void start() {
			this.nextSweepTick = this.adjustedTickDelay(10);
			Phantom.this.attackPhase = Phantom.AttackPhase.CIRCLE;
			this.setAnchorAboveTarget();
		}

		@Override
		public void stop() {
			Phantom.this.anchorPoint = Phantom.this.level().getHeightmapPos(Types.MOTION_BLOCKING, Phantom.this.anchorPoint).above(10 + Phantom.this.random.nextInt(20));
		}

		@Override
		public void tick() {
			if (Phantom.this.attackPhase == Phantom.AttackPhase.CIRCLE) {
				this.nextSweepTick--;
				if (this.nextSweepTick <= 0) {
					Phantom.this.attackPhase = Phantom.AttackPhase.SWOOP;
					this.setAnchorAboveTarget();
					this.nextSweepTick = this.adjustedTickDelay((8 + Phantom.this.random.nextInt(4)) * 20);
					Phantom.this.playSound(SoundEvents.PHANTOM_SWOOP, 10.0F, 0.95F + Phantom.this.random.nextFloat() * 0.1F);
				}
			}
		}

		private void setAnchorAboveTarget() {
			Phantom.this.anchorPoint = Phantom.this.getTarget().blockPosition().above(20 + Phantom.this.random.nextInt(20));
			if (Phantom.this.anchorPoint.getY() < Phantom.this.level().getSeaLevel()) {
				Phantom.this.anchorPoint = new BlockPos(Phantom.this.anchorPoint.getX(), Phantom.this.level().getSeaLevel() + 1, Phantom.this.anchorPoint.getZ());
			}
		}
	}

	class PhantomBodyRotationControl extends BodyRotationControl {
		public PhantomBodyRotationControl(final Mob mob) {
			super(mob);
		}

		@Override
		public void clientTick() {
			Phantom.this.yHeadRot = Phantom.this.yBodyRot;
			Phantom.this.yBodyRot = Phantom.this.getYRot();
		}
	}

	class PhantomCircleAroundAnchorGoal extends Phantom.PhantomMoveTargetGoal {
		private float angle;
		private float distance;
		private float height;
		private float clockwise;

		@Override
		public boolean canUse() {
			return Phantom.this.getTarget() == null || Phantom.this.attackPhase == Phantom.AttackPhase.CIRCLE;
		}

		@Override
		public void start() {
			this.distance = 5.0F + Phantom.this.random.nextFloat() * 10.0F;
			this.height = -4.0F + Phantom.this.random.nextFloat() * 9.0F;
			this.clockwise = Phantom.this.random.nextBoolean() ? 1.0F : -1.0F;
			this.selectNext();
		}

		@Override
		public void tick() {
			if (Phantom.this.random.nextInt(this.adjustedTickDelay(350)) == 0) {
				this.height = -4.0F + Phantom.this.random.nextFloat() * 9.0F;
			}

			if (Phantom.this.random.nextInt(this.adjustedTickDelay(250)) == 0) {
				this.distance++;
				if (this.distance > 15.0F) {
					this.distance = 5.0F;
					this.clockwise = -this.clockwise;
				}
			}

			if (Phantom.this.random.nextInt(this.adjustedTickDelay(450)) == 0) {
				this.angle = Phantom.this.random.nextFloat() * 2.0F * (float) Math.PI;
				this.selectNext();
			}

			if (this.touchingTarget()) {
				this.selectNext();
			}

			if (Phantom.this.moveTargetPoint.y < Phantom.this.getY() && !Phantom.this.level().isEmptyBlock(Phantom.this.blockPosition().below(1))) {
				this.height = Math.max(1.0F, this.height);
				this.selectNext();
			}

			if (Phantom.this.moveTargetPoint.y > Phantom.this.getY() && !Phantom.this.level().isEmptyBlock(Phantom.this.blockPosition().above(1))) {
				this.height = Math.min(-1.0F, this.height);
				this.selectNext();
			}
		}

		private void selectNext() {
			if (BlockPos.ZERO.equals(Phantom.this.anchorPoint)) {
				Phantom.this.anchorPoint = Phantom.this.blockPosition();
			}

			this.angle = this.angle + this.clockwise * 15.0F * (float) (Math.PI / 180.0);
			Phantom.this.moveTargetPoint = Vec3.atLowerCornerOf(Phantom.this.anchorPoint)
				.add(this.distance * Mth.cos(this.angle), -4.0F + this.height, this.distance * Mth.sin(this.angle));
		}
	}

	static class PhantomLookControl extends LookControl {
		public PhantomLookControl(Mob mob) {
			super(mob);
		}

		@Override
		public void tick() {
		}
	}

	class PhantomMoveControl extends MoveControl {
		private float speed = 0.1F;

		public PhantomMoveControl(final Mob mob) {
			super(mob);
		}

		@Override
		public void tick() {
			if (Phantom.this.horizontalCollision) {
				Phantom.this.setYRot(Phantom.this.getYRot() + 180.0F);
				this.speed = 0.1F;
			}

			double d = Phantom.this.moveTargetPoint.x - Phantom.this.getX();
			double e = Phantom.this.moveTargetPoint.y - Phantom.this.getY();
			double f = Phantom.this.moveTargetPoint.z - Phantom.this.getZ();
			double g = Math.sqrt(d * d + f * f);
			if (Math.abs(g) > 1.0E-5F) {
				double h = 1.0 - Math.abs(e * 0.7F) / g;
				d *= h;
				f *= h;
				g = Math.sqrt(d * d + f * f);
				double i = Math.sqrt(d * d + f * f + e * e);
				float j = Phantom.this.getYRot();
				float k = (float)Mth.atan2(f, d);
				float l = Mth.wrapDegrees(Phantom.this.getYRot() + 90.0F);
				float m = Mth.wrapDegrees(k * (180.0F / (float)Math.PI));
				Phantom.this.setYRot(Mth.approachDegrees(l, m, 4.0F) - 90.0F);
				Phantom.this.yBodyRot = Phantom.this.getYRot();
				if (Mth.degreesDifferenceAbs(j, Phantom.this.getYRot()) < 3.0F) {
					this.speed = Mth.approach(this.speed, 1.8F, 0.005F * (1.8F / this.speed));
				} else {
					this.speed = Mth.approach(this.speed, 0.2F, 0.025F);
				}

				float n = (float)(-(Mth.atan2(-e, g) * 180.0F / (float)Math.PI));
				Phantom.this.setXRot(n);
				float o = Phantom.this.getYRot() + 90.0F;
				double p = this.speed * Mth.cos(o * (float) (Math.PI / 180.0)) * Math.abs(d / i);
				double q = this.speed * Mth.sin(o * (float) (Math.PI / 180.0)) * Math.abs(f / i);
				double r = this.speed * Mth.sin(n * (float) (Math.PI / 180.0)) * Math.abs(e / i);
				Vec3 vec3 = Phantom.this.getDeltaMovement();
				Phantom.this.setDeltaMovement(vec3.add(new Vec3(p, r, q).subtract(vec3).scale(0.2)));
			}
		}
	}

	abstract class PhantomMoveTargetGoal extends Goal {
		public PhantomMoveTargetGoal() {
			this.setFlags(EnumSet.of(Flag.MOVE));
		}

		protected boolean touchingTarget() {
			return Phantom.this.moveTargetPoint.distanceToSqr(Phantom.this.getX(), Phantom.this.getY(), Phantom.this.getZ()) < 4.0;
		}
	}

	class PhantomSweepAttackGoal extends Phantom.PhantomMoveTargetGoal {
		private static final int CAT_SEARCH_TICK_DELAY = 20;
		private boolean isScaredOfCat;
		private int catSearchTick;

		@Override
		public boolean canUse() {
			return Phantom.this.getTarget() != null && Phantom.this.attackPhase == Phantom.AttackPhase.SWOOP;
		}

		@Override
		public boolean canContinueToUse() {
			LivingEntity livingEntity = Phantom.this.getTarget();
			if (livingEntity == null) {
				return false;
			} else if (!livingEntity.isAlive()) {
				return false;
			} else if (livingEntity instanceof Player player && (livingEntity.isSpectator() || player.isCreative())) {
				return false;
			} else if (!this.canUse()) {
				return false;
			} else {
				if (Phantom.this.tickCount > this.catSearchTick) {
					this.catSearchTick = Phantom.this.tickCount + 20;
					List<Cat> list = Phantom.this.level().getEntitiesOfClass(Cat.class, Phantom.this.getBoundingBox().inflate(16.0), EntitySelector.ENTITY_STILL_ALIVE);

					for (Cat cat : list) {
						cat.hiss();
					}

					this.isScaredOfCat = !list.isEmpty();
				}

				return !this.isScaredOfCat;
			}
		}

		@Override
		public void start() {
		}

		@Override
		public void stop() {
			Phantom.this.setTarget(null);
			Phantom.this.attackPhase = Phantom.AttackPhase.CIRCLE;
		}

		@Override
		public void tick() {
			LivingEntity livingEntity = Phantom.this.getTarget();
			if (livingEntity != null) {
				Phantom.this.moveTargetPoint = new Vec3(livingEntity.getX(), livingEntity.getY(0.5), livingEntity.getZ());
				if (Phantom.this.getBoundingBox().inflate(0.2F).intersects(livingEntity.getBoundingBox())) {
					Phantom.this.doHurtTarget(getServerLevel(Phantom.this.level()), livingEntity);
					Phantom.this.attackPhase = Phantom.AttackPhase.CIRCLE;
					if (!Phantom.this.isSilent()) {
						Phantom.this.level().levelEvent(1039, Phantom.this.blockPosition(), 0);
					}
				} else if (Phantom.this.horizontalCollision || Phantom.this.hurtTime > 0) {
					Phantom.this.attackPhase = Phantom.AttackPhase.CIRCLE;
				}
			}
		}
	}
}