✨ Complete serializer (#173)

Cargo.toml

@@ -17,3 +17,4 @@ thiserror = "1.0"
 serde = { version = "1.0", features = ["derive"] }
 simple_logger = "2.1"
 docmatic = "0.1"
+rstest = "0.12"

src/error.rs

@@ -47,6 +47,12 @@ pub enum Error {
         #[from]
         source: ::xml::reader::Error,
     },
+
+    #[error("Writer: {source}")]
+    Writer {
+        #[from]
+        source: ::xml::writer::Error,
+    },
 }
 
 pub type Result<T> = std::result::Result<T, Error>;

src/lib.rs

@@ -59,7 +59,7 @@
 //! }
 //!
 //! fn main() {
-//!     let src = r#"<Item><name>Banana</name><source>Store</source></Item>"#;
+//!     let src = r#"<?xml version="1.0" encoding="UTF-8"?><Item><name>Banana</name><source>Store</source></Item>"#;
 //!     let should_be = Item {
 //!         name: "Banana".to_string(),
 //!         source: "Store".to_string(),

src/ser/map.rs

@@ -0,0 +1,124 @@
+use super::{plain::to_plain_string, Serializer};
+use crate::error::{Error, Result};
+use log::debug;
+use serde::ser::Serialize;
+use std::io::Write;
+
+pub struct MapSerializer<'ser, W: 'ser + Write> {
+    ser: &'ser mut Serializer<W>,
+    must_close_tag: bool,
+}
+
+impl<'ser, W: 'ser + Write> MapSerializer<'ser, W> {
+    pub fn new(ser: &'ser mut Serializer<W>, must_close_tag: bool) -> Self {
+        MapSerializer {
+            ser,
+            must_close_tag,
+        }
+    }
+}
+
+impl<'ser, W: Write> serde::ser::SerializeMap for MapSerializer<'ser, W> {
+    type Ok = ();
+    type Error = Error;
+
+    fn serialize_key<T>(&mut self, key: &T) -> Result<()>
+    where
+        T: ?Sized + Serialize,
+    {
+        self.ser.open_tag(&to_plain_string(key)?)?;
+        Ok(())
+    }
+
+    fn serialize_value<T>(&mut self, value: &T) -> Result<()>
+    where
+        T: ?Sized + Serialize,
+    {
+        value.serialize(&mut *self.ser)?;
+        Ok(())
+    }
+
+    fn end(self) -> Result<()> {
+        if self.must_close_tag {
+            self.ser.end_tag()?;
+        }
+        Ok(())
+    }
+}
+
+pub struct StructSerializer<'ser, W: 'ser + Write> {
+    ser: &'ser mut Serializer<W>,
+    must_close_tag: bool,
+}
+
+impl<'ser, W: 'ser + Write> StructSerializer<'ser, W> {
+    pub fn new(ser: &'ser mut Serializer<W>, must_close_tag: bool) -> Self {
+        StructSerializer {
+            ser,
+            must_close_tag,
+        }
+    }
+
+    fn serialize_struct_field<T>(&mut self, key: &'static str, value: &T) -> Result<()>
+    where
+        T: ?Sized + Serialize,
+    {
+        if key.starts_with("@") {
+            debug!("attribute {}", key);
+            self.ser.add_attr(&key[1..], to_plain_string(value)?)
+        } else if key == "$value" {
+            self.ser.build_start_tag()?;
+            debug!("body");
+            value.serialize(&mut *self.ser)?;
+            Ok(())
+        } else {
+            self.ser.build_start_tag()?;
+            self.ser.open_tag(key)?;
+            debug!("field {}", key);
+            value.serialize(&mut *self.ser)?;
+            debug!("end field");
+            Ok(())
+        }
+    }
+
+    fn after_fields(self) -> Result<()> {
+        self.ser.build_start_tag()?;
+        self.ser.end_tag()?;
+        if self.must_close_tag {
+            self.ser.end_tag()?;
+        }
+        Ok(())
+    }
+}
+
+impl<'ser, W: 'ser + Write> serde::ser::SerializeStruct for StructSerializer<'ser, W> {
+    type Ok = ();
+    type Error = Error;
+
+    fn serialize_field<T>(&mut self, key: &'static str, value: &T) -> Result<()>
+    where
+        T: ?Sized + Serialize,
+    {
+        self.serialize_struct_field(key, value)
+    }
+
+    fn end(self) -> Result<()> {
+        self.after_fields()
+    }
+}
+
+impl<'ser, W: 'ser + Write> serde::ser::SerializeStructVariant for StructSerializer<'ser, W> {
+    type Ok = ();
+    type Error = Error;
+
+    fn serialize_field<T>(&mut self, key: &'static str, value: &T) -> Result<()>
+    where
+        T: ?Sized + Serialize,
+    {
+        self.serialize_struct_field(key, value)
+    }
+
+    fn end(self) -> Result<()> {
+        self.after_fields()
+    }
+}