854 lines
27 KiB
Rust
854 lines
27 KiB
Rust
use attribute::*;
|
|
use field_type::*;
|
|
use quote::TokenStreamExt;
|
|
use std::collections::BTreeMap;
|
|
use syn::Ident;
|
|
use syn::DataStruct;
|
|
use proc_macro2::{TokenStream, Span};
|
|
|
|
pub fn parse(
|
|
data_struct: &DataStruct,
|
|
name: &Ident,
|
|
root: &str,
|
|
namespaces: &BTreeMap<String, String>,
|
|
) -> TokenStream {
|
|
let validate_namespace: TokenStream = namespaces
|
|
.iter()
|
|
.map(|(_prefix, namespace)| {
|
|
Some(quote!(
|
|
|
|
let mut found = false;
|
|
println!("{:?}", namespace);
|
|
for (key, value) in namespace {
|
|
println!("{:?}", value);
|
|
if #namespace == value {
|
|
found = true;
|
|
}
|
|
}
|
|
if !found {
|
|
return Err("bad namespace".to_string());
|
|
}
|
|
// println!("{}: {}", #prefix, #namespace);
|
|
))
|
|
})
|
|
.filter(|x| x.is_some())
|
|
.map(|x| x.unwrap())
|
|
.fold(TokenStream::empty(), |mut tokens, token| {
|
|
tokens.append_all(token);
|
|
tokens
|
|
});
|
|
|
|
let variables: TokenStream = data_struct
|
|
.fields
|
|
.iter()
|
|
.map(|field| {
|
|
let label = &field.ident;
|
|
match get_field_type(field) {
|
|
Some(FieldType::FieldTypeString) => {
|
|
build_default_value(&label, "e!{String}, "e!{"".to_string()})
|
|
}
|
|
Some(FieldType::FieldTypeBool) => {
|
|
build_default_value(&label, "e!{bool}, "e!{false})
|
|
}
|
|
Some(FieldType::FieldTypeI8) => build_default_value(&label, "e!{i8}, "e!{0}),
|
|
Some(FieldType::FieldTypeU8) => build_default_value(&label, "e!{u8}, "e!{0}),
|
|
Some(FieldType::FieldTypeI16) => build_default_value(&label, "e!{i16}, "e!{0}),
|
|
Some(FieldType::FieldTypeU16) => build_default_value(&label, "e!{u16}, "e!{0}),
|
|
Some(FieldType::FieldTypeI32) => build_default_value(&label, "e!{i32}, "e!{0}),
|
|
Some(FieldType::FieldTypeU32) => build_default_value(&label, "e!{u32}, "e!{0}),
|
|
Some(FieldType::FieldTypeI64) => build_default_value(&label, "e!{i64}, "e!{0}),
|
|
Some(FieldType::FieldTypeU64) => build_default_value(&label, "e!{u64}, "e!{0}),
|
|
Some(FieldType::FieldTypeStruct { struct_name }) => Some(quote!{
|
|
#[allow(unused_mut, non_snake_case, non_camel_case_types)]
|
|
let mut #label : #struct_name = #struct_name::default();
|
|
}),
|
|
Some(FieldType::FieldTypeVec { data_type }) => {
|
|
let dt = Box::into_raw(data_type);
|
|
match unsafe { dt.as_ref() } {
|
|
Some(&FieldType::FieldTypeString) => {
|
|
build_default_value(&label, "e!{Vec<String>}, "e!{vec![]})
|
|
}
|
|
Some(&FieldType::FieldTypeBool) => {
|
|
build_default_value(&label, "e!{Vec<bool>}, "e!{vec![]})
|
|
}
|
|
Some(&FieldType::FieldTypeI8) => {
|
|
build_default_value(&label, "e!{Vec<i8>}, "e!{vec![]})
|
|
}
|
|
Some(&FieldType::FieldTypeU8) => {
|
|
build_default_value(&label, "e!{Vec<u8>}, "e!{vec![]})
|
|
}
|
|
Some(&FieldType::FieldTypeI16) => {
|
|
build_default_value(&label, "e!{Vec<i16>}, "e!{vec![]})
|
|
}
|
|
Some(&FieldType::FieldTypeU16) => {
|
|
build_default_value(&label, "e!{Vec<u16>}, "e!{vec![]})
|
|
}
|
|
Some(&FieldType::FieldTypeI32) => {
|
|
build_default_value(&label, "e!{Vec<i32>}, "e!{vec![]})
|
|
}
|
|
Some(&FieldType::FieldTypeU32) => {
|
|
build_default_value(&label, "e!{Vec<u32>}, "e!{vec![]})
|
|
}
|
|
Some(&FieldType::FieldTypeI64) => {
|
|
build_default_value(&label, "e!{Vec<i64>}, "e!{vec![]})
|
|
}
|
|
Some(&FieldType::FieldTypeU64) => {
|
|
build_default_value(&label, "e!{Vec<u64>}, "e!{vec![]})
|
|
}
|
|
Some(&FieldType::FieldTypeStruct { ref struct_name }) => Some(quote!{
|
|
#[allow(unused_mut)]
|
|
let mut #label : Vec<#struct_name> = vec![];
|
|
}),
|
|
Some(&FieldType::FieldTypeVec { .. }) => {
|
|
unimplemented!();
|
|
}
|
|
None => {
|
|
unimplemented!();
|
|
}
|
|
}
|
|
}
|
|
_ => None,
|
|
}
|
|
})
|
|
.filter(|x| x.is_some())
|
|
.map(|x| x.unwrap())
|
|
.fold(TokenStream::empty(), |mut sum, val| {
|
|
sum.append_all(val);
|
|
sum
|
|
});
|
|
|
|
let field_visitors: TokenStream = data_struct
|
|
.fields
|
|
.iter()
|
|
.map(|field| {
|
|
let field_attrs = YaSerdeAttribute::parse(&field.attrs);
|
|
let label_name = if let Some(value) = field_attrs.rename {
|
|
Ident::new(&format!("{}", value), Span::call_site()).to_string()
|
|
} else {
|
|
field.ident.clone().unwrap().to_string()
|
|
};
|
|
|
|
let visitor_label = Ident::new(&format!("__Visitor{}", label_name), Span::call_site());
|
|
|
|
match get_field_type(field) {
|
|
Some(FieldType::FieldTypeString) => {
|
|
build_declare_visitor("e!{String}, "e!{visit_str}, &visitor_label)
|
|
}
|
|
Some(FieldType::FieldTypeBool) => {
|
|
build_declare_visitor("e!{bool}, "e!{visit_bool}, &visitor_label)
|
|
}
|
|
Some(FieldType::FieldTypeI8) => {
|
|
build_declare_visitor("e!{i8}, "e!{visit_i8}, &visitor_label)
|
|
}
|
|
Some(FieldType::FieldTypeU8) => {
|
|
build_declare_visitor("e!{u8}, "e!{visit_u8}, &visitor_label)
|
|
}
|
|
Some(FieldType::FieldTypeI16) => {
|
|
build_declare_visitor("e!{i16}, "e!{visit_i16}, &visitor_label)
|
|
}
|
|
Some(FieldType::FieldTypeU16) => {
|
|
build_declare_visitor("e!{u16}, "e!{visit_u16}, &visitor_label)
|
|
}
|
|
Some(FieldType::FieldTypeI32) => {
|
|
build_declare_visitor("e!{i32}, "e!{visit_i32}, &visitor_label)
|
|
}
|
|
Some(FieldType::FieldTypeU32) => {
|
|
build_declare_visitor("e!{u32}, "e!{visit_u32}, &visitor_label)
|
|
}
|
|
Some(FieldType::FieldTypeI64) => {
|
|
build_declare_visitor("e!{i64}, "e!{visit_i64}, &visitor_label)
|
|
}
|
|
Some(FieldType::FieldTypeU64) => {
|
|
build_declare_visitor("e!{u64}, "e!{visit_u64}, &visitor_label)
|
|
}
|
|
Some(FieldType::FieldTypeStruct { struct_name }) => {
|
|
let struct_id = struct_name.to_string();
|
|
let struct_ident = Ident::new(
|
|
&format!("__Visitor_{}_{}", label_name, struct_name),
|
|
Span::call_site(),
|
|
);
|
|
|
|
Some(quote!{
|
|
#[allow(non_snake_case, non_camel_case_types)]
|
|
struct #struct_ident;
|
|
impl<'de> Visitor<'de> for #struct_ident {
|
|
type Value = #struct_name;
|
|
|
|
fn visit_str(self, v: &str) -> Result<Self::Value, String> {
|
|
let content = "<".to_string() + #struct_id + ">" + v + "</" + #struct_id + ">";
|
|
let value : Result<#struct_name, String> = yaserde::de::from_str(&content);
|
|
value
|
|
}
|
|
}
|
|
})
|
|
}
|
|
Some(FieldType::FieldTypeVec { data_type }) => {
|
|
let dt = Box::into_raw(data_type);
|
|
match unsafe { dt.as_ref() } {
|
|
Some(&FieldType::FieldTypeString) => {
|
|
build_declare_visitor("e!{String}, "e!{visit_str}, &visitor_label)
|
|
}
|
|
Some(&FieldType::FieldTypeBool) => {
|
|
build_declare_visitor("e!{bool}, "e!{visit_bool}, &visitor_label)
|
|
}
|
|
Some(&FieldType::FieldTypeI8) => {
|
|
build_declare_visitor("e!{i8}, "e!{visit_i8}, &visitor_label)
|
|
}
|
|
Some(&FieldType::FieldTypeU8) => {
|
|
build_declare_visitor("e!{u8}, "e!{visit_u8}, &visitor_label)
|
|
}
|
|
Some(&FieldType::FieldTypeI16) => {
|
|
build_declare_visitor("e!{i16}, "e!{visit_i16}, &visitor_label)
|
|
}
|
|
Some(&FieldType::FieldTypeU16) => {
|
|
build_declare_visitor("e!{u16}, "e!{visit_u16}, &visitor_label)
|
|
}
|
|
Some(&FieldType::FieldTypeI32) => {
|
|
build_declare_visitor("e!{i32}, "e!{visit_i32}, &visitor_label)
|
|
}
|
|
Some(&FieldType::FieldTypeU32) => {
|
|
build_declare_visitor("e!{u32}, "e!{visit_u32}, &visitor_label)
|
|
}
|
|
Some(&FieldType::FieldTypeI64) => {
|
|
build_declare_visitor("e!{i64}, "e!{visit_i64}, &visitor_label)
|
|
}
|
|
Some(&FieldType::FieldTypeU64) => {
|
|
build_declare_visitor("e!{u64}, "e!{visit_u64}, &visitor_label)
|
|
}
|
|
Some(&FieldType::FieldTypeStruct { ref struct_name }) => {
|
|
let struct_ident = Ident::new(&format!("{}", struct_name), Span::call_site());
|
|
Some(quote!{
|
|
#[allow(non_snake_case, non_camel_case_types)]
|
|
struct #visitor_label;
|
|
impl<'de> Visitor<'de> for #visitor_label {
|
|
type Value = #struct_ident;
|
|
}
|
|
})
|
|
}
|
|
_ => None,
|
|
}
|
|
}
|
|
_ => None,
|
|
}
|
|
})
|
|
.filter(|x| x.is_some())
|
|
.map(|x| x.unwrap())
|
|
.fold(TokenStream::empty(), |mut sum, val| {
|
|
sum.append_all(val);
|
|
sum
|
|
});
|
|
|
|
let call_visitors: TokenStream = data_struct
|
|
.fields
|
|
.iter()
|
|
.map(|field| {
|
|
let field_attrs = YaSerdeAttribute::parse(&field.attrs);
|
|
let label = &field.ident;
|
|
|
|
if field_attrs.attribute {
|
|
return None;
|
|
}
|
|
|
|
let label_name = if let Some(value) = field_attrs.rename {
|
|
Ident::new(&format!("{}", value), Span::call_site()).to_string()
|
|
} else {
|
|
field.ident.clone().unwrap().to_string()
|
|
};
|
|
|
|
let visitor_label = Ident::new(&format!("__Visitor{}", label_name), Span::call_site());
|
|
|
|
match get_field_type(field) {
|
|
Some(FieldType::FieldTypeString) => {
|
|
let visitor = Ident::new("visit_str", Span::call_site());
|
|
build_call_visitor(
|
|
"e!{String},
|
|
&visitor,
|
|
"e!{= value},
|
|
&visitor_label,
|
|
&label,
|
|
&label_name,
|
|
)
|
|
}
|
|
Some(FieldType::FieldTypeBool) => {
|
|
let visitor = Ident::new("visit_bool", Span::call_site());
|
|
build_call_visitor(
|
|
"e!{bool},
|
|
&visitor,
|
|
"e!{= value},
|
|
&visitor_label,
|
|
&label,
|
|
&label_name,
|
|
)
|
|
}
|
|
Some(FieldType::FieldTypeI8) => {
|
|
let visitor = Ident::new("visit_i8", Span::call_site());
|
|
build_call_visitor(
|
|
"e!{i8},
|
|
&visitor,
|
|
"e!{= value},
|
|
&visitor_label,
|
|
&label,
|
|
&label_name,
|
|
)
|
|
}
|
|
Some(FieldType::FieldTypeU8) => {
|
|
let visitor = Ident::new("visit_u8", Span::call_site());
|
|
build_call_visitor(
|
|
"e!{u8},
|
|
&visitor,
|
|
"e!{= value},
|
|
&visitor_label,
|
|
&label,
|
|
&label_name,
|
|
)
|
|
}
|
|
Some(FieldType::FieldTypeU16) => {
|
|
let visitor = Ident::new("visit_u16", Span::call_site());
|
|
build_call_visitor(
|
|
"e!{u16},
|
|
&visitor,
|
|
"e!{= value},
|
|
&visitor_label,
|
|
&label,
|
|
&label_name,
|
|
)
|
|
}
|
|
Some(FieldType::FieldTypeI16) => {
|
|
let visitor = Ident::new("visit_i16", Span::call_site());
|
|
build_call_visitor(
|
|
"e!{i16},
|
|
&visitor,
|
|
"e!{= value},
|
|
&visitor_label,
|
|
&label,
|
|
&label_name,
|
|
)
|
|
}
|
|
Some(FieldType::FieldTypeU32) => {
|
|
let visitor = Ident::new("visit_u32", Span::call_site());
|
|
build_call_visitor(
|
|
"e!{u32},
|
|
&visitor,
|
|
"e!{= value},
|
|
&visitor_label,
|
|
&label,
|
|
&label_name,
|
|
)
|
|
}
|
|
Some(FieldType::FieldTypeI32) => {
|
|
let visitor = Ident::new("visit_i32", Span::call_site());
|
|
build_call_visitor(
|
|
"e!{i32},
|
|
&visitor,
|
|
"e!{= value},
|
|
&visitor_label,
|
|
&label,
|
|
&label_name,
|
|
)
|
|
}
|
|
Some(FieldType::FieldTypeU64) => {
|
|
let visitor = Ident::new("visit_u64", Span::call_site());
|
|
build_call_visitor(
|
|
"e!{u64},
|
|
&visitor,
|
|
"e!{= value},
|
|
&visitor_label,
|
|
&label,
|
|
&label_name,
|
|
)
|
|
}
|
|
Some(FieldType::FieldTypeI64) => {
|
|
let visitor = Ident::new("visit_i64", Span::call_site());
|
|
build_call_visitor(
|
|
"e!{i64},
|
|
&visitor,
|
|
"e!{= value},
|
|
&visitor_label,
|
|
&label,
|
|
&label_name,
|
|
)
|
|
}
|
|
Some(FieldType::FieldTypeStruct { struct_name }) => Some(quote!{
|
|
#label_name => {
|
|
reader.set_map_value();
|
|
match #struct_name::deserialize(reader) {
|
|
Ok(parsed_item) => {
|
|
#label = parsed_item;
|
|
let _root = reader.next_event();
|
|
},
|
|
Err(msg) => {
|
|
return Err(msg);
|
|
},
|
|
}
|
|
}
|
|
}),
|
|
Some(FieldType::FieldTypeVec { data_type }) => {
|
|
let dt = Box::into_raw(data_type);
|
|
match unsafe { dt.as_ref() } {
|
|
Some(&FieldType::FieldTypeString) => {
|
|
let visitor = Ident::new("visit_str", Span::call_site());
|
|
build_call_visitor(
|
|
"e!{String},
|
|
&visitor,
|
|
"e!{.push(value)},
|
|
&visitor_label,
|
|
&label,
|
|
&label_name,
|
|
)
|
|
}
|
|
Some(&FieldType::FieldTypeBool) => {
|
|
let visitor = Ident::new("visit_bool", Span::call_site());
|
|
build_call_visitor(
|
|
"e!{bool},
|
|
&visitor,
|
|
"e!{.push(value)},
|
|
&visitor_label,
|
|
&label,
|
|
&label_name,
|
|
)
|
|
}
|
|
Some(&FieldType::FieldTypeI8) => {
|
|
let visitor = Ident::new("visit_i8", Span::call_site());
|
|
build_call_visitor(
|
|
"e!{i8},
|
|
&visitor,
|
|
"e!{.push(value)},
|
|
&visitor_label,
|
|
&label,
|
|
&label_name,
|
|
)
|
|
}
|
|
Some(&FieldType::FieldTypeU8) => {
|
|
let visitor = Ident::new("visit_u8", Span::call_site());
|
|
build_call_visitor(
|
|
"e!{u8},
|
|
&visitor,
|
|
"e!{.push(value)},
|
|
&visitor_label,
|
|
&label,
|
|
&label_name,
|
|
)
|
|
}
|
|
Some(&FieldType::FieldTypeI16) => {
|
|
let visitor = Ident::new("visit_i16", Span::call_site());
|
|
build_call_visitor(
|
|
"e!{i16},
|
|
&visitor,
|
|
"e!{.push(value)},
|
|
&visitor_label,
|
|
&label,
|
|
&label_name,
|
|
)
|
|
}
|
|
Some(&FieldType::FieldTypeU16) => {
|
|
let visitor = Ident::new("visit_u16", Span::call_site());
|
|
build_call_visitor(
|
|
"e!{u16},
|
|
&visitor,
|
|
"e!{.push(value)},
|
|
&visitor_label,
|
|
&label,
|
|
&label_name,
|
|
)
|
|
}
|
|
Some(&FieldType::FieldTypeI32) => {
|
|
let visitor = Ident::new("visit_i32", Span::call_site());
|
|
build_call_visitor(
|
|
"e!{i32},
|
|
&visitor,
|
|
"e!{.push(value)},
|
|
&visitor_label,
|
|
&label,
|
|
&label_name,
|
|
)
|
|
}
|
|
Some(&FieldType::FieldTypeU32) => {
|
|
let visitor = Ident::new("visit_u32", Span::call_site());
|
|
build_call_visitor(
|
|
"e!{u32},
|
|
&visitor,
|
|
"e!{.push(value)},
|
|
&visitor_label,
|
|
&label,
|
|
&label_name,
|
|
)
|
|
}
|
|
Some(&FieldType::FieldTypeI64) => {
|
|
let visitor = Ident::new("visit_i64", Span::call_site());
|
|
build_call_visitor(
|
|
"e!{i64},
|
|
&visitor,
|
|
"e!{.push(value)},
|
|
&visitor_label,
|
|
&label,
|
|
&label_name,
|
|
)
|
|
}
|
|
Some(&FieldType::FieldTypeU64) => {
|
|
let visitor = Ident::new("visit_u64", Span::call_site());
|
|
build_call_visitor(
|
|
"e!{u64},
|
|
&visitor,
|
|
"e!{.push(value)},
|
|
&visitor_label,
|
|
&label,
|
|
&label_name,
|
|
)
|
|
}
|
|
Some(&FieldType::FieldTypeStruct { ref struct_name }) => {
|
|
let struct_ident = Ident::new(&format!("{}", struct_name), Span::call_site());
|
|
Some(quote!{
|
|
#label_name => {
|
|
reader.set_map_value();
|
|
match #struct_ident::deserialize(reader) {
|
|
Ok(parsed_item) => {
|
|
#label.push(parsed_item);
|
|
let _root = reader.next_event();
|
|
},
|
|
Err(msg) => {
|
|
return Err(msg);
|
|
},
|
|
}
|
|
}
|
|
})
|
|
}
|
|
_ => unimplemented!(),
|
|
}
|
|
}
|
|
_ => None,
|
|
}
|
|
})
|
|
.filter(|x| x.is_some())
|
|
.map(|x| x.unwrap())
|
|
.fold(TokenStream::empty(), |mut sum, val| {
|
|
sum.append_all(val);
|
|
sum
|
|
});
|
|
|
|
let attributes_loading: TokenStream = data_struct
|
|
.fields
|
|
.iter()
|
|
.map(|field| {
|
|
let field_attrs = YaSerdeAttribute::parse(&field.attrs);
|
|
if !field_attrs.attribute {
|
|
return None;
|
|
}
|
|
|
|
let label = &field.ident;
|
|
let label_name = if let Some(value) = field_attrs.rename {
|
|
Ident::new(&format!("{}", value), Span::call_site()).to_string()
|
|
} else {
|
|
field.ident.clone().unwrap().to_string()
|
|
};
|
|
|
|
let visitor_label = Ident::new(&format!("__Visitor{}", label_name), Span::call_site());
|
|
|
|
match get_field_type(field) {
|
|
Some(FieldType::FieldTypeString) => Some(quote!{
|
|
for attr in attributes {
|
|
if attr.name.local_name == #label_name {
|
|
#label = attr.value.to_owned();
|
|
}
|
|
}
|
|
}),
|
|
Some(FieldType::FieldTypeBool) => {
|
|
build_call_visitor_for_attribute(&label, &label_name, "e!{visit_bool}, &visitor_label)
|
|
}
|
|
Some(FieldType::FieldTypeI8) => {
|
|
build_call_visitor_for_attribute(&label, &label_name, "e!{visit_i8}, &visitor_label)
|
|
}
|
|
Some(FieldType::FieldTypeU8) => {
|
|
build_call_visitor_for_attribute(&label, &label_name, "e!{visit_u8}, &visitor_label)
|
|
}
|
|
Some(FieldType::FieldTypeI16) => {
|
|
build_call_visitor_for_attribute(&label, &label_name, "e!{visit_i16}, &visitor_label)
|
|
}
|
|
Some(FieldType::FieldTypeU16) => {
|
|
build_call_visitor_for_attribute(&label, &label_name, "e!{visit_u16}, &visitor_label)
|
|
}
|
|
Some(FieldType::FieldTypeI32) => {
|
|
build_call_visitor_for_attribute(&label, &label_name, "e!{visit_i32}, &visitor_label)
|
|
}
|
|
Some(FieldType::FieldTypeU32) => {
|
|
build_call_visitor_for_attribute(&label, &label_name, "e!{visit_u32}, &visitor_label)
|
|
}
|
|
Some(FieldType::FieldTypeI64) => {
|
|
build_call_visitor_for_attribute(&label, &label_name, "e!{visit_i64}, &visitor_label)
|
|
}
|
|
Some(FieldType::FieldTypeU64) => {
|
|
build_call_visitor_for_attribute(&label, &label_name, "e!{visit_u64}, &visitor_label)
|
|
}
|
|
|
|
Some(FieldType::FieldTypeStruct { struct_name }) => {
|
|
let struct_ident = Ident::new(
|
|
&format!("__Visitor_{}_{}", label_name, struct_name),
|
|
Span::call_site(),
|
|
);
|
|
|
|
Some(quote!{
|
|
for attr in attributes {
|
|
if attr.name.local_name == #label_name {
|
|
let visitor = #struct_ident{};
|
|
match visitor.visit_str(&attr.value) {
|
|
Ok(value) => {#label = value;}
|
|
Err(msg) => {return Err(msg);}
|
|
}
|
|
}
|
|
}
|
|
})
|
|
}
|
|
_ => None,
|
|
}
|
|
})
|
|
.filter(|x| x.is_some())
|
|
.map(|x| x.unwrap())
|
|
.fold(TokenStream::empty(), |mut sum, val| {
|
|
sum.append_all(val);
|
|
sum
|
|
});
|
|
|
|
let set_text: TokenStream = data_struct
|
|
.fields
|
|
.iter()
|
|
.map(|field| {
|
|
let label = &field.ident;
|
|
let field_attrs = YaSerdeAttribute::parse(&field.attrs);
|
|
|
|
match get_field_type(field) {
|
|
Some(FieldType::FieldTypeString) => {
|
|
build_set_text_to_value(&field_attrs, &label, "e!{text_content.to_owned()})
|
|
}
|
|
Some(FieldType::FieldTypeBool) => build_set_text_to_value(
|
|
&field_attrs,
|
|
&label,
|
|
"e!{bool::from_str(text_content).unwrap()},
|
|
),
|
|
Some(FieldType::FieldTypeI8) => build_set_text_to_value(
|
|
&field_attrs,
|
|
&label,
|
|
"e!{i8::from_str(text_content).unwrap()},
|
|
),
|
|
Some(FieldType::FieldTypeU8) => build_set_text_to_value(
|
|
&field_attrs,
|
|
&label,
|
|
"e!{u8::from_str(text_content).unwrap()},
|
|
),
|
|
Some(FieldType::FieldTypeI16) => build_set_text_to_value(
|
|
&field_attrs,
|
|
&label,
|
|
"e!{i16::from_str(text_content).unwrap()},
|
|
),
|
|
Some(FieldType::FieldTypeU16) => build_set_text_to_value(
|
|
&field_attrs,
|
|
&label,
|
|
"e!{u16::from_str(text_content).unwrap()},
|
|
),
|
|
Some(FieldType::FieldTypeI32) => build_set_text_to_value(
|
|
&field_attrs,
|
|
&label,
|
|
"e!{i32::from_str(text_content).unwrap()},
|
|
),
|
|
Some(FieldType::FieldTypeU32) => build_set_text_to_value(
|
|
&field_attrs,
|
|
&label,
|
|
"e!{u32::from_str(text_content).unwrap()},
|
|
),
|
|
Some(FieldType::FieldTypeI64) => build_set_text_to_value(
|
|
&field_attrs,
|
|
&label,
|
|
"e!{i64::from_str(text_content).unwrap()},
|
|
),
|
|
Some(FieldType::FieldTypeU64) => build_set_text_to_value(
|
|
&field_attrs,
|
|
&label,
|
|
"e!{u64::from_str(text_content).unwrap()},
|
|
),
|
|
|
|
Some(FieldType::FieldTypeStruct { .. }) | Some(FieldType::FieldTypeVec { .. }) | None => {
|
|
None
|
|
}
|
|
}
|
|
})
|
|
.filter(|x| x.is_some())
|
|
.map(|x| x.unwrap())
|
|
.fold(TokenStream::empty(), |mut tokens, token| {
|
|
tokens.append_all(token);
|
|
tokens
|
|
});
|
|
|
|
let struct_builder: TokenStream = data_struct
|
|
.fields
|
|
.iter()
|
|
.map(|field| {
|
|
let label = &field.ident;
|
|
|
|
if get_field_type(field).is_some() {
|
|
Some(quote!{
|
|
#label: #label,
|
|
})
|
|
} else {
|
|
None
|
|
}
|
|
})
|
|
.filter(|x| x.is_some())
|
|
.map(|x| x.unwrap())
|
|
.fold(TokenStream::empty(), |mut tokens, token| {
|
|
tokens.append_all(token);
|
|
tokens
|
|
});
|
|
|
|
quote! {
|
|
use xml::reader::XmlEvent;
|
|
use yaserde::Visitor;
|
|
#[allow(unknown_lints, unused_imports)]
|
|
use std::str::FromStr;
|
|
|
|
impl YaDeserialize for #name {
|
|
#[allow(unused_variables)]
|
|
fn deserialize<R: Read>(reader: &mut yaserde::de::Deserializer<R>) -> Result<Self, String> {
|
|
let named_element =
|
|
if let XmlEvent::StartElement{name, ..} = reader.peek()?.to_owned() {
|
|
name.local_name.to_owned()
|
|
} else {
|
|
String::from(#root)
|
|
};
|
|
debug!("Struct: start to parse {:?}", named_element);
|
|
|
|
#variables
|
|
#field_visitors
|
|
|
|
loop {
|
|
match reader.peek()?.to_owned() {
|
|
XmlEvent::StartElement{ref name, ref attributes, ref namespace} => {
|
|
#validate_namespace
|
|
|
|
match name.local_name.as_str() {
|
|
#call_visitors
|
|
named_element => {
|
|
let _root = reader.next_event();
|
|
}
|
|
// name => {
|
|
// return Err(format!("unknown key {}", name))
|
|
// }
|
|
}
|
|
#attributes_loading
|
|
}
|
|
XmlEvent::EndElement{ref name} => {
|
|
if name.local_name == named_element {
|
|
break;
|
|
}
|
|
let _root = reader.next_event();
|
|
}
|
|
XmlEvent::Characters(ref text_content) => {
|
|
#set_text
|
|
let _root = reader.next_event();
|
|
}
|
|
event => {
|
|
return Err(format!("unknown event {:?}", event))
|
|
}
|
|
}
|
|
}
|
|
|
|
Ok(#name{#struct_builder})
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
fn build_default_value(
|
|
label: &Option<Ident>,
|
|
field_type: &TokenStream,
|
|
default: &TokenStream,
|
|
) -> Option<TokenStream> {
|
|
Some(quote!{
|
|
#[allow(unused_mut)]
|
|
let mut #label : #field_type = #default;
|
|
})
|
|
}
|
|
|
|
fn build_declare_visitor(
|
|
field_type: &TokenStream,
|
|
visitor: &TokenStream,
|
|
visitor_label: &Ident,
|
|
) -> Option<TokenStream> {
|
|
Some(quote!{
|
|
#[allow(non_snake_case, non_camel_case_types)]
|
|
struct #visitor_label;
|
|
impl<'de> Visitor<'de> for #visitor_label {
|
|
type Value = #field_type;
|
|
|
|
fn #visitor(self, v: &str) -> Result<Self::Value, String> {
|
|
Ok(#field_type::from_str(v).unwrap())
|
|
}
|
|
}
|
|
})
|
|
}
|
|
|
|
fn build_call_visitor(
|
|
field_type: &TokenStream,
|
|
visitor: &Ident,
|
|
action: &TokenStream,
|
|
visitor_label: &Ident,
|
|
label: &Option<Ident>,
|
|
label_name: &str,
|
|
) -> Option<TokenStream> {
|
|
Some(quote!{
|
|
#label_name => {
|
|
let visitor = #visitor_label{};
|
|
|
|
if let XmlEvent::StartElement { .. } = *reader.peek()? {
|
|
reader.set_map_value()
|
|
}
|
|
|
|
let result = reader.read_inner_value::<#field_type, _>(|reader| {
|
|
if let XmlEvent::EndElement { .. } = *reader.peek()? {
|
|
return visitor.#visitor("");
|
|
}
|
|
|
|
if let Ok(XmlEvent::Characters(s)) = reader.next_event() {
|
|
visitor.#visitor(&s)
|
|
} else {
|
|
Err(format!("unable to parse content for {}", #label_name))
|
|
}
|
|
});
|
|
|
|
if let Ok(value) = result {
|
|
#label#action
|
|
}
|
|
}
|
|
})
|
|
}
|
|
|
|
fn build_call_visitor_for_attribute(
|
|
label: &Option<Ident>,
|
|
label_name: &str,
|
|
visitor: &TokenStream,
|
|
visitor_label: &Ident,
|
|
) -> Option<TokenStream> {
|
|
Some(quote!{
|
|
for attr in attributes {
|
|
if attr.name.local_name == #label_name {
|
|
let visitor = #visitor_label{};
|
|
match visitor.#visitor(&attr.value) {
|
|
Ok(value) => {#label = value;}
|
|
Err(msg) => {return Err(msg);}
|
|
}
|
|
}
|
|
}
|
|
})
|
|
}
|
|
|
|
fn build_set_text_to_value(
|
|
field_attrs: &YaSerdeAttribute,
|
|
label: &Option<Ident>,
|
|
action: &TokenStream,
|
|
) -> Option<TokenStream> {
|
|
if field_attrs.text {
|
|
Some(quote!{
|
|
#label = #action;
|
|
})
|
|
} else {
|
|
None
|
|
}
|
|
}
|