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

rustical_xml/
de.rs
