package net.minecraft.world.entity.player;

import java.util.List;
import net.minecraft.core.Holder;
import net.minecraft.world.item.Item;
import net.minecraft.world.item.ItemStack;
import net.minecraft.world.item.crafting.PlacementInfo;
import net.minecraft.world.item.crafting.Recipe;
import org.jspecify.annotations.Nullable;

public class StackedItemContents {
	private final StackedContents<Holder<Item>> raw = new StackedContents<>();

	public void accountSimpleStack(final ItemStack itemStack) {
		if (Inventory.isUsableForCrafting(itemStack)) {
			this.accountStack(itemStack);
		}
	}

	public void accountStack(final ItemStack itemStack) {
		this.accountStack(itemStack, itemStack.getMaxStackSize());
	}

	public void accountStack(final ItemStack itemStack, final int maxCount) {
		if (!itemStack.isEmpty()) {
			int count = Math.min(maxCount, itemStack.getCount());
			this.raw.account(itemStack.typeHolder(), count);
		}
	}

	public boolean canCraft(final Recipe<?> recipe, @Nullable final StackedContents.Output<Holder<Item>> output) {
		return this.canCraft(recipe, 1, output);
	}

	public boolean canCraft(final Recipe<?> recipe, final int amount, @Nullable final StackedContents.Output<Holder<Item>> output) {
		PlacementInfo placementInfo = recipe.placementInfo();
		return placementInfo.isImpossibleToPlace() ? false : this.canCraft(placementInfo.ingredients(), amount, output);
	}

	public boolean canCraft(
		final List<? extends StackedContents.IngredientInfo<Holder<Item>>> contents, @Nullable final StackedContents.Output<Holder<Item>> output
	) {
		return this.canCraft(contents, 1, output);
	}

	private boolean canCraft(
		final List<? extends StackedContents.IngredientInfo<Holder<Item>>> contents, final int amount, @Nullable final StackedContents.Output<Holder<Item>> output
	) {
		return this.raw.tryPick(contents, amount, output);
	}

	public int getBiggestCraftableStack(final Recipe<?> recipe, @Nullable final StackedContents.Output<Holder<Item>> output) {
		return this.getBiggestCraftableStack(recipe, Integer.MAX_VALUE, output);
	}

	public int getBiggestCraftableStack(final Recipe<?> recipe, final int maxSize, @Nullable final StackedContents.Output<Holder<Item>> output) {
		return this.raw.tryPickAll(recipe.placementInfo().ingredients(), maxSize, output);
	}

	public void clear() {
		this.raw.clear();
	}
}