Import upstream v0.16.22, stripped
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.
This commit is contained in:
commit
7dae9b29fd
1650 files changed
+485521
No files matched your search
@@ -0,0 +1,106 @@
|
||||
/*
|
||||
* SPDX-FileCopyrightText: 2020 Stalwart Labs LLC <[email protected]>
|
||||
*
|
||||
* SPDX-License-Identifier: AGPL-3.0-only OR LicenseRef-SEL
|
||||
*/
|
||||
|
||||
use crate::{
|
||||
Command,
|
||||
protocol::{capability::Capability, enable},
|
||||
receiver::{Request, bad},
|
||||
};
|
||||
use compact_str::ToCompactString;
|
||||
|
||||
impl Request<Command> {
|
||||
pub fn parse_enable(self) -> trc::Result<enable::Arguments> {
|
||||
let len = self.tokens.len();
|
||||
if len > 0 {
|
||||
let mut capabilities = Vec::with_capacity(len);
|
||||
for capability in self.tokens {
|
||||
capabilities.push(
|
||||
Capability::parse(&capability.unwrap_bytes())
|
||||
.map_err(|v| bad(self.tag.to_compact_string(), v))?,
|
||||
);
|
||||
}
|
||||
Ok(enable::Arguments {
|
||||
tag: self.tag,
|
||||
capabilities,
|
||||
})
|
||||
} else {
|
||||
Err(self.into_error("Missing arguments."))
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl Capability {
|
||||
pub fn parse(value: &[u8]) -> super::Result<Self> {
|
||||
hashify::tiny_map_ignore_case!(value,
|
||||
"IMAP4rev2" => Self::IMAP4rev2,
|
||||
"STARTTLS" => Self::StartTLS,
|
||||
"LOGINDISABLED" => Self::LoginDisabled,
|
||||
"CONDSTORE" => Self::CondStore,
|
||||
"QRESYNC" => Self::QResync,
|
||||
"UTF8=ACCEPT" => Self::Utf8Accept,
|
||||
"OBJECTID+" => Self::ObjectIdPlus,
|
||||
"UIDONLY" => Self::UidOnly,
|
||||
)
|
||||
.ok_or_else(|| {
|
||||
format!(
|
||||
"Unsupported capability '{}'.",
|
||||
String::from_utf8_lossy(value)
|
||||
)
|
||||
.into()
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use crate::{
|
||||
protocol::{capability::Capability, enable},
|
||||
receiver::Receiver,
|
||||
};
|
||||
|
||||
#[test]
|
||||
fn parse_enable() {
|
||||
let mut receiver = Receiver::new();
|
||||
|
||||
for (command, arguments) in [
|
||||
(
|
||||
"t2 ENABLE IMAP4rev2 CONDSTORE\r\n",
|
||||
enable::Arguments {
|
||||
tag: "t2".into(),
|
||||
capabilities: vec![Capability::IMAP4rev2, Capability::CondStore],
|
||||
},
|
||||
),
|
||||
(
|
||||
"t3 ENABLE OBJECTID+\r\n",
|
||||
enable::Arguments {
|
||||
tag: "t3".into(),
|
||||
capabilities: vec![Capability::ObjectIdPlus],
|
||||
},
|
||||
),
|
||||
(
|
||||
"t4 ENABLE CONDSTORE OBJECTID+ UTF8=ACCEPT\r\n",
|
||||
enable::Arguments {
|
||||
tag: "t4".into(),
|
||||
capabilities: vec![
|
||||
Capability::CondStore,
|
||||
Capability::ObjectIdPlus,
|
||||
Capability::Utf8Accept,
|
||||
],
|
||||
},
|
||||
),
|
||||
] {
|
||||
assert_eq!(
|
||||
receiver
|
||||
.parse(&mut command.as_bytes().iter())
|
||||
.unwrap()
|
||||
.parse_enable()
|
||||
.unwrap(),
|
||||
arguments,
|
||||
"Failed to parse {command}"
|
||||
);
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in new issue
Block a user