Compare commits

...

10 Commits

Author SHA1 Message Date
pantonshire 754af04b47 version bump 2 years ago
pantonshire 605630a824
Merge pull request #4 from pantonshire/de_no_alloc
fix: deserialize all serde strings without allocating
2 years ago
pantonshire 615db6be6d tests for serde deserialize 2 years ago
pantonshire dca10eb38d fix for generated serde deserialize impl 2 years ago
pantonshire 406b42f0ca use CappedString and CowCappedString in generated unscribe fn 2 years ago
pantonshire 0483c756f0 add macro backtrace to test flags 2 years ago
pantonshire c8ca0b7e60 justfile fix 2 years ago
pantonshire 14effd44c5 replace test shell script with justfile 2 years ago
pantonshire 9487a7f533 add miri test script 2 years ago
pantonshire a799521e02 unit tests for (cow) capped string deserialisation 2 years ago

@ -1,6 +1,6 @@
[package] [package]
name = "enumscribe" name = "enumscribe"
version = "0.3.0" version = "0.3.1"
authors = ["Tom Panton <pantonshire@gmail.com>"] authors = ["Tom Panton <pantonshire@gmail.com>"]
edition = "2018" edition = "2018"
license = "MIT" license = "MIT"

@ -11,6 +11,49 @@ pub enum CowCappedString<'a, const N: usize> {
Owned(CappedString<N>), Owned(CappedString<N>),
} }
impl<'a, const N: usize> CowCappedString<'a, N> {
/// Returns the string data contained by this `CowCappedString`.
#[inline]
#[must_use]
pub fn as_str(&self) -> &str {
match self {
CowCappedString::Borrowed(s) => s,
CowCappedString::Owned(s) => s,
}
}
/// Returns a new `CappedString` with capacity `M` containing the string converted to
/// uppercase. Returns `None` if the uppercase-converted string is longer than `M` bytes.
#[inline]
#[must_use]
pub fn to_uppercase<const M: usize>(&self) -> Option<CappedString<M>> {
CappedString::<M>::uppercase_from_str(self)
}
}
impl<'a, const N: usize> Deref for CowCappedString<'a, N> {
type Target = str;
#[inline]
fn deref(&self) -> &Self::Target {
self.as_str()
}
}
impl<'a, const N: usize> AsRef<str> for CowCappedString<'a, N> {
#[inline]
fn as_ref(&self) -> &str {
self
}
}
impl<'a, const N: usize> Borrow<str> for CowCappedString<'a, N> {
#[inline]
fn borrow(&self) -> &str {
self
}
}
#[cfg(feature = "serde")] #[cfg(feature = "serde")]
impl<'de, const N: usize> serde::Deserialize<'de> for CowCappedString<'de, N> { impl<'de, const N: usize> serde::Deserialize<'de> for CowCappedString<'de, N> {
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error> fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
@ -74,16 +117,45 @@ pub struct CappedString<const N: usize> {
} }
impl<const N: usize> CappedString<N> { impl<const N: usize> CappedString<N> {
/// Returns a new `CappedString` containing a copy of the given string data. Returns an error /// Returns a new `CappedString` containing a copy of the given string data. Returns `None` if
/// if the string data is larger than `N` bytes. /// the string data is larger than `N` bytes.
#[inline] #[inline]
#[must_use] #[must_use]
pub fn from_str(s: &str) -> Option<Self> { pub fn from_str(s: &str) -> Option<Self> {
unsafe { Self::from_utf8_unchecked(s.as_bytes()) } unsafe { Self::from_utf8_unchecked(s.as_bytes()) }
} }
/// Returns a new `CappedString` containing an uppercase conversion of the given string data.
/// Returns `None` if the converted string is larger than `N` bytes.
#[inline]
#[must_use]
pub fn uppercase_from_str(s: &str) -> Option<Self> {
let mut buf = [0u8; N];
let mut cursor = 0usize;
for c_orig in s.chars() {
for c_upper in c_orig.to_uppercase() {
let encode_buf = cursor
.checked_add(c_upper.len_utf8())
.and_then(|encode_buf_end| buf.get_mut(cursor..encode_buf_end))?;
// FIXME: avoid the panic asm that gets generated for this encode (can never panic,
// as we always have at least `c_upper.len_utf8()` buffer space).
let encoded = c_upper.encode_utf8(encode_buf);
cursor = cursor.checked_add(encoded.len())?;
}
}
let filled_buf = buf.get(..cursor)?;
// SAFETY:
// `filled_buf` has been filled with a sequence of bytes obtained from `char::encode_utf8`,
// so it is valid UTF-8.
unsafe { Self::from_utf8_unchecked(filled_buf) }
}
/// Returns a new `CappedString` containing a copy of the given UTF-8 encoded string data. /// Returns a new `CappedString` containing a copy of the given UTF-8 encoded string data.
/// Returns an error if more than `N` bytes of data are given. /// Returns `None` if more than `N` bytes of data are given.
/// ///
/// # Safety /// # Safety
/// - `bs` must be valid UTF-8. /// - `bs` must be valid UTF-8.
@ -134,32 +206,11 @@ impl<const N: usize> CappedString<N> {
} }
/// Returns a new `CappedString` with capacity `M` containing the string converted to /// Returns a new `CappedString` with capacity `M` containing the string converted to
/// uppercase. Returns an error if the uppercase-converted string is longer than `M` bytes. /// uppercase. Returns `None` if the uppercase-converted string is longer than `M` bytes.
#[inline] #[inline]
#[must_use] #[must_use]
pub fn to_uppercase<const M: usize>(&self) -> Option<CappedString<M>> { pub fn to_uppercase<const M: usize>(&self) -> Option<CappedString<M>> {
let mut buf = [0u8; M]; CappedString::<M>::uppercase_from_str(self)
let mut cursor = 0usize;
for c_orig in self.as_str().chars() {
for c_upper in c_orig.to_uppercase() {
let encode_buf = cursor
.checked_add(c_upper.len_utf8())
.and_then(|encode_buf_end| buf.get_mut(cursor..encode_buf_end))?;
// FIXME: avoid the panic asm that gets generated for this encode (can never panic,
// as we always have at least `c_upper.len_utf8()` buffer space).
let encoded = c_upper.encode_utf8(encode_buf);
cursor = cursor.checked_add(encoded.len())?;
}
}
let filled_buf = buf.get(..cursor)?;
// SAFETY:
// `filled_buf` has been filled with a sequence of bytes obtained from `char::encode_utf8`,
// so it is valid UTF-8.
unsafe { CappedString::from_utf8_unchecked(filled_buf) }
} }
} }
@ -186,6 +237,20 @@ impl<const N: usize> Borrow<str> for CappedString<N> {
} }
} }
impl<const N: usize> PartialEq for CappedString<N> {
fn eq(&self, other: &Self) -> bool {
self.as_str() == other.as_str()
}
}
impl<const N: usize> Eq for CappedString<N> {}
impl<const N: usize> PartialEq<str> for CappedString<N> {
fn eq(&self, other: &str) -> bool {
self.as_str() == other
}
}
#[cfg(feature = "serde")] #[cfg(feature = "serde")]
impl<'de, const N: usize> serde::Deserialize<'de> for CappedString<N> { impl<'de, const N: usize> serde::Deserialize<'de> for CappedString<N> {
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error> fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
@ -227,7 +292,115 @@ impl<'de, const N: usize> serde::de::Visitor<'de> for CappedStringVisitor<N> {
#[cfg(test)] #[cfg(test)]
mod tests { mod tests {
use super::CappedString; use super::{CappedString, CowCappedString};
#[cfg(feature = "serde")]
#[test]
fn test_cow_capped_string_deserialize() {
struct DeBorrowedOnly<const N: usize>(String);
impl<'de, const N: usize> serde::Deserialize<'de> for DeBorrowedOnly<N> {
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
where
D: serde::Deserializer<'de>
{
match CowCappedString::<'de, N>::deserialize(deserializer)? {
CowCappedString::Borrowed(s) => Ok(Self(s.to_owned())),
CowCappedString::Owned(_) => {
Err(serde::de::Error::custom("expected borrowed CowCappedString"))
},
}
}
}
struct DeOwnedOnly<const N: usize>(String);
impl<'de, const N: usize> serde::Deserialize<'de> for DeOwnedOnly<N> {
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
where
D: serde::Deserializer<'de>
{
match CowCappedString::<'de, N>::deserialize(deserializer)? {
CowCappedString::Borrowed(_) => {
Err(serde::de::Error::custom("expected owned CowCappedString"))
},
CowCappedString::Owned(s) => Ok(Self(s.to_owned())),
}
}
}
{
let DeBorrowedOnly(s) = serde_json::from_str::<DeBorrowedOnly<5>>(
r#""hello""#
).unwrap();
assert_eq!(s, "hello");
}
{
let DeBorrowedOnly(s) = serde_json::from_str::<DeBorrowedOnly<0>>(
r#""hello""#
).unwrap();
assert_eq!(s, "hello");
}
{
let s = serde_json::from_str::<DeOwnedOnly<5>>(
r#""hello""#
);
assert!(s.is_err());
}
{
let DeOwnedOnly(s) = serde_json::from_str::<DeOwnedOnly<3>>(
r#""\u87f9""#
).unwrap();
assert_eq!(s, "蟹");
}
{
let s = serde_json::from_str::<DeBorrowedOnly<3>>(
r#""\u87f9""#
);
assert!(s.is_err());
}
}
#[cfg(feature = "serde")]
#[test]
fn test_capped_string_deserialize() {
{
let s = serde_json::from_str::<CappedString<5>>(
r#""hello""#
).unwrap();
assert_eq!(s.as_str(), "hello");
}
{
let s = serde_json::from_str::<CappedString<4>>(
r#""hello""#
);
assert!(s.is_err());
}
{
let s = serde_json::from_str::<CappedString<10>>(
r#""hello""#
).unwrap();
assert_eq!(s.as_str(), "hello");
}
{
let s = serde_json::from_str::<CappedString<12>>(
r#""hello\tworld\n""#
).unwrap();
assert_eq!(s.as_str(), "hello\tworld\n");
}
{
let s = serde_json::from_str::<CappedString<3>>(
r#""\u87f9""#
).unwrap();
assert_eq!(s.as_str(), "蟹");
}
{
let s = serde_json::from_str::<CappedString<2>>(
r#""\u87f9""#
);
assert!(s.is_err());
}
}
#[test] #[test]
fn test_capped_string_uppercase() { fn test_capped_string_uppercase() {
@ -247,8 +420,8 @@ mod tests {
assert_eq!(s2.as_str(), "HELLO"); assert_eq!(s2.as_str(), "HELLO");
} }
{ {
let s1 = CappedString::<5>::from_str("hello").unwrap(); let s = CappedString::<5>::from_str("hello").unwrap();
assert!(s1.to_uppercase::<4>().is_none()); assert!(s.to_uppercase::<4>().is_none());
} }
{ {
let s1 = CappedString::<5>::from_str("groß").unwrap(); let s1 = CappedString::<5>::from_str("groß").unwrap();

@ -1,6 +1,6 @@
[package] [package]
name = "enumscribe_derive" name = "enumscribe_derive"
version = "0.3.0" version = "0.3.1"
authors = ["Tom Panton <pantonshire@gmail.com>"] authors = ["Tom Panton <pantonshire@gmail.com>"]
edition = "2018" edition = "2018"
license = "MIT" license = "MIT"

@ -12,7 +12,45 @@ use crate::{CASE_INSENSITIVE, CRATE_ATTR, IGNORE, NAME, OTHER};
#[derive(Clone)] #[derive(Clone)]
pub(crate) struct Enum<'a> { pub(crate) struct Enum<'a> {
pub(crate) variants: Vec<Variant<'a>>, variants: Box<[Variant<'a>]>,
name_capacity: usize,
name_upper_capacity: usize,
}
impl<'a> Enum<'a> {
pub(crate) fn new(variants: Box<[Variant<'a>]>) -> Self {
let name_capacity = variants
.iter()
.filter_map(|v| v.v_type.as_named())
.map(|named| named.name().len())
.max()
.unwrap_or(0);
let name_upper_capacity = variants
.iter()
.filter_map(|v| v.v_type.as_named())
.map(|named| named.name_upper().len())
.max()
.unwrap_or(0);
Self {
variants,
name_capacity,
name_upper_capacity,
}
}
pub(crate) fn variants(&self) -> &[Variant<'a>] {
&self.variants
}
pub(crate) fn name_capacity(&self) -> usize {
self.name_capacity
}
pub(crate) fn name_upper_capacity(&self) -> usize {
self.name_upper_capacity
}
} }
#[derive(Clone)] #[derive(Clone)]
@ -25,14 +63,69 @@ pub(crate) struct Variant<'a> {
#[derive(Clone)] #[derive(Clone)]
pub(crate) enum VariantType<'a> { pub(crate) enum VariantType<'a> {
Ignore, Ignore,
Named { Named(NamedVariant),
name: String, Other(OtherVariant<'a>),
}
impl<'a> VariantType<'a> {
pub(crate) fn as_named(&self) -> Option<&NamedVariant> {
match self {
Self::Named(named) => Some(named),
_ => None,
}
}
}
#[derive(Clone)]
pub(crate) struct NamedVariant {
name: Box<str>,
name_upper: Box<str>,
constructor: VariantConstructor,
case_insensitive: bool,
}
impl NamedVariant {
pub(crate) fn new(
name: Box<str>,
constructor: VariantConstructor, constructor: VariantConstructor,
case_insensitive: bool, case_insensitive: bool
}, ) -> Self
Other { {
field_name: Option<&'a Ident>, let name_upper = char_wise_uppercase(&name);
}, Self {
name,
name_upper,
constructor,
case_insensitive,
}
}
pub(crate) fn name(&self) -> &str {
&self.name
}
pub(crate) fn name_upper(&self) -> &str {
&self.name_upper
}
pub(crate) fn constructor(&self) -> VariantConstructor {
self.constructor
}
pub(crate) fn case_insensitive(&self) -> bool {
self.case_insensitive
}
}
#[derive(Clone)]
pub(crate) struct OtherVariant<'a> {
field_name: Option<&'a Ident>,
}
impl<'a> OtherVariant<'a> {
pub(crate) fn field_name(&self) -> Option<&'a Ident> {
self.field_name
}
} }
#[derive(Clone, Copy, Debug)] #[derive(Clone, Copy, Debug)]
@ -58,20 +151,18 @@ impl<'a> Variant<'a> {
match &self.v_type { match &self.v_type {
VariantType::Ignore => Ok(None), VariantType::Ignore => Ok(None),
VariantType::Named { VariantType::Named(named) => {
name, constructor, .. let constructor_tokens = named.constructor().empty_toks();
} => {
let constructor_tokens = constructor.empty();
let pattern = quote! { #enum_ident::#variant_ident #constructor_tokens }; let pattern = quote! { #enum_ident::#variant_ident #constructor_tokens };
Ok(Some((pattern, named_fn(self, enum_ident, name)?))) Ok(Some((pattern, named_fn(self, enum_ident, named.name())?)))
} }
VariantType::Other { field_name } => { VariantType::Other(other) => {
let field_name_tokens = match field_name { let field_name_tokens = match other.field_name() {
Some(field_name) => field_name.to_token_stream(), Some(field_name) => field_name.to_token_stream(),
None => quote! { __enumscribe_other_inner }, None => quote! { __enumscribe_other_inner },
}; };
let pattern = match field_name { let pattern = match other.field_name() {
Some(_) => quote! { #enum_ident::#variant_ident{#field_name_tokens} }, Some(_) => quote! { #enum_ident::#variant_ident{#field_name_tokens} },
None => quote! { #enum_ident::#variant_ident(#field_name_tokens) }, None => quote! { #enum_ident::#variant_ident(#field_name_tokens) },
}; };
@ -85,7 +176,7 @@ impl<'a> Variant<'a> {
} }
impl VariantConstructor { impl VariantConstructor {
pub(crate) fn empty(&self) -> TokenStream2 { pub(crate) fn empty_toks(&self) -> TokenStream2 {
match self { match self {
VariantConstructor::None => quote! {}, VariantConstructor::None => quote! {},
VariantConstructor::Paren => quote! { () }, VariantConstructor::Paren => quote! { () },
@ -220,7 +311,7 @@ pub(crate) fn parse_enum<'a>(data: &'a DataEnum, attrs: &'a [Attribute]) -> Macr
Variant { Variant {
data: variant, data: variant,
v_type: VariantType::Other { field_name }, v_type: VariantType::Other(OtherVariant { field_name }),
span: variant_span, span: variant_span,
} }
} else { } else {
@ -279,13 +370,12 @@ pub(crate) fn parse_enum<'a>(data: &'a DataEnum, attrs: &'a [Attribute]) -> Macr
Fields::Unit => VariantConstructor::None, Fields::Unit => VariantConstructor::None,
}; };
let named = NamedVariant::new(name.into_boxed_str(), constructor, case_insensitive);
let v_type = VariantType::Named(named);
Variant { Variant {
data: variant, data: variant,
v_type: VariantType::Named { v_type,
name,
constructor,
case_insensitive,
},
span: variant_span, span: variant_span,
} }
}; };
@ -293,5 +383,13 @@ pub(crate) fn parse_enum<'a>(data: &'a DataEnum, attrs: &'a [Attribute]) -> Macr
variants.push(scribe_variant); variants.push(scribe_variant);
} }
Ok(Enum { variants }) Ok(Enum::new(variants.into_boxed_slice()))
} }
fn char_wise_uppercase(s: &str) -> Box<str> {
// Use the same uppercase algorithm as `enumscribe::internal::capped_string`.
s.chars()
.flat_map(char::to_uppercase)
.collect::<String>()
.into_boxed_str()
}

@ -59,9 +59,9 @@ where
let enum_ident = &input.ident; let enum_ident = &input.ident;
let mut match_arms = Vec::with_capacity(parsed_enum.variants.len()); let mut match_arms = Vec::with_capacity(parsed_enum.variants().len());
for variant in parsed_enum.variants.iter() { for variant in parsed_enum.variants().iter() {
match variant.match_variant(enum_ident, &named_fn, &other_fn) { match variant.match_variant(enum_ident, &named_fn, &other_fn) {
Ok(Some((pattern, result))) => match_arms.push(quote! { #pattern => #result }), Ok(Some((pattern, result))) => match_arms.push(quote! { #pattern => #result }),
Ok(None) => return ignore_err_fn(variant, enum_ident).into(), Ok(None) => return ignore_err_fn(variant, enum_ident).into(),
@ -102,9 +102,9 @@ where
let enum_ident = &input.ident; let enum_ident = &input.ident;
let mut ignore_variant = false; let mut ignore_variant = false;
let mut match_arms = Vec::with_capacity(parsed_enum.variants.len()); let mut match_arms = Vec::with_capacity(parsed_enum.variants().len());
for variant in parsed_enum.variants.iter() { for variant in parsed_enum.variants().iter() {
match variant.match_variant(enum_ident, &named_fn, &other_fn) { match variant.match_variant(enum_ident, &named_fn, &other_fn) {
Ok(Some((pattern, result))) => match_arms.push(quote! { #pattern => #result }), Ok(Some((pattern, result))) => match_arms.push(quote! { #pattern => #result }),
Ok(None) => ignore_variant = true, Ok(None) => ignore_variant = true,
@ -192,30 +192,28 @@ where
let mut case_sensitive_arms = Vec::new(); let mut case_sensitive_arms = Vec::new();
let mut case_insensitive_arms = Vec::new(); let mut case_insensitive_arms = Vec::new();
for variant in parsed_enum.variants.iter() { for variant in parsed_enum.variants().iter() {
let variant_ident = &variant.data.ident; let variant_ident = &variant.data.ident;
match &variant.v_type { match &variant.v_type {
VariantType::Ignore => (), VariantType::Ignore => (),
VariantType::Named { VariantType::Named(named) => {
name, let match_pattern = if named.case_insensitive() {
constructor, let uppercase_name = named.name_upper();
case_insensitive, quote! { #uppercase_name }
} => {
let match_pattern = if *case_insensitive {
let lowercase_name = name.to_lowercase();
quote! { #lowercase_name }
} else { } else {
let name = named.name();
quote! { #name } quote! { #name }
}; };
let constructor_tokens = constructor.empty(); let constructor_tokens = named.constructor().empty_toks();
let constructed_variant = let constructed_variant = quote! {
quote! { #enum_ident::#variant_ident #constructor_tokens }; #enum_ident::#variant_ident #constructor_tokens
};
let match_result = named_fn(constructed_variant); let match_result = named_fn(constructed_variant);
if *case_insensitive { if named.case_insensitive() {
&mut case_insensitive_arms &mut case_insensitive_arms
} else { } else {
&mut case_sensitive_arms &mut case_sensitive_arms
@ -223,11 +221,11 @@ where
.push(quote! { #match_pattern => #match_result }); .push(quote! { #match_pattern => #match_result });
} }
VariantType::Other { field_name } => { VariantType::Other(other) => {
let unscribe_value = let unscribe_value =
quote! { <_ as ::std::convert::Into<_>>::into(#match_against) }; quote! { <_ as ::std::convert::Into<_>>::into(#match_against) };
let constructed_variant = match field_name { let constructed_variant = match other.field_name() {
None => quote! { None => quote! {
#enum_ident::#variant_ident(#unscribe_value) #enum_ident::#variant_ident(#unscribe_value)
}, },
@ -251,10 +249,23 @@ where
let case_insensitive_match = if case_insensitive_arms.is_empty() { let case_insensitive_match = if case_insensitive_arms.is_empty() {
None None
} else { } else {
let match_against_upper_ident = quote! { __enumscribe_unscribe_uppercase };
let name_upper_cap = parsed_enum.name_upper_capacity();
Some(quote! { Some(quote! {
let __enumscribe_unscribe_lowercase = #match_against.to_lowercase(); match ::enumscribe
match __enumscribe_unscribe_lowercase.as_str() { ::internal
#(#case_insensitive_arms,)* ::capped_string
::CappedString
::<#name_upper_cap>
::uppercase_from_str(#match_against)
{
Some(#match_against_upper_ident) => {
match &*#match_against_upper_ident {
#(#case_insensitive_arms,)*
#other_arm,
}
},
#other_arm, #other_arm,
} }
}) })
@ -659,23 +670,22 @@ pub fn derive_enum_serialize(input: TokenStream) -> TokenStream {
let mut match_arms = Vec::new(); let mut match_arms = Vec::new();
let mut ignore_variant = false; let mut ignore_variant = false;
for variant in parsed_enum.variants.iter() { for variant in parsed_enum.variants().iter() {
let variant_ident = &variant.data.ident; let variant_ident = &variant.data.ident;
match &variant.v_type { match &variant.v_type {
VariantType::Ignore => ignore_variant = true, VariantType::Ignore => ignore_variant = true,
VariantType::Named { VariantType::Named(named) => {
name, constructor, .. let constructor_tokens = named.constructor().empty_toks();
} => { let name = named.name();
let constructor_tokens = constructor.empty();
match_arms.push(quote! { match_arms.push(quote! {
#enum_ident::#variant_ident #constructor_tokens => #enum_ident::#variant_ident #constructor_tokens =>
#serializer_ident.serialize_str(#name) #serializer_ident.serialize_str(#name)
}) })
} }
VariantType::Other { field_name } => match field_name { VariantType::Other(other) => match other.field_name() {
Some(field_name) => match_arms.push(quote! { Some(field_name) => match_arms.push(quote! {
#enum_ident::#variant_ident { #field_name } => #enum_ident::#variant_ident { #field_name } =>
#serializer_ident.serialize_str(&#field_name) #serializer_ident.serialize_str(&#field_name)
@ -748,13 +758,14 @@ pub fn derive_enum_deserialize(input: TokenStream) -> TokenStream {
let enum_ident = &input.ident; let enum_ident = &input.ident;
let deserializer_ident = quote! { __enumscribe_deserializer }; let deserializer_ident = quote! { __enumscribe_deserializer };
let deserialized_cow_capped_str_ident = quote! { __enumscribe_deserialized_cow_capped_str };
let deserialized_str_ident = quote! { __enumscribe_deserialized_str }; let deserialized_str_ident = quote! { __enumscribe_deserialized_str };
let variant_strings = parsed_enum let variant_strings = parsed_enum
.variants .variants()
.iter() .iter()
.map(|variant| match &variant.v_type { .map(|variant| match &variant.v_type {
VariantType::Named { name, .. } => Some(name.as_str()), VariantType::Named(named) => Some(named.name()),
_ => None, _ => None,
}) })
.filter_map(|name| name) .filter_map(|name| name)
@ -771,22 +782,32 @@ pub fn derive_enum_deserialize(input: TokenStream) -> TokenStream {
::core::result::Result::Ok(#constructed_other_variant) ::core::result::Result::Ok(#constructed_other_variant)
}, },
|_| Ok(quote! { |_| Ok(quote! {
__enumscribe_deserialize_base_case => ::core::result::Result::Err( _ => ::core::result::Result::Err(
::serde::de::Error::unknown_variant( ::serde::de::Error::unknown_variant(
__enumscribe_deserialize_base_case, #deserialized_str_ident,
&[#(#variant_strings),*] &[#(#variant_strings),*]
) )
) )
}), }),
)); ));
let name_cap = parsed_enum.name_capacity();
(quote! { (quote! {
#[automatically_derived] #[automatically_derived]
impl<'de> ::serde::Deserialize<'de> for #enum_ident { impl<'de> ::serde::Deserialize<'de> for #enum_ident {
fn deserialize<D>(#deserializer_ident: D) -> ::core::result::Result<Self, D::Error> fn deserialize<D>(#deserializer_ident: D) -> ::core::result::Result<Self, D::Error>
where D: ::serde::Deserializer<'de> where D: ::serde::Deserializer<'de>
{ {
let #deserialized_str_ident = <&str as ::serde::Deserialize<'_>>::deserialize(#deserializer_ident)?; let #deserialized_cow_capped_str_ident = <
::enumscribe
::internal
::capped_string
::CowCappedString<'de, #name_cap>
as ::serde::Deserialize<'_>
>::deserialize(#deserializer_ident)?;
let #deserialized_str_ident = &*#deserialized_cow_capped_str_ident;
#main_match #main_match
} }
} }

@ -0,0 +1,25 @@
use enumscribe::EnumDeserialize;
#[test]
fn test_deserialize() {
#[derive(EnumDeserialize, Eq, PartialEq, Debug)]
enum E0 {
V0,
#[enumscribe(str = "baa", case_insensitive)]
V1,
#[enumscribe(str = "bAz\n", case_insensitive)]
V2,
#[enumscribe(str = "")]
V3,
}
assert_eq!(serde_json::from_str::<E0>(r#""V0""#).unwrap(), E0::V0);
assert!(serde_json::from_str::<E0>(r#""v0""#).is_err());
assert_eq!(serde_json::from_str::<E0>(r#""baa""#).unwrap(), E0::V1);
assert_eq!(serde_json::from_str::<E0>(r#""BAA""#).unwrap(), E0::V1);
assert_eq!(serde_json::from_str::<E0>(r#""BaA""#).unwrap(), E0::V1);
assert_eq!(serde_json::from_str::<E0>(r#""baz\n""#).unwrap(), E0::V2);
assert_eq!(serde_json::from_str::<E0>(r#""BAZ\n""#).unwrap(), E0::V2);
assert_eq!(serde_json::from_str::<E0>(r#""BaZ\n""#).unwrap(), E0::V2);
assert_eq!(serde_json::from_str::<E0>(r#""\u87f9""#).unwrap(), E0::V3);
}

@ -0,0 +1,6 @@
nightly := 'cargo +nightly'
rustc_nightly_flags := '-Z randomize-layout -Z macro-backtrace'
test:
RUSTFLAGS='{{rustc_nightly_flags}}' {{nightly}} miri test
Loading…
Cancel
Save