Simplify deserializer macro setup
This commit is contained in:
Dirkjan Ochtman
parent
01c896a9b2
commit
674039a791
|
|
@ -3,6 +3,235 @@ use quote::quote;
|
||||||
|
|
||||||
use crate::{ContainerMeta, FieldMeta, Namespace};
|
use crate::{ContainerMeta, FieldMeta, Namespace};
|
||||||
|
|
||||||
|
pub(crate) fn from_xml(input: &syn::DeriveInput) -> TokenStream {
|
||||||
|
let ident = &input.ident;
|
||||||
|
let container_meta = ContainerMeta::from_derive(input);
|
||||||
|
let default_namespace = match &container_meta.ns.uri {
|
||||||
|
Some(ns) => quote!(#ns),
|
||||||
|
None => quote!(""),
|
||||||
|
};
|
||||||
|
|
||||||
|
let mut xml_generics = input.generics.clone();
|
||||||
|
let mut xml = syn::LifetimeDef::new(syn::Lifetime::new("'xml", Span::call_site()));
|
||||||
|
xml.bounds
|
||||||
|
.extend(xml_generics.lifetimes().map(|lt| lt.lifetime.clone()));
|
||||||
|
xml_generics.params.push(xml.into());
|
||||||
|
|
||||||
|
let (_, ty_generics, where_clause) = input.generics.split_for_impl();
|
||||||
|
let (xml_impl_generics, _, _) = xml_generics.split_for_impl();
|
||||||
|
|
||||||
|
let mut namespaces_map = quote!(let mut namespaces_map = std::collections::HashMap::new(););
|
||||||
|
for (k, v) in container_meta.ns.prefixes.iter() {
|
||||||
|
namespaces_map.extend(quote!(
|
||||||
|
namespaces_map.insert(#k, #v);
|
||||||
|
))
|
||||||
|
}
|
||||||
|
|
||||||
|
// Varying values
|
||||||
|
let mut elements_tokens = Tokens::default();
|
||||||
|
let mut attributes_tokens = Tokens::default();
|
||||||
|
|
||||||
|
// Common values
|
||||||
|
let mut declare_values = TokenStream::new();
|
||||||
|
let mut return_val = TokenStream::new();
|
||||||
|
|
||||||
|
match &input.data {
|
||||||
|
syn::Data::Struct(ref data) => {
|
||||||
|
match data.fields {
|
||||||
|
syn::Fields::Named(ref fields) => {
|
||||||
|
fields.named.iter().enumerate().for_each(|(index, field)| {
|
||||||
|
let field_meta = FieldMeta::from_field(field);
|
||||||
|
let tokens = match field_meta.attribute {
|
||||||
|
true => &mut attributes_tokens,
|
||||||
|
false => &mut elements_tokens,
|
||||||
|
};
|
||||||
|
|
||||||
|
process_field(
|
||||||
|
field,
|
||||||
|
index,
|
||||||
|
&mut declare_values,
|
||||||
|
&mut return_val,
|
||||||
|
tokens,
|
||||||
|
field_meta,
|
||||||
|
&container_meta,
|
||||||
|
);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
syn::Fields::Unnamed(_) => panic!("unamed"),
|
||||||
|
syn::Fields::Unit => {}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
_ => todo!(),
|
||||||
|
};
|
||||||
|
|
||||||
|
// Elements
|
||||||
|
let elements_enum = elements_tokens.enum_;
|
||||||
|
let elements_consts = elements_tokens.consts;
|
||||||
|
let elements_names = elements_tokens.names;
|
||||||
|
let elem_type_match = elements_tokens.match_;
|
||||||
|
|
||||||
|
// Attributes
|
||||||
|
let attributes_enum = attributes_tokens.enum_;
|
||||||
|
let attributes_consts = attributes_tokens.consts;
|
||||||
|
let attributes_names = attributes_tokens.names;
|
||||||
|
let attr_type_match = attributes_tokens.match_;
|
||||||
|
|
||||||
|
let name = ident.to_string();
|
||||||
|
quote!(
|
||||||
|
impl #xml_impl_generics FromXml<'xml> for #ident #ty_generics #where_clause {
|
||||||
|
fn deserialize<'cx>(deserializer: &'cx mut ::instant_xml::Deserializer<'cx, 'xml>) -> Result<Self, ::instant_xml::Error> {
|
||||||
|
use ::instant_xml::de::{Deserializer, Id, Node};
|
||||||
|
use ::instant_xml::Error;
|
||||||
|
use ::core::marker::PhantomData;
|
||||||
|
|
||||||
|
enum __Elements {
|
||||||
|
#elements_enum
|
||||||
|
__Ignore,
|
||||||
|
}
|
||||||
|
|
||||||
|
enum __Attributes {
|
||||||
|
#attributes_enum
|
||||||
|
__Ignore,
|
||||||
|
}
|
||||||
|
|
||||||
|
#declare_values
|
||||||
|
loop {
|
||||||
|
let node = match deserializer.next() {
|
||||||
|
Some(result) => result?,
|
||||||
|
None => break,
|
||||||
|
};
|
||||||
|
|
||||||
|
match node {
|
||||||
|
Node::Attribute(attr) => {
|
||||||
|
let id = deserializer.attribute_id(&attr)?;
|
||||||
|
let field = {
|
||||||
|
#attributes_consts
|
||||||
|
match id {
|
||||||
|
#attributes_names
|
||||||
|
_ => __Attributes::__Ignore
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
match field {
|
||||||
|
#attr_type_match
|
||||||
|
__Attributes::__Ignore => {}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Node::Open(data) => {
|
||||||
|
let id = deserializer.element_id(&data)?;
|
||||||
|
let element = {
|
||||||
|
#elements_consts
|
||||||
|
match id {
|
||||||
|
#elements_names
|
||||||
|
_ => __Elements::__Ignore
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
match element {
|
||||||
|
#elem_type_match
|
||||||
|
__Elements::__Ignore => {
|
||||||
|
let mut nested = deserializer.nested(data);
|
||||||
|
nested.ignore()?;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_ => return Err(Error::UnexpectedState),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(Self { #return_val })
|
||||||
|
}
|
||||||
|
|
||||||
|
const KIND: ::instant_xml::de::Kind = ::instant_xml::de::Kind::Element(::instant_xml::de::Id {
|
||||||
|
ns: #default_namespace,
|
||||||
|
name: #name,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
#[allow(clippy::too_many_arguments)]
|
||||||
|
fn process_field(
|
||||||
|
field: &syn::Field,
|
||||||
|
index: usize,
|
||||||
|
declare_values: &mut TokenStream,
|
||||||
|
return_val: &mut TokenStream,
|
||||||
|
tokens: &mut Tokens,
|
||||||
|
field_meta: FieldMeta,
|
||||||
|
container_meta: &ContainerMeta,
|
||||||
|
) {
|
||||||
|
let field_var = field.ident.as_ref().unwrap();
|
||||||
|
let field_var_str = field_var.to_string();
|
||||||
|
let const_field_var_str = Ident::new(&field_var_str.to_uppercase(), Span::call_site());
|
||||||
|
let mut no_lifetime_type = field.ty.clone();
|
||||||
|
discard_lifetimes(&mut no_lifetime_type);
|
||||||
|
|
||||||
|
let enum_name = Ident::new(&format!("__Value{index}"), Span::call_site());
|
||||||
|
tokens.enum_.extend(quote!(#enum_name,));
|
||||||
|
|
||||||
|
let default_ns = match &field_meta.ns.uri {
|
||||||
|
None => &container_meta.ns.uri,
|
||||||
|
_ => &field_meta.ns.uri,
|
||||||
|
};
|
||||||
|
|
||||||
|
let ns = match default_ns {
|
||||||
|
Some(Namespace::Path(path)) => quote!(#path),
|
||||||
|
Some(Namespace::Literal(ns)) => quote!(#ns),
|
||||||
|
None => quote!(""),
|
||||||
|
};
|
||||||
|
|
||||||
|
tokens.consts.extend(quote!(
|
||||||
|
const #const_field_var_str: Id<'static> = <#no_lifetime_type>::KIND.name(
|
||||||
|
Id { ns: #ns, name: #field_var_str }
|
||||||
|
);
|
||||||
|
));
|
||||||
|
|
||||||
|
if !field_meta.attribute {
|
||||||
|
tokens.names.extend(quote!(
|
||||||
|
#const_field_var_str => __Elements::#enum_name,
|
||||||
|
));
|
||||||
|
} else {
|
||||||
|
tokens.names.extend(quote!(
|
||||||
|
#const_field_var_str => __Attributes::#enum_name,
|
||||||
|
));
|
||||||
|
}
|
||||||
|
|
||||||
|
declare_values.extend(quote!(
|
||||||
|
let mut #enum_name: Option<#no_lifetime_type> = None;
|
||||||
|
));
|
||||||
|
|
||||||
|
if !field_meta.attribute {
|
||||||
|
tokens.match_.extend(quote!(
|
||||||
|
__Elements::#enum_name => {
|
||||||
|
if #enum_name.is_some() {
|
||||||
|
return Err(Error::DuplicateValue);
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut nested = deserializer.nested(data);
|
||||||
|
#enum_name = Some(<#no_lifetime_type>::deserialize(&mut nested)?);
|
||||||
|
},
|
||||||
|
));
|
||||||
|
} else {
|
||||||
|
tokens.match_.extend(quote!(
|
||||||
|
__Attributes::#enum_name => {
|
||||||
|
if #enum_name.is_some() {
|
||||||
|
return Err(Error::DuplicateValue);
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut nested = deserializer.for_attr(attr);
|
||||||
|
#enum_name = Some(<#no_lifetime_type>::deserialize(&mut nested)?);
|
||||||
|
},
|
||||||
|
));
|
||||||
|
}
|
||||||
|
|
||||||
|
return_val.extend(quote!(
|
||||||
|
#field_var: match #enum_name {
|
||||||
|
Some(v) => v,
|
||||||
|
None => <#no_lifetime_type>::missing_value()?,
|
||||||
|
},
|
||||||
|
));
|
||||||
|
}
|
||||||
|
|
||||||
struct Tokens {
|
struct Tokens {
|
||||||
enum_: TokenStream,
|
enum_: TokenStream,
|
||||||
consts: TokenStream,
|
consts: TokenStream,
|
||||||
|
|
@ -21,249 +250,6 @@ impl Default for Tokens {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub struct Deserializer {
|
|
||||||
out: TokenStream,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl quote::ToTokens for Deserializer {
|
|
||||||
fn to_tokens(&self, tokens: &mut TokenStream) {
|
|
||||||
tokens.extend(self.out.clone());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Deserializer {
|
|
||||||
pub fn new(input: &syn::DeriveInput) -> Deserializer {
|
|
||||||
let ident = &input.ident;
|
|
||||||
let container_meta = ContainerMeta::from_derive(input);
|
|
||||||
let default_namespace = match &container_meta.ns.uri {
|
|
||||||
Some(ns) => quote!(#ns),
|
|
||||||
None => quote!(""),
|
|
||||||
};
|
|
||||||
|
|
||||||
let mut xml_generics = input.generics.clone();
|
|
||||||
let mut xml = syn::LifetimeDef::new(syn::Lifetime::new("'xml", Span::call_site()));
|
|
||||||
xml.bounds
|
|
||||||
.extend(xml_generics.lifetimes().map(|lt| lt.lifetime.clone()));
|
|
||||||
xml_generics.params.push(xml.into());
|
|
||||||
|
|
||||||
let (_, ty_generics, where_clause) = input.generics.split_for_impl();
|
|
||||||
let (xml_impl_generics, _, _) = xml_generics.split_for_impl();
|
|
||||||
|
|
||||||
let mut namespaces_map = quote!(let mut namespaces_map = std::collections::HashMap::new(););
|
|
||||||
for (k, v) in container_meta.ns.prefixes.iter() {
|
|
||||||
namespaces_map.extend(quote!(
|
|
||||||
namespaces_map.insert(#k, #v);
|
|
||||||
))
|
|
||||||
}
|
|
||||||
|
|
||||||
// Varying values
|
|
||||||
let mut elements_tokens = Tokens::default();
|
|
||||||
let mut attributes_tokens = Tokens::default();
|
|
||||||
|
|
||||||
// Common values
|
|
||||||
let mut declare_values = TokenStream::new();
|
|
||||||
let mut return_val = TokenStream::new();
|
|
||||||
|
|
||||||
match &input.data {
|
|
||||||
syn::Data::Struct(ref data) => {
|
|
||||||
match data.fields {
|
|
||||||
syn::Fields::Named(ref fields) => {
|
|
||||||
fields.named.iter().enumerate().for_each(|(index, field)| {
|
|
||||||
let field_meta = FieldMeta::from_field(field);
|
|
||||||
let tokens = match field_meta.attribute {
|
|
||||||
true => &mut attributes_tokens,
|
|
||||||
false => &mut elements_tokens,
|
|
||||||
};
|
|
||||||
|
|
||||||
Self::process_field(
|
|
||||||
field,
|
|
||||||
index,
|
|
||||||
&mut declare_values,
|
|
||||||
&mut return_val,
|
|
||||||
tokens,
|
|
||||||
field_meta,
|
|
||||||
&container_meta,
|
|
||||||
);
|
|
||||||
});
|
|
||||||
}
|
|
||||||
syn::Fields::Unnamed(_) => panic!("unamed"),
|
|
||||||
syn::Fields::Unit => {}
|
|
||||||
};
|
|
||||||
}
|
|
||||||
_ => todo!(),
|
|
||||||
};
|
|
||||||
|
|
||||||
// Elements
|
|
||||||
let elements_enum = elements_tokens.enum_;
|
|
||||||
let elements_consts = elements_tokens.consts;
|
|
||||||
let elements_names = elements_tokens.names;
|
|
||||||
let elem_type_match = elements_tokens.match_;
|
|
||||||
|
|
||||||
// Attributes
|
|
||||||
let attributes_enum = attributes_tokens.enum_;
|
|
||||||
let attributes_consts = attributes_tokens.consts;
|
|
||||||
let attributes_names = attributes_tokens.names;
|
|
||||||
let attr_type_match = attributes_tokens.match_;
|
|
||||||
|
|
||||||
let name = ident.to_string();
|
|
||||||
let out = quote!(
|
|
||||||
impl #xml_impl_generics FromXml<'xml> for #ident #ty_generics #where_clause {
|
|
||||||
fn deserialize<'cx>(deserializer: &'cx mut ::instant_xml::Deserializer<'cx, 'xml>) -> Result<Self, ::instant_xml::Error> {
|
|
||||||
use ::instant_xml::de::{Deserializer, Id, Node};
|
|
||||||
use ::instant_xml::Error;
|
|
||||||
use ::core::marker::PhantomData;
|
|
||||||
|
|
||||||
enum __Elements {
|
|
||||||
#elements_enum
|
|
||||||
__Ignore,
|
|
||||||
}
|
|
||||||
|
|
||||||
enum __Attributes {
|
|
||||||
#attributes_enum
|
|
||||||
__Ignore,
|
|
||||||
}
|
|
||||||
|
|
||||||
#declare_values
|
|
||||||
loop {
|
|
||||||
let node = match deserializer.next() {
|
|
||||||
Some(result) => result?,
|
|
||||||
None => break,
|
|
||||||
};
|
|
||||||
|
|
||||||
match node {
|
|
||||||
Node::Attribute(attr) => {
|
|
||||||
let id = deserializer.attribute_id(&attr)?;
|
|
||||||
let field = {
|
|
||||||
#attributes_consts
|
|
||||||
match id {
|
|
||||||
#attributes_names
|
|
||||||
_ => __Attributes::__Ignore
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
match field {
|
|
||||||
#attr_type_match
|
|
||||||
__Attributes::__Ignore => {}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Node::Open(data) => {
|
|
||||||
let id = deserializer.element_id(&data)?;
|
|
||||||
let element = {
|
|
||||||
#elements_consts
|
|
||||||
match id {
|
|
||||||
#elements_names
|
|
||||||
_ => __Elements::__Ignore
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
match element {
|
|
||||||
#elem_type_match
|
|
||||||
__Elements::__Ignore => {
|
|
||||||
let mut nested = deserializer.nested(data);
|
|
||||||
nested.ignore()?;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
_ => return Err(Error::UnexpectedState),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(Self { #return_val })
|
|
||||||
}
|
|
||||||
|
|
||||||
const KIND: ::instant_xml::de::Kind = ::instant_xml::de::Kind::Element(::instant_xml::de::Id {
|
|
||||||
ns: #default_namespace,
|
|
||||||
name: #name,
|
|
||||||
});
|
|
||||||
}
|
|
||||||
);
|
|
||||||
|
|
||||||
Deserializer { out }
|
|
||||||
}
|
|
||||||
|
|
||||||
#[allow(clippy::too_many_arguments)]
|
|
||||||
fn process_field(
|
|
||||||
field: &syn::Field,
|
|
||||||
index: usize,
|
|
||||||
declare_values: &mut TokenStream,
|
|
||||||
return_val: &mut TokenStream,
|
|
||||||
tokens: &mut Tokens,
|
|
||||||
field_meta: FieldMeta,
|
|
||||||
container_meta: &ContainerMeta,
|
|
||||||
) {
|
|
||||||
let field_var = field.ident.as_ref().unwrap();
|
|
||||||
let field_var_str = field_var.to_string();
|
|
||||||
let const_field_var_str = Ident::new(&field_var_str.to_uppercase(), Span::call_site());
|
|
||||||
let mut no_lifetime_type = field.ty.clone();
|
|
||||||
discard_lifetimes(&mut no_lifetime_type);
|
|
||||||
|
|
||||||
let enum_name = Ident::new(&format!("__Value{index}"), Span::call_site());
|
|
||||||
tokens.enum_.extend(quote!(#enum_name,));
|
|
||||||
|
|
||||||
let default_ns = match &field_meta.ns.uri {
|
|
||||||
None => &container_meta.ns.uri,
|
|
||||||
_ => &field_meta.ns.uri,
|
|
||||||
};
|
|
||||||
|
|
||||||
let ns = match default_ns {
|
|
||||||
Some(Namespace::Path(path)) => quote!(#path),
|
|
||||||
Some(Namespace::Literal(ns)) => quote!(#ns),
|
|
||||||
None => quote!(""),
|
|
||||||
};
|
|
||||||
|
|
||||||
tokens.consts.extend(quote!(
|
|
||||||
const #const_field_var_str: Id<'static> = <#no_lifetime_type>::KIND.name(
|
|
||||||
Id { ns: #ns, name: #field_var_str }
|
|
||||||
);
|
|
||||||
));
|
|
||||||
|
|
||||||
if !field_meta.attribute {
|
|
||||||
tokens.names.extend(quote!(
|
|
||||||
#const_field_var_str => __Elements::#enum_name,
|
|
||||||
));
|
|
||||||
} else {
|
|
||||||
tokens.names.extend(quote!(
|
|
||||||
#const_field_var_str => __Attributes::#enum_name,
|
|
||||||
));
|
|
||||||
}
|
|
||||||
|
|
||||||
declare_values.extend(quote!(
|
|
||||||
let mut #enum_name: Option<#no_lifetime_type> = None;
|
|
||||||
));
|
|
||||||
|
|
||||||
if !field_meta.attribute {
|
|
||||||
tokens.match_.extend(quote!(
|
|
||||||
__Elements::#enum_name => {
|
|
||||||
if #enum_name.is_some() {
|
|
||||||
return Err(Error::DuplicateValue);
|
|
||||||
}
|
|
||||||
|
|
||||||
let mut nested = deserializer.nested(data);
|
|
||||||
#enum_name = Some(<#no_lifetime_type>::deserialize(&mut nested)?);
|
|
||||||
},
|
|
||||||
));
|
|
||||||
} else {
|
|
||||||
tokens.match_.extend(quote!(
|
|
||||||
__Attributes::#enum_name => {
|
|
||||||
if #enum_name.is_some() {
|
|
||||||
return Err(Error::DuplicateValue);
|
|
||||||
}
|
|
||||||
|
|
||||||
let mut nested = deserializer.for_attr(attr);
|
|
||||||
#enum_name = Some(<#no_lifetime_type>::deserialize(&mut nested)?);
|
|
||||||
},
|
|
||||||
));
|
|
||||||
}
|
|
||||||
|
|
||||||
return_val.extend(quote!(
|
|
||||||
#field_var: match #enum_name {
|
|
||||||
Some(v) => v,
|
|
||||||
None => <#no_lifetime_type>::missing_value()?,
|
|
||||||
},
|
|
||||||
));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn discard_lifetimes(ty: &mut syn::Type) {
|
fn discard_lifetimes(ty: &mut syn::Type) {
|
||||||
match ty {
|
match ty {
|
||||||
syn::Type::Path(ty) => discard_path_lifetimes(ty),
|
syn::Type::Path(ty) => discard_path_lifetimes(ty),
|
||||||
|
|
|
||||||
|
|
@ -6,7 +6,7 @@ mod ser;
|
||||||
use std::collections::BTreeMap;
|
use std::collections::BTreeMap;
|
||||||
|
|
||||||
use proc_macro2::{Delimiter, Group, Ident, Literal, Punct, TokenStream, TokenTree};
|
use proc_macro2::{Delimiter, Group, Ident, Literal, Punct, TokenStream, TokenTree};
|
||||||
use quote::{quote, ToTokens};
|
use quote::ToTokens;
|
||||||
use syn::parse_macro_input;
|
use syn::parse_macro_input;
|
||||||
use syn::punctuated::Punctuated;
|
use syn::punctuated::Punctuated;
|
||||||
use syn::token::Colon2;
|
use syn::token::Colon2;
|
||||||
|
|
@ -20,12 +20,7 @@ pub fn to_xml(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
|
||||||
#[proc_macro_derive(FromXml, attributes(xml))]
|
#[proc_macro_derive(FromXml, attributes(xml))]
|
||||||
pub fn from_xml(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
|
pub fn from_xml(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
|
||||||
let ast = parse_macro_input!(input as syn::DeriveInput);
|
let ast = parse_macro_input!(input as syn::DeriveInput);
|
||||||
|
proc_macro::TokenStream::from(de::from_xml(&ast))
|
||||||
let deserializer = de::Deserializer::new(&ast);
|
|
||||||
|
|
||||||
proc_macro::TokenStream::from(quote!(
|
|
||||||
#deserializer
|
|
||||||
))
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Default)]
|
#[derive(Default)]
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue