basin2-protocol 578.1.0

Basin2 subpackage for MC protocol.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84

use crate::network::*;
use crate::packet::*;
use basin2_lib::result::*;
use bytes::BytesMut;

#[derive(PartialEq, Clone, Debug)]
pub struct UpdateAttributesPacket {
    pub entityId: i32,
    pub attributes: Vec<EntityAttribute>,
}

impl CodablePacket for UpdateAttributesPacket {
    fn encode(self, buf: &mut BytesMut) {
        buf.set_mc_var_int(self.entityId);
        buf.set_mc_var_int(self.attributes.len() as i32);
        for attribute in self.attributes {
            buf.set_mc_string(attribute.name);
            buf.set_mc_f64(attribute.base);
            buf.set_mc_var_int(attribute.modifiers.len() as i32);
            for modifier in attribute.modifiers {
                buf.set_mc_uuid(modifier.uuid);
                buf.set_mc_f64(modifier.amount);
                buf.set_mc_u8(modifier.operation as u8);
            }
        }
    }

    fn decode(buf: &mut BytesMut) -> Result<Self>
    where
        Self: Sized,
    {
        let entityId = buf.get_mc_var_int()?;
        let attribute_count = buf.get_mc_var_int()?;
        let mut attributes: Vec<EntityAttribute> = vec![];
        for _ in 0..attribute_count {
            let name = buf.get_mc_string(64)?;
            let base = buf.get_mc_f64()?;
            let modifier_count = buf.get_mc_var_int()?;
            let mut modifiers: Vec<EntityAttributeModifier> = vec![];
            for _ in 0..modifier_count {
                let uuid = buf.get_mc_uuid()?;
                let amount = buf.get_mc_f64()?;
                let operation: EntityAttributeModifierOperation = buf.get_mc_enum_u8()?;
                modifiers.push(EntityAttributeModifier {
                    uuid,
                    amount,
                    operation,
                });
            }
            attributes.push(EntityAttribute {
                name,
                base,
                modifiers,
            });
        }
        return Ok(UpdateAttributesPacket {
            entityId,
            attributes,
        });
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::packet::test::*;
    use uuid::Uuid;

    #[test]
    fn test_cycle() -> Result<()> {
        cycle(UpdateAttributesPacket {
            entityId: 4235356,
            attributes: vec![EntityAttribute {
                name: "attribute_name".to_string(),
                base: 120.0,
                modifiers: vec![EntityAttributeModifier {
                    uuid: Uuid::new_v4(),
                    amount: 140.0,
                    operation: EntityAttributeModifierOperation::Addition,
                }],
            }],
        })
    }
}