package net.minecraft.commands.arguments.item;

import com.mojang.brigadier.exceptions.CommandSyntaxException;
import com.mojang.brigadier.exceptions.Dynamic2CommandExceptionType;
import com.mojang.serialization.DynamicOps;
import java.util.Optional;
import java.util.stream.Collectors;
import java.util.stream.Stream;
import net.minecraft.core.Holder;
import net.minecraft.core.HolderLookup;
import net.minecraft.core.component.DataComponentPatch;
import net.minecraft.core.component.DataComponentType;
import net.minecraft.core.component.TypedDataComponent;
import net.minecraft.core.registries.BuiltInRegistries;
import net.minecraft.nbt.NbtOps;
import net.minecraft.nbt.Tag;
import net.minecraft.network.chat.Component;
import net.minecraft.resources.ResourceKey;
import net.minecraft.resources.ResourceLocation;
import net.minecraft.world.item.Item;
import net.minecraft.world.item.ItemStack;

public class ItemInput {
	private static final Dynamic2CommandExceptionType ERROR_STACK_TOO_BIG = new Dynamic2CommandExceptionType(
		(object, object2) -> Component.translatableEscape("arguments.item.overstacked", object, object2)
	);
	private final Holder<Item> item;
	private final DataComponentPatch components;

	public ItemInput(Holder<Item> item, DataComponentPatch components) {
		this.item = item;
		this.components = components;
	}

	public Item getItem() {
		return this.item.value();
	}

	public ItemStack createItemStack(int count, boolean allowOversizedStacks) throws CommandSyntaxException {
		ItemStack itemStack = new ItemStack(this.item, count);
		itemStack.applyComponents(this.components);
		if (allowOversizedStacks && count > itemStack.getMaxStackSize()) {
			throw ERROR_STACK_TOO_BIG.create(this.getItemName(), itemStack.getMaxStackSize());
		} else {
			return itemStack;
		}
	}

	public String serialize(HolderLookup.Provider levelRegistry) {
		StringBuilder stringBuilder = new StringBuilder(this.getItemName());
		String string = this.serializeComponents(levelRegistry);
		if (!string.isEmpty()) {
			stringBuilder.append('[');
			stringBuilder.append(string);
			stringBuilder.append(']');
		}

		return stringBuilder.toString();
	}

	private String serializeComponents(HolderLookup.Provider levelRegistries) {
		DynamicOps<Tag> dynamicOps = levelRegistries.createSerializationContext(NbtOps.INSTANCE);
		return (String)this.components.entrySet().stream().flatMap(entry -> {
			DataComponentType<?> dataComponentType = (DataComponentType<?>)entry.getKey();
			ResourceLocation resourceLocation = BuiltInRegistries.DATA_COMPONENT_TYPE.getKey(dataComponentType);
			if (resourceLocation == null) {
				return Stream.empty();
			} else {
				Optional<?> optional = (Optional<?>)entry.getValue();
				if (optional.isPresent()) {
					TypedDataComponent<?> typedDataComponent = TypedDataComponent.createUnchecked(dataComponentType, optional.get());
					return typedDataComponent.encodeValue(dynamicOps).result().stream().map(tag -> resourceLocation.toString() + "=" + tag);
				} else {
					return Stream.of("!" + resourceLocation.toString());
				}
			}
		}).collect(Collectors.joining(String.valueOf(',')));
	}

	private String getItemName() {
		return this.item.unwrapKey().map(ResourceKey::location).orElseGet(() -> "unknown[" + this.item + "]").toString();
	}
}