azalea_protocol/packets/game/
c_set_equipment.rs

1use std::io::{self, Cursor, Write};
2
3use azalea_buf::{AzBuf, AzaleaRead, AzaleaWrite, BufReadError};
4use azalea_inventory::{ItemStack, components::EquipmentSlot};
5use azalea_protocol_macros::ClientboundGamePacket;
6use azalea_world::MinecraftEntityId;
7
8#[derive(Clone, Debug, AzBuf, PartialEq, ClientboundGamePacket)]
9pub struct ClientboundSetEquipment {
10    #[var]
11    pub entity_id: MinecraftEntityId,
12    pub slots: EquipmentSlots,
13}
14
15#[derive(Clone, Debug, PartialEq)]
16pub struct EquipmentSlots {
17    pub slots: Vec<(EquipmentSlot, ItemStack)>,
18}
19
20impl AzaleaRead for EquipmentSlots {
21    fn azalea_read(buf: &mut Cursor<&[u8]>) -> Result<Self, BufReadError> {
22        let mut slots = vec![];
23
24        loop {
25            let equipment_byte = u8::azalea_read(buf)?;
26            let equipment_slot =
27                EquipmentSlot::from_byte(equipment_byte & 127).ok_or_else(|| {
28                    BufReadError::UnexpectedEnumVariant {
29                        id: equipment_byte.into(),
30                    }
31                })?;
32            let item = ItemStack::azalea_read(buf)?;
33            slots.push((equipment_slot, item));
34            if equipment_byte & 128 == 0 {
35                break;
36            };
37        }
38
39        Ok(EquipmentSlots { slots })
40    }
41}
42impl AzaleaWrite for EquipmentSlots {
43    fn azalea_write(&self, buf: &mut impl Write) -> io::Result<()> {
44        for i in 0..self.slots.len() {
45            let (equipment_slot, item) = &self.slots[i];
46            let mut equipment_byte = *equipment_slot as u8;
47            if i != self.slots.len() - 1 {
48                equipment_byte |= 128;
49            }
50            equipment_byte.azalea_write(buf)?;
51            item.azalea_write(buf)?;
52        }
53
54        Ok(())
55    }
56}