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.
147 lines
4.6 KiB
Rust
147 lines
4.6 KiB
Rust
/*
|
|
* SPDX-FileCopyrightText: 2020 Stalwart Labs LLC <[email protected]>
|
|
*
|
|
* SPDX-License-Identifier: AGPL-3.0-only OR LicenseRef-SEL
|
|
*/
|
|
|
|
use super::proppatch::FilePropPatchRequestHandler;
|
|
use crate::{
|
|
DavMethod, PropStatBuilder,
|
|
common::{
|
|
ExtractETag,
|
|
acl::ResourceAcl,
|
|
lock::{LockRequestHandler, ResourceState},
|
|
uri::DavUriResource,
|
|
},
|
|
file::DavFileResource,
|
|
};
|
|
use common::{Server, auth::AccessToken, storage::index::ObjectIndexBuilder};
|
|
use dav_proto::{
|
|
RequestHeaders, Return,
|
|
schema::{Namespace, request::MkCol, response::MkColResponse},
|
|
};
|
|
use groupware::{cache::GroupwareCache, file::FileNode};
|
|
use http_proto::HttpResponse;
|
|
use hyper::StatusCode;
|
|
use store::write::{BatchBuilder, now};
|
|
use trc::AddContext;
|
|
use types::{
|
|
acl::Acl,
|
|
collection::{Collection, SyncCollection},
|
|
};
|
|
|
|
pub(crate) trait FileMkColRequestHandler: Sync + Send {
|
|
fn handle_file_mkcol_request(
|
|
&self,
|
|
access_token: &AccessToken,
|
|
headers: &RequestHeaders<'_>,
|
|
request: Option<MkCol>,
|
|
) -> impl Future<Output = crate::Result<HttpResponse>> + Send;
|
|
}
|
|
|
|
impl FileMkColRequestHandler for Server {
|
|
async fn handle_file_mkcol_request(
|
|
&self,
|
|
access_token: &AccessToken,
|
|
headers: &RequestHeaders<'_>,
|
|
request: Option<MkCol>,
|
|
) -> crate::Result<HttpResponse> {
|
|
// Validate URI
|
|
let resource_ = self
|
|
.validate_uri(access_token, headers.uri)
|
|
.await?
|
|
.into_owned_uri()?;
|
|
let account_id = resource_.account_id;
|
|
let resources = self
|
|
.fetch_dav_resources(
|
|
access_token.account_id(),
|
|
account_id,
|
|
SyncCollection::FileNode,
|
|
)
|
|
.await
|
|
.caused_by(trc::location!())?;
|
|
let resource = resources.map_parent_resource(&resource_)?;
|
|
|
|
// Validate and map parent ACL
|
|
let parent_id = resources.validate_and_map_parent_acl(
|
|
access_token,
|
|
access_token.is_member(account_id),
|
|
resource.resource.0,
|
|
Acl::AddItems,
|
|
)?;
|
|
|
|
// Validate headers
|
|
self.validate_headers(
|
|
access_token,
|
|
headers,
|
|
vec![ResourceState {
|
|
account_id,
|
|
collection: resource.collection,
|
|
document_id: Some(u32::MAX),
|
|
path: resource_.resource.unwrap(),
|
|
..Default::default()
|
|
}],
|
|
Default::default(),
|
|
DavMethod::MKCOL,
|
|
)
|
|
.await?;
|
|
|
|
// Build file container
|
|
let now = now();
|
|
let mut node = FileNode {
|
|
parent_id,
|
|
name: resource.resource.1.to_string(),
|
|
display_name: None,
|
|
file: None,
|
|
created: now as i64,
|
|
modified: now as i64,
|
|
dead_properties: Default::default(),
|
|
acls: Default::default(),
|
|
};
|
|
|
|
// Apply MKCOL properties
|
|
let mut return_prop_stat = None;
|
|
if let Some(mkcol) = request {
|
|
let mut prop_stat = PropStatBuilder::default();
|
|
if !self.apply_file_properties(&mut node, false, mkcol.props, &mut prop_stat) {
|
|
return Ok(HttpResponse::new(StatusCode::FORBIDDEN).with_xml_body(
|
|
MkColResponse::new(prop_stat.build())
|
|
.with_namespace(Namespace::Dav)
|
|
.to_string(),
|
|
));
|
|
}
|
|
if headers.ret != Return::Minimal {
|
|
return_prop_stat = Some(prop_stat);
|
|
}
|
|
}
|
|
|
|
// Prepare write batch
|
|
let document_id = self
|
|
.store()
|
|
.assign_document_ids(account_id, Collection::FileNode, 1)
|
|
.await
|
|
.caused_by(trc::location!())?;
|
|
let mut batch = BatchBuilder::new();
|
|
batch
|
|
.with_account_id(account_id)
|
|
.with_collection(Collection::FileNode)
|
|
.with_document(document_id)
|
|
.custom(ObjectIndexBuilder::<(), _>::new().with_changes(node))
|
|
.caused_by(trc::location!())?;
|
|
let etag = batch.etag();
|
|
self.commit_batch(batch).await.caused_by(trc::location!())?;
|
|
|
|
if let Some(prop_stat) = return_prop_stat {
|
|
Ok(HttpResponse::new(StatusCode::CREATED)
|
|
.with_xml_body(
|
|
MkColResponse::new(prop_stat.build())
|
|
.with_namespace(Namespace::Dav)
|
|
.to_string(),
|
|
)
|
|
.with_etag_opt(etag))
|
|
} else {
|
|
Ok(HttpResponse::new(StatusCode::CREATED).with_etag_opt(etag))
|
|
}
|
|
}
|
|
}
|