nikdoof/rustical
1
1
Fork 0
mirror of https://github.com/lennart-k/rustical.git synced 2025-12-13 22:52:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

xml: EnumUnitVariants support untagged enums

Browse Source
This commit is contained in:
Lennart
2025-01-18 21:51:30 +01:00
parent e9610dc974
commit 8d7574290c
3 changed files with 179 additions and 65 deletions
Download Patch File Download Diff File Expand all files Collapse all files

199
crates/xml/derive/src/xml_enum.rs
View File

@@ -240,87 +240,164 @@ impl Enum {
}
pub fn impl_enum_unit_variants(&self) -> proc_macro2::TokenStream {
let ident = &self.ident;
if self.attrs.untagged.is_present() {
panic!("EnumUnitVariants not implemented for untagged enums");
}
let unit_enum_ident = self
.attrs
.unit_variants_ident
.as_ref()
.expect("unit_variants_ident no set");
let ident = &self.ident;
let tagged_variants: Vec<_> = self
.variants
.iter()
.filter(|variant| !variant.attrs.other.is_present())
.collect();
if self.attrs.untagged.is_present() {
let variant_branches: Vec<_> = self
.variants
.iter()
.map(|variant| {
let variant_type = variant.deserializer_type();
let variant_ident = &variant.variant.ident;
quote! {
#variant_ident (<#variant_type as ::rustical_xml::EnumUnitVariants>::UnitVariants)
}
})
.collect();
let variant_outputs: Vec<_> = tagged_variants
.iter()
.map(|variant| {
let ns = match &variant.attrs.common.ns {
Some(ns) => quote! { Some(#ns) },
None => quote! { None },
};
let variant_idents: Vec<_> = self
.variants
.iter()
.map(|variant| &variant.variant.ident)
.collect();
let unit_to_output_branches = variant_idents.iter().map(|variant_ident| {
quote! { #unit_enum_ident::#variant_ident(val) => val.into() }
});
let str_to_unit_branches = self.variants.iter().map(|variant| {
let variant_type = variant.deserializer_type();
let variant_ident = &variant.variant.ident;
quote! {
if let Ok(name) = <#variant_type as ::rustical_xml::EnumUnitVariants>::UnitVariants::from_str(val) {
return Ok(Self::#variant_ident(name))
}
}
});
let from_enum_to_unit_branches = variant_idents.iter().map(|variant_ident| {
quote! { #ident::#variant_ident(val) => #unit_enum_ident::#variant_ident(val.into()) }
});
quote! {
#[derive(Clone, Debug, PartialEq)]
pub enum #unit_enum_ident {
#(#variant_branches),*
}
impl ::rustical_xml::EnumUnitVariants for #ident {
type UnitVariants = #unit_enum_ident;
}
impl From<#unit_enum_ident> for (Option<::quick_xml::name::Namespace<'static>>, &'static str) {
fn from(val: #unit_enum_ident) -> Self {
match val {
#(#unit_to_output_branches),*
}
}
}
impl From<#ident> for #unit_enum_ident {
fn from(val: #ident) -> Self {
match val {
#(#from_enum_to_unit_branches),*
}
}
}
impl ::std::str::FromStr for #unit_enum_ident {
type Err = ::rustical_xml::FromStrError;
fn from_str(val: &str) -> Result<Self, Self::Err> {
#(#str_to_unit_branches);*
Err(::rustical_xml::FromStrError)
}
}
}
} else {
let tagged_variants: Vec<_> = self
.variants
.iter()
.filter(|variant| !variant.attrs.other.is_present())
.collect();
let variant_outputs: Vec<_> = tagged_variants
.iter()
.map(|variant| {
let ns = match &variant.attrs.common.ns {
Some(ns) => quote! { Some(#ns) },
None => quote! { None },
};
let b_xml_name = variant.xml_name().value();
let xml_name = String::from_utf8_lossy(&b_xml_name);
quote! {(#ns, #xml_name)}
})
.collect();
let variant_idents: Vec<_> = tagged_variants
.iter()
.map(|variant| &variant.variant.ident)
.collect();
let unit_to_output_branches =
variant_idents
.iter()
.zip(&variant_outputs)
.map(|(variant_ident, out)| {
quote! { #unit_enum_ident::#variant_ident => #out }
});
let from_enum_to_unit_branches = variant_idents.iter().map(|variant_ident| {
quote! { #ident::#variant_ident { .. } => #unit_enum_ident::#variant_ident }
});
let str_to_unit_branches = tagged_variants.iter().map(|variant| {
let variant_ident = &variant.variant.ident;
let b_xml_name = variant.xml_name().value();
let xml_name = String::from_utf8_lossy(&b_xml_name);
quote! {(#ns, #xml_name)}
})
.collect();
quote! { #xml_name => Ok(#unit_enum_ident::#variant_ident) }
});
let variant_idents: Vec<_> = tagged_variants
.iter()
.map(|variant| &variant.variant.ident)
.collect();
quote! {
#[derive(Clone, Debug, PartialEq)]
pub enum #unit_enum_ident {
#(#variant_idents),*
}
let unit_to_output_branches =
variant_idents
.iter()
.zip(&variant_outputs)
.map(|(variant_ident, out)| {
quote! { #unit_enum_ident::#variant_ident => #out }
});
let from_enum_to_unit_branches = variant_idents.iter().map(|variant_ident| {
quote! { #ident::#variant_ident { .. } => #unit_enum_ident::#variant_ident }
});
impl ::rustical_xml::EnumUnitVariants for #ident {
type UnitVariants = #unit_enum_ident;
}
let str_to_unit_branches = tagged_variants.iter().map(|variant| {
let variant_ident = &variant.variant.ident;
let b_xml_name = variant.xml_name().value();
let xml_name = String::from_utf8_lossy(&b_xml_name);
quote! { #xml_name => Ok(#unit_enum_ident::#variant_ident) }
});
quote! {
pub enum #unit_enum_ident {
#(#variant_idents),*
}
impl From<#unit_enum_ident> for (Option<::quick_xml::name::Namespace<'static>>, &'static str) {
fn from(val: #unit_enum_ident) -> Self {
match val {
#(#unit_to_output_branches),*
impl From<#unit_enum_ident> for (Option<::quick_xml::name::Namespace<'static>>, &'static str) {
fn from(val: #unit_enum_ident) -> Self {
match val {
#(#unit_to_output_branches),*
}
}
}
}
impl From<#ident> for #unit_enum_ident {
fn from(val: #ident) -> Self {
match val {
#(#from_enum_to_unit_branches),*
impl From<#ident> for #unit_enum_ident {
fn from(val: #ident) -> Self {
match val {
#(#from_enum_to_unit_branches),*
}
}
}
}
impl ::std::str::FromStr for #unit_enum_ident {
type Err = ::rustical_xml::FromStrError;
impl ::std::str::FromStr for #unit_enum_ident {
type Err = ::rustical_xml::FromStrError;
fn from_str(val: &str) -> Result<Self, Self::Err> {
match val {
#(#str_to_unit_branches),*,
_ => Err(::rustical_xml::FromStrError)
fn from_str(val: &str) -> Result<Self, Self::Err> {
match val {
#(#str_to_unit_branches),*,
_ => Err(::rustical_xml::FromStrError)
}
}
}
}