package net.minecraft.world.item;

import java.util.Collection;
import java.util.Set;
import java.util.function.Supplier;
import net.minecraft.core.HolderLookup;
import net.minecraft.network.chat.Component;
import net.minecraft.resources.Identifier;
import net.minecraft.world.flag.FeatureFlagSet;
import net.minecraft.world.level.ItemLike;
import org.jspecify.annotations.Nullable;

public class CreativeModeTab {
	private static final Identifier DEFAULT_BACKGROUND = createTextureLocation("items");
	private final Component displayName;
	private Identifier backgroundTexture = DEFAULT_BACKGROUND;
	private boolean canScroll = true;
	private boolean showTitle = true;
	private boolean alignedRight = false;
	private final CreativeModeTab.Row row;
	private final int column;
	private final CreativeModeTab.Type type;
	@Nullable
	private ItemStack iconItemStack;
	private Collection<ItemStack> displayItems = ItemStackLinkedSet.createTypeAndComponentsSet();
	private Set<ItemStack> displayItemsSearchTab = ItemStackLinkedSet.createTypeAndComponentsSet();
	private final Supplier<ItemStack> iconGenerator;
	private final CreativeModeTab.DisplayItemsGenerator displayItemsGenerator;

	private CreativeModeTab(
		final CreativeModeTab.Row row,
		final int column,
		final CreativeModeTab.Type type,
		final Component displayName,
		final Supplier<ItemStack> iconGenerator,
		final CreativeModeTab.DisplayItemsGenerator displayItemsGenerator
	) {
		this.row = row;
		this.column = column;
		this.displayName = displayName;
		this.iconGenerator = iconGenerator;
		this.displayItemsGenerator = displayItemsGenerator;
		this.type = type;
	}

	public static Identifier createTextureLocation(final String name) {
		return Identifier.withDefaultNamespace("textures/gui/container/creative_inventory/tab_" + name + ".png");
	}

	public static CreativeModeTab.Builder builder(final CreativeModeTab.Row row, final int column) {
		return new CreativeModeTab.Builder(row, column);
	}

	public Component getDisplayName() {
		return this.displayName;
	}

	public ItemStack getIconItem() {
		if (this.iconItemStack == null) {
			this.iconItemStack = (ItemStack)this.iconGenerator.get();
		}

		return this.iconItemStack;
	}

	public Identifier getBackgroundTexture() {
		return this.backgroundTexture;
	}

	public boolean showTitle() {
		return this.showTitle;
	}

	public boolean canScroll() {
		return this.canScroll;
	}

	public int column() {
		return this.column;
	}

	public CreativeModeTab.Row row() {
		return this.row;
	}

	public boolean hasAnyItems() {
		return !this.displayItems.isEmpty();
	}

	public boolean shouldDisplay() {
		return this.type != CreativeModeTab.Type.CATEGORY || this.hasAnyItems();
	}

	public boolean isAlignedRight() {
		return this.alignedRight;
	}

	public CreativeModeTab.Type getType() {
		return this.type;
	}

	public void buildContents(final CreativeModeTab.ItemDisplayParameters parameters) {
		CreativeModeTab.ItemDisplayBuilder displayList = new CreativeModeTab.ItemDisplayBuilder(this, parameters.enabledFeatures);
		this.displayItemsGenerator.accept(parameters, displayList);
		this.displayItems = displayList.tabContents;
		this.displayItemsSearchTab = displayList.searchTabContents;
	}

	public Collection<ItemStack> getDisplayItems() {
		return this.displayItems;
	}

	public Collection<ItemStack> getSearchTabDisplayItems() {
		return this.displayItemsSearchTab;
	}

	public boolean contains(final ItemStack stack) {
		return this.displayItemsSearchTab.contains(stack);
	}

	public static class Builder {
		private static final CreativeModeTab.DisplayItemsGenerator EMPTY_GENERATOR = (parameters, output) -> {};
		private final CreativeModeTab.Row row;
		private final int column;
		private Component displayName = Component.empty();
		private Supplier<ItemStack> iconGenerator = () -> ItemStack.EMPTY;
		private CreativeModeTab.DisplayItemsGenerator displayItemsGenerator = EMPTY_GENERATOR;
		private boolean canScroll = true;
		private boolean showTitle = true;
		private boolean alignedRight = false;
		private CreativeModeTab.Type type = CreativeModeTab.Type.CATEGORY;
		private Identifier backgroundTexture = CreativeModeTab.DEFAULT_BACKGROUND;

		public Builder(final CreativeModeTab.Row row, final int column) {
			this.row = row;
			this.column = column;
		}

		public CreativeModeTab.Builder title(final Component displayName) {
			this.displayName = displayName;
			return this;
		}

		public CreativeModeTab.Builder icon(final Supplier<ItemStack> iconGenerator) {
			this.iconGenerator = iconGenerator;
			return this;
		}

		public CreativeModeTab.Builder displayItems(final CreativeModeTab.DisplayItemsGenerator displayItemsGenerator) {
			this.displayItemsGenerator = displayItemsGenerator;
			return this;
		}

		public CreativeModeTab.Builder alignedRight() {
			this.alignedRight = true;
			return this;
		}

		public CreativeModeTab.Builder hideTitle() {
			this.showTitle = false;
			return this;
		}

		public CreativeModeTab.Builder noScrollBar() {
			this.canScroll = false;
			return this;
		}

		protected CreativeModeTab.Builder type(final CreativeModeTab.Type type) {
			this.type = type;
			return this;
		}

		public CreativeModeTab.Builder backgroundTexture(final Identifier backgroundTexture) {
			this.backgroundTexture = backgroundTexture;
			return this;
		}

		public CreativeModeTab build() {
			if ((this.type == CreativeModeTab.Type.HOTBAR || this.type == CreativeModeTab.Type.INVENTORY) && this.displayItemsGenerator != EMPTY_GENERATOR) {
				throw new IllegalStateException("Special tabs can't have display items");
			} else {
				CreativeModeTab tab = new CreativeModeTab(this.row, this.column, this.type, this.displayName, this.iconGenerator, this.displayItemsGenerator);
				tab.alignedRight = this.alignedRight;
				tab.showTitle = this.showTitle;
				tab.canScroll = this.canScroll;
				tab.backgroundTexture = this.backgroundTexture;
				return tab;
			}
		}
	}

	@FunctionalInterface
	public interface DisplayItemsGenerator {
		void accept(CreativeModeTab.ItemDisplayParameters parameters, CreativeModeTab.Output output);
	}

	private static class ItemDisplayBuilder implements CreativeModeTab.Output {
		public final Collection<ItemStack> tabContents = ItemStackLinkedSet.createTypeAndComponentsSet();
		public final Set<ItemStack> searchTabContents = ItemStackLinkedSet.createTypeAndComponentsSet();
		private final CreativeModeTab tab;
		private final FeatureFlagSet featureFlagSet;

		public ItemDisplayBuilder(final CreativeModeTab tab, final FeatureFlagSet featureFlagSet) {
			this.tab = tab;
			this.featureFlagSet = featureFlagSet;
		}

		@Override
		public void accept(final ItemStack stack, final CreativeModeTab.TabVisibility tabVisibility) {
			if (stack.getCount() != 1) {
				throw new IllegalArgumentException("Stack size must be exactly 1");
			} else {
				boolean foundDuplicateStack = this.tabContents.contains(stack) && tabVisibility != CreativeModeTab.TabVisibility.SEARCH_TAB_ONLY;
				if (foundDuplicateStack) {
					throw new IllegalStateException(
						"Accidentally adding the same item stack twice "
							+ stack.getDisplayName().getString()
							+ " to a Creative Mode Tab: "
							+ this.tab.getDisplayName().getString()
					);
				} else {
					if (stack.getItem().isEnabled(this.featureFlagSet)) {
						switch (tabVisibility) {
							case PARENT_AND_SEARCH_TABS:
								this.tabContents.add(stack);
								this.searchTabContents.add(stack);
								break;
							case PARENT_TAB_ONLY:
								this.tabContents.add(stack);
								break;
							case SEARCH_TAB_ONLY:
								this.searchTabContents.add(stack);
						}
					}
				}
			}
		}
	}

	public record ItemDisplayParameters(FeatureFlagSet enabledFeatures, boolean hasPermissions, HolderLookup.Provider holders) {
		public boolean needsUpdate(final FeatureFlagSet enabledFeatures, final boolean hasPermissions, final HolderLookup.Provider holders) {
			return !this.enabledFeatures.equals(enabledFeatures) || this.hasPermissions != hasPermissions || this.holders != holders;
		}
	}

	/**
	 * Access widened by fabric-creative-tab-api-v1 to accessible
	 */
	public interface Output {
		void accept(final ItemStack stack, final CreativeModeTab.TabVisibility tabVisibility);

		default void accept(final ItemStack stack) {
			this.accept(stack, CreativeModeTab.TabVisibility.PARENT_AND_SEARCH_TABS);
		}

		default void accept(final ItemLike item, final CreativeModeTab.TabVisibility tabVisibility) {
			this.accept(new ItemStack(item), tabVisibility);
		}

		default void accept(final ItemLike item) {
			this.accept(new ItemStack(item), CreativeModeTab.TabVisibility.PARENT_AND_SEARCH_TABS);
		}

		default void acceptAll(final Collection<ItemStack> stacks, final CreativeModeTab.TabVisibility tabVisibility) {
			stacks.forEach(stack -> this.accept(stack, tabVisibility));
		}

		default void acceptAll(final Collection<ItemStack> stacks) {
			this.acceptAll(stacks, CreativeModeTab.TabVisibility.PARENT_AND_SEARCH_TABS);
		}
	}

	public static enum Row {
		TOP,
		BOTTOM;
	}

	/**
	 * Access widened by fabric-creative-tab-api-v1 to accessible
	 */
	public static enum TabVisibility {
		PARENT_AND_SEARCH_TABS,
		PARENT_TAB_ONLY,
		SEARCH_TAB_ONLY;
	}

	public static enum Type {
		CATEGORY,
		INVENTORY,
		HOTBAR,
		SEARCH;
	}
}