Import upstream v0.16.22, stripped

Upstream commit: 474dd0229cb20cf513036619781ed97bd8073c3f Enterprise-only files removed or emptied: 63 Enterprise-only snippets removed: 117 in 50 files Dangling module declarations removed: 5 Cargo edits turning enterprise off: 14 Verification: clean Enterprise feature gates left for rebuilt features: 19 in 18 files Produced by tools/fork/strip.py. The full report is in docs/fork/strip-reports/ on main.
2026-09-18 10:21:56 -07:00
commit 7dae9b29fd
1650 changed files with 485521 additions and 0 deletions
@@ -0,0 +1,100 @@
+/*
+ * SPDX-FileCopyrightText: 2020 Stalwart Labs LLC <[email protected]>
+ *
+ * SPDX-License-Identifier: AGPL-3.0-only OR LicenseRef-SEL
+ */
+
+use super::propfind::PrincipalPropFind;
+use crate::{
+    DavError,
+    common::{
+        DavQuery, DavQueryResource,
+        propfind::PropFindRequestHandler,
+        uri::{DavUriResource, UriResource},
+    },
+};
+use common::{Server, auth::AccessToken};
+use dav_proto::{
+    RequestHeaders,
+    schema::{
+        property::{DavProperty, WebDavProperty},
+        request::{PrincipalMatch, PropFind},
+        response::MultiStatus,
+    },
+};
+use http_proto::HttpResponse;
+use hyper::StatusCode;
+use store::roaring::RoaringBitmap;
+use types::collection::Collection;
+
+pub(crate) trait PrincipalMatching: Sync + Send {
+    fn handle_principal_match(
+        &self,
+        access_token: &AccessToken,
+        headers: &RequestHeaders<'_>,
+        request: PrincipalMatch,
+    ) -> impl Future<Output = crate::Result<HttpResponse>> + Send;
+}
+
+impl PrincipalMatching for Server {
+    async fn handle_principal_match(
+        &self,
+        access_token: &AccessToken,
+        headers: &RequestHeaders<'_>,
+        mut request: PrincipalMatch,
+    ) -> crate::Result<HttpResponse> {
+        let resource = self.validate_uri(access_token, headers.uri).await?;
+
+        match resource.collection {
+            Collection::AddressBook | Collection::Calendar | Collection::FileNode => {
+                if request.properties.is_empty() {
+                    request
+                        .properties
+                        .push(DavProperty::WebDav(WebDavProperty::Owner));
+                }
+                if let Some(account_id) = resource.account_id {
+                    return self
+                        .handle_dav_query(
+                            access_token,
+                            DavQuery {
+                                resource: DavQueryResource::Uri(UriResource {
+                                    collection: resource.collection,
+                                    account_id,
+                                    resource: resource.resource,
+                                }),
+                                propfind: PropFind::Prop(request.properties),
+                                depth: usize::MAX,
+                                ret: headers.ret,
+                                depth_no_root: headers.depth_no_root,
+                                uri: headers.uri,
+                                sync_type: Default::default(),
+                                limit: Default::default(),
+                                vcard_version: Default::default(),
+                                expand: Default::default(),
+                            },
+                        )
+                        .await;
+                }
+            }
+            Collection::Principal => {}
+            _ => return Err(DavError::Code(StatusCode::METHOD_NOT_ALLOWED)),
+        }
+
+        let mut response = MultiStatus::new(Vec::with_capacity(16));
+        if request.properties.is_empty() {
+            request
+                .properties
+                .push(DavProperty::WebDav(WebDavProperty::DisplayName));
+        }
+        let request = PropFind::Prop(request.properties);
+        self.prepare_principal_propfind_response(
+            access_token,
+            resource.collection,
+            RoaringBitmap::from_iter(access_token.all_ids()).into_iter(),
+            &request,
+            &mut response,
+        )
+        .await?;
+        Ok(HttpResponse::new(StatusCode::MULTI_STATUS).with_xml_body(response.to_string()))
+    }
+}