package net.minecraft.client.renderer.entity;

import com.mojang.blaze3d.vertex.PoseStack;
import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.model.BreezeModel;
import net.minecraft.client.model.geom.ModelLayers;
import net.minecraft.client.model.geom.ModelPart;
import net.minecraft.client.renderer.MultiBufferSource;
import net.minecraft.client.renderer.entity.layers.BreezeEyesLayer;
import net.minecraft.client.renderer.entity.layers.BreezeWindLayer;
import net.minecraft.client.renderer.entity.state.BreezeRenderState;
import net.minecraft.resources.ResourceLocation;
import net.minecraft.world.entity.monster.breeze.Breeze;

@Environment(EnvType.CLIENT)
public class BreezeRenderer extends MobRenderer<Breeze, BreezeRenderState, BreezeModel> {
	private static final ResourceLocation TEXTURE_LOCATION = ResourceLocation.withDefaultNamespace("textures/entity/breeze/breeze.png");

	public BreezeRenderer(EntityRendererProvider.Context context) {
		super(context, new BreezeModel(context.bakeLayer(ModelLayers.BREEZE)), 0.5F);
		this.addLayer(new BreezeWindLayer(context, this));
		this.addLayer(new BreezeEyesLayer(this));
	}

	public void render(BreezeRenderState renderState, PoseStack poseStack, MultiBufferSource bufferSource, int packedLight) {
		BreezeModel breezeModel = this.getModel();
		enable(breezeModel, breezeModel.head(), breezeModel.rods());
		super.render(renderState, poseStack, bufferSource, packedLight);
	}

	public ResourceLocation getTextureLocation(BreezeRenderState renderState) {
		return TEXTURE_LOCATION;
	}

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

	public void extractRenderState(Breeze entity, BreezeRenderState reusedState, float partialTick) {
		super.extractRenderState(entity, reusedState, partialTick);
		reusedState.idle.copyFrom(entity.idle);
		reusedState.shoot.copyFrom(entity.shoot);
		reusedState.slide.copyFrom(entity.slide);
		reusedState.slideBack.copyFrom(entity.slideBack);
		reusedState.inhale.copyFrom(entity.inhale);
		reusedState.longJump.copyFrom(entity.longJump);
	}

	public static BreezeModel enable(BreezeModel model, ModelPart... parts) {
		model.head().visible = false;
		model.eyes().visible = false;
		model.rods().visible = false;
		model.wind().visible = false;

		for (ModelPart modelPart : parts) {
			modelPart.visible = true;
		}

		return model;
	}
}