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.
95 lines
3.1 KiB
Rust
95 lines
3.1 KiB
Rust
/*
|
|
* SPDX-FileCopyrightText: 2020 Stalwart Labs LLC <[email protected]>
|
|
*
|
|
* SPDX-License-Identifier: AGPL-3.0-only OR LicenseRef-SEL
|
|
*/
|
|
|
|
use crate::core::{Command, ResponseCode, Session, StatusResponse};
|
|
use common::network::SessionStream;
|
|
use email::sieve::SieveScript;
|
|
use imap_proto::receiver::Request;
|
|
use registry::schema::enums::Permission;
|
|
use std::time::Instant;
|
|
use store::{
|
|
ValueKey,
|
|
write::{AlignedBytes, Archive},
|
|
};
|
|
use trc::AddContext;
|
|
use types::{blob::BlobSection, blob_hash::BlobHash, collection::Collection};
|
|
|
|
impl<T: SessionStream> Session<T> {
|
|
pub async fn handle_getscript(&mut self, request: Request<Command>) -> trc::Result<Vec<u8>> {
|
|
// Validate access
|
|
self.assert_has_permission(Permission::SieveGetScript)?;
|
|
|
|
let op_start = Instant::now();
|
|
let name = request
|
|
.tokens
|
|
.into_iter()
|
|
.next()
|
|
.and_then(|s| s.unwrap_string().ok())
|
|
.ok_or_else(|| {
|
|
trc::ManageSieveEvent::Error
|
|
.into_err()
|
|
.details("Expected script name as a parameter.")
|
|
})?;
|
|
let account_id = self.state.access_token().account_id();
|
|
let document_id = self.get_script_id(account_id, &name).await?;
|
|
let sieve_ = self
|
|
.server
|
|
.store()
|
|
.get_value::<Archive<AlignedBytes>>(ValueKey::archive(
|
|
account_id,
|
|
Collection::SieveScript,
|
|
document_id,
|
|
))
|
|
.await
|
|
.caused_by(trc::location!())?
|
|
.ok_or_else(|| {
|
|
trc::ManageSieveEvent::Error
|
|
.into_err()
|
|
.details("Script not found")
|
|
.code(ResponseCode::NonExistent)
|
|
})?;
|
|
let sieve = sieve_
|
|
.unarchive::<SieveScript>()
|
|
.caused_by(trc::location!())?;
|
|
let blob_size = u32::from(sieve.size) as usize;
|
|
let script = self
|
|
.server
|
|
.get_blob_section(
|
|
&BlobHash::from(&sieve.blob_hash),
|
|
&BlobSection {
|
|
size: blob_size,
|
|
..Default::default()
|
|
},
|
|
)
|
|
.await
|
|
.caused_by(trc::location!())?
|
|
.ok_or_else(|| {
|
|
trc::ManageSieveEvent::Error
|
|
.into_err()
|
|
.details("Script blob not found")
|
|
.code(ResponseCode::NonExistent)
|
|
})?;
|
|
debug_assert_eq!(script.len(), blob_size);
|
|
|
|
let mut response = Vec::with_capacity(script.len() + 32);
|
|
response.push(b'{');
|
|
response.extend_from_slice(blob_size.to_string().as_bytes());
|
|
response.extend_from_slice(b"}\r\n");
|
|
response.extend(script);
|
|
response.extend_from_slice(b"\r\n");
|
|
|
|
trc::event!(
|
|
ManageSieve(trc::ManageSieveEvent::GetScript),
|
|
SpanId = self.session_id,
|
|
Id = name,
|
|
DocumentId = document_id,
|
|
Elapsed = op_start.elapsed()
|
|
);
|
|
|
|
Ok(StatusResponse::ok("").serialize(response))
|
|
}
|
|
}
|