package net.minecraft.entity.ai.brain.task;

import java.util.List;
import java.util.Optional;
import net.minecraft.entity.LivingEntity;
import net.minecraft.entity.ai.brain.MemoryModuleType;
import net.minecraft.entity.ai.pathing.Path;
import net.minecraft.entity.mob.PathAwareEntity;
import net.minecraft.entity.passive.VillagerEntity;
import net.minecraft.registry.entry.RegistryEntry;
import net.minecraft.server.network.DebugInfoSender;
import net.minecraft.util.math.BlockPos;
import net.minecraft.util.math.GlobalPos;
import net.minecraft.village.VillagerProfession;
import net.minecraft.world.poi.PointOfInterestType;

public class TakeJobSiteTask {
	public static Task<VillagerEntity> create(float speed) {
		return TaskTriggerer.task(
			context -> context.group(
					context.queryMemoryValue(MemoryModuleType.POTENTIAL_JOB_SITE),
					context.queryMemoryAbsent(MemoryModuleType.JOB_SITE),
					context.queryMemoryValue(MemoryModuleType.MOBS),
					context.queryMemoryOptional(MemoryModuleType.WALK_TARGET),
					context.queryMemoryOptional(MemoryModuleType.LOOK_TARGET)
				)
				.apply(
					context,
					(potentialJobSite, jobSite, mobs, walkTarget, lookTarget) -> (world, entity, time) -> {
						if (entity.isBaby()) {
							return false;
						} else if (entity.getVillagerData().getProfession() != VillagerProfession.NONE) {
							return false;
						} else {
							BlockPos blockPos = context.<GlobalPos>getValue(potentialJobSite).getPos();
							Optional<RegistryEntry<PointOfInterestType>> optional = world.getPointOfInterestStorage().getType(blockPos);
							if (optional.isEmpty()) {
								return true;
							} else {
								context.<List>getValue(mobs)
									.stream()
									.filter(mob -> mob instanceof VillagerEntity && mob != entity)
									.map(villager -> (VillagerEntity)villager)
									.filter(LivingEntity::isAlive)
									.filter(villager -> canUseJobSite((RegistryEntry<PointOfInterestType>)optional.get(), villager, blockPos))
									.findFirst()
									.ifPresent(villager -> {
										walkTarget.forget();
										lookTarget.forget();
										potentialJobSite.forget();
										if (villager.getBrain().getOptionalRegisteredMemory(MemoryModuleType.JOB_SITE).isEmpty()) {
											LookTargetUtil.walkTowards(villager, blockPos, speed, 1);
											villager.getBrain().remember(MemoryModuleType.POTENTIAL_JOB_SITE, GlobalPos.create(world.getRegistryKey(), blockPos));
											DebugInfoSender.sendPointOfInterest(world, blockPos);
										}
									});
								return true;
							}
						}
					}
				)
		);
	}

	private static boolean canUseJobSite(RegistryEntry<PointOfInterestType> poiType, VillagerEntity villager, BlockPos pos) {
		boolean bl = villager.getBrain().getOptionalRegisteredMemory(MemoryModuleType.POTENTIAL_JOB_SITE).isPresent();
		if (bl) {
			return false;
		} else {
			Optional<GlobalPos> optional = villager.getBrain().getOptionalRegisteredMemory(MemoryModuleType.JOB_SITE);
			VillagerProfession villagerProfession = villager.getVillagerData().getProfession();
			if (villagerProfession.heldWorkstation().test(poiType)) {
				return optional.isEmpty() ? canReachJobSite(villager, pos, poiType.value()) : ((GlobalPos)optional.get()).getPos().equals(pos);
			} else {
				return false;
			}
		}
	}

	private static boolean canReachJobSite(PathAwareEntity entity, BlockPos pos, PointOfInterestType poiType) {
		Path path = entity.getNavigation().findPathTo(pos, poiType.searchDistance());
		return path != null && path.reachesTarget();
	}
}
