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

import net.minecraft.core.BlockPos;
import net.minecraft.core.GlobalPos;
import net.minecraft.world.entity.LivingEntity;
import net.minecraft.world.entity.ai.behavior.declarative.BehaviorBuilder;
import net.minecraft.world.entity.ai.memory.MemoryModuleType;
import org.apache.commons.lang3.mutable.MutableInt;

public class SetHiddenState {
	private static final int HIDE_TIMEOUT = 300;

	public static BehaviorControl<LivingEntity> create(int stayHiddenSeconds, int closeEnoughdist) {
		int i = stayHiddenSeconds * 20;
		MutableInt mutableInt = new MutableInt(0);
		return BehaviorBuilder.create(
			instance -> instance.group(instance.present(MemoryModuleType.HIDING_PLACE), instance.present(MemoryModuleType.HEARD_BELL_TIME))
				.apply(instance, (memoryAccessor, memoryAccessor2) -> (serverLevel, livingEntity, l) -> {
					long m = instance.<Long>get(memoryAccessor2);
					boolean bl = m + 300L <= l;
					if (mutableInt.getValue() <= i && !bl) {
						BlockPos blockPos = instance.<GlobalPos>get(memoryAccessor).pos();
						if (blockPos.closerThan(livingEntity.blockPosition(), closeEnoughdist)) {
							mutableInt.increment();
						}

						return true;
					} else {
						memoryAccessor2.erase();
						memoryAccessor.erase();
						livingEntity.getBrain().updateActivityFromSchedule(serverLevel.getDayTime(), serverLevel.getGameTime());
						mutableInt.setValue(0);
						return true;
					}
				})
		);
	}
}