Add initial carddav support

crates/carddav/src/lib.rs

@@ -1,17 +1,81 @@
-use actix_web::{web, HttpResponse, Responder};
+use actix_web::{
+    dev::Service,
+    http::{
+        header::{HeaderName, HeaderValue},
+        Method, StatusCode,
+    },
+    web::{self, Data},
+};
+use address_object::resource::AddressObjectResourceService;
+use addressbook::resource::AddressbookResourceService;
+pub use error::Error;
+use futures_util::FutureExt;
+use principal::PrincipalResourceService;
+use root::RootResourceService;
+use rustical_dav::resource::ResourceService;
+use rustical_store::{
+    auth::{AuthenticationMiddleware, AuthenticationProvider},
+    AddressbookStore,
+};
+use std::sync::Arc;
+use tokio::sync::RwLock;
+
+pub mod address_object;
+pub mod addressbook;
+pub mod error;
+pub mod principal;
+pub mod root;
 
 pub fn configure_well_known(cfg: &mut web::ServiceConfig, carddav_root: String) {
     cfg.service(web::redirect("/carddav", carddav_root).permanent());
 }
 
-pub fn configure_dav(_cfg: &mut web::ServiceConfig, _prefix: String) {}
-
-pub async fn options_handler() -> impl Responder {
-    HttpResponse::Ok()
-        .insert_header((
-            "Allow",
-            "OPTIONS, GET, HEAD, POST, PUT, REPORT, PROPFIND, PROPPATCH, MKCOL",
-        ))
-        .insert_header(("DAV", "1, 2, 3, addressbook, extended-mkcol"))
-        .body("options")
+pub fn configure_dav<AP: AuthenticationProvider, A: AddressbookStore + ?Sized>(
+    cfg: &mut web::ServiceConfig,
+    auth_provider: Arc<AP>,
+    store: Arc<RwLock<A>>,
+) {
+    cfg.service(
+        web::scope("")
+            .wrap(AuthenticationMiddleware::new(auth_provider))
+            .wrap_fn(|req, srv| {
+                // Middleware to set the DAV header
+                // Could be more elegant if actix_web::guard::RegisteredMethods was public :(
+                let method = req.method().clone();
+                srv.call(req).map(move |res| {
+                    if method == Method::OPTIONS {
+                        return res.map(|mut response| {
+                            if response.status() == StatusCode::METHOD_NOT_ALLOWED {
+                                response.headers_mut().insert(
+                                    HeaderName::from_static("dav"),
+                                    HeaderValue::from_static(
+                                        "1, 2, 3, access-control, addressbook, extended-mkcol",
+                                    ),
+                                );
+                                *response.response_mut().status_mut() = StatusCode::OK;
+                            }
+                            response
+                        });
+                    }
+                    res
+                })
+            })
+            .app_data(Data::from(store.clone()))
+            .service(RootResourceService::actix_resource())
+            .service(
+                web::scope("/user").service(
+                    web::scope("/{principal}")
+                        .service(PrincipalResourceService::<A>::actix_resource())
+                        .service(
+                            web::scope("/{addressbook}")
+                                .service(AddressbookResourceService::<A>::actix_resource())
+                                .service(
+                                    web::scope("/{object}").service(
+                                        AddressObjectResourceService::<A>::actix_resource(),
+                                    ),
+                                ),
+                        ),
+                ),
+            ),
+    );
 }