package net.minecraft.world.item.component;

import com.mojang.serialization.Codec;
import com.mojang.serialization.codecs.RecordCodecBuilder;
import io.netty.buffer.ByteBuf;
import it.unimi.dsi.fastutil.ints.IntArrayList;
import it.unimi.dsi.fastutil.ints.IntList;
import java.util.ArrayList;
import java.util.function.Consumer;
import java.util.function.IntFunction;
import net.minecraft.ChatFormatting;
import net.minecraft.core.component.DataComponentGetter;
import net.minecraft.network.chat.CommonComponents;
import net.minecraft.network.chat.Component;
import net.minecraft.network.chat.MutableComponent;
import net.minecraft.network.codec.ByteBufCodecs;
import net.minecraft.network.codec.StreamCodec;
import net.minecraft.util.ByIdMap;
import net.minecraft.util.StringRepresentable;
import net.minecraft.util.ByIdMap.OutOfBoundsStrategy;
import net.minecraft.world.item.DyeColor;
import net.minecraft.world.item.Item;
import net.minecraft.world.item.TooltipFlag;

public record FireworkExplosion(FireworkExplosion.Shape shape, IntList colors, IntList fadeColors, boolean hasTrail, boolean hasTwinkle)
	implements TooltipProvider {
	public static final FireworkExplosion DEFAULT = new FireworkExplosion(FireworkExplosion.Shape.SMALL_BALL, IntList.of(), IntList.of(), false, false);
	public static final Codec<IntList> COLOR_LIST_CODEC = Codec.INT.listOf().xmap(IntArrayList::new, ArrayList::new);
	public static final Codec<FireworkExplosion> CODEC = RecordCodecBuilder.create(
		instance -> instance.group(
				FireworkExplosion.Shape.CODEC.fieldOf("shape").forGetter(FireworkExplosion::shape),
				COLOR_LIST_CODEC.optionalFieldOf("colors", IntList.of()).forGetter(FireworkExplosion::colors),
				COLOR_LIST_CODEC.optionalFieldOf("fade_colors", IntList.of()).forGetter(FireworkExplosion::fadeColors),
				Codec.BOOL.optionalFieldOf("has_trail", false).forGetter(FireworkExplosion::hasTrail),
				Codec.BOOL.optionalFieldOf("has_twinkle", false).forGetter(FireworkExplosion::hasTwinkle)
			)
			.apply(instance, FireworkExplosion::new)
	);
	private static final StreamCodec<ByteBuf, IntList> COLOR_LIST_STREAM_CODEC = ByteBufCodecs.INT
		.apply(ByteBufCodecs.list())
		.map(IntArrayList::new, ArrayList::new);
	public static final StreamCodec<ByteBuf, FireworkExplosion> STREAM_CODEC = StreamCodec.composite(
		FireworkExplosion.Shape.STREAM_CODEC,
		FireworkExplosion::shape,
		COLOR_LIST_STREAM_CODEC,
		FireworkExplosion::colors,
		COLOR_LIST_STREAM_CODEC,
		FireworkExplosion::fadeColors,
		ByteBufCodecs.BOOL,
		FireworkExplosion::hasTrail,
		ByteBufCodecs.BOOL,
		FireworkExplosion::hasTwinkle,
		FireworkExplosion::new
	);
	private static final Component CUSTOM_COLOR_NAME = Component.translatable("item.minecraft.firework_star.custom_color");

	@Override
	public void addToTooltip(Item.TooltipContext context, Consumer<Component> tooltipAdder, TooltipFlag flag, DataComponentGetter componentGetter) {
		tooltipAdder.accept(this.shape.getName().withStyle(ChatFormatting.GRAY));
		this.addAdditionalTooltip(tooltipAdder);
	}

	public void addAdditionalTooltip(Consumer<Component> tooltipAdder) {
		if (!this.colors.isEmpty()) {
			tooltipAdder.accept(appendColors(Component.empty().withStyle(ChatFormatting.GRAY), this.colors));
		}

		if (!this.fadeColors.isEmpty()) {
			tooltipAdder.accept(
				appendColors(Component.translatable("item.minecraft.firework_star.fade_to").append(CommonComponents.SPACE).withStyle(ChatFormatting.GRAY), this.fadeColors)
			);
		}

		if (this.hasTrail) {
			tooltipAdder.accept(Component.translatable("item.minecraft.firework_star.trail").withStyle(ChatFormatting.GRAY));
		}

		if (this.hasTwinkle) {
			tooltipAdder.accept(Component.translatable("item.minecraft.firework_star.flicker").withStyle(ChatFormatting.GRAY));
		}
	}

	private static Component appendColors(MutableComponent component, IntList colors) {
		for (int i = 0; i < colors.size(); i++) {
			if (i > 0) {
				component.append(", ");
			}

			component.append(getColorName(colors.getInt(i)));
		}

		return component;
	}

	private static Component getColorName(int id) {
		DyeColor dyeColor = DyeColor.byFireworkColor(id);
		return (Component)(dyeColor == null ? CUSTOM_COLOR_NAME : Component.translatable("item.minecraft.firework_star." + dyeColor.getName()));
	}

	public FireworkExplosion withFadeColors(IntList fadeColors) {
		return new FireworkExplosion(this.shape, this.colors, new IntArrayList(fadeColors), this.hasTrail, this.hasTwinkle);
	}

	public static enum Shape implements StringRepresentable {
		SMALL_BALL(0, "small_ball"),
		LARGE_BALL(1, "large_ball"),
		STAR(2, "star"),
		CREEPER(3, "creeper"),
		BURST(4, "burst");

		private static final IntFunction<FireworkExplosion.Shape> BY_ID = ByIdMap.continuous(FireworkExplosion.Shape::getId, values(), OutOfBoundsStrategy.ZERO);
		public static final StreamCodec<ByteBuf, FireworkExplosion.Shape> STREAM_CODEC = ByteBufCodecs.idMapper(BY_ID, FireworkExplosion.Shape::getId);
		public static final Codec<FireworkExplosion.Shape> CODEC = StringRepresentable.fromValues(FireworkExplosion.Shape::values);
		private final int id;
		private final String name;

		private Shape(final int id, final String name) {
			this.id = id;
			this.name = name;
		}

		public MutableComponent getName() {
			return Component.translatable("item.minecraft.firework_star.shape." + this.name);
		}

		public int getId() {
			return this.id;
		}

		public static FireworkExplosion.Shape byId(int id) {
			return (FireworkExplosion.Shape)BY_ID.apply(id);
		}

		@Override
		public String getSerializedName() {
			return this.name;
		}
	}
}