package net.minecraft.world.inventory;

import java.util.Optional;
import net.minecraft.advancements.CriteriaTriggers;
import net.minecraft.core.Holder;
import net.minecraft.core.component.DataComponents;
import net.minecraft.resources.ResourceLocation;
import net.minecraft.server.level.ServerPlayer;
import net.minecraft.tags.ItemTags;
import net.minecraft.world.Container;
import net.minecraft.world.SimpleContainer;
import net.minecraft.world.entity.player.Inventory;
import net.minecraft.world.entity.player.Player;
import net.minecraft.world.item.ItemStack;
import net.minecraft.world.item.Items;
import net.minecraft.world.item.alchemy.Potion;
import net.minecraft.world.item.alchemy.PotionBrewing;
import net.minecraft.world.item.alchemy.PotionContents;

public class BrewingStandMenu extends AbstractContainerMenu {
	static final ResourceLocation EMPTY_SLOT_FUEL = ResourceLocation.withDefaultNamespace("container/slot/brewing_fuel");
	static final ResourceLocation EMPTY_SLOT_POTION = ResourceLocation.withDefaultNamespace("container/slot/potion");
	private static final int BOTTLE_SLOT_START = 0;
	private static final int BOTTLE_SLOT_END = 2;
	private static final int INGREDIENT_SLOT = 3;
	private static final int FUEL_SLOT = 4;
	private static final int SLOT_COUNT = 5;
	private static final int DATA_COUNT = 2;
	private static final int INV_SLOT_START = 5;
	private static final int INV_SLOT_END = 32;
	private static final int USE_ROW_SLOT_START = 32;
	private static final int USE_ROW_SLOT_END = 41;
	private final Container brewingStand;
	private final ContainerData brewingStandData;
	private final Slot ingredientSlot;

	public BrewingStandMenu(int containerId, Inventory playerInventory) {
		this(containerId, playerInventory, new SimpleContainer(5), new SimpleContainerData(2));
	}

	public BrewingStandMenu(int containerId, Inventory playerInventory, Container brewingStandContainer, ContainerData brewingStandData) {
		super(MenuType.BREWING_STAND, containerId);
		checkContainerSize(brewingStandContainer, 5);
		checkContainerDataCount(brewingStandData, 2);
		this.brewingStand = brewingStandContainer;
		this.brewingStandData = brewingStandData;
		PotionBrewing potionBrewing = playerInventory.player.level().potionBrewing();
		this.addSlot(new BrewingStandMenu.PotionSlot(brewingStandContainer, 0, 56, 51));
		this.addSlot(new BrewingStandMenu.PotionSlot(brewingStandContainer, 1, 79, 58));
		this.addSlot(new BrewingStandMenu.PotionSlot(brewingStandContainer, 2, 102, 51));
		this.ingredientSlot = this.addSlot(new BrewingStandMenu.IngredientsSlot(potionBrewing, brewingStandContainer, 3, 79, 17));
		this.addSlot(new BrewingStandMenu.FuelSlot(brewingStandContainer, 4, 17, 17));
		this.addDataSlots(brewingStandData);
		this.addStandardInventorySlots(playerInventory, 8, 84);
	}

	@Override
	public boolean stillValid(Player player) {
		return this.brewingStand.stillValid(player);
	}

	@Override
	public ItemStack quickMoveStack(Player player, int index) {
		ItemStack itemStack = ItemStack.EMPTY;
		Slot slot = this.slots.get(index);
		if (slot != null && slot.hasItem()) {
			ItemStack itemStack2 = slot.getItem();
			itemStack = itemStack2.copy();
			if ((index < 0 || index > 2) && index != 3 && index != 4) {
				if (BrewingStandMenu.FuelSlot.mayPlaceItem(itemStack)) {
					if (this.moveItemStackTo(itemStack2, 4, 5, false) || this.ingredientSlot.mayPlace(itemStack2) && !this.moveItemStackTo(itemStack2, 3, 4, false)) {
						return ItemStack.EMPTY;
					}
				} else if (this.ingredientSlot.mayPlace(itemStack2)) {
					if (!this.moveItemStackTo(itemStack2, 3, 4, false)) {
						return ItemStack.EMPTY;
					}
				} else if (BrewingStandMenu.PotionSlot.mayPlaceItem(itemStack)) {
					if (!this.moveItemStackTo(itemStack2, 0, 3, false)) {
						return ItemStack.EMPTY;
					}
				} else if (index >= 5 && index < 32) {
					if (!this.moveItemStackTo(itemStack2, 32, 41, false)) {
						return ItemStack.EMPTY;
					}
				} else if (index >= 32 && index < 41) {
					if (!this.moveItemStackTo(itemStack2, 5, 32, false)) {
						return ItemStack.EMPTY;
					}
				} else if (!this.moveItemStackTo(itemStack2, 5, 41, false)) {
					return ItemStack.EMPTY;
				}
			} else {
				if (!this.moveItemStackTo(itemStack2, 5, 41, true)) {
					return ItemStack.EMPTY;
				}

				slot.onQuickCraft(itemStack2, itemStack);
			}

			if (itemStack2.isEmpty()) {
				slot.setByPlayer(ItemStack.EMPTY);
			} else {
				slot.setChanged();
			}

			if (itemStack2.getCount() == itemStack.getCount()) {
				return ItemStack.EMPTY;
			}

			slot.onTake(player, itemStack);
		}

		return itemStack;
	}

	public int getFuel() {
		return this.brewingStandData.get(1);
	}

	public int getBrewingTicks() {
		return this.brewingStandData.get(0);
	}

	static class FuelSlot extends Slot {
		public FuelSlot(Container container, int i, int j, int k) {
			super(container, i, j, k);
		}

		@Override
		public boolean mayPlace(ItemStack stack) {
			return mayPlaceItem(stack);
		}

		/**
		 * Returns {@code true} if the given {@link net.minecraft.world.item.ItemStack} is usable as fuel in the brewing stand.
		 */
		public static boolean mayPlaceItem(ItemStack itemStack) {
			return itemStack.is(ItemTags.BREWING_FUEL);
		}

		@Override
		public ResourceLocation getNoItemIcon() {
			return BrewingStandMenu.EMPTY_SLOT_FUEL;
		}
	}

	static class IngredientsSlot extends Slot {
		private final PotionBrewing potionBrewing;

		public IngredientsSlot(PotionBrewing potionBrewing, Container container, int slot, int x, int y) {
			super(container, slot, x, y);
			this.potionBrewing = potionBrewing;
		}

		@Override
		public boolean mayPlace(ItemStack stack) {
			return this.potionBrewing.isIngredient(stack);
		}
	}

	static class PotionSlot extends Slot {
		public PotionSlot(Container container, int i, int j, int k) {
			super(container, i, j, k);
		}

		@Override
		public boolean mayPlace(ItemStack stack) {
			return mayPlaceItem(stack);
		}

		@Override
		public int getMaxStackSize() {
			return 1;
		}

		@Override
		public void onTake(Player player, ItemStack stack) {
			Optional<Holder<Potion>> optional = stack.getOrDefault(DataComponents.POTION_CONTENTS, PotionContents.EMPTY).potion();
			if (optional.isPresent() && player instanceof ServerPlayer serverPlayer) {
				CriteriaTriggers.BREWED_POTION.trigger(serverPlayer, (Holder<Potion>)optional.get());
			}

			super.onTake(player, stack);
		}

		/**
		 * Returns {@code true} if this {@link net.minecraft.world.item.ItemStack} can be filled with a potion.
		 */
		public static boolean mayPlaceItem(ItemStack stack) {
			return stack.is(Items.POTION) || stack.is(Items.SPLASH_POTION) || stack.is(Items.LINGERING_POTION) || stack.is(Items.GLASS_BOTTLE);
		}

		@Override
		public ResourceLocation getNoItemIcon() {
			return BrewingStandMenu.EMPTY_SLOT_POTION;
		}
	}
}