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 breezeRenderState, PoseStack poseStack, MultiBufferSource multiBufferSource, int i) {
		BreezeModel breezeModel = this.getModel();
		enable(breezeModel, breezeModel.head(), breezeModel.rods());
		super.render(breezeRenderState, poseStack, multiBufferSource, i);
	}

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

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

	public void extractRenderState(Breeze breeze, BreezeRenderState breezeRenderState, float f) {
		super.extractRenderState(breeze, breezeRenderState, f);
		breezeRenderState.idle.copyFrom(breeze.idle);
		breezeRenderState.shoot.copyFrom(breeze.shoot);
		breezeRenderState.slide.copyFrom(breeze.slide);
		breezeRenderState.slideBack.copyFrom(breeze.slideBack);
		breezeRenderState.inhale.copyFrom(breeze.inhale);
		breezeRenderState.longJump.copyFrom(breeze.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;
	}
}