jcoffey-dev
7dae9b29fd
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.
101 lines
3.5 KiB
Rust
101 lines
3.5 KiB
Rust
/*
|
|
* 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()))
|
|
}
|
|
}
|