package net.minecraft.client.model;

import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.model.geom.ModelPart;
import net.minecraft.client.model.geom.PartNames;
import net.minecraft.client.model.geom.PartPose;
import net.minecraft.client.model.geom.builders.CubeDeformation;
import net.minecraft.client.model.geom.builders.CubeListBuilder;
import net.minecraft.client.model.geom.builders.LayerDefinition;
import net.minecraft.client.model.geom.builders.MeshDefinition;
import net.minecraft.client.model.geom.builders.MeshTransformer;
import net.minecraft.client.model.geom.builders.PartDefinition;
import net.minecraft.client.renderer.entity.state.HappyGhastRenderState;

@Environment(EnvType.CLIENT)
public class HappyGhastModel extends EntityModel<HappyGhastRenderState> {
	public static final MeshTransformer BABY_TRANSFORMER = MeshTransformer.scaling(0.2375F);
	private static final float BODY_SQUEEZE = 0.9375F;
	private final ModelPart[] tentacles = new ModelPart[9];
	private final ModelPart body;

	public HappyGhastModel(ModelPart root) {
		super(root);
		this.body = root.getChild("body");

		for (int i = 0; i < this.tentacles.length; i++) {
			this.tentacles[i] = this.body.getChild(PartNames.tentacle(i));
		}
	}

	public static LayerDefinition createBodyLayer(boolean baby, CubeDeformation cubeDeformation) {
		MeshDefinition meshDefinition = new MeshDefinition();
		PartDefinition partDefinition = meshDefinition.getRoot();
		PartDefinition partDefinition2 = partDefinition.addOrReplaceChild(
			"body", CubeListBuilder.create().texOffs(0, 0).addBox(-8.0F, -8.0F, -8.0F, 16.0F, 16.0F, 16.0F, cubeDeformation), PartPose.offset(0.0F, 16.0F, 0.0F)
		);
		if (baby) {
			partDefinition2.addOrReplaceChild(
				"inner_body",
				CubeListBuilder.create().texOffs(0, 32).addBox(-8.0F, -16.0F, -8.0F, 16.0F, 16.0F, 16.0F, cubeDeformation.extend(-0.5F)),
				PartPose.offset(0.0F, 8.0F, 0.0F)
			);
		}

		partDefinition2.addOrReplaceChild(
			PartNames.tentacle(0),
			CubeListBuilder.create().texOffs(0, 0).addBox(-1.0F, 0.0F, -1.0F, 2.0F, 5.0F, 2.0F, cubeDeformation),
			PartPose.offset(-3.75F, 7.0F, -5.0F)
		);
		partDefinition2.addOrReplaceChild(
			PartNames.tentacle(1),
			CubeListBuilder.create().texOffs(0, 0).addBox(-1.0F, 0.0F, -1.0F, 2.0F, 7.0F, 2.0F, cubeDeformation),
			PartPose.offset(1.25F, 7.0F, -5.0F)
		);
		partDefinition2.addOrReplaceChild(
			PartNames.tentacle(2),
			CubeListBuilder.create().texOffs(0, 0).addBox(-1.0F, 0.0F, -1.0F, 2.0F, 4.0F, 2.0F, cubeDeformation),
			PartPose.offset(6.25F, 7.0F, -5.0F)
		);
		partDefinition2.addOrReplaceChild(
			PartNames.tentacle(3),
			CubeListBuilder.create().texOffs(0, 0).addBox(-1.0F, 0.0F, -1.0F, 2.0F, 5.0F, 2.0F, cubeDeformation),
			PartPose.offset(-6.25F, 7.0F, 0.0F)
		);
		partDefinition2.addOrReplaceChild(
			PartNames.tentacle(4),
			CubeListBuilder.create().texOffs(0, 0).addBox(-1.0F, 0.0F, -1.0F, 2.0F, 5.0F, 2.0F, cubeDeformation),
			PartPose.offset(-1.25F, 7.0F, 0.0F)
		);
		partDefinition2.addOrReplaceChild(
			PartNames.tentacle(5),
			CubeListBuilder.create().texOffs(0, 0).addBox(-1.0F, 0.0F, -1.0F, 2.0F, 7.0F, 2.0F, cubeDeformation),
			PartPose.offset(3.75F, 7.0F, 0.0F)
		);
		partDefinition2.addOrReplaceChild(
			PartNames.tentacle(6),
			CubeListBuilder.create().texOffs(0, 0).addBox(-1.0F, 0.0F, -1.0F, 2.0F, 8.0F, 2.0F, cubeDeformation),
			PartPose.offset(-3.75F, 7.0F, 5.0F)
		);
		partDefinition2.addOrReplaceChild(
			PartNames.tentacle(7),
			CubeListBuilder.create().texOffs(0, 0).addBox(-1.0F, 0.0F, -1.0F, 2.0F, 8.0F, 2.0F, cubeDeformation),
			PartPose.offset(1.25F, 7.0F, 5.0F)
		);
		partDefinition2.addOrReplaceChild(
			PartNames.tentacle(8),
			CubeListBuilder.create().texOffs(0, 0).addBox(-1.0F, 0.0F, -1.0F, 2.0F, 5.0F, 2.0F, cubeDeformation),
			PartPose.offset(6.25F, 7.0F, 5.0F)
		);
		return LayerDefinition.create(meshDefinition, 64, 64).apply(MeshTransformer.scaling(4.0F));
	}

	public void setupAnim(HappyGhastRenderState renderState) {
		super.setupAnim(renderState);
		if (!renderState.bodyItem.isEmpty()) {
			this.body.xScale = 0.9375F;
			this.body.yScale = 0.9375F;
			this.body.zScale = 0.9375F;
		}

		GhastModel.animateTentacles(renderState, this.tentacles);
	}
}