mirror of
https://github.com/lennart-k/rustical.git
synced 2025-12-14 07:02:24 +00:00
xml: Add tag_name extractor
This commit is contained in:
@@ -47,6 +47,8 @@ pub enum FieldType {
|
|||||||
Attr,
|
Attr,
|
||||||
Text,
|
Text,
|
||||||
Untagged,
|
Untagged,
|
||||||
|
TagName,
|
||||||
|
Namespace,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Default, FromField)]
|
#[derive(Default, FromField)]
|
||||||
|
|||||||
@@ -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) }
|
quote! { return Err(XmlDeError::InvalidFieldName(format!("[{ns:?}]{tag:?}"))) }
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -80,6 +80,7 @@ impl NamedStruct {
|
|||||||
}
|
}
|
||||||
let text_field_branches = self.fields.iter().filter_map(Field::text_branch);
|
let text_field_branches = self.fields.iter().filter_map(Field::text_branch);
|
||||||
let attr_field_branches = self.fields.iter().filter_map(Field::attr_branch);
|
let attr_field_branches = self.fields.iter().filter_map(Field::attr_branch);
|
||||||
|
let tagname_field_branches = self.fields.iter().filter_map(Field::tagname_branch);
|
||||||
|
|
||||||
let builder_field_builds = self.fields.iter().map(Field::builder_field_build);
|
let builder_field_builds = self.fields.iter().map(Field::builder_field_build);
|
||||||
|
|
||||||
@@ -106,11 +107,14 @@ impl NamedStruct {
|
|||||||
#(#builder_field_inits),*
|
#(#builder_field_inits),*
|
||||||
};
|
};
|
||||||
|
|
||||||
|
let (ns, name) = reader.resolve_element(start.name());
|
||||||
|
#(#tagname_field_branches);*
|
||||||
|
|
||||||
for attr in start.attributes() {
|
for attr in start.attributes() {
|
||||||
let attr = attr?;
|
let attr = attr?;
|
||||||
match attr.key.as_ref() {
|
match attr.key.as_ref() {
|
||||||
#(#attr_field_branches),*
|
#(#attr_field_branches),*
|
||||||
_ => { #invalid_field_branch }
|
_ => { /* ignore */ }
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -129,7 +133,7 @@ impl NamedStruct {
|
|||||||
match (ns, name.as_ref()) {
|
match (ns, name.as_ref()) {
|
||||||
#(#named_field_branches),*
|
#(#named_field_branches),*
|
||||||
#(#untagged_field_branches),*
|
#(#untagged_field_branches),*
|
||||||
_ => { #invalid_field_branch }
|
(ns, tag) => { #invalid_field_branch }
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
Event::Text(bytes_text) => {
|
Event::Text(bytes_text) => {
|
||||||
|
|||||||
@@ -219,6 +219,24 @@ impl Field {
|
|||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub fn tagname_branch(&self) -> Option<proc_macro2::TokenStream> {
|
||||||
|
if self.attrs.xml_ty != FieldType::TagName {
|
||||||
|
return None;
|
||||||
|
}
|
||||||
|
let field_ident = self.field_ident();
|
||||||
|
|
||||||
|
let value = wrap_option_if_no_default(
|
||||||
|
quote! {
|
||||||
|
rustical_xml::Value::deserialize(&String::from_utf8_lossy(name.as_ref()))?
|
||||||
|
},
|
||||||
|
self.attrs.default.is_some(),
|
||||||
|
);
|
||||||
|
|
||||||
|
Some(quote! {
|
||||||
|
builder.#field_ident = #value;
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
pub fn tag_writer(&self) -> Option<proc_macro2::TokenStream> {
|
pub fn tag_writer(&self) -> Option<proc_macro2::TokenStream> {
|
||||||
let field_ident = self.field_ident();
|
let field_ident = self.field_ident();
|
||||||
let field_name = self.xml_name();
|
let field_name = self.xml_name();
|
||||||
@@ -235,6 +253,8 @@ impl Field {
|
|||||||
// TODO: untag!
|
// TODO: untag!
|
||||||
self.#field_ident.serialize(None, writer)?;
|
self.#field_ident.serialize(None, writer)?;
|
||||||
}),
|
}),
|
||||||
|
// TODO: Think about what to do here
|
||||||
|
FieldType::TagName | FieldType::Namespace => None,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -29,8 +29,8 @@ pub enum XmlDeError {
|
|||||||
Other(String),
|
Other(String),
|
||||||
#[error("Invalid variant: {0}")]
|
#[error("Invalid variant: {0}")]
|
||||||
InvalidVariant(String),
|
InvalidVariant(String),
|
||||||
#[error("Invalid field name: ")]
|
#[error("Invalid field name: {0}")]
|
||||||
InvalidFieldName,
|
InvalidFieldName(String),
|
||||||
#[error(transparent)]
|
#[error(transparent)]
|
||||||
InvalidValue(#[from] crate::value::ParseValueError),
|
InvalidValue(#[from] crate::value::ParseValueError),
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user