package net.minecraft.client.gui.screens.recipebook;

import com.google.common.collect.ImmutableList;
import com.google.common.collect.Lists;
import com.google.common.collect.Sets;
import java.util.List;
import java.util.Set;
import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.core.RegistryAccess;
import net.minecraft.stats.RecipeBook;
import net.minecraft.world.entity.player.StackedContents;
import net.minecraft.world.item.ItemStack;
import net.minecraft.world.item.crafting.RecipeHolder;

@Environment(EnvType.CLIENT)
public class RecipeCollection {
	private final RegistryAccess registryAccess;
	private final List<RecipeHolder<?>> recipes;
	private final boolean singleResultItem;
	private final Set<RecipeHolder<?>> craftable = Sets.<RecipeHolder<?>>newHashSet();
	private final Set<RecipeHolder<?>> fitsDimensions = Sets.<RecipeHolder<?>>newHashSet();
	private final Set<RecipeHolder<?>> known = Sets.<RecipeHolder<?>>newHashSet();

	public RecipeCollection(RegistryAccess registryAccess, List<RecipeHolder<?>> recipes) {
		this.registryAccess = registryAccess;
		this.recipes = ImmutableList.copyOf(recipes);
		if (recipes.size() <= 1) {
			this.singleResultItem = true;
		} else {
			this.singleResultItem = allRecipesHaveSameResult(registryAccess, recipes);
		}
	}

	private static boolean allRecipesHaveSameResult(RegistryAccess registryAccess, List<RecipeHolder<?>> recipes) {
		int i = recipes.size();
		ItemStack itemStack = ((RecipeHolder)recipes.get(0)).value().getResultItem(registryAccess);

		for (int j = 1; j < i; j++) {
			ItemStack itemStack2 = ((RecipeHolder)recipes.get(j)).value().getResultItem(registryAccess);
			if (!ItemStack.isSameItemSameComponents(itemStack, itemStack2)) {
				return false;
			}
		}

		return true;
	}

	public RegistryAccess registryAccess() {
		return this.registryAccess;
	}

	/**
	 * Checks if recipebook is not empty
	 */
	public boolean hasKnownRecipes() {
		return !this.known.isEmpty();
	}

	public void updateKnownRecipes(RecipeBook book) {
		for (RecipeHolder<?> recipeHolder : this.recipes) {
			if (book.contains(recipeHolder)) {
				this.known.add(recipeHolder);
			}
		}
	}

	public void canCraft(StackedContents handler, int width, int height, RecipeBook book) {
		for (RecipeHolder<?> recipeHolder : this.recipes) {
			boolean bl = recipeHolder.value().canCraftInDimensions(width, height) && book.contains(recipeHolder);
			if (bl) {
				this.fitsDimensions.add(recipeHolder);
			} else {
				this.fitsDimensions.remove(recipeHolder);
			}

			if (bl && handler.canCraft(recipeHolder.value(), null)) {
				this.craftable.add(recipeHolder);
			} else {
				this.craftable.remove(recipeHolder);
			}
		}
	}

	public boolean isCraftable(RecipeHolder<?> recipe) {
		return this.craftable.contains(recipe);
	}

	public boolean hasCraftable() {
		return !this.craftable.isEmpty();
	}

	public boolean hasFitting() {
		return !this.fitsDimensions.isEmpty();
	}

	public List<RecipeHolder<?>> getRecipes() {
		return this.recipes;
	}

	public List<RecipeHolder<?>> getRecipes(boolean onlyCraftable) {
		List<RecipeHolder<?>> list = Lists.<RecipeHolder<?>>newArrayList();
		Set<RecipeHolder<?>> set = onlyCraftable ? this.craftable : this.fitsDimensions;

		for (RecipeHolder<?> recipeHolder : this.recipes) {
			if (set.contains(recipeHolder)) {
				list.add(recipeHolder);
			}
		}

		return list;
	}

	/**
	 * @param craftable If true, this method will only return craftable recipes. If false, this method will only return uncraftable recipes.
	 */
	public List<RecipeHolder<?>> getDisplayRecipes(boolean craftable) {
		List<RecipeHolder<?>> list = Lists.<RecipeHolder<?>>newArrayList();

		for (RecipeHolder<?> recipeHolder : this.recipes) {
			if (this.fitsDimensions.contains(recipeHolder) && this.craftable.contains(recipeHolder) == craftable) {
				list.add(recipeHolder);
			}
		}

		return list;
	}

	public boolean hasSingleResultItem() {
		return this.singleResultItem;
	}
}