beginning...
This commit is contained in:
parent
ad25912a0f
commit
c9edf6c37c
2 changed files with 88 additions and 100 deletions
|
@ -1,17 +1,18 @@
|
|||
use std::borrow::Cow;
|
||||
|
||||
use im::HashMap;
|
||||
use quick_xml::events::{BytesStart, BytesText};
|
||||
use quick_xml::events::{Event, BytesStart, BytesDecl, BytesText};
|
||||
use quick_xml::events::attributes::AttrError;
|
||||
use quick_xml::name::PrefixDeclaration;
|
||||
use quick_xml::reader::Reader;
|
||||
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)
|
||||
}
|
||||
impl From<AttrError> for ParsingError {
|
||||
|
@ -29,112 +30,98 @@ 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";
|
||||
const DAV_NS: ResolveResult = Bound(Namespace(DAV_URN));
|
||||
|
||||
#[derive(PartialEq, Clone)]
|
||||
pub enum XmlNamespace {
|
||||
None,
|
||||
Dav,
|
||||
CalDav,
|
||||
CardDav,
|
||||
Xml,
|
||||
Unknown(Vec<u8>),
|
||||
pub struct PeekRead<T: AsyncBufRead+Unpin> {
|
||||
evt: Event<'static>,
|
||||
rdr: NsReader<T>,
|
||||
buf: Vec<u8>,
|
||||
}
|
||||
impl From<&[u8]> for XmlNamespace {
|
||||
fn from(value: &[u8]) -> Self {
|
||||
match value {
|
||||
[] => Self::None,
|
||||
DAV_URN => Self::Dav,
|
||||
CALDAV_URN => Self::CalDav,
|
||||
CARDDAV_URN => Self::CardDav,
|
||||
XML_URN => Self::Xml,
|
||||
v => Self::Unknown(v.into()),
|
||||
}
|
||||
impl<T: AsyncBufRead+Unpin> PeekRead<T> {
|
||||
async fn new(mut rdr: NsReader<T>) -> Result<Self, ParsingError> {
|
||||
let mut buf: Vec<u8> = vec![];
|
||||
let evt = rdr.read_event_into_async(&mut buf).await?.into_owned();
|
||||
buf.clear();
|
||||
Ok(Self { evt, rdr, buf })
|
||||
}
|
||||
}
|
||||
|
||||
/// Context must stay cheap to clone
|
||||
/// as we are cloning it from one fonction to another
|
||||
#[derive(Clone)]
|
||||
pub struct Context<'a, E: Extension + Clone> {
|
||||
pub aliases: HashMap<&'a [u8], XmlNamespace>,
|
||||
phantom: std::marker::PhantomData<E>,
|
||||
}
|
||||
impl<'a, E: Extension + Clone> Context<'a, E> {
|
||||
/// External buffer
|
||||
pub fn new() -> Self {
|
||||
Self {
|
||||
aliases: HashMap::new(),
|
||||
phantom: std::marker::PhantomData
|
||||
fn peek(&self) -> &Event<'static> {
|
||||
&self.evt
|
||||
}
|
||||
// skip tag, some tags can't be skipped like end, text, cdata
|
||||
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::Empty(_) | Event::Comment(_) | Event::PI(_) | Event::Decl(_) | Event::DocType(_) => self.next().await,
|
||||
_ => return Err(ParsingError::WrongToken),
|
||||
}
|
||||
}
|
||||
|
||||
pub fn ns_scan(&mut self, token: &'a BytesStart<'a>) -> Result<(XmlNamespace, &[u8]), ParsingError> {
|
||||
// Register namespace aliases from attributes (aka namespace bindings)
|
||||
for attr_res in token.attributes() {
|
||||
let attr = attr_res?;
|
||||
match attr.key.as_namespace_binding() {
|
||||
None => (),
|
||||
Some(PrefixDeclaration::Named(prefix)) => self.ns_alias(attr.value.as_ref(), prefix.as_ref())?,
|
||||
Some(PrefixDeclaration::Default) => self.ns_default(attr.value.as_ref())?,
|
||||
// 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();
|
||||
self.evt = evt;
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
|
||||
pub trait QReadable<T: AsyncBufRead+Unpin>: Sized {
|
||||
async fn read(xml: &mut PeekRead<T>) -> Result<Self, ParsingError>;
|
||||
}
|
||||
|
||||
impl<E: Extension, T: AsyncBufRead+Unpin> QReadable<T> for PropFind<E> {
|
||||
async fn read(xml: &mut PeekRead<T>) -> Result<PropFind<E>, ParsingError> {
|
||||
|
||||
// Find propfind
|
||||
loop {
|
||||
match xml.peek() {
|
||||
Event::Start(b) if b.local_name().into_inner() == &b"propfind"[..] => break,
|
||||
_ => xml.skip().await?,
|
||||
}
|
||||
}
|
||||
xml.next().await?;
|
||||
|
||||
// Find any tag
|
||||
let propfind = loop {
|
||||
match xml.peek() {
|
||||
Event::Start(b) | Event::Empty(b) if b.local_name().into_inner() == &b"allprop"[..] => {
|
||||
unimplemented!()
|
||||
},
|
||||
Event::Start(b) if b.local_name().into_inner() == &b"prop"[..] => {
|
||||
unimplemented!();
|
||||
},
|
||||
Event::Empty(b) if b.local_name().into_inner() == &b"propname"[..] => break PropFind::PropName,
|
||||
_ => xml.skip().await?,
|
||||
}
|
||||
};
|
||||
xml.next().await?;
|
||||
|
||||
// Close tag
|
||||
loop {
|
||||
match xml.peek() {
|
||||
Event::End(b) if b.local_name().into_inner() == &b"propfind"[..] => break,
|
||||
_ => xml.skip().await?,
|
||||
}
|
||||
}
|
||||
|
||||
// Decompose tag name
|
||||
let (key, maybe_prefix) = token.name().decompose();
|
||||
let ns = self.ns_resolve(maybe_prefix.map(|p| p.into_inner()).unwrap_or(&b""[..]));
|
||||
|
||||
Ok((ns, key.into_inner()))
|
||||
}
|
||||
|
||||
fn ns_default(&mut self, fqns: &[u8]) -> Result<(), ParsingError> {
|
||||
self.ns_alias(fqns, &b""[..])
|
||||
}
|
||||
|
||||
fn ns_alias(&mut self, fqns: &[u8], alias: &'a [u8]) -> Result<(), ParsingError> {
|
||||
let parsed_ns = XmlNamespace::from(fqns);
|
||||
if let Some(reg_fqns) = self.aliases.get(alias) {
|
||||
if *reg_fqns != parsed_ns {
|
||||
return Err(ParsingError::NamespacePrefixAlreadyUsed)
|
||||
}
|
||||
}
|
||||
self.aliases.insert(alias, parsed_ns);
|
||||
Ok(())
|
||||
}
|
||||
|
||||
// If the namespace is not found in the alias table (binding table)
|
||||
// we suppose it's a fully qualified namespace (fqns)
|
||||
fn ns_resolve(&self, prefix: &[u8]) -> XmlNamespace {
|
||||
match self.aliases.get(prefix) {
|
||||
Some(fqns) => fqns.clone(),
|
||||
None => XmlNamespace::from(prefix),
|
||||
}
|
||||
Ok(propfind)
|
||||
}
|
||||
}
|
||||
|
||||
trait DavReader<'a> {
|
||||
async fn doctype(&self) -> Result<(), ParsingError>;
|
||||
async fn tag(&self) -> Result<BytesStart<'a>, ParsingError>;
|
||||
async fn txt(&self) -> Result<Cow<'a, u8>, ParsingError>;
|
||||
}
|
||||
/*impl<'a, I: AsyncBufRead+Unpin> DavReader<'a> for Reader<I> {
|
||||
async fn doctype(&self) -> Result<(), ParsingError> {
|
||||
}
|
||||
async fn tag(&self) -> Result<BytesStart<'a>, ParsingError> {
|
||||
}
|
||||
async fn txt(&self) -> Result<Cow<'a, u8>, ParsingError> {
|
||||
}
|
||||
}*/
|
||||
|
||||
pub async fn propfind<E: Extension+Clone>(
|
||||
xml: &mut Reader<impl AsyncBufRead+Unpin>,
|
||||
ctx: Context<'_, E>,
|
||||
buf: &mut Vec<u8>,
|
||||
) -> Result<PropFind<E>, ParsingError> {
|
||||
let local = ctx.clone();
|
||||
|
||||
match xml.read_event_into_async(buf).await? {
|
||||
_ => unimplemented!(),
|
||||
}
|
||||
|
||||
unimplemented!();
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[tokio::test]
|
||||
async fn basic_propfind() {
|
||||
let src = r#"<?xml version="1.0" encoding="utf-8" ?><rando/><garbage><old/></garbage><D:propfind xmlns:D="DAV:"><D:propname/></D:propfind>"#;
|
||||
|
||||
let mut rdr = PeekRead::new(NsReader::from_reader(src.as_bytes())).await.unwrap();
|
||||
let got = PropFind::<NoExtension>::read(&mut rdr).await.unwrap();
|
||||
assert!(matches!(got, PropFind::PropName));
|
||||
}
|
||||
}
|
||||
|
|
|
@ -20,7 +20,8 @@ impl Extension for NoExtension {
|
|||
type Property = Disabled;
|
||||
type PropertyRequest = Disabled;
|
||||
type ResourceType = Disabled;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
/// 14.1. activelock XML Element
|
||||
///
|
||||
|
|
Loading…
Reference in a new issue