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

Go to download
Show more of this group Show more artifacts with this name
Show all versions of powernukkitx Show documentation
Make Nukkit great again!
There is a newer version: 1.20.40-r1
package cn.nukkit.inventory;

import cn.nukkit.Player;
import cn.nukkit.api.PowerNukkitOnly;
import cn.nukkit.api.PowerNukkitXOnly;
import cn.nukkit.api.Since;
import cn.nukkit.entity.Entity;
import cn.nukkit.item.Item;
import cn.nukkit.network.protocol.MobEquipmentPacket;

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

@PowerNukkitOnly
@Since("1.4.0.0-PN")
public class EntityEquipmentInventory extends BaseInventory {

    private final Entity entity;

    @PowerNukkitOnly @Since("1.4.0.0-PN") public static final int MAIN_HAND = 0;
    @PowerNukkitOnly @Since("1.4.0.0-PN") public static final int OFFHAND = 1;

    /**
     * @param entity an Entity which implements {@link InventoryHolder}.
     * @throws ClassCastException if the entity does not implements {@link InventoryHolder}
     */
    @PowerNukkitOnly
    @Since("1.4.0.0-PN")
    public EntityEquipmentInventory(Entity entity) {
        super((InventoryHolder) entity, InventoryType.ENTITY_EQUIPMENT);
        this.entity = entity;
    }

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

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

    @PowerNukkitOnly
    @Since("1.4.0.0-PN")
    public Entity getEntity() {
        return entity;
    }
    
    @Override
    public InventoryHolder getHolder() {
        return this.holder;
    }

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

    @Override
    public void sendSlot( int index, Player player ) {
        MobEquipmentPacket mobEquipmentPacket = new MobEquipmentPacket();
        mobEquipmentPacket.eid = this.entity.getId();
        mobEquipmentPacket.inventorySlot = mobEquipmentPacket.hotbarSlot = index;
        mobEquipmentPacket.item = this.getItem( index );
        player.dataPacket( mobEquipmentPacket );
    }

    @Override
    public Set getViewers() {
        Set viewers = new HashSet<>(this.viewers);
        viewers.addAll(entity.getViewers().values());
        return viewers;
    }

    @Override
    public boolean open( Player who ) {
        return this.viewers.add( who );
    }

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

    @PowerNukkitOnly
    @Since("1.4.0.0-PN")
    public Item getItemInHand() {
        return this.getItem( MAIN_HAND);
    }

    @PowerNukkitOnly
    @Since("1.4.0.0-PN")
    public Item getItemInOffhand() {
        return this.getItem( OFFHAND );
    }


    @PowerNukkitXOnly
    @Since("1.6.0.0-PNX")
    public boolean setItemInHand(Item item){
        return this.setItem(MAIN_HAND,item);
    }

    @PowerNukkitOnly
    @Since("1.4.0.0-PN")
    public boolean setItemInHand( Item item, boolean send ) {
        return this.setItem( MAIN_HAND, item, send );
    }

    @PowerNukkitOnly
    @Since("1.4.0.0-PN")
    public boolean setItemInOffhand(Item item, boolean send ) {
        return this.setItem( OFFHAND, item, send );
    }

    @Override
    public void sendContents( Player target ) {
        this.sendSlot( MAIN_HAND, target );
        this.sendSlot( OFFHAND, target );
    }

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