use std::borrow::Cow; use std::future::Future; use quick_xml::events::{Event, BytesStart, BytesDecl, BytesText}; use quick_xml::events::attributes::AttrError; use quick_xml::name::{Namespace, QName, PrefixDeclaration, ResolveResult, ResolveResult::*}; use quick_xml::reader::NsReader; use tokio::io::AsyncBufRead; use super::types::*; #[derive(Debug)] pub enum ParsingError { NamespacePrefixAlreadyUsed, WrongToken, TagNotFound, QuickXml(quick_xml::Error), Eof } impl From for ParsingError { fn from(value: AttrError) -> Self { Self::QuickXml(value.into()) } } impl From for ParsingError { fn from(value: quick_xml::Error) -> Self { Self::QuickXml(value) } } const DAV_URN: &[u8] = b"DAV:"; const CALDAV_URN: &[u8] = b"urn:ietf:params:xml:ns:caldav"; const CARDDAV_URN: &[u8] = b"urn:ietf:params:xml:ns:carddav"; //const XML_URN: &[u8] = b"xml"; trait Reader = AsyncBufRead+Unpin+'static; pub struct PeekRead { evt: Event<'static>, rdr: NsReader, buf: Vec, } impl PeekRead { async fn new(mut rdr: NsReader) -> Result { let mut buf: Vec = vec![]; let evt = rdr.read_event_into_async(&mut buf).await?.into_owned(); buf.clear(); Ok(Self { evt, rdr, buf }) } fn peek(&self) -> &Event<'static> { &self.evt } /// skip tag. Can't skip end, can't skip eof. async fn skip(&mut self) -> Result, ParsingError> { match &self.evt { Event::Start(b) => { let _span = self.rdr.read_to_end_into_async(b.to_end().name(), &mut self.buf).await?; self.next().await }, Event::End(_) => Err(ParsingError::WrongToken), Event::Eof => Err(ParsingError::Eof), _ => self.next().await, } } /// read one more tag async fn next(&mut self) -> Result, ParsingError> { let evt = self.rdr.read_event_into_async(&mut self.buf).await?.into_owned(); self.buf.clear(); let old_evt = std::mem::replace(&mut self.evt, evt); Ok(old_evt) } /// check if this is the desired tag fn is_tag(&self, ns: &[u8], key: &str) -> bool { let qname = match self.peek() { Event::Start(bs) | Event::Empty(bs) => bs.name(), Event::End(be) => be.name(), _ => return false, }; let (extr_ns, local) = self.rdr.resolve_element(qname); if local.into_inner() != key.as_bytes() { return false } match extr_ns { ResolveResult::Bound(v) => v.into_inner() == ns, _ => false, } } /// find start tag async fn tag_start(&mut self, ns: &[u8], key: &str) -> Result, ParsingError> { loop { match self.peek() { Event::Start(b) if self.is_tag(ns, key) => break, _ => { self.skip().await?; }, } } self.next().await } // find stop tag async fn tag_stop(&mut self, ns: &[u8], key: &str) -> Result, ParsingError> { loop { match self.peek() { Event::End(b) if self.is_tag(ns, key) => break, _ => { self.skip().await?; }, } } self.next().await } } pub trait QReadable: Sized { async fn read(xml: &mut PeekRead) -> Result; } impl QReadable for PropFind { async fn read(xml: &mut PeekRead) -> Result, ParsingError> { // Find propfind xml.tag_start(DAV_URN, "propfind").await?; // Find any tag let propfind: PropFind = loop { match xml.peek() { Event::Start(_) if xml.is_tag(DAV_URN, "allprop") => { xml.tag_start(DAV_URN, "allprop").await?; let r = PropFind::AllProp(Some(Include::read(xml).await?)); xml.tag_stop(DAV_URN, "allprop").await?; break r }, Event::Start(_) if xml.is_tag(DAV_URN, "prop") => { xml.tag_start(DAV_URN, "prop").await?; let r = PropFind::Prop(PropName::read(xml).await?); xml.tag_stop(DAV_URN, "prop").await?; break r }, Event::Empty(_) if xml.is_tag(DAV_URN, "allprop") => { xml.next().await?; break PropFind::AllProp(None) }, Event::Empty(_) if xml.is_tag(DAV_URN, "propname") => { xml.next().await?; break PropFind::PropName }, _ => { xml.skip().await?; }, } }; // Close tag xml.tag_stop(DAV_URN, "propfind").await?; Ok(propfind) } } impl QReadable for Include { async fn read(xml: &mut PeekRead) -> Result, ParsingError> { xml.tag_start(DAV_URN, "include").await?; let mut acc: Vec> = Vec::new(); loop { match xml.peek() { Event::Start(_) => acc.push(PropertyRequest::read(xml).await?), Event::End(_) if xml.is_tag(DAV_URN, "include") => break, _ => { xml.skip().await?; }, } } xml.tag_stop(DAV_URN, "include").await?; Ok(Include(acc)) } } impl QReadable for PropName { async fn read(xml: &mut PeekRead) -> Result, ParsingError> { xml.tag_start(DAV_URN, "prop").await?; let mut acc: Vec> = Vec::new(); loop { match xml.peek() { Event::Start(_) => acc.push(PropertyRequest::read(xml).await?), Event::End(_) if xml.is_tag(DAV_URN, "prop") => break, _ => { xml.skip().await?; }, } } xml.tag_stop(DAV_URN, "prop").await?; Ok(PropName(acc)) } } impl QReadable for PropertyRequest { async fn read(xml: &mut PeekRead) -> Result, ParsingError> { /*match xml.peek() { }*/ unimplemented!(); } } #[cfg(test)] mod tests { use super::*; #[tokio::test] async fn basic_propfind() { let src = r#" "#; let mut rdr = PeekRead::new(NsReader::from_reader(src.as_bytes())).await.unwrap(); let got = PropFind::::read(&mut rdr).await.unwrap(); assert!(matches!(got, PropFind::PropName)); } }