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.
171 lines
4.8 KiB
Rust
171 lines
4.8 KiB
Rust
/*
|
|
* SPDX-FileCopyrightText: 2020 Stalwart Labs LLC <[email protected]>
|
|
*
|
|
* SPDX-License-Identifier: AGPL-3.0-only OR LicenseRef-SEL
|
|
*/
|
|
|
|
use super::{ArchivedFileNode, FileNode};
|
|
use crate::DestroyArchive;
|
|
use common::{Server, auth::AccountTenantIds, storage::index::ObjectIndexBuilder};
|
|
use store::{
|
|
ValueKey,
|
|
write::{AlignedBytes, Archive, BatchBuilder, now},
|
|
};
|
|
use trc::AddContext;
|
|
use types::collection::{Collection, VanishedCollection};
|
|
|
|
impl FileNode {
|
|
pub fn insert(
|
|
self,
|
|
changed_by: AccountTenantIds,
|
|
account_id: u32,
|
|
document_id: u32,
|
|
set_created: bool,
|
|
set_modified: bool,
|
|
batch: &mut BatchBuilder,
|
|
) -> trc::Result<&mut BatchBuilder> {
|
|
let mut node = self;
|
|
let now = now() as i64;
|
|
if set_created {
|
|
node.created = now;
|
|
}
|
|
if set_modified {
|
|
node.modified = now;
|
|
}
|
|
|
|
// Prepare write batch
|
|
batch
|
|
.with_account_id(account_id)
|
|
.with_collection(Collection::FileNode)
|
|
.with_document(document_id)
|
|
.custom(
|
|
ObjectIndexBuilder::<(), _>::new()
|
|
.with_changes(node)
|
|
.with_changed_by(changed_by),
|
|
)
|
|
.map(|b| b.commit_point())
|
|
}
|
|
|
|
pub fn update<'x>(
|
|
self,
|
|
changed_by: AccountTenantIds,
|
|
node: Archive<&ArchivedFileNode>,
|
|
account_id: u32,
|
|
document_id: u32,
|
|
set_modified: bool,
|
|
batch: &'x mut BatchBuilder,
|
|
) -> trc::Result<&'x mut BatchBuilder> {
|
|
let mut new_node = self;
|
|
if set_modified {
|
|
new_node.modified = now() as i64;
|
|
}
|
|
batch
|
|
.with_account_id(account_id)
|
|
.with_collection(Collection::FileNode)
|
|
.with_document(document_id)
|
|
.custom(
|
|
ObjectIndexBuilder::new()
|
|
.with_current(node)
|
|
.with_changes(new_node)
|
|
.with_changed_by(changed_by),
|
|
)
|
|
.map(|b| b.commit_point())
|
|
}
|
|
}
|
|
|
|
impl DestroyArchive<Archive<&ArchivedFileNode>> {
|
|
pub fn delete(
|
|
self,
|
|
changed_by: AccountTenantIds,
|
|
account_id: u32,
|
|
document_id: u32,
|
|
batch: &mut BatchBuilder,
|
|
path: String,
|
|
) -> trc::Result<()> {
|
|
// Prepare write batch
|
|
batch
|
|
.with_account_id(account_id)
|
|
.with_collection(Collection::FileNode)
|
|
.with_document(document_id)
|
|
.custom(
|
|
ObjectIndexBuilder::<_, ()>::new()
|
|
.with_current(self.0)
|
|
.with_changed_by(changed_by),
|
|
)?
|
|
.log_vanished_item(VanishedCollection::FileNode, path)
|
|
.commit_point();
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
impl DestroyArchive<Vec<u32>> {
|
|
pub async fn delete(
|
|
self,
|
|
server: &Server,
|
|
changed_by: AccountTenantIds,
|
|
account_id: u32,
|
|
delete_path: Option<String>,
|
|
) -> trc::Result<()> {
|
|
// Process deletions
|
|
let mut batch = BatchBuilder::new();
|
|
self.delete_batch(server, changed_by, account_id, delete_path, &mut batch)
|
|
.await?;
|
|
// Write changes
|
|
if !batch.is_empty() {
|
|
server
|
|
.commit_batch(batch)
|
|
.await
|
|
.caused_by(trc::location!())?;
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
|
|
pub async fn delete_batch(
|
|
self,
|
|
server: &Server,
|
|
changed_by: AccountTenantIds,
|
|
account_id: u32,
|
|
delete_path: Option<String>,
|
|
batch: &mut BatchBuilder,
|
|
) -> trc::Result<()> {
|
|
// Process deletions
|
|
batch
|
|
.with_account_id(account_id)
|
|
.with_collection(Collection::FileNode);
|
|
for document_id in self.0 {
|
|
if let Some(node) = server
|
|
.store()
|
|
.get_value::<Archive<AlignedBytes>>(ValueKey::archive(
|
|
account_id,
|
|
Collection::FileNode,
|
|
document_id,
|
|
))
|
|
.await?
|
|
{
|
|
// Delete record
|
|
batch
|
|
.with_document(document_id)
|
|
.custom(
|
|
ObjectIndexBuilder::<_, ()>::new()
|
|
.with_changed_by(changed_by)
|
|
.with_current(
|
|
node.to_unarchived::<FileNode>()
|
|
.caused_by(trc::location!())?,
|
|
),
|
|
)
|
|
.caused_by(trc::location!())?
|
|
.commit_point();
|
|
}
|
|
}
|
|
|
|
if !batch.is_empty()
|
|
&& let Some(delete_path) = delete_path
|
|
{
|
|
batch.log_vanished_item(VanishedCollection::FileNode, delete_path);
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
}
|