use serde::{
de::{Deserializer, Visitor},
ser::Serializer,
};
use std::borrow::Cow;
use std::convert::AsRef;
use std::error::Error;
use std::fmt::{self, Formatter};
use std::marker::PhantomData;
pub type CowStr = Cow<'static, str>;
struct GimmeBytesVisitor<T>(PhantomData<T>);
impl<'de, T> Visitor<'de> for GimmeBytesVisitor<T>
where
for<'a> T: From<Vec<u8>> + From<&'a [u8]>,
{
type Value = T;
fn expecting(&self, formatter: &mut Formatter) -> fmt::Result {
write!(formatter, "some sorta bytes")
}
fn visit_bytes<E: Error>(self, v: &[u8]) -> Result<Self::Value, E> {
Ok(v.into())
}
fn visit_borrowed_bytes<E: Error>(self, v: &'de [u8]) -> Result<Self::Value, E> {
Ok(v.into())
}
fn visit_byte_buf<E: Error>(self, v: Vec<u8>) -> Result<Self::Value, E> {
Ok(v.into())
}
}
pub fn deserialize_owned_bytes<'de, D>(deserializer: D) -> Result<Vec<u8>, D::Error>
where
D: Deserializer<'de>,
{
deserializer.deserialize_byte_buf(GimmeBytesVisitor(PhantomData))
}
pub fn serialize_bytes<T, S>(t: &T, serializer: S) -> Result<S::Ok, S::Error>
where
S: Serializer,
T: AsRef<[u8]>,
{
serializer.serialize_bytes(t.as_ref())
}
pub trait MessageTypeId {
const MSG_TYPE_ID: u8;
fn msg_type_id(&self) -> u8 {
Self::MSG_TYPE_ID
}
}
macro_rules! message_type_id {
($mtype:ty, $id:expr) => {
impl $crate::common::MessageTypeId for $mtype {
const MSG_TYPE_ID: u8 = $id;
}
};
}
macro_rules! message_type_ids {
{$($mtype:ty = $id:expr),*} => {
$(
message_type_id!($mtype, $id);
)*
}
}