package net.minecraft.world.entity.boss.enderdragon.phases;

import net.minecraft.core.particles.ParticleTypes;
import net.minecraft.server.level.ServerLevel;
import net.minecraft.util.RandomSource;
import net.minecraft.world.entity.boss.enderdragon.EnderDragon;
import net.minecraft.world.level.levelgen.Heightmap.Types;
import net.minecraft.world.level.levelgen.feature.EndPodiumFeature;
import net.minecraft.world.phys.Vec3;
import org.jetbrains.annotations.Nullable;

public class DragonLandingPhase extends AbstractDragonPhaseInstance {
	@Nullable
	private Vec3 targetLocation;

	public DragonLandingPhase(EnderDragon enderDragon) {
		super(enderDragon);
	}

	@Override
	public void doClientTick() {
		Vec3 vec3 = this.dragon.getHeadLookVector(1.0F).normalize();
		vec3.yRot((float) (-Math.PI / 4));
		double d = this.dragon.head.getX();
		double e = this.dragon.head.getY(0.5);
		double f = this.dragon.head.getZ();

		for (int i = 0; i < 8; i++) {
			RandomSource randomSource = this.dragon.getRandom();
			double g = d + randomSource.nextGaussian() / 2.0;
			double h = e + randomSource.nextGaussian() / 2.0;
			double j = f + randomSource.nextGaussian() / 2.0;
			Vec3 vec32 = this.dragon.getDeltaMovement();
			this.dragon.level().addParticle(ParticleTypes.DRAGON_BREATH, g, h, j, -vec3.x * 0.08F + vec32.x, -vec3.y * 0.3F + vec32.y, -vec3.z * 0.08F + vec32.z);
			vec3.yRot((float) (Math.PI / 16));
		}
	}

	@Override
	public void doServerTick(ServerLevel level) {
		if (this.targetLocation == null) {
			this.targetLocation = Vec3.atBottomCenterOf(
				level.getHeightmapPos(Types.MOTION_BLOCKING_NO_LEAVES, EndPodiumFeature.getLocation(this.dragon.getFightOrigin()))
			);
		}

		if (this.targetLocation.distanceToSqr(this.dragon.getX(), this.dragon.getY(), this.dragon.getZ()) < 1.0) {
			this.dragon.getPhaseManager().getPhase(EnderDragonPhase.SITTING_FLAMING).resetFlameCount();
			this.dragon.getPhaseManager().setPhase(EnderDragonPhase.SITTING_SCANNING);
		}
	}

	@Override
	public float getFlySpeed() {
		return 1.5F;
	}

	@Override
	public float getTurnSpeed() {
		float f = (float)this.dragon.getDeltaMovement().horizontalDistance() + 1.0F;
		float g = Math.min(f, 40.0F);
		return g / f;
	}

	@Override
	public void begin() {
		this.targetLocation = null;
	}

	@Nullable
	@Override
	public Vec3 getFlyTargetLocation() {
		return this.targetLocation;
	}

	@Override
	public EnderDragonPhase<DragonLandingPhase> getPhase() {
		return EnderDragonPhase.LANDING;
	}
}