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;
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}
57
58#[derive(Clone, Debug, Copy, AzBuf, PartialEq)]
59pub enum EquipmentSlot {
60    MainHand = 0,
61    OffHand = 1,
62    Feet = 2,
63    Legs = 3,
64    Chest = 4,
65    Head = 5,
66}
67
68impl EquipmentSlot {
69    #[must_use]
70    pub fn from_byte(byte: u8) -> Option<Self> {
71        match byte {
72            0 => Some(EquipmentSlot::MainHand),
73            1 => Some(EquipmentSlot::OffHand),
74            2 => Some(EquipmentSlot::Feet),
75            3 => Some(EquipmentSlot::Legs),
76            4 => Some(EquipmentSlot::Chest),
77            5 => Some(EquipmentSlot::Head),
78            _ => None,
79        }
80    }
81}
azalea_protocol/packets/game/c_set_equipment.rs

azalea_protocol/packets/game/
c_set_equipment.rs
