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.network.protocol.game.DebugPackets;
import net.minecraft.server.level.ServerLevel;
import net.minecraft.world.entity.ai.memory.MemoryModuleType;
import net.minecraft.world.entity.ai.memory.MemoryStatus;
import net.minecraft.world.entity.ai.village.poi.PoiManager;
import net.minecraft.world.entity.npc.Villager;
import net.minecraft.world.entity.schedule.Activity;

public class GoToPotentialJobSite extends Behavior<Villager> {
	private static final int TICKS_UNTIL_TIMEOUT = 1200;
	final float speedModifier;

	public GoToPotentialJobSite(float speedModifier) {
		super(ImmutableMap.of(MemoryModuleType.POTENTIAL_JOB_SITE, MemoryStatus.VALUE_PRESENT), 1200);
		this.speedModifier = speedModifier;
	}

	protected boolean checkExtraStartConditions(ServerLevel level, Villager owner) {
		return (Boolean)owner.getBrain()
			.getActiveNonCoreActivity()
			.map(activity -> activity == Activity.IDLE || activity == Activity.WORK || activity == Activity.PLAY)
			.orElse(true);
	}

	protected boolean canStillUse(ServerLevel level, Villager entity, long gameTime) {
		return entity.getBrain().hasMemoryValue(MemoryModuleType.POTENTIAL_JOB_SITE);
	}

	protected void tick(ServerLevel level, Villager owner, long gameTime) {
		BehaviorUtils.setWalkAndLookTargetMemories(
			owner, ((GlobalPos)owner.getBrain().getMemory(MemoryModuleType.POTENTIAL_JOB_SITE).get()).pos(), this.speedModifier, 1
		);
	}

	protected void stop(ServerLevel level, Villager entity, long gameTime) {
		Optional<GlobalPos> optional = entity.getBrain().getMemory(MemoryModuleType.POTENTIAL_JOB_SITE);
		optional.ifPresent(globalPos -> {
			BlockPos blockPos = globalPos.pos();
			ServerLevel serverLevel2 = level.getServer().getLevel(globalPos.dimension());
			if (serverLevel2 != null) {
				PoiManager poiManager = serverLevel2.getPoiManager();
				if (poiManager.exists(blockPos, holder -> true)) {
					poiManager.release(blockPos);
				}

				DebugPackets.sendPoiTicketCountPacket(level, blockPos);
			}
		});
		entity.getBrain().eraseMemory(MemoryModuleType.POTENTIAL_JOB_SITE);
	}
}