rustical_xml/
de.rs

1use crate::XmlError;
2use crate::XmlRootTag;
3use quick_xml::events::{BytesStart, Event};
4use quick_xml::name::ResolveResult;
5use std::io::BufRead;
6pub use xml_derive::XmlDeserialize;
7pub use xml_derive::XmlDocument;
8
9pub trait XmlDeserialize: Sized {
10    fn deserialize<R: BufRead>(
11        reader: &mut quick_xml::NsReader<R>,
12        start: &BytesStart,
13        empty: bool,
14    ) -> Result<Self, XmlError>;
15}
16
17pub trait XmlDocument: XmlDeserialize {
18    fn parse<R: BufRead>(reader: quick_xml::NsReader<R>) -> Result<Self, XmlError>;
19
20    #[inline]
21    fn parse_reader<R: BufRead>(input: R) -> Result<Self, XmlError> {
22        let mut reader = quick_xml::NsReader::from_reader(input);
23        reader.config_mut().trim_text(true);
24        Self::parse(reader)
25    }
26
27    #[inline]
28    fn parse_str(s: &str) -> Result<Self, XmlError> {
29        Self::parse_reader(s.as_bytes())
30    }
31}
32
33impl<T: XmlRootTag + XmlDeserialize> XmlDocument for T {
34    fn parse<R: BufRead>(mut reader: quick_xml::NsReader<R>) -> Result<Self, XmlError>
35    where
36        Self: XmlDeserialize,
37    {
38        let mut buf = Vec::new();
39        loop {
40            let event = reader.read_event_into(&mut buf)?;
41            let empty = matches!(event, Event::Empty(_));
42            match event {
43                Event::Decl(_) | Event::Comment(_) => { /*  ignore this */ }
44                Event::Start(start) | Event::Empty(start) => {
45                    let (ns, name) = reader.resolve_element(start.name());
46                    let matches = match (Self::root_ns(), &ns, name) {
47                        // Wrong tag
48                        (_, _, name) if name.as_ref() != Self::root_tag().as_bytes() => false,
49                        // Wrong namespace
50                        (Some(root_ns), ns, _) if &ResolveResult::Bound(root_ns) != ns => false,
51                        _ => true,
52                    };
53                    if !matches {
54                        let root_ns = Self::root_ns();
55                        return Err(XmlError::InvalidTag(
56                            format!("{ns:?}"),
57                            String::from_utf8_lossy(name.as_ref()).to_string(),
58                            format!("{root_ns:?}"),
59                            Self::root_tag().to_owned(),
60                        ));
61                    }
62
63                    return Self::deserialize(&mut reader, &start, empty);
64                }
65                Event::Eof => return Err(XmlError::Eof),
66                _ => return Err(XmlError::UnsupportedEvent("unknown, todo")),
67            }
68        }
69    }
70}
71
72impl XmlDeserialize for () {
73    fn deserialize<R: BufRead>(
74        reader: &mut quick_xml::NsReader<R>,
75        start: &BytesStart,
76        empty: bool,
77    ) -> Result<Self, XmlError> {
78        if empty {
79            return Ok(());
80        }
81        let mut buf = Vec::new();
82        loop {
83            match reader.read_event_into(&mut buf)? {
84                Event::End(e) if e.name() == start.name() => return Ok(()),
85                Event::Eof => return Err(XmlError::Eof),
86                _ => {}
87            }
88        }
89    }
90}
rustical_xml/de.rs

rustical_xml/
de.rs
