package net.minecraft.client.renderer.entity.state;

import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.model.HumanoidModel.ArmPose;
import net.minecraft.client.renderer.item.ItemModelResolver;
import net.minecraft.client.renderer.item.ItemStackRenderState;
import net.minecraft.world.entity.HumanoidArm;
import net.minecraft.world.entity.LivingEntity;
import net.minecraft.world.item.ItemDisplayContext;

@Environment(EnvType.CLIENT)
public class ArmedEntityRenderState extends LivingEntityRenderState {
	public HumanoidArm mainArm = HumanoidArm.RIGHT;
	public ArmPose rightArmPose = ArmPose.EMPTY;
	public final ItemStackRenderState rightHandItem = new ItemStackRenderState();
	public ArmPose leftArmPose = ArmPose.EMPTY;
	public final ItemStackRenderState leftHandItem = new ItemStackRenderState();

	public ItemStackRenderState getMainHandItem() {
		return this.mainArm == HumanoidArm.RIGHT ? this.rightHandItem : this.leftHandItem;
	}

	public static void extractArmedEntityRenderState(LivingEntity entity, ArmedEntityRenderState reusedState, ItemModelResolver itemModelResolver) {
		reusedState.mainArm = entity.getMainArm();
		itemModelResolver.updateForLiving(reusedState.rightHandItem, entity.getItemHeldByArm(HumanoidArm.RIGHT), ItemDisplayContext.THIRD_PERSON_RIGHT_HAND, entity);
		itemModelResolver.updateForLiving(reusedState.leftHandItem, entity.getItemHeldByArm(HumanoidArm.LEFT), ItemDisplayContext.THIRD_PERSON_LEFT_HAND, entity);
	}
}