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

import com.google.common.collect.ImmutableMap;
import java.util.Optional;
import net.minecraft.core.BlockPos;
import net.minecraft.core.GlobalPos;
import net.minecraft.core.Holder;
import net.minecraft.server.level.ServerLevel;
import net.minecraft.world.entity.AgeableMob;
import net.minecraft.world.entity.EntityTypes;
import net.minecraft.world.entity.ai.Brain;
import net.minecraft.world.entity.ai.memory.MemoryModuleType;
import net.minecraft.world.entity.ai.memory.MemoryStatus;
import net.minecraft.world.entity.ai.village.poi.PoiType;
import net.minecraft.world.entity.ai.village.poi.PoiTypes;
import net.minecraft.world.entity.npc.villager.Villager;
import net.minecraft.world.level.pathfinder.Path;

public class VillagerMakeLove extends Behavior<Villager> {
	private long birthTimestamp;

	public VillagerMakeLove() {
		super(
			ImmutableMap.of(MemoryModuleType.BREED_TARGET, MemoryStatus.VALUE_PRESENT, MemoryModuleType.NEAREST_VISIBLE_LIVING_ENTITIES, MemoryStatus.VALUE_PRESENT),
			350,
			350
		);
	}

	protected boolean checkExtraStartConditions(final ServerLevel level, final Villager body) {
		return this.isBreedingPossible(body);
	}

	protected boolean canStillUse(final ServerLevel level, final Villager body, final long timestamp) {
		return timestamp <= this.birthTimestamp && this.isBreedingPossible(body);
	}

	protected void start(final ServerLevel level, final Villager body, final long timestamp) {
		AgeableMob breedTarget = (AgeableMob)body.getBrain().getMemory(MemoryModuleType.BREED_TARGET).get();
		BehaviorUtils.lockGazeAndWalkToEachOther(body, breedTarget, 0.5F, 2);
		level.broadcastEntityEvent(breedTarget, (byte)18);
		level.broadcastEntityEvent(body, (byte)18);
		int duration = 275 + body.getRandom().nextInt(50);
		this.birthTimestamp = timestamp + duration;
	}

	protected void tick(final ServerLevel level, final Villager body, final long timestamp) {
		Villager target = (Villager)body.getBrain().getMemory(MemoryModuleType.BREED_TARGET).get();
		if (!(body.distanceToSqr(target) > 5.0)) {
			BehaviorUtils.lockGazeAndWalkToEachOther(body, target, 0.5F, 2);
			if (timestamp >= this.birthTimestamp) {
				body.eatAndDigestFood();
				target.eatAndDigestFood();
				this.tryToGiveBirth(level, body, target);
			} else if (body.getRandom().nextInt(35) == 0) {
				level.broadcastEntityEvent(target, (byte)12);
				level.broadcastEntityEvent(body, (byte)12);
			}
		}
	}

	private void tryToGiveBirth(final ServerLevel level, final Villager body, final Villager target) {
		Optional<BlockPos> childsBed = this.takeVacantBed(level, body);
		if (childsBed.isEmpty()) {
			level.broadcastEntityEvent(target, (byte)13);
			level.broadcastEntityEvent(body, (byte)13);
		} else {
			Optional<Villager> child = this.breed(level, body, target);
			if (child.isPresent()) {
				this.giveBedToChild(level, (Villager)child.get(), (BlockPos)childsBed.get());
			} else {
				level.getPoiManager().release((BlockPos)childsBed.get());
				level.debugSynchronizers().updatePoi((BlockPos)childsBed.get());
			}
		}
	}

	protected void stop(final ServerLevel level, final Villager body, final long timestamp) {
		body.getBrain().eraseMemory(MemoryModuleType.BREED_TARGET);
	}

	private boolean isBreedingPossible(final Villager myBody) {
		Brain<Villager> brain = myBody.getBrain();
		Optional<AgeableMob> breedTarget = brain.getMemory(MemoryModuleType.BREED_TARGET).filter(entity -> entity.is(EntityTypes.VILLAGER));
		return breedTarget.isEmpty()
			? false
			: BehaviorUtils.targetIsValid(brain, MemoryModuleType.BREED_TARGET, EntityTypes.VILLAGER) && myBody.canBreed() && ((AgeableMob)breedTarget.get()).canBreed();
	}

	private Optional<BlockPos> takeVacantBed(final ServerLevel level, final Villager body) {
		return level.getPoiManager().take(p -> p.is(PoiTypes.HOME), (poiType, poiPos) -> this.canReach(body, poiPos, poiType), body.blockPosition(), 48);
	}

	private boolean canReach(final Villager body, final BlockPos poiPos, final Holder<PoiType> poiType) {
		Path path = body.getNavigation().createPath(poiPos, poiType.value().validRange());
		return path != null && path.canReach();
	}

	private Optional<Villager> breed(final ServerLevel level, final Villager source, final Villager target) {
		Villager child = source.getBreedOffspring(level, target);
		if (child == null) {
			return Optional.empty();
		} else {
			source.setAge(6000);
			target.setAge(6000);
			child.setAge(-24000);
			child.snapTo(source.getX(), source.getY(), source.getZ(), 0.0F, 0.0F);
			level.addFreshEntityWithPassengers(child);
			level.broadcastEntityEvent(child, (byte)12);
			return Optional.of(child);
		}
	}

	private void giveBedToChild(final ServerLevel level, final Villager child, final BlockPos bedPos) {
		GlobalPos globalBedPos = GlobalPos.of(level.dimension(), bedPos);
		child.getBrain().setMemory(MemoryModuleType.HOME, globalBedPos);
	}
}