package net.minecraft.world.item.crafting;

import java.util.ArrayList;
import java.util.List;
import net.minecraft.world.entity.player.StackedItemContents;
import net.minecraft.world.item.ItemStack;

public class CraftingInput implements RecipeInput {
	public static final CraftingInput EMPTY = new CraftingInput(0, 0, List.of());
	private final int width;
	private final int height;
	private final List<ItemStack> items;
	private final StackedItemContents stackedContents = new StackedItemContents();
	private final int ingredientCount;

	private CraftingInput(final int width, final int height, final List<ItemStack> items) {
		this.width = width;
		this.height = height;
		this.items = items;
		int ingredientCount = 0;

		for (ItemStack item : items) {
			if (!item.isEmpty()) {
				ingredientCount++;
				this.stackedContents.accountStack(item, 1);
			}
		}

		this.ingredientCount = ingredientCount;
	}

	public static CraftingInput of(final int width, final int height, final List<ItemStack> items) {
		return ofPositioned(width, height, items).input();
	}

	public static CraftingInput.Positioned ofPositioned(final int width, final int height, final List<ItemStack> items) {
		if (width != 0 && height != 0) {
			int left = width - 1;
			int right = 0;
			int top = height - 1;
			int bottom = 0;

			for (int y = 0; y < height; y++) {
				boolean rowEmpty = true;

				for (int x = 0; x < width; x++) {
					ItemStack item = (ItemStack)items.get(x + y * width);
					if (!item.isEmpty()) {
						left = Math.min(left, x);
						right = Math.max(right, x);
						rowEmpty = false;
					}
				}

				if (!rowEmpty) {
					top = Math.min(top, y);
					bottom = Math.max(bottom, y);
				}
			}

			int newWidth = right - left + 1;
			int newHeight = bottom - top + 1;
			if (newWidth <= 0 || newHeight <= 0) {
				return CraftingInput.Positioned.EMPTY;
			} else if (newWidth == width && newHeight == height) {
				return new CraftingInput.Positioned(new CraftingInput(width, height, items), left, top);
			} else {
				List<ItemStack> newItems = new ArrayList(newWidth * newHeight);

				for (int y = 0; y < newHeight; y++) {
					for (int xx = 0; xx < newWidth; xx++) {
						int index = xx + left + (y + top) * width;
						newItems.add((ItemStack)items.get(index));
					}
				}

				return new CraftingInput.Positioned(new CraftingInput(newWidth, newHeight, newItems), left, top);
			}
		} else {
			return CraftingInput.Positioned.EMPTY;
		}
	}

	@Override
	public ItemStack getItem(final int index) {
		return (ItemStack)this.items.get(index);
	}

	public ItemStack getItem(final int x, final int y) {
		return (ItemStack)this.items.get(x + y * this.width);
	}

	@Override
	public int size() {
		return this.items.size();
	}

	@Override
	public boolean isEmpty() {
		return this.ingredientCount == 0;
	}

	public StackedItemContents stackedContents() {
		return this.stackedContents;
	}

	public List<ItemStack> items() {
		return this.items;
	}

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

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

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

	public boolean equals(final Object obj) {
		if (obj == this) {
			return true;
		} else {
			return !(obj instanceof CraftingInput input)
				? false
				: this.width == input.width
					&& this.height == input.height
					&& this.ingredientCount == input.ingredientCount
					&& ItemStack.listMatches(this.items, input.items);
		}
	}

	public int hashCode() {
		int result = ItemStack.hashStackList(this.items);
		result = 31 * result + this.width;
		return 31 * result + this.height;
	}

	public record Positioned(CraftingInput input, int left, int top) {
		public static final CraftingInput.Positioned EMPTY = new CraftingInput.Positioned(CraftingInput.EMPTY, 0, 0);
	}
}