package net.minecraft.client.renderer.entity;

import com.mojang.blaze3d.vertex.PoseStack;
import com.mojang.math.Axis;
import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.renderer.MultiBufferSource;
import net.minecraft.client.renderer.entity.state.FireworkRocketRenderState;
import net.minecraft.client.renderer.texture.OverlayTexture;
import net.minecraft.world.entity.projectile.FireworkRocketEntity;
import net.minecraft.world.item.ItemDisplayContext;
import net.minecraft.world.item.ItemStack;

@Environment(EnvType.CLIENT)
public class FireworkEntityRenderer extends EntityRenderer<FireworkRocketEntity, FireworkRocketRenderState> {
	private final ItemRenderer itemRenderer;

	public FireworkEntityRenderer(EntityRendererProvider.Context context) {
		super(context);
		this.itemRenderer = context.getItemRenderer();
	}

	public void render(FireworkRocketRenderState fireworkRocketRenderState, PoseStack poseStack, MultiBufferSource multiBufferSource, int i) {
		poseStack.pushPose();
		poseStack.mulPose(this.entityRenderDispatcher.cameraOrientation());
		if (fireworkRocketRenderState.isShotAtAngle) {
			poseStack.mulPose(Axis.ZP.rotationDegrees(180.0F));
			poseStack.mulPose(Axis.YP.rotationDegrees(180.0F));
			poseStack.mulPose(Axis.XP.rotationDegrees(90.0F));
		}

		this.itemRenderer
			.render(
				fireworkRocketRenderState.item,
				ItemDisplayContext.GROUND,
				false,
				poseStack,
				multiBufferSource,
				i,
				OverlayTexture.NO_OVERLAY,
				fireworkRocketRenderState.itemModel
			);
		poseStack.popPose();
		super.render(fireworkRocketRenderState, poseStack, multiBufferSource, i);
	}

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

	public void extractRenderState(FireworkRocketEntity fireworkRocketEntity, FireworkRocketRenderState fireworkRocketRenderState, float f) {
		super.extractRenderState(fireworkRocketEntity, fireworkRocketRenderState, f);
		fireworkRocketRenderState.isShotAtAngle = fireworkRocketEntity.isShotAtAngle();
		ItemStack itemStack = fireworkRocketEntity.getItem();
		fireworkRocketRenderState.item = itemStack.copy();
		fireworkRocketRenderState.itemModel = !itemStack.isEmpty()
			? this.itemRenderer.getModel(itemStack, fireworkRocketEntity.level(), null, fireworkRocketEntity.getId())
			: null;
	}
}