aboutsummaryrefslogtreecommitdiff
path: root/src/message/handshake/sessioninit.rs
blob: eca4c108f6eff4968be6d4e674481444d7a2e6f8 (plain)
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
use crate::error::ProtocolError;
use crate::primitive::{Variant, VariantList, VariantMap};
use crate::{HandshakeDeserialize, HandshakeSerialize};

use failure::Error;

/// SessionInit is received along with ClientLoginAck to initialize that user Session
// TODO Replace with proper types
#[derive(Debug)]
pub struct SessionInit {
    /// List of all configured identities
    identities: VariantList,
    /// List of all existing buffers
    buffers: VariantList,
    /// Ids of all networks
    network_ids: VariantList,
}

impl HandshakeSerialize for SessionInit {
    fn serialize(&self) -> Result<Vec<u8>, Error> {
        let mut values: VariantMap = VariantMap::with_capacity(1);
        values.insert(
            "MsgType".to_string(),
            Variant::String("SessionInit".to_string()),
        );
        values.insert(
            "Identities".to_string(),
            Variant::VariantList(self.identities.clone()),
        );
        values.insert(
            "BufferInfos".to_string(),
            Variant::VariantList(self.buffers.clone()),
        );
        values.insert(
            "NetworkIds".to_string(),
            Variant::VariantList(self.network_ids.clone()),
        );
        return HandshakeSerialize::serialize(&values);
    }
}

impl HandshakeDeserialize for SessionInit {
    fn parse(b: &[u8]) -> Result<(usize, Self), Error> {
        let (len, values): (usize, VariantMap) = HandshakeDeserialize::parse(b)?;

        let msgtype = match_variant!(&values["MsgType"], Variant::StringUTF8);

        if msgtype == "ClientLogin" {
            return Ok((
                len,
                Self {
                    identities: match_variant!(values["Identities"], Variant::VariantList),
                    buffers: match_variant!(values["BufferInfos"], Variant::VariantList),
                    network_ids: match_variant!(values["NetworkIds"], Variant::VariantList),
                },
            ));
        } else {
            bail!(ProtocolError::WrongMsgType);
        }
    }
}