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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147

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

#[derive(PartialEq, Clone, Debug)]
pub struct SetPlayerTeamPacket {
    pub name: String,
    pub displayName: Option<ChatComponent>,
    pub playerPrefix: Option<ChatComponent>,
    pub playerSuffix: Option<ChatComponent>,
    pub nametagVisibility: Option<String>,
    pub collisionRule: Option<String>,
    pub color: Option<ChatFormatting>,
    pub players: Option<Vec<String>>,
    pub method: i8,
    pub options: Option<i8>,
}

impl CodablePacket for SetPlayerTeamPacket {
    fn encode(self, buf: &mut BytesMut) {
        buf.set_mc_string(self.name);
        buf.set_mc_i8(self.method);
        if self.method == 0 || self.method == 2 {
            buf.set_mc_chat_component(self.displayName.unwrap());
            buf.set_mc_i8(self.options.unwrap());
            buf.set_mc_string(self.nametagVisibility.unwrap());
            buf.set_mc_string(self.collisionRule.unwrap());
            buf.set_mc_var_int(self.color.unwrap() as i32);
            buf.set_mc_chat_component(self.playerPrefix.unwrap());
            buf.set_mc_chat_component(self.playerSuffix.unwrap());
        }

        if self.method == 0 || self.method == 3 || self.method == 4 {
            let players = self.players.unwrap();
            buf.set_mc_var_int(players.len() as i32);
            for player in players {
                buf.set_mc_string(player);
            }
        }
    }

    fn decode(buf: &mut BytesMut) -> Result<Self>
    where
        Self: Sized,
    {
        let name = buf.get_mc_string(16)?;
        let method = buf.get_mc_i8()?;
        let (
            displayName,
            options,
            nametagVisibility,
            collisionRule,
            color,
            playerPrefix,
            playerSuffix,
        ) = match method {
            0 | 2 => (
                Some(buf.get_mc_chat_component()?),
                Some(buf.get_mc_i8()?),
                Some(buf.get_mc_string(40)?),
                Some(buf.get_mc_string(40)?),
                Some(buf.get_mc_enum::<ChatFormatting>()?),
                Some(buf.get_mc_chat_component()?),
                Some(buf.get_mc_chat_component()?),
            ),
            _ => (None, None, None, None, None, None, None),
        };
        let players = match method {
            0 | 3 | 4 => {
                let mut players: Vec<String> = vec![];
                let count = buf.get_mc_var_int()?;
                for _ in 0..count {
                    players.push(buf.get_mc_string(40)?);
                }
                Some(players)
            }
            _ => None,
        };
        return Ok(SetPlayerTeamPacket {
            name,
            displayName,
            playerPrefix,
            playerSuffix,
            nametagVisibility,
            collisionRule,
            color,
            players,
            method,
            options,
        });
    }
}

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

    #[test]
    fn test_cycle_method0() -> Result<()> {
        cycle(SetPlayerTeamPacket {
            name: "team name".to_string(),
            displayName: Some(ChatComponent::from("display".to_string())),
            playerPrefix: Some(ChatComponent::from("prefix".to_string())),
            playerSuffix: Some(ChatComponent::from("suffix".to_string())),
            nametagVisibility: Some("nametag".to_string()),
            collisionRule: Some("collision".to_string()),
            color: Some(ChatFormatting::Green),
            players: Some(vec!["player1".to_string(), "player2".to_string()]),
            method: 0,
            options: Some(12),
        })
    }

    #[test]
    fn test_cycle_method2() -> Result<()> {
        cycle(SetPlayerTeamPacket {
            name: "team name".to_string(),
            displayName: Some(ChatComponent::from("display".to_string())),
            playerPrefix: Some(ChatComponent::from("prefix".to_string())),
            playerSuffix: Some(ChatComponent::from("suffix".to_string())),
            nametagVisibility: Some("nametag".to_string()),
            collisionRule: Some("collision".to_string()),
            color: Some(ChatFormatting::Green),
            players: None,
            method: 2,
            options: Some(12),
        })
    }

    #[test]
    fn test_cycle_method3() -> Result<()> {
        cycle(SetPlayerTeamPacket {
            name: "team name".to_string(),
            displayName: None,
            playerPrefix: None,
            playerSuffix: None,
            nametagVisibility: None,
            collisionRule: None,
            color: None,
            players: Some(vec!["player1".to_string(), "player2".to_string()]),
            method: 3,
            options: None,
        })
    }
}