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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
/*
* Copyright (c) 2020-2022, Stalwart Labs Ltd.
*
* This file is part of the Stalwart IMAP Server.
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as
* published by the Free Software Foundation, either version 3 of
* the License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
* in the LICENSE file at the top-level directory of this distribution.
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*
* You can be released from the requirements of the AGPLv3 license by
* purchasing a commercial license. Please contact licensing@stalw.art
* for more details.
*/
use crate::utf7::utf7_encode;
use super::quoted_string;
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct Arguments {
pub tag: String,
pub mailbox_name: String,
pub items: Vec<Status>,
}
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum Status {
Messages,
UidNext,
UidValidity,
Unseen,
Deleted,
Size,
Recent,
HighestModSeq,
MailboxId,
}
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct StatusItem {
pub mailbox_name: String,
pub items: Vec<(Status, StatusItemType)>,
}
#[derive(Debug, Clone, PartialEq, Eq)]
pub enum StatusItemType {
Number(u32),
String(String),
}
impl StatusItem {
pub fn serialize(&self, buf: &mut Vec<u8>, is_rev2: bool) {
buf.extend_from_slice(b"* STATUS ");
if is_rev2 {
quoted_string(buf, &self.mailbox_name);
} else {
quoted_string(buf, &utf7_encode(&self.mailbox_name));
}
buf.extend_from_slice(b" (");
for (pos, (status_item, value)) in self.items.iter().enumerate() {
if pos > 0 {
buf.push(b' ');
}
buf.extend_from_slice(match status_item {
Status::Messages => b"MESSAGES ",
Status::UidNext => b"UIDNEXT ",
Status::UidValidity => b"UIDVALIDITY ",
Status::Unseen => b"UNSEEN ",
Status::Deleted => b"DELETED ",
Status::Size => b"SIZE ",
Status::HighestModSeq => b"HIGHESTMODSEQ ",
Status::MailboxId => b"MAILBOXID ",
Status::Recent => b"RECENT ",
});
match value {
StatusItemType::Number(num) => {
buf.extend_from_slice(num.to_string().as_bytes());
}
StatusItemType::String(str) => {
buf.push(b'(');
buf.extend_from_slice(str.as_bytes());
buf.push(b')');
}
}
}
buf.extend_from_slice(b")\r\n");
}
}
#[cfg(test)]
mod tests {
use crate::protocol::status::{Status, StatusItem, StatusItemType};
#[test]
fn serialize_status() {
let mut buf = Vec::new();
StatusItem {
mailbox_name: "blurdybloop".to_string(),
items: vec![
(Status::Messages, StatusItemType::Number(231)),
(Status::UidNext, StatusItemType::Number(44292)),
(
Status::MailboxId,
StatusItemType::String("abc-123".to_string()),
),
],
}
.serialize(&mut buf, true);
assert_eq!(
String::from_utf8(buf).unwrap(),
concat!(
"* STATUS \"blurdybloop\" (MESSAGES 231 UIDNEXT 44292 MAILBOXID (abc-123))\r\n",
)
);
}
}
|