package net.minecraft.world.item.component;

import com.google.common.collect.Lists;
import com.mojang.serialization.Codec;
import java.util.List;
import java.util.function.Consumer;
import net.minecraft.ChatFormatting;
import net.minecraft.core.component.DataComponentGetter;
import net.minecraft.network.RegistryFriendlyByteBuf;
import net.minecraft.network.chat.Component;
import net.minecraft.network.chat.ComponentSerialization;
import net.minecraft.network.chat.ComponentUtils;
import net.minecraft.network.chat.Style;
import net.minecraft.network.codec.ByteBufCodecs;
import net.minecraft.network.codec.StreamCodec;
import net.minecraft.util.Util;
import net.minecraft.world.item.Item;
import net.minecraft.world.item.TooltipFlag;

public record ItemLore(List<Component> lines, List<Component> styledLines) implements TooltipProvider {
	public static final ItemLore EMPTY = new ItemLore(List.of());
	public static final int MAX_LINES = 256;
	private static final Style LORE_STYLE = Style.EMPTY.withColor(ChatFormatting.DARK_PURPLE).withItalic(true);
	public static final Codec<ItemLore> CODEC = ComponentSerialization.CODEC.sizeLimitedListOf(256).xmap(ItemLore::new, ItemLore::lines);
	public static final StreamCodec<RegistryFriendlyByteBuf, ItemLore> STREAM_CODEC = ComponentSerialization.STREAM_CODEC
		.apply(ByteBufCodecs.list(256))
		.map(ItemLore::new, ItemLore::lines);

	public ItemLore(final List<Component> lines) {
		this(lines, Lists.transform(lines, component -> ComponentUtils.mergeStyles(component, LORE_STYLE)));
	}

	public ItemLore(List<Component> lines, List<Component> styledLines) {
		if (lines.size() > 256) {
			throw new IllegalArgumentException("Got " + lines.size() + " lines, but maximum is 256");
		} else {
			this.lines = lines;
			this.styledLines = styledLines;
		}
	}

	public ItemLore withLineAdded(final Component component) {
		return new ItemLore(Util.copyAndAdd(this.lines, component));
	}

	@Override
	public void addToTooltip(final Item.TooltipContext context, final Consumer<Component> consumer, final TooltipFlag flag, final DataComponentGetter components) {
		this.styledLines.forEach(consumer);
	}
}