rustical/crates/xml/src/value.rs

use crate::{XmlDeserialize, XmlError, XmlSerialize};
use quick_xml::events::{BytesEnd, BytesStart, BytesText, Event};
use quick_xml::name::Namespace;
use std::collections::HashMap;
use std::num::{ParseFloatError, ParseIntError};
use std::{convert::Infallible, io::BufRead};
use thiserror::Error;

pub trait ValueSerialize {
    fn serialize(&self) -> String;
}

pub trait ValueDeserialize: Sized {
    fn deserialize(val: &str) -> Result<Self, XmlError>;
}

#[derive(Debug, Error)]
pub enum ParseValueError {
    #[error(transparent)]
    Infallible(#[from] Infallible),
    #[error(transparent)]
    ParseIntError(#[from] ParseIntError),
    #[error(transparent)]
    ParseFloatError(#[from] ParseFloatError),
    #[error("{0}")]
    Other(String),
}

macro_rules! impl_value_parse {
    ($t:ty) => {
        impl ValueSerialize for $t {
            fn serialize(&self) -> String {
                self.to_string()
            }
        }

        impl ValueDeserialize for $t {
            fn deserialize(val: &str) -> Result<Self, XmlError> {
                val.parse()
                    .map_err(ParseValueError::from)
                    .map_err(XmlError::from)
            }
        }
    };
}

impl_value_parse!(String);
impl_value_parse!(i8);
impl_value_parse!(u8);
impl_value_parse!(i16);
impl_value_parse!(u16);
impl_value_parse!(f32);
impl_value_parse!(i32);
impl_value_parse!(u32);
impl_value_parse!(f64);
impl_value_parse!(i64);
impl_value_parse!(u64);
impl_value_parse!(isize);
impl_value_parse!(usize);

impl ValueSerialize for &str {
    fn serialize(&self) -> String {
        self.to_string()
    }
}

impl<T: ValueDeserialize> XmlDeserialize for T {
    fn deserialize<R: BufRead>(
        reader: &mut quick_xml::NsReader<R>,
        _start: &BytesStart,
        empty: bool,
    ) -> Result<Self, XmlError> {
        let mut string = String::new();

        if !empty {
            let mut buf = Vec::new();
            loop {
                match reader.read_event_into(&mut buf)? {
                    Event::Text(bytes_text) => {
                        let text = bytes_text.decode()?;
                        string.push_str(&text);
                    }
                    Event::CData(cdata) => {
                        let text = String::from_utf8(cdata.to_vec())?;
                        string.push_str(&text);
                    }
                    Event::GeneralRef(gref) => {
                        if let Some(char) = gref.resolve_char_ref()? {
                            string.push(char);
                        } else if let Some(text) =
                            quick_xml::escape::resolve_xml_entity(&gref.xml_content()?)
                        {
                            string.push_str(text);
                        } else {
                            return Err(XmlError::UnsupportedEvent("invalid XML ref"));
                        }
                    }
                    Event::End(_) => break,
                    Event::Eof => return Err(XmlError::Eof),
                    _ => return Err(XmlError::UnsupportedEvent("todo")),
                };
            }
        }

        ValueDeserialize::deserialize(&string)
    }
}

impl<T: ValueSerialize> XmlSerialize for T {
    fn serialize(
        &self,
        ns: Option<Namespace>,
        tag: Option<&str>,
        namespaces: &HashMap<Namespace, &str>,
        writer: &mut quick_xml::Writer<&mut Vec<u8>>,
    ) -> std::io::Result<()> {
        let prefix = ns
            .map(|ns| namespaces.get(&ns))
            .unwrap_or(None)
            .map(|prefix| {
                if !prefix.is_empty() {
                    [*prefix, ":"].concat()
                } else {
                    String::new()
                }
            });
        let has_prefix = prefix.is_some();
        let tagname = tag.map(|tag| [&prefix.unwrap_or_default(), tag].concat());
        if let Some(tagname) = tagname.as_ref() {
            let mut bytes_start = BytesStart::new(tagname);
            if !has_prefix && let Some(ns) = &ns {
                bytes_start.push_attribute((b"xmlns".as_ref(), ns.as_ref()));
            }
            writer.write_event(Event::Start(bytes_start))?;
        }
        writer.write_event(Event::Text(BytesText::new(&self.serialize())))?;
        if let Some(tagname) = tagname {
            writer.write_event(Event::End(BytesEnd::new(tagname)))?;
        }
        Ok(())
    }

    fn attributes<'a>(&self) -> Option<Vec<quick_xml::events::attributes::Attribute<'a>>> {
        None
    }
}