aboutsummaryrefslogtreecommitdiff
path: root/src/message/signalproxy/objects/buffersyncer.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/message/signalproxy/objects/buffersyncer.rs')
-rw-r--r--src/message/signalproxy/objects/buffersyncer.rs263
1 files changed, 255 insertions, 8 deletions
diff --git a/src/message/signalproxy/objects/buffersyncer.rs b/src/message/signalproxy/objects/buffersyncer.rs
index 7ee4ca5..cc26310 100644
--- a/src/message/signalproxy/objects/buffersyncer.rs
+++ b/src/message/signalproxy/objects/buffersyncer.rs
@@ -1,13 +1,148 @@
-use crate::message::signalproxy::MessageType;
-use std::collections::HashMap;
+use std::{collections::HashMap, convert::TryInto};
-// use default_macro::default;
-// #[default(crate::message::signalproxy::objects::BufferSyncerClient)]
+use num_traits::{FromPrimitive, ToPrimitive};
+
+use itertools::Itertools;
+
+use crate::{
+ message::signalproxy::Network,
+ primitive::{MessageType, Variant, VariantList},
+};
+
+#[derive(Debug, Clone, PartialEq)]
pub struct BufferSyncer {
- pub activities: HashMap<u32, MessageType>,
- pub highlight_counts: HashMap<u32, u32>,
- pub last_seen_msg: HashMap<u32, u32>,
- pub marker_line: HashMap<u32, u32>,
+ pub activities: HashMap<i32, MessageType>,
+ pub highlight_counts: HashMap<i32, i32>,
+ pub last_seen_msg: HashMap<i32, i64>,
+ pub marker_line: HashMap<i32, i64>,
+}
+
+impl Network for BufferSyncer {
+ type Item = VariantList;
+
+ fn to_network(&self) -> Self::Item {
+ let mut res = Self::Item::new();
+
+ res.push(Variant::ByteArray(s!("Activities")));
+ res.push(Variant::VariantList({
+ let mut res = VariantList::new();
+
+ self.activities.iter().for_each(|(k, v)| {
+ res.push(Variant::i32(*k));
+ res.push(Variant::i32(v.to_i32().unwrap()));
+ });
+
+ res
+ }));
+
+ res.push(Variant::ByteArray(s!("HighlightCounts")));
+ res.push(Variant::VariantList({
+ let mut res = VariantList::new();
+
+ self.highlight_counts.iter().for_each(|(k, v)| {
+ res.push(Variant::i32(*k));
+ res.push(Variant::i32(*v));
+ });
+
+ res
+ }));
+
+ res.push(Variant::ByteArray(s!("LastSeenMsg")));
+ res.push(Variant::VariantList({
+ let mut res = VariantList::new();
+
+ self.last_seen_msg.iter().for_each(|(k, v)| {
+ res.push(Variant::i32(*k));
+ res.push(Variant::i64(*v));
+ });
+
+ res
+ }));
+
+ res.push(Variant::ByteArray(s!("MarkerLines")));
+ res.push(Variant::VariantList({
+ let mut res = VariantList::new();
+
+ self.marker_line.iter().for_each(|(k, v)| {
+ res.push(Variant::i32(*k));
+ res.push(Variant::i64(*v));
+ });
+
+ res
+ }));
+
+ res
+ }
+
+ fn from_network(input: &mut Self::Item) -> Self {
+ let mut i = input.iter().cycle();
+
+ i.position(|x| *x == crate::primitive::Variant::ByteArray(s!("Activities")))
+ .unwrap();
+ let activities: VariantList = i.next().unwrap().try_into().unwrap();
+ let activities = activities
+ .iter()
+ .batching(|it| match it.next() {
+ None => None,
+ Some(x) => match it.next() {
+ None => None,
+ Some(y) => Some((
+ x.try_into().unwrap(),
+ MessageType::from_i32(y.try_into().unwrap()).unwrap(),
+ )),
+ },
+ })
+ .collect();
+
+ i.position(|x| *x == crate::primitive::Variant::ByteArray(s!("HighlightCounts")))
+ .unwrap();
+ let highlight_counts: VariantList = i.next().unwrap().try_into().unwrap();
+ let highlight_counts = highlight_counts
+ .iter()
+ .batching(|it| match it.next() {
+ None => None,
+ Some(x) => match it.next() {
+ None => None,
+ Some(y) => Some((x.try_into().unwrap(), y.try_into().unwrap())),
+ },
+ })
+ .collect();
+
+ i.position(|x| *x == crate::primitive::Variant::ByteArray(s!("LastSeenMsg")))
+ .unwrap();
+ let last_seen_msg: VariantList = i.next().unwrap().try_into().unwrap();
+ let last_seen_msg = last_seen_msg
+ .iter()
+ .batching(|it| match it.next() {
+ None => None,
+ Some(x) => match it.next() {
+ None => None,
+ Some(y) => Some((x.try_into().unwrap(), y.try_into().unwrap())),
+ },
+ })
+ .collect();
+
+ i.position(|x| *x == crate::primitive::Variant::ByteArray(s!("MarkerLines")))
+ .unwrap();
+ let marker_line: VariantList = i.next().unwrap().try_into().unwrap();
+ let marker_line = marker_line
+ .iter()
+ .batching(|it| match it.next() {
+ None => None,
+ Some(x) => match it.next() {
+ None => None,
+ Some(y) => Some((x.try_into().unwrap(), y.try_into().unwrap())),
+ },
+ })
+ .collect();
+
+ Self {
+ activities,
+ highlight_counts,
+ last_seen_msg,
+ marker_line,
+ }
+ }
}
pub trait BufferSyncerServer {
@@ -72,3 +207,115 @@ pub trait BufferSyncerClient {
self.marker_line_mut().insert(buffer, msgid);
}
}
+
+#[cfg(test)]
+mod tests {
+ use super::*;
+ use crate::message::signalproxy::translation::Network;
+ use crate::primitive::{Variant, VariantList};
+ use pretty_assertions::assert_eq;
+
+ fn get_network() -> VariantList {
+ vec![
+ Variant::ByteArray(s!("Activities")),
+ Variant::VariantList(vec![
+ Variant::i32(1),
+ Variant::i32(0),
+ Variant::i32(2),
+ Variant::i32(0),
+ Variant::i32(3),
+ Variant::i32(0),
+ Variant::i32(4),
+ Variant::i32(0),
+ Variant::i32(5),
+ Variant::i32(0),
+ ]),
+ Variant::ByteArray(s!("HighlightCounts")),
+ Variant::VariantList(vec![
+ Variant::i32(1),
+ Variant::i32(0),
+ Variant::i32(2),
+ Variant::i32(0),
+ Variant::i32(3),
+ Variant::i32(0),
+ Variant::i32(4),
+ Variant::i32(0),
+ Variant::i32(5),
+ Variant::i32(0),
+ ]),
+ Variant::ByteArray(s!("LastSeenMsg")),
+ Variant::VariantList(vec![
+ Variant::i32(1),
+ Variant::i64(2185),
+ Variant::i32(2),
+ Variant::i64(2188),
+ Variant::i32(3),
+ Variant::i64(860),
+ Variant::i32(4),
+ Variant::i64(2183),
+ Variant::i32(5),
+ Variant::i64(2180),
+ ]),
+ Variant::ByteArray(s!("MarkerLines")),
+ Variant::VariantList(vec![
+ Variant::i32(1),
+ Variant::i64(2185),
+ Variant::i32(2),
+ Variant::i64(2188),
+ Variant::i32(3),
+ Variant::i64(860),
+ Variant::i32(4),
+ Variant::i64(1527),
+ Variant::i32(5),
+ Variant::i64(2180),
+ ]),
+ ]
+ }
+
+ fn get_runtime() -> BufferSyncer {
+ BufferSyncer {
+ activities: map! {
+ 1 => MessageType::None,
+ 2 => MessageType::None,
+ 3 => MessageType::None,
+ 4 => MessageType::None,
+ 5 => MessageType::None,
+ },
+ highlight_counts: map! {
+ 1 => 0,
+ 2 => 0,
+ 3 => 0,
+ 4 => 0,
+ 5 => 0,
+ },
+ last_seen_msg: map! {
+ 1 => 2185,
+ 2 => 2188,
+ 3 => 860,
+ 4 => 2183,
+ 5 => 2180,
+ },
+ marker_line: map! {
+ 1 => 2185,
+ 2 => 2188,
+ 3 => 860,
+ 4 => 1527,
+ 5 => 2180,
+ },
+ }
+ }
+
+ // Disabled cus not sorted
+ // #[test]
+ // fn buffersyncer_to_network() {
+ // assert_eq!(get_runtime().to_network(), get_network())
+ // }
+
+ #[test]
+ fn buffersyncer_from_network() {
+ assert_eq!(
+ BufferSyncer::from_network(&mut get_network()),
+ get_runtime()
+ )
+ }
+}
generated by cgit v1.2.3 (git 2.46.0) at 2026-03-02 07:21:02 +0000