use {
crate::{
de::Deserializer,
error::{Error, ErrorCode::*, Result},
},
serde::de::{self, DeserializeSeed, EnumAccess, VariantAccess, Visitor},
};
pub struct EnumReader<'a, 'de: 'a> {
de: &'a mut Deserializer<'de>,
}
impl<'a, 'de> EnumReader<'a, 'de> {
pub fn new(de: &'a mut Deserializer<'de>) -> Self {
EnumReader { de }
}
}
impl<'de> EnumAccess<'de> for EnumReader<'_, 'de> {
type Error = Error;
type Variant = Self;
fn variant_seed<V>(self, seed: V) -> Result<(V::Value, Self::Variant)>
where
V: DeserializeSeed<'de>,
{
let val = seed.deserialize(&mut *self.de)?;
self.de.eat_shit()?;
if self.de.next_byte()? == b':' {
Ok((val, self))
} else {
self.de.fail(ExpectedMapColon)
}
}
}
impl<'de> VariantAccess<'de> for EnumReader<'_, 'de> {
type Error = Error;
fn unit_variant(self) -> Result<()> {
self.de.fail(ExpectedString)
}
fn newtype_variant_seed<T>(self, seed: T) -> Result<T::Value>
where
T: DeserializeSeed<'de>,
{
seed.deserialize(self.de)
}
fn tuple_variant<V>(self, _len: usize, visitor: V) -> Result<V::Value>
where
V: Visitor<'de>,
{
de::Deserializer::deserialize_seq(self.de, visitor)
}
fn struct_variant<V>(self, _fields: &'static [&'static str], visitor: V) -> Result<V::Value>
where
V: Visitor<'de>,
{
de::Deserializer::deserialize_map(self.de, visitor)
}
}