Minestom/src/main/java/net/minestom/server/entity/type/decoration/EntityArmorStand.java

package net.minestom.server.entity.type.decoration;

import net.minestom.server.entity.*;
import net.minestom.server.event.EventDispatcher;
import net.minestom.server.event.item.EntityEquipEvent;
import net.minestom.server.inventory.EquipmentHandler;
import net.minestom.server.item.ItemStack;
import net.minestom.server.utils.Position;
import net.minestom.server.utils.Vector;
import net.minestom.server.utils.binary.BitmaskUtil;
import org.jetbrains.annotations.NotNull;

/**
 * @deprecated Use {@link net.minestom.server.entity.metadata.other.ArmorStandMeta} instead.
 */
@Deprecated
public class EntityArmorStand extends ObjectEntity implements EquipmentHandler {

    // Equipments
    private ItemStack mainHandItem;
    private ItemStack offHandItem;

    private ItemStack helmet;
    private ItemStack chestplate;
    private ItemStack leggings;
    private ItemStack boots;

    public EntityArmorStand(Position spawnPosition) {
        super(EntityType.ARMOR_STAND, spawnPosition);

        // Refresh BoundingBox
        setSmall(false);

        setHeadRotation(new Vector(0, 0, 0));
        setBodyRotation(new Vector(0, 0, 0));
        setLeftArmRotation(new Vector(-10f, 0, -10f));
        setRightArmRotation(new Vector(-15f, 0, -10f));
        setLeftLegRotation(new Vector(-1f, 0, -1f));
        setRightLegRotation(new Vector(1, 0, 1));

        this.mainHandItem = ItemStack.AIR;
        this.offHandItem = ItemStack.AIR;

        this.helmet = ItemStack.AIR;
        this.chestplate = ItemStack.AIR;
        this.leggings = ItemStack.AIR;
        this.boots = ItemStack.AIR;

    }

    @Override
    protected boolean addViewer0(@NotNull Player player) {
        if (!super.addViewer0(player)) {
            return false;
        }
        player.getPlayerConnection().sendPacket(getEquipmentsPacket());
        return true;
    }

    @Override
    public int getObjectData() {
        return 0;
    }

    @NotNull
    @Override
    public ItemStack getItemInMainHand() {
        return mainHandItem;
    }

    @Override
    public void setItemInMainHand(@NotNull ItemStack itemStack) {
        this.mainHandItem = getEquipmentItem(itemStack, EquipmentSlot.MAIN_HAND);
        syncEquipment(EquipmentSlot.MAIN_HAND);
    }

    @NotNull
    @Override
    public ItemStack getItemInOffHand() {
        return offHandItem;
    }

    @Override
    public void setItemInOffHand(@NotNull ItemStack itemStack) {
        this.offHandItem = getEquipmentItem(itemStack, EquipmentSlot.OFF_HAND);
        syncEquipment(EquipmentSlot.OFF_HAND);
    }

    @NotNull
    @Override
    public ItemStack getHelmet() {
        return helmet;
    }

    @Override
    public void setHelmet(@NotNull ItemStack itemStack) {
        this.helmet = getEquipmentItem(itemStack, EquipmentSlot.HELMET);
        syncEquipment(EquipmentSlot.HELMET);
    }

    @NotNull
    @Override
    public ItemStack getChestplate() {
        return chestplate;
    }

    @Override
    public void setChestplate(@NotNull ItemStack itemStack) {
        this.chestplate = getEquipmentItem(itemStack, EquipmentSlot.CHESTPLATE);
        syncEquipment(EquipmentSlot.CHESTPLATE);
    }

    @NotNull
    @Override
    public ItemStack getLeggings() {
        return leggings;
    }

    @Override
    public void setLeggings(@NotNull ItemStack itemStack) {
        this.leggings = getEquipmentItem(itemStack, EquipmentSlot.LEGGINGS);
        syncEquipment(EquipmentSlot.LEGGINGS);
    }

    @NotNull
    @Override
    public ItemStack getBoots() {
        return boots;
    }

    @Override
    public void setBoots(@NotNull ItemStack itemStack) {
        this.boots = getEquipmentItem(itemStack, EquipmentSlot.BOOTS);
        syncEquipment(EquipmentSlot.BOOTS);
    }

    public boolean isSmall() {
        return (getStateMeta() & 0x01) != 0;
    }

    public void setSmall(boolean small) {
        final byte state = BitmaskUtil.changeBit(getStateMeta(), (byte) 0x01, (byte) (small ? 1 : 0), (byte) 0);
        this.metadata.setIndex((byte) 14, Metadata.Byte(state));

        if (small) {
            setBoundingBox(0.25f, 0.9875f, 0.25f);
        } else {
            setBoundingBox(0.5f, 1.975f, 0.5f);
        }
    }

    public boolean hasArms() {
        return (getStateMeta() & 0x04) != 0;
    }

    public void setHasArms(boolean hasArms) {
        final byte state = BitmaskUtil.changeBit(getStateMeta(), (byte) 0x08, (byte) (hasArms ? 1 : 0), (byte) 2);
        this.metadata.setIndex((byte) 14, Metadata.Byte(state));
    }

    public boolean hasNoBasePlate() {
        return (getStateMeta() & 0x08) != 0;
    }

    public void setNoBasePlate(boolean noBasePlate) {
        final byte state = BitmaskUtil.changeBit(getStateMeta(), (byte) 0x10, (byte) (noBasePlate ? 1 : 0), (byte) 3);
        this.metadata.setIndex((byte) 14, Metadata.Byte(state));
    }

    public boolean hasMarker() {
        return (getStateMeta() & 0x10) != 0;
    }

    public void setMarker(boolean setMarker) {
        final byte state = BitmaskUtil.changeBit(getStateMeta(), (byte) 0x20, (byte) (setMarker ? 1 : 0), (byte) 4);
        this.metadata.setIndex((byte) 14, Metadata.Byte(state));
    }

    @NotNull
    public Vector getHeadRotation() {
        return metadata.getIndex((byte) 15, new Vector(0, 0, 0));
    }

    public void setHeadRotation(@NotNull Vector headRotation) {
        this.metadata.setIndex((byte) 15, Metadata.Rotation(headRotation));
    }

    @NotNull
    public Vector getBodyRotation() {
        return metadata.getIndex((byte) 16, new Vector(0, 0, 0));
    }

    public void setBodyRotation(@NotNull Vector bodyRotation) {
        this.metadata.setIndex((byte) 16, Metadata.Rotation(bodyRotation));
    }

    @NotNull
    public Vector getLeftArmRotation() {
        return metadata.getIndex((byte) 17, new Vector(-10, 0, -10));
    }

    public void setLeftArmRotation(@NotNull Vector leftArmRotation) {
        this.metadata.setIndex((byte) 17, Metadata.Rotation(leftArmRotation));
    }

    @NotNull
    public Vector getRightArmRotation() {
        return metadata.getIndex((byte) 18, new Vector(-15, 0, 10));
    }

    public void setRightArmRotation(@NotNull Vector rightArmRotation) {
        this.metadata.setIndex((byte) 18, Metadata.Rotation(rightArmRotation));
    }

    @NotNull
    public Vector getLeftLegRotation() {
        return metadata.getIndex((byte) 19, new Vector(-1, 0, -1));
    }

    public void setLeftLegRotation(@NotNull Vector leftLegRotation) {
        this.metadata.setIndex((byte) 19, Metadata.Rotation(leftLegRotation));
    }

    @NotNull
    public Vector getRightLegRotation() {
        return metadata.getIndex((byte) 20, new Vector(1, 0, 1));
    }

    public void setRightLegRotation(@NotNull Vector rightLegRotation) {
        this.metadata.setIndex((byte) 20, Metadata.Rotation(rightLegRotation));
    }

    private byte getStateMeta() {
        return metadata.getIndex((byte) 14, (byte) 0);
    }

    // Equipments

    private ItemStack getEquipmentItem(@NotNull ItemStack itemStack, @NotNull EquipmentSlot slot) {
        EntityEquipEvent entityEquipEvent = new EntityEquipEvent(this, itemStack, slot);
        EventDispatcher.call(entityEquipEvent);
        return entityEquipEvent.getEquippedItem();
    }
}