package net.minecraft.client.renderer.special;

import com.mojang.blaze3d.vertex.PoseStack;
import com.mojang.serialization.MapCodec;
import com.mojang.serialization.codecs.RecordCodecBuilder;
import java.util.Optional;
import java.util.function.Consumer;
import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.model.Model;
import net.minecraft.client.renderer.Sheets;
import net.minecraft.client.renderer.SubmitNodeCollector;
import net.minecraft.client.renderer.blockentity.HangingSignRenderer;
import net.minecraft.client.resources.model.sprite.SpriteGetter;
import net.minecraft.client.resources.model.sprite.SpriteId;
import net.minecraft.resources.Identifier;
import net.minecraft.world.level.block.HangingSignBlock;
import net.minecraft.world.level.block.state.properties.WoodType;
import org.joml.Vector3fc;

@Environment(EnvType.CLIENT)
public class HangingSignSpecialRenderer implements NoDataSpecialModelRenderer {
	private final SpriteGetter sprites;
	private final Model.Simple model;
	private final SpriteId sprite;

	public HangingSignSpecialRenderer(final SpriteGetter sprites, final Model.Simple model, final SpriteId sprite) {
		this.sprites = sprites;
		this.model = model;
		this.sprite = sprite;
	}

	@Override
	public void submit(
		final PoseStack poseStack,
		final SubmitNodeCollector submitNodeCollector,
		final int lightCoords,
		final int overlayCoords,
		final boolean hasFoil,
		final int outlineColor
	) {
		HangingSignRenderer.submitSpecial(this.sprites, poseStack, submitNodeCollector, lightCoords, overlayCoords, this.model, this.sprite);
	}

	@Override
	public void getExtents(final Consumer<Vector3fc> output) {
		PoseStack poseStack = new PoseStack();
		this.model.root().getExtentsForGui(poseStack, output);
	}

	@Environment(EnvType.CLIENT)
	public record Unbaked(WoodType woodType, HangingSignBlock.Attachment attachment, Optional<Identifier> texture) implements NoDataSpecialModelRenderer.Unbaked {
		public static final MapCodec<HangingSignSpecialRenderer.Unbaked> MAP_CODEC = RecordCodecBuilder.mapCodec(
			i -> i.group(
					WoodType.CODEC.fieldOf("wood_type").forGetter(HangingSignSpecialRenderer.Unbaked::woodType),
					HangingSignBlock.Attachment.CODEC
						.optionalFieldOf("attachment", HangingSignBlock.Attachment.CEILING_MIDDLE)
						.forGetter(HangingSignSpecialRenderer.Unbaked::attachment),
					Identifier.CODEC.optionalFieldOf("texture").forGetter(HangingSignSpecialRenderer.Unbaked::texture)
				)
				.apply(i, HangingSignSpecialRenderer.Unbaked::new)
		);

		public Unbaked(final WoodType woodType, final HangingSignBlock.Attachment attachment) {
			this(woodType, attachment, Optional.empty());
		}

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

		public HangingSignSpecialRenderer bake(final SpecialModelRenderer.BakingContext context) {
			Model.Simple model = HangingSignRenderer.createSignModel(context.entityModelSet(), this.woodType, this.attachment);
			SpriteId sprite = (SpriteId)this.texture.map(Sheets.HANGING_SIGN_MAPPER::apply).orElseGet(() -> Sheets.getHangingSignSprite(this.woodType));
			return new HangingSignSpecialRenderer(context.sprites(), model, sprite);
		}
	}
}