use anytime FieldType methods

This commit is contained in:
Marc-Antoine Arnaud 2018-04-10 11:52:37 +02:00
parent 1c867a6ff1
commit 513e546836
3 changed files with 75 additions and 95 deletions

View File

@ -4,45 +4,29 @@ use field_type::*;
use quote::Tokens; use quote::Tokens;
use syn::Ident; use syn::Ident;
use syn::DataStruct; use syn::DataStruct;
use syn::punctuated::Pair;
use syn::Type::Path;
use proc_macro2::Span; use proc_macro2::Span;
pub fn parse(data_struct: &DataStruct, name: &Ident, root: &String) -> Tokens { pub fn parse(data_struct: &DataStruct, name: &Ident, root: &String) -> Tokens {
let variables : Tokens = data_struct.fields.iter().map(|ref field| let variables : Tokens = data_struct.fields.iter().map(|ref field|
{ {
let label = field.ident; let label = field.ident;
match get_field_type(field) {
match field.ty { Some(FieldType::FieldTypeString) => {
Path(ref path) => {
match path.path.segments.first() {
Some(Pair::End(t)) => {
let pair = path.path.segments.first().unwrap();
match t.ident.to_string().as_str() {
"String" => {
Some(quote!{ Some(quote!{
let mut #label : #pair = "".to_string(); let mut #label : String = "".to_string();
}) })
}, },
"Vec" => { Some(FieldType::FieldTypeVec{data_type}) => {
Some(quote!{ Some(quote!{
let mut #label : #pair = vec![]; let mut #label : Vec<#data_type> = vec![];
}) })
}, },
_ => { Some(FieldType::FieldTypeStruct{struct_name}) => {
Some(quote!{ Some(quote!{
let mut #label : #pair = #pair::default(); let mut #label : #struct_name = #struct_name::default();
}) })
},
} }
}, _ => None
_ => {
None
},
}
},
_ => {None},
} }
}) })
.filter(|x| x.is_some()) .filter(|x| x.is_some())
@ -142,8 +126,8 @@ pub fn parse(data_struct: &DataStruct, name: &Ident, root: &String) -> Tokens {
}, },
}) })
}, },
Some(FieldType::FieldTypeStruct{name}) => { Some(FieldType::FieldTypeStruct{struct_name}) => {
let struct_ident = Ident::new(&format!("{}", name), Span::def_site()); let struct_ident = Ident::new(&format!("{}", struct_name), Span::def_site());
Some(quote!{ Some(quote!{
#label_name => { #label_name => {
@ -159,10 +143,8 @@ pub fn parse(data_struct: &DataStruct, name: &Ident, root: &String) -> Tokens {
}, },
}) })
}, },
Some(FieldType::FieldTypeVec) => { Some(FieldType::FieldTypeVec{data_type}) => {
match get_vec_type(field) { match data_type.to_string().as_str() {
Some(identifier) => {
match identifier.to_string().as_str() {
"String" => { "String" => {
Some(quote!{ Some(quote!{
#label_name => { #label_name => {
@ -193,9 +175,6 @@ pub fn parse(data_struct: &DataStruct, name: &Ident, root: &String) -> Tokens {
} }
} }
}, },
None => None
}
},
_ => None _ => None
} }
}) })
@ -210,7 +189,7 @@ pub fn parse(data_struct: &DataStruct, name: &Ident, root: &String) -> Tokens {
match get_field_type(field) { match get_field_type(field) {
Some(FieldType::FieldTypeString) | Some(FieldType::FieldTypeString) |
Some(FieldType::FieldTypeStruct{..}) | Some(FieldType::FieldTypeStruct{..}) |
Some(FieldType::FieldTypeVec) => Some(FieldType::FieldTypeVec{..}) =>
Some(quote!{ Some(quote!{
#label: #label, #label: #label,
}), }),

View File

@ -6,8 +6,8 @@ use syn::Type::Path;
#[derive(Debug)] #[derive(Debug)]
pub enum FieldType { pub enum FieldType {
FieldTypeString, FieldTypeString,
FieldTypeVec, FieldTypeVec{data_type: syn::Ident},
FieldTypeStruct{name: String}, FieldTypeStruct{struct_name: syn::Ident},
} }
pub fn get_field_type(field: &syn::Field) -> Option<FieldType> { pub fn get_field_type(field: &syn::Field) -> Option<FieldType> {
@ -17,8 +17,19 @@ pub fn get_field_type(field: &syn::Field) -> Option<FieldType> {
Some(Pair::End(t)) => { Some(Pair::End(t)) => {
match t.ident.to_string().as_str() { match t.ident.to_string().as_str() {
"String" => Some(FieldType::FieldTypeString), "String" => Some(FieldType::FieldTypeString),
"Vec" => Some(FieldType::FieldTypeVec), "Vec" => {
name => Some(FieldType::FieldTypeStruct{name: name.to_string()}), match get_vec_type(t) {
Some(data_type) =>
Some(FieldType::FieldTypeVec{
data_type: data_type
}),
None => None,
}
},
_struct_name =>
Some(FieldType::FieldTypeStruct{
struct_name: t.ident
}),
} }
}, },
_ => { _ => {
@ -30,11 +41,7 @@ pub fn get_field_type(field: &syn::Field) -> Option<FieldType> {
} }
} }
pub fn get_vec_type(field: &syn::Field) -> Option<syn::Ident> { fn get_vec_type(t: &syn::PathSegment) -> Option<syn::Ident> {
match field.ty {
Path(ref path) => {
match path.path.segments.first() {
Some(Pair::End(t)) => {
match t.arguments { match t.arguments {
syn::PathArguments::AngleBracketed(ref args) => { syn::PathArguments::AngleBracketed(ref args) => {
match args.args.first() { match args.args.first() {
@ -61,10 +68,4 @@ pub fn get_vec_type(field: &syn::Field) -> Option<syn::Ident> {
}, },
_ => None _ => None
} }
},
_ => None
}
}
_ => None
}
} }

View File

@ -75,7 +75,7 @@ pub fn serialize(data_struct: &DataStruct, name: &Ident, root: &String) -> Token
}, },
}; };
}), }),
Some(FieldType::FieldTypeVec) => Some(FieldType::FieldTypeVec{..}) =>
Some(quote!{ Some(quote!{
for item in &self.#label { for item in &self.#label {
let start_event = XmlEvent::start_element(#label_name); let start_event = XmlEvent::start_element(#label_name);