package net.minecraft.world.inventory;

import java.util.List;
import java.util.Map;
import java.util.Objects;
import net.minecraft.resources.Identifier;
import net.minecraft.server.level.ServerLevel;
import net.minecraft.world.Container;
import net.minecraft.world.entity.EquipmentSlot;
import net.minecraft.world.entity.player.Inventory;
import net.minecraft.world.entity.player.Player;
import net.minecraft.world.item.ItemStack;

public class InventoryMenu extends AbstractCraftingMenu {
	public static final int CONTAINER_ID = 0;
	public static final int RESULT_SLOT = 0;
	private static final int CRAFTING_GRID_WIDTH = 2;
	private static final int CRAFTING_GRID_HEIGHT = 2;
	public static final int CRAFT_SLOT_START = 1;
	public static final int CRAFT_SLOT_COUNT = 4;
	public static final int CRAFT_SLOT_END = 5;
	public static final int ARMOR_SLOT_START = 5;
	public static final int ARMOR_SLOT_COUNT = 4;
	public static final int ARMOR_SLOT_END = 9;
	public static final int INV_SLOT_START = 9;
	public static final int INV_SLOT_END = 36;
	public static final int USE_ROW_SLOT_START = 36;
	public static final int USE_ROW_SLOT_END = 45;
	public static final int SHIELD_SLOT = 45;
	public static final Identifier EMPTY_ARMOR_SLOT_HELMET = Identifier.withDefaultNamespace("container/slot/helmet");
	public static final Identifier EMPTY_ARMOR_SLOT_CHESTPLATE = Identifier.withDefaultNamespace("container/slot/chestplate");
	public static final Identifier EMPTY_ARMOR_SLOT_LEGGINGS = Identifier.withDefaultNamespace("container/slot/leggings");
	public static final Identifier EMPTY_ARMOR_SLOT_BOOTS = Identifier.withDefaultNamespace("container/slot/boots");
	public static final Identifier EMPTY_ARMOR_SLOT_SHIELD = Identifier.withDefaultNamespace("container/slot/shield");
	private static final Map<EquipmentSlot, Identifier> TEXTURE_EMPTY_SLOTS = Map.of(
		EquipmentSlot.FEET,
		EMPTY_ARMOR_SLOT_BOOTS,
		EquipmentSlot.LEGS,
		EMPTY_ARMOR_SLOT_LEGGINGS,
		EquipmentSlot.CHEST,
		EMPTY_ARMOR_SLOT_CHESTPLATE,
		EquipmentSlot.HEAD,
		EMPTY_ARMOR_SLOT_HELMET
	);
	private static final EquipmentSlot[] SLOT_IDS = new EquipmentSlot[]{EquipmentSlot.HEAD, EquipmentSlot.CHEST, EquipmentSlot.LEGS, EquipmentSlot.FEET};
	public final boolean active;
	private final Player owner;

	public InventoryMenu(final Inventory inventory, final boolean active, final Player owner) {
		super(null, 0, 2, 2);
		this.active = active;
		this.owner = owner;
		this.addResultSlot(owner, 154, 28);
		this.addCraftingGridSlots(98, 18);

		for (int i = 0; i < 4; i++) {
			EquipmentSlot slot = SLOT_IDS[i];
			Identifier emptyIcon = (Identifier)TEXTURE_EMPTY_SLOTS.get(slot);
			this.addSlot(new ArmorSlot(inventory, owner, slot, 39 - i, 8, 8 + i * 18, emptyIcon));
		}

		this.addStandardInventorySlots(inventory, 8, 84);
		this.addSlot(new Slot(inventory, 40, 77, 62) {
			{
				Objects.requireNonNull(InventoryMenu.this);
			}

			@Override
			public void setByPlayer(final ItemStack itemStack, final ItemStack previous) {
				owner.onEquipItem(EquipmentSlot.OFFHAND, previous, itemStack);
				super.setByPlayer(itemStack, previous);
			}

			@Override
			public Identifier getNoItemIcon() {
				return InventoryMenu.EMPTY_ARMOR_SLOT_SHIELD;
			}
		});
	}

	public static boolean isHotbarSlot(final int slot) {
		return slot >= 36 && slot < 45 || slot == 45;
	}

	@Override
	public void slotsChanged(final Container container) {
		if (this.owner.level() instanceof ServerLevel level) {
			CraftingMenu.slotChangedCraftingGrid(this, level, this.owner, this.craftSlots, this.resultSlots, null);
		}
	}

	@Override
	public void removed(final Player player) {
		super.removed(player);
		this.resultSlots.clearContent();
		if (!player.level().isClientSide()) {
			this.clearContainer(player, this.craftSlots);
		}
	}

	@Override
	public boolean stillValid(final Player player) {
		return true;
	}

	@Override
	public ItemStack quickMoveStack(final Player player, final int slotIndex) {
		ItemStack clicked = ItemStack.EMPTY;
		Slot slot = this.slots.get(slotIndex);
		if (slot.hasItem()) {
			ItemStack stack = slot.getItem();
			clicked = stack.copy();
			EquipmentSlot eqSlot = player.getEquipmentSlotForItem(clicked);
			if (slotIndex == 0) {
				if (!this.moveItemStackTo(stack, 9, 45, true)) {
					return ItemStack.EMPTY;
				}

				slot.onQuickCraft(stack, clicked);
			} else if (slotIndex >= 1 && slotIndex < 5) {
				if (!this.moveItemStackTo(stack, 9, 45, false)) {
					return ItemStack.EMPTY;
				}
			} else if (slotIndex >= 5 && slotIndex < 9) {
				if (!this.moveItemStackTo(stack, 9, 45, false)) {
					return ItemStack.EMPTY;
				}
			} else if (eqSlot.getType() == EquipmentSlot.Type.HUMANOID_ARMOR && !this.slots.get(8 - eqSlot.getIndex()).hasItem()) {
				int pos = 8 - eqSlot.getIndex();
				if (!this.moveItemStackTo(stack, pos, pos + 1, false)) {
					return ItemStack.EMPTY;
				}
			} else if (eqSlot == EquipmentSlot.OFFHAND && !this.slots.get(45).hasItem()) {
				if (!this.moveItemStackTo(stack, 45, 46, false)) {
					return ItemStack.EMPTY;
				}
			} else if (slotIndex >= 9 && slotIndex < 36) {
				if (!this.moveItemStackTo(stack, 36, 45, false)) {
					return ItemStack.EMPTY;
				}
			} else if (slotIndex >= 36 && slotIndex < 45) {
				if (!this.moveItemStackTo(stack, 9, 36, false)) {
					return ItemStack.EMPTY;
				}
			} else if (!this.moveItemStackTo(stack, 9, 45, false)) {
				return ItemStack.EMPTY;
			}

			if (stack.isEmpty()) {
				slot.setByPlayer(ItemStack.EMPTY, clicked);
			} else {
				slot.setChanged();
			}

			if (stack.getCount() == clicked.getCount()) {
				return ItemStack.EMPTY;
			}

			slot.onTake(player, stack);
			if (slotIndex == 0) {
				player.drop(stack, false);
			}
		}

		return clicked;
	}

	@Override
	public boolean canTakeItemForPickAll(final ItemStack carried, final Slot target) {
		return target.container != this.resultSlots && super.canTakeItemForPickAll(carried, target);
	}

	@Override
	public Slot getResultSlot() {
		return this.slots.get(0);
	}

	@Override
	public List<Slot> getInputGridSlots() {
		return this.slots.subList(1, 5);
	}

	public CraftingContainer getCraftSlots() {
		return this.craftSlots;
	}

	@Override
	public RecipeBookType getRecipeBookType() {
		return RecipeBookType.CRAFTING;
	}

	@Override
	protected Player owner() {
		return this.owner;
	}
}