use super::EventV2;
use crate::types::{Filter, SubscriptionId};
use serde::de::Error as DeError;
use serde::de::{Deserialize, Deserializer, IgnoredAny, SeqAccess, Visitor};
use serde::ser::{Serialize, SerializeSeq, Serializer};
#[cfg(feature = "speedy")]
use speedy::{Readable, Writable};
use std::fmt;
#[derive(Clone, Debug, Eq, PartialEq)]
#[cfg_attr(feature = "speedy", derive(Readable, Writable))]
pub enum ClientMessageV2 {
Event(Box<EventV2>),
Req(SubscriptionId, Vec<Filter>),
Close(SubscriptionId),
Auth(Box<EventV2>),
}
impl ClientMessageV2 {
#[allow(dead_code)]
pub(crate) fn mock() -> ClientMessageV2 {
ClientMessageV2::Event(Box::new(EventV2::mock()))
}
}
impl Serialize for ClientMessageV2 {
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
where
S: Serializer,
{
match self {
ClientMessageV2::Event(event) => {
let mut seq = serializer.serialize_seq(Some(2))?;
seq.serialize_element("EVENT")?;
seq.serialize_element(&event)?;
seq.end()
}
ClientMessageV2::Req(id, filters) => {
let mut seq = serializer.serialize_seq(Some(3))?;
seq.serialize_element("REQ")?;
seq.serialize_element(&id)?;
for filter in filters {
seq.serialize_element(&filter)?;
}
seq.end()
}
ClientMessageV2::Close(id) => {
let mut seq = serializer.serialize_seq(Some(2))?;
seq.serialize_element("CLOSE")?;
seq.serialize_element(&id)?;
seq.end()
}
ClientMessageV2::Auth(event) => {
let mut seq = serializer.serialize_seq(Some(2))?;
seq.serialize_element("AUTH")?;
seq.serialize_element(&event)?;
seq.end()
}
}
}
}
impl<'de> Deserialize<'de> for ClientMessageV2 {
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
where
D: Deserializer<'de>,
{
deserializer.deserialize_seq(ClientMessageVisitor)
}
}
struct ClientMessageVisitor;
impl<'de> Visitor<'de> for ClientMessageVisitor {
type Value = ClientMessageV2;
fn expecting(&self, f: &mut fmt::Formatter) -> fmt::Result {
write!(f, "a sequence of strings")
}
fn visit_seq<A>(self, mut seq: A) -> Result<ClientMessageV2, A::Error>
where
A: SeqAccess<'de>,
{
let word: &str = seq
.next_element()?
.ok_or_else(|| DeError::custom("Message missing initial string field"))?;
let mut output: Option<ClientMessageV2> = None;
if word == "EVENT" {
let event: EventV2 = seq
.next_element()?
.ok_or_else(|| DeError::custom("Message missing event field"))?;
output = Some(ClientMessageV2::Event(Box::new(event)))
} else if word == "REQ" {
let id: SubscriptionId = seq
.next_element()?
.ok_or_else(|| DeError::custom("Message missing id field"))?;
let mut filters: Vec<Filter> = vec![];
loop {
let f: Option<Filter> = seq.next_element()?;
match f {
None => break,
Some(fil) => filters.push(fil),
}
}
output = Some(ClientMessageV2::Req(id, filters))
} else if word == "CLOSE" {
let id: SubscriptionId = seq
.next_element()?
.ok_or_else(|| DeError::custom("Message missing id field"))?;
output = Some(ClientMessageV2::Close(id))
} else if word == "AUTH" {
let event: EventV2 = seq
.next_element()?
.ok_or_else(|| DeError::custom("Message missing event field"))?;
output = Some(ClientMessageV2::Auth(Box::new(event)))
}
while let Some(_ignored) = seq.next_element::<IgnoredAny>()? {}
match output {
Some(cm) => Ok(cm),
None => Err(DeError::custom(format!("Unknown Message: {word}"))),
}
}
}
#[cfg(test)]
mod test {
use super::*;
test_serde! {ClientMessageV2, test_client_message_serde}
}