diff --git a/src/main/java/net/minestom/server/inventory/EquipmentHandler.java b/src/main/java/net/minestom/server/inventory/EquipmentHandler.java index 504b484bf..932355205 100644 --- a/src/main/java/net/minestom/server/inventory/EquipmentHandler.java +++ b/src/main/java/net/minestom/server/inventory/EquipmentHandler.java @@ -1,5 +1,6 @@ package net.minestom.server.inventory; +import net.minestom.server.entity.Player; import net.minestom.server.item.ItemStack; import net.minestom.server.network.packet.server.play.EntityEquipmentPacket; @@ -13,6 +14,31 @@ public interface EquipmentHandler { void setItemInOffHand(ItemStack itemStack); + default ItemStack getItemInHand(Player.Hand hand) { + switch (hand) { + case MAIN: + return getItemInMainHand(); + + case OFF: + return getItemInOffHand(); + + default: + return null; + } + } + + default void setItemInHand(Player.Hand hand, ItemStack stack) { + switch (hand) { + case MAIN: + setItemInMainHand(stack); + break; + + case OFF: + setItemInOffHand(stack); + break; + } + } + ItemStack getHelmet(); void setHelmet(ItemStack itemStack);