Impl `EhloOkResp::{new, serialize}`

This commit is contained in:
Damian Poddebniak 2021-05-22 22:56:45 +02:00
parent 54a0311fc6
commit 7cac2a2903
1 changed files with 39 additions and 0 deletions

View File

@ -282,6 +282,45 @@ pub struct EhloOkResp {
pub lines: Vec<Capability>, pub lines: Vec<Capability>,
} }
impl EhloOkResp {
pub fn new<D, G>(domain: D, greet: Option<G>, capabilities: Vec<Capability>) -> EhloOkResp
where
D: Into<String>,
G: Into<String>,
{
EhloOkResp {
domain: domain.into(),
greet: greet.map(|inner| inner.into()),
lines: capabilities,
}
}
pub fn serialize(&self, writer: &mut impl Write) -> std::io::Result<()> {
let greet = match self.greet {
Some(ref greet) => format!(" {}", greet),
None => "".to_string(),
};
if let Some((tail, head)) = self.lines.split_last() {
writer.write_all(format!("250-{}{}\r\n", self.domain, greet).as_bytes())?;
for capability in head {
writer.write_all(b"250-")?;
capability.serialize(writer)?;
writer.write_all(b"\r\n")?;
}
writer.write_all(b"250 ")?;
tail.serialize(writer)?;
writer.write_all(b"\r\n")
} else {
writer.write_all(format!("250 {}{}\r\n", self.domain, greet).as_bytes())
}
}
}
// -------------------------------------------------------------------------------------------------
#[derive(Debug, Clone, PartialEq, Eq)] #[derive(Debug, Clone, PartialEq, Eq)]
pub enum Capability { pub enum Capability {
// Send as mail [RFC821] // Send as mail [RFC821]