package net.minecraft.client.renderer.special;

import com.mojang.blaze3d.vertex.PoseStack;
import com.mojang.blaze3d.vertex.VertexConsumer;
import com.mojang.serialization.Codec;
import com.mojang.serialization.MapCodec;
import com.mojang.serialization.codecs.RecordCodecBuilder;
import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.model.ChestModel;
import net.minecraft.client.model.geom.EntityModelSet;
import net.minecraft.client.model.geom.ModelLayers;
import net.minecraft.client.renderer.MultiBufferSource;
import net.minecraft.client.renderer.RenderType;
import net.minecraft.client.renderer.Sheets;
import net.minecraft.client.resources.model.Material;
import net.minecraft.resources.ResourceLocation;
import net.minecraft.world.item.ItemDisplayContext;

@Environment(EnvType.CLIENT)
public class ChestSpecialRenderer implements NoDataSpecialModelRenderer {
	public static final ResourceLocation GIFT_CHEST_TEXTURE = ResourceLocation.withDefaultNamespace("christmas");
	public static final ResourceLocation NORMAL_CHEST_TEXTURE = ResourceLocation.withDefaultNamespace("normal");
	public static final ResourceLocation TRAPPED_CHEST_TEXTURE = ResourceLocation.withDefaultNamespace("trapped");
	public static final ResourceLocation ENDER_CHEST_TEXTURE = ResourceLocation.withDefaultNamespace("ender");
	private final ChestModel model;
	private final Material material;
	private final float openness;

	public ChestSpecialRenderer(ChestModel model, Material material, float openness) {
		this.model = model;
		this.material = material;
		this.openness = openness;
	}

	@Override
	public void render(
		ItemDisplayContext displayContext, PoseStack poseStack, MultiBufferSource bufferSource, int packedLight, int packedOverlay, boolean hasFoilType
	) {
		VertexConsumer vertexConsumer = this.material.buffer(bufferSource, RenderType::entitySolid);
		this.model.setupAnim(this.openness);
		this.model.renderToBuffer(poseStack, vertexConsumer, packedLight, packedOverlay);
	}

	@Environment(EnvType.CLIENT)
	public record Unbaked(ResourceLocation texture, float openness) implements net.minecraft.client.renderer.special.SpecialModelRenderer.Unbaked {
		public static final MapCodec<ChestSpecialRenderer.Unbaked> MAP_CODEC = RecordCodecBuilder.mapCodec(
			instance -> instance.group(
					ResourceLocation.CODEC.fieldOf("texture").forGetter(ChestSpecialRenderer.Unbaked::texture),
					Codec.FLOAT.optionalFieldOf("openness", 0.0F).forGetter(ChestSpecialRenderer.Unbaked::openness)
				)
				.apply(instance, ChestSpecialRenderer.Unbaked::new)
		);

		public Unbaked(ResourceLocation texture) {
			this(texture, 0.0F);
		}

		@Override
		public MapCodec<ChestSpecialRenderer.Unbaked> type() {
			return MAP_CODEC;
		}

		@Override
		public SpecialModelRenderer<?> bake(EntityModelSet modelSet) {
			ChestModel chestModel = new ChestModel(modelSet.bakeLayer(ModelLayers.CHEST));
			Material material = Sheets.CHEST_MAPPER.apply(this.texture);
			return new ChestSpecialRenderer(chestModel, material, this.openness);
		}
	}
}