package net.minecraft.client.renderer.entity;

import com.google.common.collect.Maps;
import java.util.Map;
import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.model.EquineSaddleModel;
import net.minecraft.client.model.HorseModel;
import net.minecraft.client.model.geom.ModelLayers;
import net.minecraft.client.renderer.entity.EntityRendererProvider.Context;
import net.minecraft.client.renderer.entity.layers.HorseMarkingLayer;
import net.minecraft.client.renderer.entity.layers.SimpleEquipmentLayer;
import net.minecraft.client.renderer.entity.state.HorseRenderState;
import net.minecraft.client.resources.model.EquipmentClientInfo;
import net.minecraft.resources.ResourceLocation;
import net.minecraft.world.entity.animal.horse.Horse;
import net.minecraft.world.entity.animal.horse.Variant;

@Environment(EnvType.CLIENT)
public final class HorseRenderer extends AbstractHorseRenderer<Horse, HorseRenderState, HorseModel> {
	private static final Map<Variant, ResourceLocation> LOCATION_BY_VARIANT = Maps.newEnumMap(
		Map.of(
			Variant.WHITE,
			ResourceLocation.withDefaultNamespace("textures/entity/horse/horse_white.png"),
			Variant.CREAMY,
			ResourceLocation.withDefaultNamespace("textures/entity/horse/horse_creamy.png"),
			Variant.CHESTNUT,
			ResourceLocation.withDefaultNamespace("textures/entity/horse/horse_chestnut.png"),
			Variant.BROWN,
			ResourceLocation.withDefaultNamespace("textures/entity/horse/horse_brown.png"),
			Variant.BLACK,
			ResourceLocation.withDefaultNamespace("textures/entity/horse/horse_black.png"),
			Variant.GRAY,
			ResourceLocation.withDefaultNamespace("textures/entity/horse/horse_gray.png"),
			Variant.DARK_BROWN,
			ResourceLocation.withDefaultNamespace("textures/entity/horse/horse_darkbrown.png")
		)
	);

	public HorseRenderer(Context context) {
		super(context, new HorseModel(context.bakeLayer(ModelLayers.HORSE)), new HorseModel(context.bakeLayer(ModelLayers.HORSE_BABY)));
		this.addLayer(new HorseMarkingLayer(this));
		this.addLayer(
			new SimpleEquipmentLayer<>(
				this,
				context.getEquipmentRenderer(),
				EquipmentClientInfo.LayerType.HORSE_BODY,
				horseRenderState -> horseRenderState.bodyArmorItem,
				new HorseModel(context.bakeLayer(ModelLayers.HORSE_ARMOR)),
				new HorseModel(context.bakeLayer(ModelLayers.HORSE_BABY_ARMOR))
			)
		);
		this.addLayer(
			new SimpleEquipmentLayer<>(
				this,
				context.getEquipmentRenderer(),
				EquipmentClientInfo.LayerType.HORSE_SADDLE,
				horseRenderState -> horseRenderState.saddle,
				new EquineSaddleModel(context.bakeLayer(ModelLayers.HORSE_SADDLE)),
				new EquineSaddleModel(context.bakeLayer(ModelLayers.HORSE_BABY_SADDLE))
			)
		);
	}

	public ResourceLocation getTextureLocation(HorseRenderState horseRenderState) {
		return (ResourceLocation)LOCATION_BY_VARIANT.get(horseRenderState.variant);
	}

	public HorseRenderState createRenderState() {
		return new HorseRenderState();
	}

	public void extractRenderState(Horse horse, HorseRenderState horseRenderState, float f) {
		super.extractRenderState(horse, horseRenderState, f);
		horseRenderState.variant = horse.getVariant();
		horseRenderState.markings = horse.getMarkings();
		horseRenderState.bodyArmorItem = horse.getBodyArmorItem().copy();
	}
}