package net.minecraft.world.entity.ai.util;

import net.minecraft.core.BlockPos;
import net.minecraft.world.entity.PathfinderMob;
import net.minecraft.world.phys.Vec3;
import org.jspecify.annotations.Nullable;

public class DefaultRandomPos {
	@Nullable
	public static Vec3 getPos(final PathfinderMob mob, final int horizontalDist, final int verticalDist) {
		boolean restrict = GoalUtils.mobRestricted(mob, horizontalDist);
		return RandomPos.generateRandomPos(mob, () -> {
			BlockPos direction = RandomPos.generateRandomDirection(mob.getRandom(), horizontalDist, verticalDist);
			return generateRandomPosTowardDirection(mob, horizontalDist, restrict, direction);
		});
	}

	@Nullable
	public static Vec3 getPosTowards(
		final PathfinderMob mob, final int horizontalDist, final int verticalDist, final Vec3 towardsPos, final double maxXzRadiansFromDir
	) {
		Vec3 dir = towardsPos.subtract(mob.getX(), mob.getY(), mob.getZ());
		boolean restrict = GoalUtils.mobRestricted(mob, horizontalDist);
		return RandomPos.generateRandomPos(
			mob,
			() -> {
				BlockPos direction = RandomPos.generateRandomDirectionWithinRadians(
					mob.getRandom(), 0.0, horizontalDist, verticalDist, 0, dir.x, dir.z, maxXzRadiansFromDir
				);
				return direction == null ? null : generateRandomPosTowardDirection(mob, horizontalDist, restrict, direction);
			}
		);
	}

	@Nullable
	public static Vec3 getPosAway(final PathfinderMob mob, final int horizontalDist, final int verticalDist, final Vec3 avoidPos) {
		Vec3 dirAway = mob.position().subtract(avoidPos);
		boolean restrict = GoalUtils.mobRestricted(mob, horizontalDist);
		return RandomPos.generateRandomPos(
			mob,
			() -> {
				BlockPos direction = RandomPos.generateRandomDirectionWithinRadians(
					mob.getRandom(), 0.0, horizontalDist, verticalDist, 0, dirAway.x, dirAway.z, (float) (Math.PI / 2)
				);
				return direction == null ? null : generateRandomPosTowardDirection(mob, horizontalDist, restrict, direction);
			}
		);
	}

	@Nullable
	private static BlockPos generateRandomPosTowardDirection(final PathfinderMob mob, final int horizontalDist, final boolean restrict, final BlockPos direction) {
		BlockPos pos = RandomPos.generateRandomPosTowardDirection(mob, horizontalDist, mob.getRandom(), direction);
		return !GoalUtils.isOutsideLimits(pos, mob)
				&& !GoalUtils.isRestricted(restrict, mob, pos)
				&& !GoalUtils.isNotStable(mob.getNavigation(), pos)
				&& !GoalUtils.hasMalus(mob, pos)
			? pos
			: null;
	}
}