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

use anyhow::Context as _;
use batbox_time::Timer;
use futures::prelude::*;
use serde::{Deserialize, Serialize};
use std::fmt::Debug;
use std::pin::Pin;
use std::sync::Arc;

pub mod client;
#[cfg(not(target_arch = "wasm32"))]
pub mod server;
mod traffic;

#[cfg(not(target_arch = "wasm32"))]
pub use server::{Server, ServerHandle};
pub use traffic::*;

pub trait Message: Debug + Serialize + for<'de> Deserialize<'de> + Send + 'static + Unpin {}

impl<T: Debug + Serialize + for<'de> Deserialize<'de> + Send + 'static + Unpin> Message for T {}

pub fn serialize_message<T: Message>(message: T) -> Vec<u8> {
    let mut buf = Vec::new();
    let writer = flate2::write::GzEncoder::new(&mut buf, flate2::Compression::best());
    bincode::serialize_into(writer, &message).unwrap();
    buf
}

pub fn deserialize_message<T: Message>(data: &[u8]) -> anyhow::Result<T> {
    let reader = flate2::read::GzDecoder::new(data);
    bincode::deserialize_from(reader).context("Failed to deserialize message")
}

#[cfg(target_arch = "wasm32")]
pub trait Sender<T> {
    fn send(&mut self, message: T)
    where
        T: Message,
    {
        self.send_serialized(Arc::new(serialize_message(message)))
    }
    fn send_serialized(&mut self, data: Arc<Vec<u8>>);
}

#[cfg(not(target_arch = "wasm32"))]
pub trait Sender<T>: Send {
    fn send(&mut self, message: T)
    where
        T: Message,
    {
        self.send_serialized(Arc::new(serialize_message(message)))
    }
    fn send_serialized(&mut self, data: Arc<Vec<u8>>);
}

pub trait Receiver<T> {
    fn handle(&mut self, message: T);
}