package net.minecraft.world.item.crafting;

import com.mojang.serialization.MapCodec;
import com.mojang.serialization.codecs.RecordCodecBuilder;
import java.util.List;
import java.util.Objects;
import java.util.Optional;
import net.minecraft.core.Holder;
import net.minecraft.core.HolderLookup;
import net.minecraft.core.component.DataComponents;
import net.minecraft.network.RegistryFriendlyByteBuf;
import net.minecraft.network.codec.StreamCodec;
import net.minecraft.world.item.ItemStack;
import net.minecraft.world.item.Items;
import net.minecraft.world.item.crafting.display.RecipeDisplay;
import net.minecraft.world.item.crafting.display.SlotDisplay;
import net.minecraft.world.item.crafting.display.SmithingRecipeDisplay;
import net.minecraft.world.item.equipment.trim.ArmorTrim;
import net.minecraft.world.item.equipment.trim.TrimMaterial;
import net.minecraft.world.item.equipment.trim.TrimMaterials;
import net.minecraft.world.item.equipment.trim.TrimPattern;
import org.jetbrains.annotations.Nullable;

public class SmithingTrimRecipe implements SmithingRecipe {
	final Ingredient template;
	final Ingredient base;
	final Ingredient addition;
	final Holder<TrimPattern> pattern;
	@Nullable
	private PlacementInfo placementInfo;

	public SmithingTrimRecipe(Ingredient template, Ingredient base, Ingredient addition, Holder<TrimPattern> pattern) {
		this.template = template;
		this.base = base;
		this.addition = addition;
		this.pattern = pattern;
	}

	public ItemStack assemble(SmithingRecipeInput smithingRecipeInput, HolderLookup.Provider provider) {
		return applyTrim(provider, smithingRecipeInput.base(), smithingRecipeInput.addition(), this.pattern);
	}

	public static ItemStack applyTrim(HolderLookup.Provider registries, ItemStack base, ItemStack addition, Holder<TrimPattern> pattern) {
		Optional<Holder<TrimMaterial>> optional = TrimMaterials.getFromIngredient(registries, addition);
		if (optional.isPresent()) {
			ArmorTrim armorTrim = base.get(DataComponents.TRIM);
			ArmorTrim armorTrim2 = new ArmorTrim((Holder<TrimMaterial>)optional.get(), pattern);
			if (Objects.equals(armorTrim, armorTrim2)) {
				return ItemStack.EMPTY;
			} else {
				ItemStack itemStack = base.copyWithCount(1);
				itemStack.set(DataComponents.TRIM, armorTrim2);
				return itemStack;
			}
		} else {
			return ItemStack.EMPTY;
		}
	}

	@Override
	public Optional<Ingredient> templateIngredient() {
		return Optional.of(this.template);
	}

	@Override
	public Ingredient baseIngredient() {
		return this.base;
	}

	@Override
	public Optional<Ingredient> additionIngredient() {
		return Optional.of(this.addition);
	}

	@Override
	public RecipeSerializer<SmithingTrimRecipe> getSerializer() {
		return RecipeSerializer.SMITHING_TRIM;
	}

	@Override
	public PlacementInfo placementInfo() {
		if (this.placementInfo == null) {
			this.placementInfo = PlacementInfo.create(List.of(this.template, this.base, this.addition));
		}

		return this.placementInfo;
	}

	@Override
	public List<RecipeDisplay> display() {
		SlotDisplay slotDisplay = this.base.display();
		SlotDisplay slotDisplay2 = this.addition.display();
		SlotDisplay slotDisplay3 = this.template.display();
		return List.of(
			new SmithingRecipeDisplay(
				slotDisplay3,
				slotDisplay,
				slotDisplay2,
				new SlotDisplay.SmithingTrimDemoSlotDisplay(slotDisplay, slotDisplay2, this.pattern),
				new SlotDisplay.ItemSlotDisplay(Items.SMITHING_TABLE)
			)
		);
	}

	public static class Serializer implements RecipeSerializer<SmithingTrimRecipe> {
		private static final MapCodec<SmithingTrimRecipe> CODEC = RecordCodecBuilder.mapCodec(
			instance -> instance.group(
					Ingredient.CODEC.fieldOf("template").forGetter(smithingTrimRecipe -> smithingTrimRecipe.template),
					Ingredient.CODEC.fieldOf("base").forGetter(smithingTrimRecipe -> smithingTrimRecipe.base),
					Ingredient.CODEC.fieldOf("addition").forGetter(smithingTrimRecipe -> smithingTrimRecipe.addition),
					TrimPattern.CODEC.fieldOf("pattern").forGetter(smithingTrimRecipe -> smithingTrimRecipe.pattern)
				)
				.apply(instance, SmithingTrimRecipe::new)
		);
		public static final StreamCodec<RegistryFriendlyByteBuf, SmithingTrimRecipe> STREAM_CODEC = StreamCodec.composite(
			Ingredient.CONTENTS_STREAM_CODEC,
			smithingTrimRecipe -> smithingTrimRecipe.template,
			Ingredient.CONTENTS_STREAM_CODEC,
			smithingTrimRecipe -> smithingTrimRecipe.base,
			Ingredient.CONTENTS_STREAM_CODEC,
			smithingTrimRecipe -> smithingTrimRecipe.addition,
			TrimPattern.STREAM_CODEC,
			smithingTrimRecipe -> smithingTrimRecipe.pattern,
			SmithingTrimRecipe::new
		);

		@Override
		public MapCodec<SmithingTrimRecipe> codec() {
			return CODEC;
		}

		@Override
		public StreamCodec<RegistryFriendlyByteBuf, SmithingTrimRecipe> streamCodec() {
			return STREAM_CODEC;
		}
	}
}