From 3bdae21716d10032f70a0a889070766bbbe4d141 Mon Sep 17 00:00:00 2001
From: Max Audron <audron@cocaine.farm>
Date: Sat, 26 Sep 2020 12:01:27 +0200
Subject: split handshake.rs

---
 src/message/handshake/sessioninit.rs | 61 ++++++++++++++++++++++++++++++++++++
 1 file changed, 61 insertions(+)
 create mode 100644 src/message/handshake/sessioninit.rs

(limited to 'src/message/handshake/sessioninit.rs')

diff --git a/src/message/handshake/sessioninit.rs b/src/message/handshake/sessioninit.rs
new file mode 100644
index 0000000..eca4c10
--- /dev/null
+++ b/src/message/handshake/sessioninit.rs
@@ -0,0 +1,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);
+        }
+    }
+}
-- 
cgit v1.2.3