package net.minecraft.world.item.crafting;

import net.minecraft.core.HolderLookup;
import net.minecraft.core.component.DataComponents;
import net.minecraft.world.item.ItemStack;
import net.minecraft.world.item.Items;
import net.minecraft.world.level.Level;

public class TippedArrowRecipe extends CustomRecipe {
	public TippedArrowRecipe(CraftingBookCategory category) {
		super(category);
	}

	public boolean matches(CraftingInput input, Level level) {
		if (input.width() == 3 && input.height() == 3) {
			for (int i = 0; i < input.height(); i++) {
				for (int j = 0; j < input.width(); j++) {
					ItemStack itemStack = input.getItem(j, i);
					if (itemStack.isEmpty()) {
						return false;
					}

					if (j == 1 && i == 1) {
						if (!itemStack.is(Items.LINGERING_POTION)) {
							return false;
						}
					} else if (!itemStack.is(Items.ARROW)) {
						return false;
					}
				}
			}

			return true;
		} else {
			return false;
		}
	}

	public ItemStack assemble(CraftingInput input, HolderLookup.Provider registries) {
		ItemStack itemStack = input.getItem(1, 1);
		if (!itemStack.is(Items.LINGERING_POTION)) {
			return ItemStack.EMPTY;
		} else {
			ItemStack itemStack2 = new ItemStack(Items.TIPPED_ARROW, 8);
			itemStack2.set(DataComponents.POTION_CONTENTS, itemStack.get(DataComponents.POTION_CONTENTS));
			return itemStack2;
		}
	}

	@Override
	public boolean canCraftInDimensions(int width, int height) {
		return width >= 3 && height >= 3;
	}

	@Override
	public RecipeSerializer<?> getSerializer() {
		return RecipeSerializer.TIPPED_ARROW;
	}
}