use core::fmt;
use std::fmt::Formatter;
use std::marker::PhantomData;
use serde::de::Visitor;
use serde::{de, Deserialize, Deserializer, Serialize, Serializer};
use crate::{AnyStr, BinFactory, NewSStr, NewStr, SStr, Str, StrFactory};
impl Serialize for Str {
#[inline]
fn serialize<S>(&self, serializer: S) -> Result<<S as Serializer>::Ok, <S as Serializer>::Error>
where
S: Serializer,
{
serializer.serialize_str(self.as_str())
}
}
impl<'de> Deserialize<'de> for Str {
#[inline]
fn deserialize<D>(deserializer: D) -> Result<Self, <D as Deserializer<'de>>::Error>
where
D: Deserializer<'de>,
{
deserializer.deserialize_str(RcStrVisitor::<NewStr>::new())
}
}
impl Serialize for SStr {
#[inline]
fn serialize<S>(&self, serializer: S) -> Result<<S as Serializer>::Ok, <S as Serializer>::Error>
where
S: Serializer,
{
serializer.serialize_str(self.as_str())
}
}
impl<'de> Deserialize<'de> for SStr {
#[inline]
fn deserialize<D>(deserializer: D) -> Result<Self, <D as Deserializer<'de>>::Error>
where
D: Deserializer<'de>,
{
deserializer.deserialize_str(RcStrVisitor::<NewSStr>::new())
}
}
struct RcStrVisitor<T> {
_phantom: PhantomData<T>,
}
impl<T> RcStrVisitor<T> {
pub fn new() -> Self {
Self {
_phantom: PhantomData::default(),
}
}
}
impl<'de, T> Visitor<'de> for RcStrVisitor<T>
where
T: StrFactory,
{
type Value = AnyStr<<T::TBinFactory as BinFactory>::T>;
fn expecting(&self, formatter: &mut Formatter<'_>) -> fmt::Result {
formatter.write_str("expecting a string")
}
#[inline]
fn visit_str<E>(self, v: &str) -> Result<Self::Value, E>
where
E: de::Error,
{
Ok(T::copy_from_str(v))
}
#[inline]
fn visit_string<E>(self, v: String) -> Result<Self::Value, E>
where
E: de::Error,
{
Ok(T::from_given_string(v))
}
}