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

import com.google.common.collect.ImmutableMap;
import com.google.common.collect.Lists;
import java.util.List;
import net.minecraft.server.level.ServerLevel;
import net.minecraft.world.entity.EntityTypes;
import net.minecraft.world.entity.EquipmentSlot;
import net.minecraft.world.entity.LivingEntity;
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.npc.villager.Villager;
import net.minecraft.world.item.ItemStack;
import net.minecraft.world.item.trading.MerchantOffer;
import org.jspecify.annotations.Nullable;

public class ShowTradesToPlayer extends Behavior<Villager> {
	private static final int MAX_LOOK_TIME = 900;
	private static final int STARTING_LOOK_TIME = 40;
	@Nullable
	private ItemStack playerItemStack;
	private final List<ItemStack> displayItems = Lists.<ItemStack>newArrayList();
	private int cycleCounter;
	private int displayIndex;
	private int lookTime;

	public ShowTradesToPlayer(final int minDuration, final int maxDuration) {
		super(ImmutableMap.of(MemoryModuleType.INTERACTION_TARGET, MemoryStatus.VALUE_PRESENT), minDuration, maxDuration);
	}

	public boolean checkExtraStartConditions(final ServerLevel level, final Villager body) {
		Brain<?> brain = body.getBrain();
		if (brain.getMemory(MemoryModuleType.INTERACTION_TARGET).isEmpty()) {
			return false;
		} else {
			LivingEntity target = (LivingEntity)brain.getMemory(MemoryModuleType.INTERACTION_TARGET).get();
			return target.is(EntityTypes.PLAYER) && body.isAlive() && target.isAlive() && !body.isBaby() && body.distanceToSqr(target) <= 17.0;
		}
	}

	public boolean canStillUse(final ServerLevel level, final Villager body, final long timestamp) {
		return this.checkExtraStartConditions(level, body) && this.lookTime > 0 && body.getBrain().getMemory(MemoryModuleType.INTERACTION_TARGET).isPresent();
	}

	public void start(final ServerLevel level, final Villager body, final long timestamp) {
		super.start(level, body, timestamp);
		this.lookAtTarget(body);
		this.cycleCounter = 0;
		this.displayIndex = 0;
		this.lookTime = 40;
	}

	public void tick(final ServerLevel level, final Villager body, final long timestamp) {
		LivingEntity target = this.lookAtTarget(body);
		this.findItemsToDisplay(target, body);
		if (!this.displayItems.isEmpty()) {
			this.displayCyclingItems(body);
		} else {
			clearHeldItem(body);
			this.lookTime = Math.min(this.lookTime, 40);
		}

		this.lookTime--;
	}

	public void stop(final ServerLevel level, final Villager body, final long timestamp) {
		super.stop(level, body, timestamp);
		body.getBrain().eraseMemory(MemoryModuleType.INTERACTION_TARGET);
		clearHeldItem(body);
		this.playerItemStack = null;
	}

	private void findItemsToDisplay(final LivingEntity player, final Villager villager) {
		boolean changed = false;
		ItemStack currentPlayerItemStack = player.getMainHandItem();
		if (this.playerItemStack == null || !ItemStack.isSameItem(this.playerItemStack, currentPlayerItemStack)) {
			this.playerItemStack = currentPlayerItemStack;
			changed = true;
			this.displayItems.clear();
		}

		if (changed && !this.playerItemStack.isEmpty()) {
			this.updateDisplayItems(villager);
			if (!this.displayItems.isEmpty()) {
				this.lookTime = 900;
				this.displayFirstItem(villager);
			}
		}
	}

	private void displayFirstItem(final Villager villager) {
		displayAsHeldItem(villager, (ItemStack)this.displayItems.get(0));
	}

	private void updateDisplayItems(final Villager villager) {
		for (MerchantOffer offer : villager.getOffers()) {
			if (!offer.isOutOfStock() && this.playerItemStackMatchesCostOfOffer(offer)) {
				this.displayItems.add(offer.assemble());
			}
		}
	}

	private boolean playerItemStackMatchesCostOfOffer(final MerchantOffer offer) {
		return ItemStack.isSameItem(this.playerItemStack, offer.getCostA()) || ItemStack.isSameItem(this.playerItemStack, offer.getCostB());
	}

	private static void clearHeldItem(final Villager body) {
		body.setItemSlot(EquipmentSlot.MAINHAND, ItemStack.EMPTY);
		body.setDropChance(EquipmentSlot.MAINHAND, 0.085F);
	}

	private static void displayAsHeldItem(final Villager body, final ItemStack itemStack) {
		body.setItemSlot(EquipmentSlot.MAINHAND, itemStack);
		body.setDropChance(EquipmentSlot.MAINHAND, 0.0F);
	}

	private LivingEntity lookAtTarget(final Villager myBody) {
		Brain<?> brain = myBody.getBrain();
		LivingEntity target = (LivingEntity)brain.getMemory(MemoryModuleType.INTERACTION_TARGET).get();
		brain.setMemory(MemoryModuleType.LOOK_TARGET, new EntityTracker(target, true));
		return target;
	}

	private void displayCyclingItems(final Villager villager) {
		if (this.displayItems.size() >= 2 && ++this.cycleCounter >= 40) {
			this.displayIndex++;
			this.cycleCounter = 0;
			if (this.displayIndex > this.displayItems.size() - 1) {
				this.displayIndex = 0;
			}

			displayAsHeldItem(villager, (ItemStack)this.displayItems.get(this.displayIndex));
		}
	}
}