package net.minecraft.client.renderer.item.properties.select;

import com.google.common.collect.HashMultiset;
import com.google.common.collect.Multiset;
import com.mojang.serialization.Codec;
import com.mojang.serialization.DataResult;
import com.mojang.serialization.MapCodec;
import com.mojang.serialization.codecs.RecordCodecBuilder;
import java.util.List;
import java.util.stream.Collectors;
import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.multiplayer.ClientLevel;
import net.minecraft.client.renderer.item.SelectItemModel;
import net.minecraft.world.entity.LivingEntity;
import net.minecraft.world.item.ItemDisplayContext;
import net.minecraft.world.item.ItemStack;
import org.jetbrains.annotations.Nullable;

@Environment(EnvType.CLIENT)
public interface SelectItemModelProperty<T> {
	@Nullable
	T get(ItemStack stack, @Nullable ClientLevel level, @Nullable LivingEntity entity, int seed, ItemDisplayContext displayContext);

	SelectItemModelProperty.Type<? extends SelectItemModelProperty<T>, T> type();

	@Environment(EnvType.CLIENT)
	public record Type<P extends SelectItemModelProperty<T>, T>(MapCodec<SelectItemModel.UnbakedSwitch<P, T>> switchCodec) {
		public static <P extends SelectItemModelProperty<T>, T> SelectItemModelProperty.Type<P, T> create(MapCodec<P> mapCodec, Codec<T> codec) {
			Codec<List<SelectItemModel.SwitchCase<T>>> codec2 = SelectItemModel.SwitchCase.codec(codec)
				.listOf()
				.validate(
					list -> {
						if (list.isEmpty()) {
							return DataResult.error(() -> "Empty case list");
						} else {
							Multiset<T> multiset = HashMultiset.create();

							for (SelectItemModel.SwitchCase<T> switchCase : list) {
								multiset.addAll(switchCase.values());
							}

							return multiset.size() != multiset.entrySet().size()
								? DataResult.error(
									() -> "Duplicate case conditions: "
										+ (String)multiset.entrySet()
											.stream()
											.filter(entry -> entry.getCount() > 1)
											.map(entry -> entry.getElement().toString())
											.collect(Collectors.joining(", "))
								)
								: DataResult.success(list);
						}
					}
				);
			MapCodec<SelectItemModel.UnbakedSwitch<P, T>> mapCodec2 = RecordCodecBuilder.mapCodec(
				instance -> instance.group(
						mapCodec.forGetter(SelectItemModel.UnbakedSwitch::property), codec2.fieldOf("cases").forGetter(SelectItemModel.UnbakedSwitch::cases)
					)
					.apply(instance, SelectItemModel.UnbakedSwitch::new)
			);
			return new SelectItemModelProperty.Type<>(mapCodec2);
		}
	}
}