1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
|
/*
* SPDX-FileCopyrightText: 2020 Stalwart Labs Ltd <hello@stalw.art>
*
* SPDX-License-Identifier: AGPL-3.0-only OR LicenseRef-SEL
*/
use std::time::Instant;
use imap_proto::receiver::Request;
use jmap::sieve::set::ObjectBlobId;
use jmap_proto::{
object::Object,
types::{collection::Collection, property::Property, value::Value},
};
use tokio::io::{AsyncRead, AsyncWrite};
use trc::AddContext;
use crate::core::{Command, ResponseCode, Session, StatusResponse};
impl<T: AsyncRead + AsyncWrite> Session<T> {
pub async fn handle_getscript(&mut self, request: Request<Command>) -> trc::Result<Vec<u8>> {
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().primary_id();
let document_id = self.get_script_id(account_id, &name).await?;
let (blob_section, blob_hash) = self
.jmap
.get_property::<Object<Value>>(
account_id,
Collection::SieveScript,
document_id,
Property::Value,
)
.await
.caused_by(trc::location!())?
.ok_or_else(|| {
trc::ManageSieveEvent::Error
.into_err()
.details("Script not found")
.code(ResponseCode::NonExistent)
})?
.blob_id()
.and_then(|id| (id.section.as_ref()?.clone(), id.hash.clone()).into())
.ok_or_else(|| {
trc::ManageSieveEvent::Error
.into_err()
.details("Failed to retrieve blobId")
.code(ResponseCode::TryLater)
})?;
let script = self
.jmap
.get_blob_section(&blob_hash, &blob_section)
.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_section.size);
let mut response = Vec::with_capacity(script.len() + 32);
response.push(b'{');
response.extend_from_slice(blob_section.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))
}
}
|
