Initial commit

crates/dav/src/propfind.rs

@@ -0,0 +1,122 @@
+use actix_web::http::StatusCode;
+use anyhow::{anyhow, Result};
+use quick_xml::{
+    events::{attributes::Attribute, BytesText},
+    Writer,
+};
+
+pub fn parse_propfind(body: &str) -> Result<Vec<&str>> {
+    if body.is_empty() {
+        // if body is empty, allprops must be returned (RFC 4918)
+        return Ok(vec!["allprops"]);
+    }
+    let doc = roxmltree::Document::parse(body)?;
+
+    let propfind_node = doc.root_element();
+    if propfind_node.tag_name().name() != "propfind" {
+        return Err(anyhow!("invalid tag"));
+    }
+
+    let prop_node = if let Some(el) = propfind_node.first_element_child() {
+        el
+    } else {
+        return Ok(Vec::new());
+    };
+
+    match prop_node.tag_name().name() {
+        "prop" => Ok(prop_node
+            .children()
+            .map(|node| node.tag_name().name())
+            .collect()),
+        _ => Err(anyhow!("invalid prop tag")),
+    }
+}
+
+pub fn write_resourcetype(
+    writer: &mut Writer<&mut Vec<u8>>,
+    types: Vec<&str>,
+) -> Result<(), quick_xml::Error> {
+    writer
+        .create_element("resourcetype")
+        .write_inner_content(|writer| {
+            for resourcetype in types {
+                writer.create_element(resourcetype).write_empty()?;
+            }
+            Ok(())
+        })?;
+    Ok(())
+}
+
+pub fn write_invalid_props_response(
+    writer: &mut Writer<&mut Vec<u8>>,
+    href: &str,
+    invalid_props: Vec<&str>,
+) -> Result<(), quick_xml::Error> {
+    if invalid_props.is_empty() {
+        return Ok(());
+    };
+
+    write_propstat_response(writer, href, StatusCode::NOT_FOUND, |writer| {
+        for prop in invalid_props {
+            writer.create_element(prop).write_empty()?;
+        }
+        Ok(())
+    })?;
+
+    Ok(())
+}
+
+// Writes a propstat response into a multistatus
+// closure hooks into the <prop> element
+pub fn write_propstat_response<F>(
+    writer: &mut Writer<&mut Vec<u8>>,
+    href: &str,
+    status: StatusCode,
+    prop_closure: F,
+) -> Result<(), quick_xml::Error>
+where
+    F: FnOnce(&mut Writer<&mut Vec<u8>>) -> Result<(), quick_xml::Error>,
+{
+    writer
+        .create_element("response")
+        .write_inner_content(|writer| {
+            writer
+                .create_element("href")
+                .write_text_content(BytesText::new(href))?;
+
+            writer
+                .create_element("propstat")
+                .write_inner_content(|writer| {
+                    writer
+                        .create_element("prop")
+                        .write_inner_content(prop_closure)?;
+
+                    writer
+                        .create_element("status")
+                        .write_text_content(BytesText::new(&format!("HTTP/1.1 {}", status)))?;
+                    Ok(())
+                })?;
+
+            Ok(())
+        })?;
+    Ok(())
+}
+
+pub fn generate_multistatus<'a, F, A>(namespaces: A, closure: F) -> Result<String>
+where
+    F: FnOnce(&mut Writer<&mut Vec<u8>>) -> Result<(), quick_xml::Error>,
+    A: IntoIterator,
+    A::Item: Into<Attribute<'a>>,
+{
+    let mut output_buffer = Vec::new();
+    let mut writer = Writer::new_with_indent(&mut output_buffer, b' ', 2);
+    writer
+        .create_element("multistatus")
+        .with_attributes(namespaces)
+        .write_inner_content(closure)?;
+
+    Ok(format!(
+        "<?xml version=\"1.0\" encoding=\"utf-8\"?>\n{}",
+        std::str::from_utf8(&output_buffer)?
+    ))
+}