xml: Add tag_name extractor

This commit is contained in:
Lennart
2024-12-23 11:57:01 +01:00
parent 67ab2ff62b
commit fa2851dc81
4 changed files with 31 additions and 5 deletions

View File

@@ -47,6 +47,8 @@ pub enum FieldType {
Attr, Attr,
Text, Text,
Untagged, Untagged,
TagName,
Namespace,
} }
#[derive(Default, FromField)] #[derive(Default, FromField)]

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) } 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) => {

View File

@@ -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,
} }
} }
} }

View File

@@ -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),
} }