cn.nukkit.inventory.EntityArmorInventory Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of powernukkit Show documentation
A Minecraft Bedrock Edition server software implementation made in Java from scratch which supports all new features.
There is a newer version: 1.6.0.1-PN
package cn.nukkit.inventory;

import cn.nukkit.Player;
import cn.nukkit.entity.EntityLiving;

import cn.nukkit.item.Item;
import cn.nukkit.network.protocol.InventoryContentPacket;
import cn.nukkit.network.protocol.InventorySlotPacket;
import cn.nukkit.network.protocol.MobArmorEquipmentPacket;
import cn.nukkit.entity.item.EntityArmorStand;

import java.util.HashSet;
import java.util.Set;

public class EntityArmorInventory extends BaseInventory {

    private EntityLiving entityLiving;

    private final Set viewers = new HashSet<>();

    public static final int SLOT_HEAD = 0;
    public static final int  SLOT_CHEST = 1;
    public static final int  SLOT_LEGS = 2;
    public static final int  SLOT_FEET = 3;

    public EntityArmorInventory(EntityArmorStand entityArmorStand) {
        super(entityArmorStand, InventoryType.PLAYER);
        this.entityLiving = entityArmorStand;
    }

    @Override
    public InventoryHolder getHolder() {
        return this.holder;
    }

    @Override
    public String getName() {
        return "Armor";
    }

    @Override
    public int getSize() {
        return 4;
    }

    public Item getHelmet(){
        return this.getItem(SLOT_HEAD);
    }

    public Item getChestplate(){
        return this.getItem(SLOT_CHEST);
    }

    public Item getLeggings(){
        return this.getItem(SLOT_LEGS);
    }

    public Item getBoots(){
        return this.getItem(SLOT_FEET);
    }

    public void setHelmet(Item item) {
        this.setItem(SLOT_CHEST, item);
    }

    public void setChestplate(Item item) {
        this.setItem(SLOT_CHEST, item);
    }

    public void setLeggings(Item item) {
        this.setItem(SLOT_LEGS, item);
    }

    public void setBoots(Item item) {
        this.setItem(SLOT_FEET, item);
    }

    @Override
    public void sendSlot(int index, Player... players) {
        for (Player player : players) {
            this.sendSlot(index, player);
        }
    }

    @Override
    public void sendSlot(int index, Player player) {
        MobArmorEquipmentPacket mobArmorEquipmentPacket = new MobArmorEquipmentPacket();
        mobArmorEquipmentPacket.eid = this.entityLiving.getId();
        mobArmorEquipmentPacket.slots = new Item[]{this.getHelmet(), this.getChestplate(), this.getLeggings(), this.getBoots()};

        if(player == this.getHolder()){
            InventorySlotPacket inventorySlotPacket = new InventorySlotPacket();
            inventorySlotPacket.inventoryId = player.getWindowId(this);
            inventorySlotPacket.slot = index;
            inventorySlotPacket.item = this.getItem(index);
            player.dataPacket(inventorySlotPacket);
        }else{
            player.dataPacket(mobArmorEquipmentPacket);
        }
    }

    @Override
    public void sendContents(Player... players) {
        for (Player player : players) {
            this.sendContents(player);
        }
    }

    @Override
    public void sendContents(Player player) {
        MobArmorEquipmentPacket mobArmorEquipmentPacket = new MobArmorEquipmentPacket();
        mobArmorEquipmentPacket.eid = this.entityLiving.getId();
        mobArmorEquipmentPacket.slots = new Item[]{this.getHelmet(), this.getChestplate(), this.getLeggings(), this.getBoots()};

        if(player == this.getHolder()){
            InventoryContentPacket inventoryContentPacket = new InventoryContentPacket();
            inventoryContentPacket.inventoryId = player.getWindowId(this);
            inventoryContentPacket.slots = new Item[]{this.getHelmet(), this.getChestplate(), this.getLeggings(), this.getBoots()};
            player.dataPacket(inventoryContentPacket);
        }else{
            player.dataPacket(mobArmorEquipmentPacket);
        }
    }

    @Override
    public void onOpen(Player who) {
        this.viewers.add(who);
    }

    @Override
    public void onClose(Player who) {
        this.viewers.remove(who);
    }

    @Override
    public Set getViewers() {
        return this.viewers;
    }
}