Revision control
Copy as Markdown
Other Tools
/*
* Copyright Stalwart Labs Ltd. See the COPYING
* file at the top-level directory of this distribution.
*
* Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
* option. This file may not be copied, modified, or distributed
* except according to those terms.
*/
pub mod address;
pub mod content_type;
pub mod date;
pub mod message_id;
pub mod raw;
pub mod text;
pub mod url;
use std::io::{self, Write};
use self::{
address::Address, content_type::ContentType, date::Date, message_id::MessageId, raw::Raw,
text::Text, url::URL,
};
pub trait Header {
fn write_header(&self, output: impl Write, bytes_written: usize) -> io::Result<usize>;
}
#[derive(Debug, Clone, PartialEq, Eq, PartialOrd, Ord)]
pub enum HeaderType<'x> {
Address(Address<'x>),
Date(Date),
MessageId(MessageId<'x>),
Raw(Raw<'x>),
Text(Text<'x>),
URL(URL<'x>),
ContentType(ContentType<'x>),
}
impl<'x> From<Address<'x>> for HeaderType<'x> {
fn from(value: Address<'x>) -> Self {
HeaderType::Address(value)
}
}
impl<'x> From<ContentType<'x>> for HeaderType<'x> {
fn from(value: ContentType<'x>) -> Self {
HeaderType::ContentType(value)
}
}
impl<'x> From<Date> for HeaderType<'x> {
fn from(value: Date) -> Self {
HeaderType::Date(value)
}
}
impl<'x> From<MessageId<'x>> for HeaderType<'x> {
fn from(value: MessageId<'x>) -> Self {
HeaderType::MessageId(value)
}
}
impl<'x> From<Raw<'x>> for HeaderType<'x> {
fn from(value: Raw<'x>) -> Self {
HeaderType::Raw(value)
}
}
impl<'x> From<Text<'x>> for HeaderType<'x> {
fn from(value: Text<'x>) -> Self {
HeaderType::Text(value)
}
}
impl<'x> From<URL<'x>> for HeaderType<'x> {
fn from(value: URL<'x>) -> Self {
HeaderType::URL(value)
}
}
impl<'x> Header for HeaderType<'x> {
fn write_header(&self, output: impl Write, bytes_written: usize) -> io::Result<usize> {
match self {
HeaderType::Address(value) => value.write_header(output, bytes_written),
HeaderType::Date(value) => value.write_header(output, bytes_written),
HeaderType::MessageId(value) => value.write_header(output, bytes_written),
HeaderType::Raw(value) => value.write_header(output, bytes_written),
HeaderType::Text(value) => value.write_header(output, bytes_written),
HeaderType::URL(value) => value.write_header(output, bytes_written),
HeaderType::ContentType(value) => value.write_header(output, bytes_written),
}
}
}
impl<'x> HeaderType<'x> {
pub fn as_content_type(&self) -> Option<&ContentType> {
match self {
HeaderType::ContentType(value) => Some(value),
_ => None,
}
}
}