80 lines
2.4 KiB
Rust
80 lines
2.4 KiB
Rust
/*
|
|
* SPDX-FileCopyrightText: 2020 Stalwart Labs Ltd <hello@stalw.art>
|
|
*
|
|
* SPDX-License-Identifier: AGPL-3.0-only OR LicenseRef-SEL
|
|
*/
|
|
|
|
use std::time::Instant;
|
|
|
|
use common::listener::SessionStream;
|
|
use directory::Permission;
|
|
use jmap_proto::{
|
|
object::Object,
|
|
types::{collection::Collection, property::Property, value::Value},
|
|
};
|
|
use trc::AddContext;
|
|
|
|
use crate::core::{Session, StatusResponse};
|
|
|
|
impl<T: SessionStream> Session<T> {
|
|
pub async fn handle_listscripts(&mut self) -> trc::Result<Vec<u8>> {
|
|
// Validate access
|
|
self.assert_has_permission(Permission::SieveListScripts)?;
|
|
|
|
let op_start = Instant::now();
|
|
let account_id = self.state.access_token().primary_id();
|
|
let document_ids = self
|
|
.jmap
|
|
.get_document_ids(account_id, Collection::SieveScript)
|
|
.await
|
|
.caused_by(trc::location!())?
|
|
.unwrap_or_default();
|
|
|
|
if document_ids.is_empty() {
|
|
return Ok(StatusResponse::ok("").into_bytes());
|
|
}
|
|
|
|
let mut response = Vec::with_capacity(128);
|
|
let count = document_ids.len();
|
|
|
|
for document_id in document_ids {
|
|
if let Some(script) = self
|
|
.jmap
|
|
.get_property::<Object<Value>>(
|
|
account_id,
|
|
Collection::SieveScript,
|
|
document_id,
|
|
Property::Value,
|
|
)
|
|
.await
|
|
.caused_by(trc::location!())?
|
|
{
|
|
response.push(b'\"');
|
|
if let Some(name) = script.get(&Property::Name).as_string() {
|
|
for ch in name.as_bytes() {
|
|
if [b'\\', b'\"'].contains(ch) {
|
|
response.push(b'\\');
|
|
}
|
|
response.push(*ch);
|
|
}
|
|
}
|
|
|
|
if script.get(&Property::IsActive).as_bool() == Some(true) {
|
|
response.extend_from_slice(b"\" ACTIVE\r\n");
|
|
} else {
|
|
response.extend_from_slice(b"\"\r\n");
|
|
}
|
|
}
|
|
}
|
|
|
|
trc::event!(
|
|
ManageSieve(trc::ManageSieveEvent::ListScripts),
|
|
SpanId = self.session_id,
|
|
Total = count,
|
|
Elapsed = op_start.elapsed()
|
|
);
|
|
|
|
Ok(StatusResponse::ok("").serialize(response))
|
|
}
|
|
}
|