package net.minecraft.world.item.component;

import com.mojang.serialization.Codec;
import com.mojang.serialization.codecs.RecordCodecBuilder;
import it.unimi.dsi.fastutil.objects.ReferenceLinkedOpenHashSet;
import it.unimi.dsi.fastutil.objects.ReferenceSortedSets;
import java.util.List;
import java.util.SequencedSet;
import net.minecraft.core.component.DataComponentType;
import net.minecraft.network.RegistryFriendlyByteBuf;
import net.minecraft.network.codec.ByteBufCodecs;
import net.minecraft.network.codec.StreamCodec;

public record TooltipDisplay(boolean hideTooltip, SequencedSet<DataComponentType<?>> hiddenComponents) {
	private static final Codec<SequencedSet<DataComponentType<?>>> COMPONENT_SET_CODEC = DataComponentType.CODEC
		.listOf()
		.xmap(ReferenceLinkedOpenHashSet::new, List::copyOf);
	public static final Codec<TooltipDisplay> CODEC = RecordCodecBuilder.create(
		instance -> instance.group(
				Codec.BOOL.optionalFieldOf("hide_tooltip", false).forGetter(TooltipDisplay::hideTooltip),
				COMPONENT_SET_CODEC.optionalFieldOf("hidden_components", ReferenceSortedSets.<DataComponentType<?>>emptySet()).forGetter(TooltipDisplay::hiddenComponents)
			)
			.apply(instance, TooltipDisplay::new)
	);
	public static final StreamCodec<RegistryFriendlyByteBuf, TooltipDisplay> STREAM_CODEC = StreamCodec.composite(
		ByteBufCodecs.BOOL,
		TooltipDisplay::hideTooltip,
		DataComponentType.STREAM_CODEC.apply(ByteBufCodecs.collection(ReferenceLinkedOpenHashSet::new)),
		TooltipDisplay::hiddenComponents,
		TooltipDisplay::new
	);
	public static final TooltipDisplay DEFAULT = new TooltipDisplay(false, ReferenceSortedSets.<DataComponentType<?>>emptySet());

	public TooltipDisplay withHidden(DataComponentType<?> component, boolean isHidden) {
		if (this.hiddenComponents.contains(component) == isHidden) {
			return this;
		} else {
			SequencedSet<DataComponentType<?>> sequencedSet = new ReferenceLinkedOpenHashSet<DataComponentType<?>>(this.hiddenComponents);
			if (isHidden) {
				sequencedSet.add(component);
			} else {
				sequencedSet.remove(component);
			}

			return new TooltipDisplay(this.hideTooltip, sequencedSet);
		}
	}

	public boolean shows(DataComponentType<?> component) {
		return !this.hideTooltip && !this.hiddenComponents.contains(component);
	}
}