xml: new variant flag and fixes to generic types

This commit is contained in:
Lennart
2024-12-25 10:21:09 +01:00
parent 92b7be5591
commit 5e2717e130
8 changed files with 194 additions and 82 deletions

View File

@@ -19,6 +19,7 @@ pub struct VariantAttrs {
#[darling(flatten)] #[darling(flatten)]
pub common: TagAttrs, pub common: TagAttrs,
pub other: Flag, pub other: Flag,
pub skip_deserializing: Flag,
} }
#[derive(Default, FromDeriveInput, Clone)] #[derive(Default, FromDeriveInput, Clone)]

View File

@@ -1,6 +1,6 @@
use core::panic; use core::panic;
use super::attrs::EnumAttrs; use super::{attrs::EnumAttrs, get_generic_type};
use crate::de::attrs::VariantAttrs; use crate::de::attrs::VariantAttrs;
use darling::{FromDeriveInput, FromVariant}; use darling::{FromDeriveInput, FromVariant};
use heck::ToKebabCase; use heck::ToKebabCase;
@@ -28,29 +28,92 @@ impl Variant {
)) ))
} }
pub fn tagged_branch(&self) -> proc_macro2::TokenStream { fn skip_de(&self) -> bool {
let ident = self.ident(); self.attrs.skip_deserializing.is_present()
let variant_name = self.xml_name();
match (self.attrs.other.is_present(), &self.variant.fields) {
(_, Fields::Named(_)) => {
panic!(
"struct variants are not supported, please use a tuple variant with a struct"
)
} }
(false, Fields::Unnamed(FieldsUnnamed { unnamed, .. })) => {
fn variant_type(&self) -> syn::Type {
match &self.variant.fields {
Fields::Named(_) => panic!(
"struct variants are not supported, please use a tuple variant with a struct"
),
Fields::Unit => syn::Type::Path(syn::parse_str("::rustical_xml::Unit").unwrap()),
Fields::Unnamed(FieldsUnnamed { unnamed, .. }) => {
if unnamed.len() != 1 { if unnamed.len() != 1 {
panic!("tuple variants should contain exactly one element"); panic!("tuple variants should contain exactly one element");
} }
let field = unnamed.iter().next().unwrap(); let field = unnamed.iter().next().unwrap();
field.ty.to_owned()
}
}
}
fn is_optional(&self) -> bool {
if let syn::Type::Path(syn::TypePath { path, .. }) = self.variant_type() {
if path.segments.len() != 1 {
return false;
}
let type_ident = &path.segments.first().unwrap().ident;
let option: syn::Ident = syn::parse_str("Option").unwrap();
return type_ident == &option;
}
false
}
/// The type to deserialize to
/// - type Option<T> => optional: deserialize with T
/// - flatten Vec<T>: deserialize with T
/// - deserialize with T
pub fn deserializer_type(&self) -> syn::Type {
let ty = self.variant_type();
if self.is_optional() {
return get_generic_type(&ty).unwrap().to_owned();
}
ty
}
pub fn tagged_branch(&self) -> Option<proc_macro2::TokenStream> {
if self.skip_de() {
return None;
}
let ident = self.ident();
let variant_name = self.xml_name();
let deserializer_type = self.deserializer_type();
Some(
match (
self.attrs.other.is_present(),
&self.variant.fields,
self.is_optional(),
) {
(_, Fields::Named(_), _) => {
panic!(
"struct variants are not supported, please use a tuple variant with a struct"
)
}
(false, Fields::Unnamed(FieldsUnnamed { unnamed, .. }), true) => {
if unnamed.len() != 1 {
panic!("tuple variants should contain exactly one element");
}
quote! { quote! {
#variant_name => { #variant_name => {
let val = #field::deserialize(reader, start, empty)?; let val = Some(<#deserializer_type as ::rustical_xml::XmlDeserialize>::deserialize(reader, start, empty)?);
Ok(Self::#ident(val)) Ok(Self::#ident(val))
} }
} }
} }
(false, Fields::Unit) => { (false, Fields::Unnamed(FieldsUnnamed { unnamed, .. }), false) => {
if unnamed.len() != 1 {
panic!("tuple variants should contain exactly one element");
}
quote! {
#variant_name => {
let val = <#deserializer_type as ::rustical_xml::XmlDeserialize>::deserialize(reader, start, empty)?;
Ok(Self::#ident(val))
}
}
}
(false, Fields::Unit, _) => {
quote! { quote! {
#variant_name => { #variant_name => {
// Make sure that content is still consumed // Make sure that content is still consumed
@@ -59,10 +122,10 @@ impl Variant {
} }
} }
} }
(true, Fields::Unnamed(_)) => { (true, Fields::Unnamed(_), _) => {
panic!("other for tuple enums not implemented yet") panic!("other for tuple enums not implemented yet")
} }
(true, Fields::Unit) => { (true, Fields::Unit, _) => {
quote! { quote! {
_ => { _ => {
// Make sure that content is still consumed // Make sure that content is still consumed
@@ -71,17 +134,21 @@ impl Variant {
} }
} }
} }
} },
)
} }
pub fn untagged_branch(&self) -> proc_macro2::TokenStream { pub fn untagged_branch(&self) -> Option<proc_macro2::TokenStream> {
if self.skip_de() {
return None;
}
if self.attrs.other.is_present() { if self.attrs.other.is_present() {
panic!("using the other flag on an untagged variant is futile"); panic!("using the other flag on an untagged variant is futile");
} }
let ident = self.ident(); let ident = self.ident();
match &self.variant.fields { Some(match &self.variant.fields {
Fields::Named(_) => { Fields::Named(_) => {
panic!( panic!(
"struct variants are not supported, please use a tuple variant with a struct" "struct variants are not supported, please use a tuple variant with a struct"
@@ -93,7 +160,7 @@ impl Variant {
} }
let field = unnamed.iter().next().unwrap(); let field = unnamed.iter().next().unwrap();
quote! { quote! {
if let Ok(val) = #field::deserialize(reader, start, empty) { if let Ok(val) = <#field as ::rustical_xml::XmlDeserialize>::deserialize(reader, start, empty) {
return Ok(Self::#ident(val)); return Ok(Self::#ident(val));
} }
} }
@@ -106,7 +173,7 @@ impl Variant {
} }
} }
} }
} })
} }
} }
@@ -125,7 +192,7 @@ impl Enum {
let variant_branches = self let variant_branches = self
.variants .variants
.iter() .iter()
.map(|variant| variant.untagged_branch()); .filter_map(|variant| variant.untagged_branch());
quote! { quote! {
impl #impl_generics ::rustical_xml::XmlDeserialize for #name #type_generics #where_clause { impl #impl_generics ::rustical_xml::XmlDeserialize for #name #type_generics #where_clause {
@@ -146,7 +213,7 @@ impl Enum {
let (impl_generics, type_generics, where_clause) = self.generics.split_for_impl(); let (impl_generics, type_generics, where_clause) = self.generics.split_for_impl();
let name = &self.ident; let name = &self.ident;
let variant_branches = self.variants.iter().map(Variant::tagged_branch); let variant_branches = self.variants.iter().filter_map(Variant::tagged_branch);
quote! { quote! {
impl #impl_generics ::rustical_xml::XmlDeserialize for #name #type_generics #where_clause { impl #impl_generics ::rustical_xml::XmlDeserialize for #name #type_generics #where_clause {
@@ -246,7 +313,7 @@ impl Enum {
Event::Decl(_) => { /* <?xml ... ?> ignore this */ } Event::Decl(_) => { /* <?xml ... ?> ignore this */ }
Event::Comment(_) => { /* ignore this */ } Event::Comment(_) => { /* ignore this */ }
Event::Start(start) | Event::Empty(start) => { Event::Start(start) | Event::Empty(start) => {
return Self::deserialize(&mut reader, &start, empty); return <Self as ::rustical_xml::XmlDeserialize>::deserialize(&mut reader, &start, empty);
} }
_ => return Err(::rustical_xml::XmlDeError::UnknownError), _ => return Err(::rustical_xml::XmlDeError::UnknownError),
}; };

View File

@@ -9,7 +9,7 @@ fn invalid_field_branch(allow: bool) -> proc_macro2::TokenStream {
if allow { if allow {
quote! {} quote! {}
} else { } else {
quote! { return Err(XmlDeError::InvalidFieldName(format!("[{ns:?}]{tag:?}"))) } quote! { return Err(XmlDeError::InvalidFieldName(format!("[{ns:?}]{tag}", tag = String::from_utf8_lossy(tag)))) }
} }
} }
@@ -64,6 +64,7 @@ impl NamedStruct {
} }
pub fn impl_de(&self) -> proc_macro2::TokenStream { pub fn impl_de(&self) -> proc_macro2::TokenStream {
let builder_generics = &self.generics;
let (impl_generics, type_generics, where_clause) = self.generics.split_for_impl(); let (impl_generics, type_generics, where_clause) = self.generics.split_for_impl();
let ident = &self.ident; let ident = &self.ident;
@@ -99,7 +100,7 @@ impl NamedStruct {
let mut buf = Vec::new(); let mut buf = Vec::new();
// initialise fields // initialise fields
struct StructBuilder #type_generics #where_clause { struct StructBuilder #builder_generics {
#(#builder_fields),* #(#builder_fields),*
} }

View File

@@ -1,4 +1,7 @@
use super::attrs::{ContainerAttrs, FieldAttrs, FieldType}; use super::{
attrs::{ContainerAttrs, FieldAttrs, FieldType},
get_generic_type,
};
use darling::FromField; use darling::FromField;
use heck::ToKebabCase; use heck::ToKebabCase;
use quote::quote; use quote::quote;
@@ -14,23 +17,6 @@ fn wrap_option_if_no_default(
} }
} }
fn get_generic_type(ty: &syn::Type) -> Option<&syn::Type> {
if let syn::Type::Path(syn::TypePath { path, .. }) = ty {
if let Some(seg) = path.segments.last() {
if let syn::PathArguments::AngleBracketed(syn::AngleBracketedGenericArguments {
args,
..
}) = &seg.arguments
{
if let Some(syn::GenericArgument::Type(t)) = &args.first() {
return Some(t);
}
}
}
}
None
}
pub struct Field { pub struct Field {
pub field: syn::Field, pub field: syn::Field,
pub attrs: FieldAttrs, pub attrs: FieldAttrs,

View File

@@ -6,3 +6,20 @@ mod field;
pub use de_enum::Enum; pub use de_enum::Enum;
pub use de_struct::NamedStruct; pub use de_struct::NamedStruct;
pub use field::Field; pub use field::Field;
pub fn get_generic_type(ty: &syn::Type) -> Option<&syn::Type> {
if let syn::Type::Path(syn::TypePath { path, .. }) = ty {
if let Some(seg) = path.segments.last() {
if let syn::PathArguments::AngleBracketed(syn::AngleBracketedGenericArguments {
args,
..
}) = &seg.arguments
{
if let Some(syn::GenericArgument::Type(t)) = &args.first() {
return Some(t);
}
}
}
}
None
}

View File

@@ -10,8 +10,6 @@ use thiserror::Error;
#[derive(Debug, Error)] #[derive(Debug, Error)]
pub enum XmlDeError { pub enum XmlDeError {
#[error(transparent)]
QuickXmlDeError(#[from] quick_xml::de::DeError),
#[error(transparent)] #[error(transparent)]
QuickXmlError(#[from] quick_xml::Error), QuickXmlError(#[from] quick_xml::Error),
#[error(transparent)] #[error(transparent)]

View File

@@ -20,21 +20,6 @@ pub trait Value: Sized {
fn deserialize(val: &str) -> Result<Self, XmlDeError>; fn deserialize(val: &str) -> Result<Self, XmlDeError>;
} }
// impl<T: Value> Value for Option<T> {
// fn serialize(&self) -> String {
// match self {
// Some(inner) => inner.serialize(),
// None => "".to_owned(),
// }
// }
// fn deserialize(val: &str) -> Result<Self, XmlDeError> {
// match val {
// "" => Ok(None),
// val => Ok(Some(T::deserialize(val)?)),
// }
// }
// }
macro_rules! impl_value_parse { macro_rules! impl_value_parse {
($t:ty) => { ($t:ty) => {
impl Value for $t { impl Value for $t {

View File

@@ -0,0 +1,57 @@
use rustical_xml::{Unparsed, XmlDeserialize, XmlDocument, XmlRootTag};
#[test]
fn test_propertyupdate() {
#[derive(XmlDeserialize)]
struct SetPropertyElement<T: XmlDeserialize> {
prop: T,
}
#[derive(XmlDeserialize)]
struct TagName {
#[xml(ty = "tag_name")]
name: String,
}
#[derive(XmlDeserialize)]
struct PropertyElement {
#[xml(ty = "untagged")]
property: TagName,
}
#[derive(XmlDeserialize)]
struct RemovePropertyElement {
prop: PropertyElement,
}
#[derive(XmlDeserialize)]
enum Operation<T: XmlDeserialize> {
Set(SetPropertyElement<T>),
Remove(RemovePropertyElement),
}
#[derive(XmlDeserialize, XmlRootTag)]
#[xml(root = b"propertyupdate")]
struct PropertyupdateElement<T: XmlDeserialize> {
#[xml(ty = "untagged", flatten)]
operations: Vec<Operation<T>>,
}
let doc = PropertyupdateElement::<Unparsed>::parse_str(
r#"
<propertyupdate>
<set>
<prop>
<displayname>Hello</displayname>
</prop>
</set>
<remove>
<prop>
<displayname />
</prop>
</remove>
</propertyupdate>
"#,
)
.unwrap();
}