From 2b2e3c032cb67668395c7301962669bc927bd854 Mon Sep 17 00:00:00 2001 From: Quentin Dufour Date: Sat, 2 Mar 2024 18:19:03 +0100 Subject: [PATCH] Encode Calendar Properties --- src/dav/calencoder.rs | 169 +++++++++++++++++++++++++++++++++++++++--- src/dav/caltypes.rs | 22 +++--- 2 files changed, 171 insertions(+), 20 deletions(-) diff --git a/src/dav/calencoder.rs b/src/dav/calencoder.rs index a2d9a74..5cd40ef 100644 --- a/src/dav/calencoder.rs +++ b/src/dav/calencoder.rs @@ -8,6 +8,8 @@ use quick_xml::writer::{ElementWriter, Writer}; use quick_xml::name::PrefixDeclaration; use tokio::io::AsyncWrite; +const ICAL_DATETIME_FMT: &str = "%Y%m%dT%H%M%SZ"; + // =============== Calendar Trait =========================== pub trait CalContext: Context { fn create_cal_element(&self, name: &str) -> BytesStart; @@ -61,13 +63,25 @@ impl CalExtension { // -------------------- MKCALENDAR METHOD ------------------------------------ impl QuickWritable for MkCalendar { async fn write(&self, xml: &mut Writer, ctx: C) -> Result<(), QError> { - unimplemented!(); + let start = ctx.create_cal_element("mkcalendar"); + let end = start.to_end(); + + xml.write_event_async(Event::Start(start.clone())).await?; + self.0.write(xml, ctx.child()).await?; + xml.write_event_async(Event::End(end)).await } } impl QuickWritable for MkCalendarResponse { async fn write(&self, xml: &mut Writer, ctx: C) -> Result<(), QError> { - unimplemented!(); + let start = ctx.create_cal_element("mkcalendar-response"); + let end = start.to_end(); + + xml.write_event_async(Event::Start(start.clone())).await?; + for propstat in self.0.iter() { + propstat.write(xml, ctx.child()).await?; + } + xml.write_event_async(Event::End(end)).await } } @@ -75,31 +89,164 @@ impl QuickWritable for MkCalendarResponse { impl QuickWritable for CalendarQuery { async fn write(&self, xml: &mut Writer, ctx: C) -> Result<(), QError> { - unimplemented!(); + let start = ctx.create_cal_element("calendar-query"); + let end = start.to_end(); + + xml.write_event_async(Event::Start(start.clone())).await?; + if let Some(selector) = &self.selector { + selector.write(xml, ctx.child()).await?; + } + self.filter.write(xml, ctx.child()).await?; + if let Some(tz) = &self.timezone { + tz.write(xml, ctx.child()).await?; + } + xml.write_event_async(Event::End(end)).await } } impl QuickWritable for CalendarMultiget { async fn write(&self, xml: &mut Writer, ctx: C) -> Result<(), QError> { - unimplemented!(); + let start = ctx.create_cal_element("calendar-multiget"); + let end = start.to_end(); + + xml.write_event_async(Event::Start(start.clone())).await?; + if let Some(selector) = &self.selector { + selector.write(xml, ctx.child()).await?; + } + for href in self.href.iter() { + href.write(xml, ctx.child()).await?; + } + xml.write_event_async(Event::End(end)).await } } impl QuickWritable for FreeBusyQuery { async fn write(&self, xml: &mut Writer, ctx: C) -> Result<(), QError> { - unimplemented!(); + let start = ctx.create_cal_element("free-busy-query"); + let end = start.to_end(); + + xml.write_event_async(Event::Start(start.clone())).await?; + self.0.write(xml, ctx.child()).await?; + xml.write_event_async(Event::End(end)).await } } // -------------------------- DAV::prop -------------------------------------- -impl QuickWritable for Property { - async fn write(&self, xml: &mut Writer, ctx: C) -> Result<(), QError> { - unimplemented!(); - } -} impl QuickWritable for PropertyRequest { async fn write(&self, xml: &mut Writer, ctx: C) -> Result<(), QError> { - unimplemented!(); + let mut atom = async |c| xml.write_event_async(Event::Empty(ctx.create_cal_element(c))).await; + + match self { + Self::CalendarDescription => atom("calendar-description").await, + Self::CalendarTimezone => atom("calendar-timezone").await, + Self::SupportedCalendarComponentSet => atom("supported-calendar-component-set").await, + Self::SupportedCalendarData => atom("supported-calendar-data").await, + Self::MaxResourceSize => atom("max-resource-size").await, + Self::MinDateTime => atom("min-date-time").await, + Self::MaxDateTime => atom("max-date-time").await, + Self::MaxInstances => atom("max-instances").await, + Self::MaxAttendeesPerInstance => atom("max-attendees-per-instance").await, + Self::SupportedCollationSet => atom("supported-collation-set").await, + Self::CalendarData(req) => req.write(xml, ctx).await, + } + } +} +impl QuickWritable for Property { + async fn write(&self, xml: &mut Writer, ctx: C) -> Result<(), QError> { + match self { + Self::CalendarDescription { lang, text } => { + let mut start = ctx.create_cal_element("calendar-description"); + if let Some(the_lang) = lang { + start.push_attribute(("xml:lang", the_lang.as_str())); + } + let end = start.to_end(); + + xml.write_event_async(Event::Start(start.clone())).await?; + xml.write_event_async(Event::Text(BytesText::new(text))).await?; + xml.write_event_async(Event::End(end)).await + }, + Self::CalendarTimezone(payload) => { + let start = ctx.create_cal_element("calendar-timezone"); + let end = start.to_end(); + + xml.write_event_async(Event::Start(start.clone())).await?; + xml.write_event_async(Event::Text(BytesText::new(payload))).await?; + xml.write_event_async(Event::End(end)).await + }, + Self::SupportedCalendarComponentSet(many_comp) => { + let start = ctx.create_cal_element("supported-calendar-component-set"); + let end = start.to_end(); + + xml.write_event_async(Event::Start(start.clone())).await?; + for comp in many_comp.iter() { + comp.write(xml, ctx.child()).await?; + } + xml.write_event_async(Event::End(end)).await + }, + Self::SupportedCalendarData(many_mime) => { + let start = ctx.create_cal_element("supported-calendar-data"); + let end = start.to_end(); + + xml.write_event_async(Event::Start(start.clone())).await?; + for mime in many_mime.iter() { + mime.write(xml, ctx.child()).await?; + } + xml.write_event_async(Event::End(end)).await + }, + Self::MaxResourceSize(bytes) => { + let start = ctx.create_cal_element("max-resource-size"); + let end = start.to_end(); + + xml.write_event_async(Event::Start(start.clone())).await?; + xml.write_event_async(Event::Text(BytesText::new(bytes.to_string().as_str()))).await?; + xml.write_event_async(Event::End(end)).await + }, + Self::MinDateTime(dt) => { + let start = ctx.create_cal_element("min-date-time"); + let end = start.to_end(); + + let dtstr = format!("{}", dt.format(ICAL_DATETIME_FMT)); + xml.write_event_async(Event::Start(start.clone())).await?; + xml.write_event_async(Event::Text(BytesText::new(dtstr.as_str()))).await?; + xml.write_event_async(Event::End(end)).await + }, + Self::MaxDateTime(dt) => { + let start = ctx.create_cal_element("max-date-time"); + let end = start.to_end(); + + let dtstr = format!("{}", dt.format(ICAL_DATETIME_FMT)); + xml.write_event_async(Event::Start(start.clone())).await?; + xml.write_event_async(Event::Text(BytesText::new(dtstr.as_str()))).await?; + xml.write_event_async(Event::End(end)).await + }, + Self::MaxInstances(count) => { + let start = ctx.create_cal_element("max-instances"); + let end = start.to_end(); + + xml.write_event_async(Event::Start(start.clone())).await?; + xml.write_event_async(Event::Text(BytesText::new(count.to_string().as_str()))).await?; + xml.write_event_async(Event::End(end)).await + }, + Self::MaxAttendeesPerInstance(count) => { + let start = ctx.create_cal_element("max-attendees-per-instance"); + let end = start.to_end(); + + xml.write_event_async(Event::Start(start.clone())).await?; + xml.write_event_async(Event::Text(BytesText::new(count.to_string().as_str()))).await?; + xml.write_event_async(Event::End(end)).await + }, + Self::SupportedCollationSet(many_collations) => { + let start = ctx.create_cal_element("supported-collation-set"); + let end = start.to_end(); + + xml.write_event_async(Event::Start(start.clone())).await?; + for collation in many_collations.iter() { + collation.write(xml, ctx.child()).await?; + } + xml.write_event_async(Event::End(end)).await + }, + Self::CalendarData(inner) => inner.write(xml, ctx).await, + } } } diff --git a/src/dav/caltypes.rs b/src/dav/caltypes.rs index 3b65fe2..c803c0c 100644 --- a/src/dav/caltypes.rs +++ b/src/dav/caltypes.rs @@ -35,7 +35,7 @@ impl Dav::Extension for CalExtension { /// instruction in Section 12.13.2 of [RFC2518]. /// /// -pub struct MkCalendar(Dav::Set); +pub struct MkCalendar(pub Dav::Set); /// If a response body for a successful request is included, it MUST @@ -51,7 +51,7 @@ pub struct MkCalendar(Dav::Set); /// Definition: /// /// -pub struct MkCalendarResponse(Vec>); +pub struct MkCalendarResponse(pub Vec>); // --- (REPORT PART) --- @@ -69,9 +69,9 @@ pub struct MkCalendarResponse(Vec>); /// DAV:propname | /// DAV:prop)?, filter, timezone?)> pub struct CalendarQuery { - selector: Option>, - filter: Filter, - timezone: Option, + pub selector: Option>, + pub filter: Filter, + pub timezone: Option, } /// Name: calendar-multiget @@ -89,8 +89,8 @@ pub struct CalendarQuery { /// DAV:propname | /// DAV:prop)?, DAV:href+)> pub struct CalendarMultiget { - selector: Option>, - href: Vec, + pub selector: Option>, + pub href: Vec, } /// Name: free-busy-query @@ -104,7 +104,7 @@ pub struct CalendarMultiget { /// /// Definition: /// -pub struct FreeBusyQuery(TimeRange); +pub struct FreeBusyQuery(pub TimeRange); // ----- Hooks ----- pub enum ResourceType { @@ -784,7 +784,10 @@ pub enum Collation { /// when nested in the DAV:prop XML element in a calendaring /// REPORT response to specify the content of a returned /// calendar object resource. -pub struct CalendarDataPayload(String); +pub struct CalendarDataPayload { + pub mime: Option, + pub payload: String, +} /// , comp: Option, reccurence: Option, limit_freebusy_set: Option,