split signalproxy.rs

1 files changed, 65 insertions, 0 deletions

diff --git a/src/message/signalproxy/heartbeat.rs b/src/message/signalproxy/heartbeat.rs
new file mode 100644
index 0000000..46bfd51
--- /dev/null
+++ b/src/message/signalproxy/heartbeat.rs
@@ -0,0 +1,65 @@
+use crate::message::MessageType;
+use crate::primitive::{DateTime, Variant, VariantList};
+use crate::{Deserialize, Serialize};
+
+#[derive(Clone, Debug, std::cmp::PartialEq)]
+pub struct HeartBeat {
+    timestamp: DateTime,
+}
+
+impl Serialize for HeartBeat {
+    fn serialize(&self) -> Result<Vec<std::primitive::u8>, failure::Error> {
+        let mut res = VariantList::new();
+
+        res.push(Variant::i32(MessageType::HeartBeat as i32));
+        res.push(Variant::DateTime(self.timestamp.clone()));
+
+        res.serialize()
+    }
+}
+
+impl Deserialize for HeartBeat {
+    fn parse(b: &[std::primitive::u8]) -> Result<(std::primitive::usize, Self), failure::Error> {
+        let (size, mut res) = VariantList::parse(&b)?;
+
+        res.remove(0);
+
+        Ok((
+            size,
+            Self {
+                timestamp: match_variant!(res.remove(0), Variant::DateTime),
+            },
+        ))
+    }
+}
+
+#[derive(Clone, Debug, std::cmp::PartialEq)]
+pub struct HeartBeatReply {
+    timestamp: DateTime,
+}
+
+impl Serialize for HeartBeatReply {
+    fn serialize(&self) -> Result<Vec<std::primitive::u8>, failure::Error> {
+        let mut res = VariantList::new();
+
+        res.push(Variant::i32(MessageType::HeartBeatReply as i32));
+        res.push(Variant::DateTime(self.timestamp.clone()));
+
+        res.serialize()
+    }
+}
+
+impl Deserialize for HeartBeatReply {
+    fn parse(b: &[std::primitive::u8]) -> Result<(std::primitive::usize, Self), failure::Error> {
+        let (size, mut res) = VariantList::parse(&b)?;
+
+        res.remove(0);
+
+        Ok((
+            size,
+            Self {
+                timestamp: match_variant!(res.remove(0), Variant::DateTime),
+            },
+        ))
+    }
+}