mirror of
https://github.com/lennart-k/rustical.git
synced 2025-12-14 08:12:24 +00:00
xml: tag_name serialization in untagged fields
This commit is contained in:
@@ -185,6 +185,7 @@ impl NamedStruct {
|
|||||||
.fields
|
.fields
|
||||||
.iter()
|
.iter()
|
||||||
.filter(|field| field.attrs.xml_ty == FieldType::Untagged)
|
.filter(|field| field.attrs.xml_ty == FieldType::Untagged)
|
||||||
|
.filter(|field| !field.attrs.flatten.is_present())
|
||||||
.map(|field| {
|
.map(|field| {
|
||||||
let field_ident = field.field_ident();
|
let field_ident = field.field_ident();
|
||||||
quote! {
|
quote! {
|
||||||
@@ -209,6 +210,17 @@ impl NamedStruct {
|
|||||||
}
|
}
|
||||||
});
|
});
|
||||||
|
|
||||||
|
let tag_name_field = self
|
||||||
|
.fields
|
||||||
|
.iter()
|
||||||
|
.find(|field| field.attrs.xml_ty == FieldType::TagName)
|
||||||
|
.map(|field| {
|
||||||
|
let field_ident = field.field_ident();
|
||||||
|
quote! {
|
||||||
|
let tag_str = Some(tag_str.unwrap_or(self.#field_ident.to_string()));
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
quote! {
|
quote! {
|
||||||
impl #impl_generics ::rustical_xml::XmlSerialize for #ident #type_generics #where_clause {
|
impl #impl_generics ::rustical_xml::XmlSerialize for #ident #type_generics #where_clause {
|
||||||
fn serialize<W: ::std::io::Write>(
|
fn serialize<W: ::std::io::Write>(
|
||||||
@@ -219,14 +231,15 @@ impl NamedStruct {
|
|||||||
) -> ::std::io::Result<()> {
|
) -> ::std::io::Result<()> {
|
||||||
use ::quick_xml::events::{BytesEnd, BytesStart, BytesText, Event};
|
use ::quick_xml::events::{BytesEnd, BytesStart, BytesText, Event};
|
||||||
|
|
||||||
let tag_str = tag.map(String::from_utf8_lossy);
|
let tag_str = tag.map(|a| String::from_utf8_lossy(a).to_string());
|
||||||
|
#tag_name_field;
|
||||||
|
|
||||||
if let Some(tag) = &tag_str {
|
if let Some(tag) = &tag_str {
|
||||||
let mut bytes_start = BytesStart::new(tag.to_owned());
|
let mut bytes_start = BytesStart::new(tag.to_owned());
|
||||||
if let Some(attrs) = self.attributes() {
|
if let Some(attrs) = self.attributes() {
|
||||||
bytes_start.extend_attributes(attrs);
|
bytes_start.extend_attributes(attrs);
|
||||||
}
|
}
|
||||||
// #(#untagged_attributes);*
|
#(#untagged_attributes);*
|
||||||
writer.write_event(Event::Start(bytes_start))?;
|
writer.write_event(Event::Start(bytes_start))?;
|
||||||
}
|
}
|
||||||
#(#tag_writers);*
|
#(#tag_writers);*
|
||||||
|
|||||||
@@ -50,6 +50,16 @@ impl_value_parse!(u64);
|
|||||||
impl_value_parse!(isize);
|
impl_value_parse!(isize);
|
||||||
impl_value_parse!(usize);
|
impl_value_parse!(usize);
|
||||||
|
|
||||||
|
impl Value for &str {
|
||||||
|
fn serialize(&self) -> String {
|
||||||
|
self.to_string()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn deserialize(val: &str) -> Result<Self, XmlDeError> {
|
||||||
|
Err(XmlDeError::Other("TODO: Handle this error".to_owned()))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
impl<T: Value> XmlDeserialize for T {
|
impl<T: Value> XmlDeserialize for T {
|
||||||
fn deserialize<R: BufRead>(
|
fn deserialize<R: BufRead>(
|
||||||
reader: &mut quick_xml::NsReader<R>,
|
reader: &mut quick_xml::NsReader<R>,
|
||||||
|
|||||||
@@ -1,5 +1,7 @@
|
|||||||
|
use std::borrow::{Borrow, Cow};
|
||||||
|
|
||||||
use quick_xml::Writer;
|
use quick_xml::Writer;
|
||||||
use rustical_xml::{XmlRootTag, XmlSerialize, XmlSerializeRoot};
|
use rustical_xml::{XmlDocument, XmlRootTag, XmlSerialize, XmlSerializeRoot};
|
||||||
use xml_derive::XmlDeserialize;
|
use xml_derive::XmlDeserialize;
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
@@ -132,7 +134,7 @@ fn test_struct_serialize_with() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn serialize_href<W: ::std::io::Write>(
|
fn serialize_href<W: ::std::io::Write>(
|
||||||
val: &String,
|
val: &str,
|
||||||
ns: Option<&[u8]>,
|
ns: Option<&[u8]>,
|
||||||
tag: Option<&[u8]>,
|
tag: Option<&[u8]>,
|
||||||
writer: &mut Writer<W>,
|
writer: &mut Writer<W>,
|
||||||
@@ -150,3 +152,39 @@ fn test_struct_serialize_with() {
|
|||||||
let out = String::from_utf8(buf).unwrap();
|
let out = String::from_utf8(buf).unwrap();
|
||||||
assert_eq!(out, "<document><href>OKAY</href></document>");
|
assert_eq!(out, "<document><href>OKAY</href></document>");
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_struct_tag_list() {
|
||||||
|
#[derive(Debug, XmlRootTag, XmlSerialize, XmlDeserialize, PartialEq)]
|
||||||
|
#[xml(root = b"document")]
|
||||||
|
struct Document {
|
||||||
|
#[xml(ty = "untagged", flatten)]
|
||||||
|
tags: Vec<Tag>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, XmlSerialize, XmlDeserialize, PartialEq)]
|
||||||
|
struct Tag {
|
||||||
|
#[xml(ty = "tag_name")]
|
||||||
|
name: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut buf = Vec::new();
|
||||||
|
let mut writer = quick_xml::Writer::new(&mut buf);
|
||||||
|
Document {
|
||||||
|
tags: vec![
|
||||||
|
Tag {
|
||||||
|
name: "hello".to_owned(),
|
||||||
|
},
|
||||||
|
Tag {
|
||||||
|
name: "ok".to_owned(),
|
||||||
|
},
|
||||||
|
Tag {
|
||||||
|
name: "wow".to_owned(),
|
||||||
|
},
|
||||||
|
],
|
||||||
|
}
|
||||||
|
.serialize_root(&mut writer)
|
||||||
|
.unwrap();
|
||||||
|
let out = String::from_utf8(buf).unwrap();
|
||||||
|
dbg!(out);
|
||||||
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user