EntityEquipmentCodec.java

package net.glowstone.net.codec.play.entity;

import com.flowpowered.networking.Codec;
import com.flowpowered.networking.util.ByteBufUtils;
import io.netty.buffer.ByteBuf;
import net.glowstone.net.GlowBufUtils;
import net.glowstone.net.message.play.entity.EntityEquipmentMessage;
import org.bukkit.inventory.ItemStack;

import java.io.IOException;

public final class EntityEquipmentCodec implements Codec<EntityEquipmentMessage> {
    @Override
    public EntityEquipmentMessage decode(ByteBuf buf) throws IOException {
        int id = ByteBufUtils.readVarInt(buf);
        int slot = buf.readShort();
        ItemStack stack = GlowBufUtils.readSlot(buf);
        return new EntityEquipmentMessage(id, slot, stack);
    }

    @Override
    public ByteBuf encode(ByteBuf buf, EntityEquipmentMessage message) throws IOException {
        ByteBufUtils.writeVarInt(buf, message.getId());
        buf.writeShort(message.getSlot());
        GlowBufUtils.writeSlot(buf, message.getStack());
        return buf;
    }
}