Enable support for scalar Vec elements

This commit is contained in:
Dirkjan Ochtman 2022-11-25 21:04:09 -08:00
parent 526414cd4b
commit c15a0a985c
5 changed files with 87 additions and 40 deletions

View File

@ -49,9 +49,24 @@ fn serialize_scalar_enum(
impl #impl_generics ToXml for #ident #ty_generics #where_clause { impl #impl_generics ToXml for #ident #ty_generics #where_clause {
fn serialize<W: ::core::fmt::Write + ?::core::marker::Sized>( fn serialize<W: ::core::fmt::Write + ?::core::marker::Sized>(
&self, &self,
field: Option<::instant_xml::Id<'_>>,
serializer: &mut instant_xml::Serializer<W>, serializer: &mut instant_xml::Serializer<W>,
) -> Result<(), instant_xml::Error> { ) -> Result<(), instant_xml::Error> {
serializer.write_str(match self { #variants }) let prefix = match field {
Some(id) => {
let prefix = serializer.write_start(id.name, id.ns, true)?;
serializer.end_start()?;
Some((prefix, id.name))
}
None => None,
};
serializer.write_str(match self { #variants })?;
if let Some((prefix, name)) = prefix {
serializer.write_close(prefix, name)?;
}
Ok(())
} }
const KIND: ::instant_xml::Kind<'static> = ::instant_xml::Kind::Scalar; const KIND: ::instant_xml::Kind<'static> = ::instant_xml::Kind::Scalar;
@ -95,7 +110,7 @@ fn serialize_wrapped_enum(
} }
let v_ident = &variant.ident; let v_ident = &variant.ident;
variants.extend(quote!(#ident::#v_ident(inner) => inner.serialize(serializer)?,)); variants.extend(quote!(#ident::#v_ident(inner) => inner.serialize(None, serializer)?,));
} }
let default_namespace = meta.default_namespace(); let default_namespace = meta.default_namespace();
@ -124,6 +139,7 @@ fn serialize_wrapped_enum(
impl #impl_generics ToXml for #ident #ty_generics #where_clause { impl #impl_generics ToXml for #ident #ty_generics #where_clause {
fn serialize<W: ::core::fmt::Write + ?::core::marker::Sized>( fn serialize<W: ::core::fmt::Write + ?::core::marker::Sized>(
&self, &self,
field: Option<::instant_xml::Id<'_>>,
serializer: &mut instant_xml::Serializer<W>, serializer: &mut instant_xml::Serializer<W>,
) -> Result<(), instant_xml::Error> { ) -> Result<(), instant_xml::Error> {
// Start tag // Start tag
@ -204,6 +220,7 @@ fn serialize_struct(
impl #impl_generics ToXml for #ident #ty_generics #where_clause { impl #impl_generics ToXml for #ident #ty_generics #where_clause {
fn serialize<W: ::core::fmt::Write + ?::core::marker::Sized>( fn serialize<W: ::core::fmt::Write + ?::core::marker::Sized>(
&self, &self,
field: Option<::instant_xml::Id<'_>>,
serializer: &mut instant_xml::Serializer<W>, serializer: &mut instant_xml::Serializer<W>,
) -> Result<(), instant_xml::Error> { ) -> Result<(), instant_xml::Error> {
// Start tag // Start tag
@ -318,17 +335,7 @@ fn process_named_field(
let mut no_lifetime_type = field.ty.clone(); let mut no_lifetime_type = field.ty.clone();
discard_lifetimes(&mut no_lifetime_type); discard_lifetimes(&mut no_lifetime_type);
body.extend(quote!( body.extend(quote!(
match <#no_lifetime_type as ToXml>::KIND { self.#field_name.serialize(Some(::instant_xml::Id { ns: #ns, name: #tag }), serializer)?;
::instant_xml::Kind::Element(_) | ::instant_xml::Kind::Vec(_) => {
self.#field_name.serialize(serializer)?;
}
::instant_xml::Kind::Scalar => {
let prefix = serializer.write_start(#tag, #ns, true)?;
serializer.end_start()?;
self.#field_name.serialize(serializer)?;
serializer.write_close(prefix, #tag)?;
}
}
)); ));
Ok(()) Ok(())

View File

@ -5,7 +5,7 @@ use std::str::FromStr;
#[cfg(feature = "chrono")] #[cfg(feature = "chrono")]
use chrono::{DateTime, Utc}; use chrono::{DateTime, Utc};
use crate::{Deserializer, Error, FromXml, Kind, Serializer, ToXml}; use crate::{Deserializer, Error, FromXml, Id, Kind, Serializer, ToXml};
// Deserializer // Deserializer
struct FromXmlStr<T: FromStr>(T); struct FromXmlStr<T: FromStr>(T);
@ -64,9 +64,24 @@ where
{ {
fn serialize<W: fmt::Write + ?Sized>( fn serialize<W: fmt::Write + ?Sized>(
&self, &self,
field: Option<Id<'_>>,
serializer: &mut Serializer<W>, serializer: &mut Serializer<W>,
) -> Result<(), Error> { ) -> Result<(), Error> {
serializer.write_str(self.0) let prefix = match field {
Some(id) => {
let prefix = serializer.write_start(id.name, id.ns, true)?;
serializer.end_start()?;
Some((prefix, id.name))
}
None => None,
};
serializer.write_str(self.0)?;
if let Some((prefix, name)) = prefix {
serializer.write_close(prefix, name)?;
}
Ok(())
} }
const KIND: Kind<'static> = Kind::Scalar; const KIND: Kind<'static> = Kind::Scalar;
@ -77,9 +92,10 @@ macro_rules! to_xml_for_number {
impl ToXml for $typ { impl ToXml for $typ {
fn serialize<W: fmt::Write + ?Sized>( fn serialize<W: fmt::Write + ?Sized>(
&self, &self,
field: Option<Id<'_>>,
serializer: &mut Serializer<W>, serializer: &mut Serializer<W>,
) -> Result<(), Error> { ) -> Result<(), Error> {
DisplayToXml(self).serialize(serializer) DisplayToXml(self).serialize(field, serializer)
} }
const KIND: Kind<'static> = DisplayToXml::<Self>::KIND; const KIND: Kind<'static> = DisplayToXml::<Self>::KIND;
@ -257,6 +273,7 @@ to_xml_for_number!(f64);
impl ToXml for bool { impl ToXml for bool {
fn serialize<W: fmt::Write + ?Sized>( fn serialize<W: fmt::Write + ?Sized>(
&self, &self,
field: Option<Id<'_>>,
serializer: &mut Serializer<W>, serializer: &mut Serializer<W>,
) -> Result<(), Error> { ) -> Result<(), Error> {
let value = match self { let value = match self {
@ -264,7 +281,7 @@ impl ToXml for bool {
false => "false", false => "false",
}; };
DisplayToXml(&value).serialize(serializer) DisplayToXml(&value).serialize(field, serializer)
} }
const KIND: Kind<'static> = DisplayToXml::<Self>::KIND; const KIND: Kind<'static> = DisplayToXml::<Self>::KIND;
@ -273,9 +290,10 @@ impl ToXml for bool {
impl ToXml for String { impl ToXml for String {
fn serialize<W: fmt::Write + ?Sized>( fn serialize<W: fmt::Write + ?Sized>(
&self, &self,
field: Option<Id<'_>>,
serializer: &mut Serializer<W>, serializer: &mut Serializer<W>,
) -> Result<(), Error> { ) -> Result<(), Error> {
DisplayToXml(&encode(self)?).serialize(serializer) DisplayToXml(&encode(self)?).serialize(field, serializer)
} }
const KIND: Kind<'static> = DisplayToXml::<Self>::KIND; const KIND: Kind<'static> = DisplayToXml::<Self>::KIND;
@ -284,10 +302,11 @@ impl ToXml for String {
impl ToXml for char { impl ToXml for char {
fn serialize<W: fmt::Write + ?Sized>( fn serialize<W: fmt::Write + ?Sized>(
&self, &self,
field: Option<Id<'_>>,
serializer: &mut Serializer<W>, serializer: &mut Serializer<W>,
) -> Result<(), Error> { ) -> Result<(), Error> {
let mut tmp = [0u8; 4]; let mut tmp = [0u8; 4];
DisplayToXml(&encode(&*self.encode_utf8(&mut tmp))?).serialize(serializer) DisplayToXml(&encode(&*self.encode_utf8(&mut tmp))?).serialize(field, serializer)
} }
const KIND: Kind<'static> = DisplayToXml::<Self>::KIND; const KIND: Kind<'static> = DisplayToXml::<Self>::KIND;
@ -296,9 +315,10 @@ impl ToXml for char {
impl ToXml for &str { impl ToXml for &str {
fn serialize<W: fmt::Write + ?Sized>( fn serialize<W: fmt::Write + ?Sized>(
&self, &self,
field: Option<Id<'_>>,
serializer: &mut Serializer<W>, serializer: &mut Serializer<W>,
) -> Result<(), Error> { ) -> Result<(), Error> {
DisplayToXml(&encode(self)?).serialize(serializer) DisplayToXml(&encode(self)?).serialize(field, serializer)
} }
const KIND: Kind<'static> = DisplayToXml::<Self>::KIND; const KIND: Kind<'static> = DisplayToXml::<Self>::KIND;
@ -307,9 +327,10 @@ impl ToXml for &str {
impl ToXml for Cow<'_, str> { impl ToXml for Cow<'_, str> {
fn serialize<W: fmt::Write + ?Sized>( fn serialize<W: fmt::Write + ?Sized>(
&self, &self,
field: Option<Id<'_>>,
serializer: &mut Serializer<W>, serializer: &mut Serializer<W>,
) -> Result<(), Error> { ) -> Result<(), Error> {
DisplayToXml(&encode(self)?).serialize(serializer) DisplayToXml(&encode(self)?).serialize(field, serializer)
} }
const KIND: Kind<'static> = DisplayToXml::<Self>::KIND; const KIND: Kind<'static> = DisplayToXml::<Self>::KIND;
@ -318,10 +339,11 @@ impl ToXml for Cow<'_, str> {
impl<T: ToXml> ToXml for Option<T> { impl<T: ToXml> ToXml for Option<T> {
fn serialize<W: fmt::Write + ?Sized>( fn serialize<W: fmt::Write + ?Sized>(
&self, &self,
field: Option<Id<'_>>,
serializer: &mut Serializer<W>, serializer: &mut Serializer<W>,
) -> Result<(), Error> { ) -> Result<(), Error> {
match self { match self {
Some(v) => v.serialize(serializer), Some(v) => v.serialize(field, serializer),
None => Ok(()), None => Ok(()),
} }
} }
@ -429,36 +451,48 @@ impl<'xml, T: FromXml<'xml>> FromXml<'xml> for Vec<T> {
Ok(Vec::new()) Ok(Vec::new())
} }
const KIND: Kind<'static> = Kind::Vec(T::KIND.element()); const KIND: Kind<'static> = T::KIND;
const WRAPPED: bool = true;
} }
impl<T: ToXml> ToXml for Vec<T> { impl<T: ToXml> ToXml for Vec<T> {
fn serialize<W: fmt::Write + ?Sized>( fn serialize<W: fmt::Write + ?Sized>(
&self, &self,
field: Option<Id<'_>>,
serializer: &mut Serializer<W>, serializer: &mut Serializer<W>,
) -> Result<(), Error> { ) -> Result<(), Error> {
match T::KIND {
Kind::Element(_) => {}
_ => return Err(Error::UnexpectedState("only elements allowed in `Vec`")),
}
for i in self { for i in self {
i.serialize(serializer)?; i.serialize(field, serializer)?;
} }
Ok(()) Ok(())
} }
const KIND: Kind<'static> = Kind::Vec(T::KIND.element()); const KIND: Kind<'static> = T::KIND;
} }
#[cfg(feature = "chrono")] #[cfg(feature = "chrono")]
impl ToXml for DateTime<Utc> { impl ToXml for DateTime<Utc> {
fn serialize<W: fmt::Write + ?Sized>( fn serialize<W: fmt::Write + ?Sized>(
&self, &self,
field: Option<Id<'_>>,
serializer: &mut Serializer<W>, serializer: &mut Serializer<W>,
) -> Result<(), Error> { ) -> Result<(), Error> {
serializer.write_str(&self.to_rfc3339()) let prefix = match field {
Some(id) => {
let prefix = serializer.write_start(id.name, id.ns, true)?;
serializer.end_start()?;
Some((prefix, id.name))
}
None => None,
};
serializer.write_str(&self.to_rfc3339())?;
if let Some((prefix, name)) = prefix {
serializer.write_close(prefix, name)?;
}
Ok(())
} }
const KIND: Kind<'static> = Kind::Scalar; const KIND: Kind<'static> = Kind::Scalar;

View File

@ -16,6 +16,7 @@ pub use ser::Serializer;
pub trait ToXml { pub trait ToXml {
fn serialize<W: fmt::Write + ?Sized>( fn serialize<W: fmt::Write + ?Sized>(
&self, &self,
field: Option<Id<'_>>,
serializer: &mut Serializer<W>, serializer: &mut Serializer<W>,
) -> Result<(), Error>; ) -> Result<(), Error>;
@ -25,9 +26,10 @@ pub trait ToXml {
impl<'a, T: ToXml + ?Sized> ToXml for &'a T { impl<'a, T: ToXml + ?Sized> ToXml for &'a T {
fn serialize<W: fmt::Write + ?Sized>( fn serialize<W: fmt::Write + ?Sized>(
&self, &self,
field: Option<Id<'_>>,
serializer: &mut Serializer<W>, serializer: &mut Serializer<W>,
) -> Result<(), Error> { ) -> Result<(), Error> {
(*self).serialize(serializer) (*self).serialize(field, serializer)
} }
const KIND: Kind<'static> = T::KIND; const KIND: Kind<'static> = T::KIND;
@ -46,6 +48,7 @@ pub trait FromXml<'xml>: Sized {
} }
const KIND: Kind<'static>; const KIND: Kind<'static>;
const WRAPPED: bool = false;
} }
pub fn from_str<'xml, T: FromXml<'xml>>(input: &'xml str) -> Result<T, Error> { pub fn from_str<'xml, T: FromXml<'xml>>(input: &'xml str) -> Result<T, Error> {
@ -53,7 +56,6 @@ pub fn from_str<'xml, T: FromXml<'xml>>(input: &'xml str) -> Result<T, Error> {
let id = context.element_id(&root)?; let id = context.element_id(&root)?;
let expected = match T::KIND { let expected = match T::KIND {
Kind::Scalar => return Err(Error::UnexpectedState("found scalar as root")), Kind::Scalar => return Err(Error::UnexpectedState("found scalar as root")),
Kind::Vec(_) => return Err(Error::UnexpectedState("found list as root")),
Kind::Element(expected) => expected, Kind::Element(expected) => expected,
}; };
@ -79,7 +81,7 @@ pub fn to_writer(
value: &(impl ToXml + ?Sized), value: &(impl ToXml + ?Sized),
output: &mut (impl fmt::Write + ?Sized), output: &mut (impl fmt::Write + ?Sized),
) -> Result<(), Error> { ) -> Result<(), Error> {
value.serialize(&mut Serializer::new(output)) value.serialize(None, &mut Serializer::new(output))
} }
pub trait FromXmlOwned: for<'xml> FromXml<'xml> {} pub trait FromXmlOwned: for<'xml> FromXml<'xml> {}
@ -124,7 +126,6 @@ pub enum Error {
pub enum Kind<'a> { pub enum Kind<'a> {
Scalar, Scalar,
Element(Id<'a>), Element(Id<'a>),
Vec(Id<'a>),
} }
impl<'a> Kind<'a> { impl<'a> Kind<'a> {
@ -140,7 +141,6 @@ impl<'a> Kind<'a> {
match self { match self {
Kind::Scalar => id == field, Kind::Scalar => id == field,
Kind::Element(name) => id == *name, Kind::Element(name) => id == *name,
Kind::Vec(inner) => id == *inner,
} }
} }
} }

View File

@ -85,7 +85,7 @@ impl<'xml, W: fmt::Write + ?Sized> Serializer<'xml, W> {
} }
self.state = State::Scalar; self.state = State::Scalar;
value.serialize(self)?; value.serialize(None, self)?;
self.state = State::Attribute; self.state = State::Attribute;
self.output.write_char('"')?; self.output.write_char('"')?;
Ok(()) Ok(())

View File

@ -8,26 +8,32 @@ struct Foo {
#[derive(Debug, Eq, FromXml, PartialEq, ToXml)] #[derive(Debug, Eq, FromXml, PartialEq, ToXml)]
struct Bar { struct Bar {
foo: Vec<Foo>, foo: Vec<Foo>,
baz: Vec<String>,
} }
#[test] #[test]
fn vec() { fn vec() {
let val = Bar { foo: vec![] }; let val = Bar {
foo: vec![],
baz: vec![],
};
let xml = "<Bar></Bar>"; let xml = "<Bar></Bar>";
assert_eq!(xml, to_string(&val).unwrap()); assert_eq!(xml, to_string(&val).unwrap());
assert_eq!(val, from_str(xml).unwrap()); assert_eq!(val, from_str(xml).unwrap());
let val = Bar { let val = Bar {
foo: vec![Foo { bar: 42 }], foo: vec![Foo { bar: 42 }],
baz: vec!["hello".to_owned()],
}; };
let xml = "<Bar><Foo><bar>42</bar></Foo></Bar>"; let xml = "<Bar><Foo><bar>42</bar></Foo><baz>hello</baz></Bar>";
assert_eq!(xml, to_string(&val).unwrap()); assert_eq!(xml, to_string(&val).unwrap());
assert_eq!(val, from_str(xml).unwrap()); assert_eq!(val, from_str(xml).unwrap());
let val = Bar { let val = Bar {
foo: vec![Foo { bar: 42 }, Foo { bar: 73 }], foo: vec![Foo { bar: 42 }, Foo { bar: 73 }],
baz: vec!["hello".to_owned(), "world".to_owned()],
}; };
let xml = "<Bar><Foo><bar>42</bar></Foo><Foo><bar>73</bar></Foo></Bar>"; let xml = "<Bar><Foo><bar>42</bar></Foo><Foo><bar>73</bar></Foo><baz>hello</baz><baz>world</baz></Bar>";
assert_eq!(xml, to_string(&val).unwrap()); assert_eq!(xml, to_string(&val).unwrap());
assert_eq!(val, from_str(xml).unwrap()); assert_eq!(val, from_str(xml).unwrap());
} }