Add message::link using generic "T: Display + PartialEq", provide impl's

for "&'a str" and String
This commit is contained in:
Nathan Fisher 2023-05-31 11:20:02 -04:00
parent 477de69ca3
commit 42afaf4e6a
5 changed files with 83 additions and 17 deletions

View file

@ -6,3 +6,6 @@ pub mod verifier;
pub struct Connection { pub struct Connection {
pub inner: rustls::ServerConnection, pub inner: rustls::ServerConnection,
} }
impl Connection {
}

View file

@ -8,6 +8,7 @@ pub enum Error {
EmptyHost, EmptyHost,
EmptyMessage, EmptyMessage,
ParseHostError(ParseHostError), ParseHostError(ParseHostError),
MalformedLink,
} }
impl fmt::Display for Error { impl fmt::Display for Error {

59
src/message/link.rs Normal file
View file

@ -0,0 +1,59 @@
use std::{fmt::Display, str::FromStr};
#[cfg(feature = "serde")]
use serde::{Deserialize, Serialize};
#[derive(Clone, Debug, PartialEq)]
#[cfg_attr(feature = "serde", derive(Deserialize, Serialize))]
pub struct Link<T: Display + PartialEq> {
pub url: T,
pub display: Option<T>,
}
impl<T> Display for Link<T>
where T: Display + PartialEq {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
match &self.display {
Some(d) => write!(f, "=> {} {d}", self.url),
None => write!(f, "=> {}", self.url),
}
}
}
impl FromStr for Link<String> {
type Err = super::Error;
fn from_str(s: &str) -> Result<Self, Self::Err> {
let Some(s) = s.strip_prefix("=> ") else {
return Err(super::Error::MalformedLink);
};
if let Some((url, display)) = s.split_once(char::is_whitespace) {
Ok(Self { url: url.to_string(), display: Some(display.to_string()) })
} else {
Ok(Self { url: s.to_string(), display: None })
}
}
}
impl TryFrom<&str> for Link<String> {
type Error = super::Error;
fn try_from(value: &str) -> Result<Self, Self::Error> {
value.parse()
}
}
impl<'a> TryFrom<&'a str> for Link<&'a str> {
type Error = super::Error;
fn try_from(value: &'a str) -> Result<Self, Self::Error> {
let Some(s) = value.strip_prefix("=> ") else {
return Err(super::Error::MalformedLink);
};
if let Some((url, display)) = s.split_once(char::is_whitespace) {
Ok(Self { url, display: Some(display) })
} else {
Ok(Self { url: s , display: None })
}
}
}

View file

@ -1,11 +1,13 @@
use crate::prelude::{Host, Mailbox}; use crate::prelude::{Host, Mailbox};
use std::{fmt, str::FromStr}; use std::{fmt, str::FromStr, io::BufRead};
#[cfg(feature = "serde")] #[cfg(feature = "serde")]
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
mod error; mod error;
pub use error::Error; mod link;
mod parser;
pub use {error::Error, link::Link, parser::Parser};
#[derive(Clone, Debug, PartialEq)] #[derive(Clone, Debug, PartialEq)]
pub struct Recipients { pub struct Recipients {
@ -20,19 +22,21 @@ impl fmt::Display for Recipients {
} }
#[derive(Clone, Debug, PartialEq)] #[derive(Clone, Debug, PartialEq)]
pub enum Lines { pub enum Lines<T: fmt::Display + PartialEq> {
Sender(Mailbox), Sender(Mailbox),
Recipients(Recipients), Recipients(Recipients),
Timestamp(String), Timestamp(T),
Text(String), Text(T),
Heading1(String), Heading1(T),
Heading2(String), Heading2(T),
Heading3(String), Heading3(T),
Quote(String), Quote(T),
Preformatted(String), Preformatted(T),
Link(Link<T>),
} }
impl fmt::Display for Lines { impl<T> fmt::Display for Lines<T>
where T: fmt::Display + PartialEq + BufRead {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
match self { match self {
Self::Sender(m) => writeln!(f, "< {m}"), Self::Sender(m) => writeln!(f, "< {m}"),
@ -42,13 +46,9 @@ impl fmt::Display for Lines {
Self::Heading1(h) => writeln!(f, "# {h}"), Self::Heading1(h) => writeln!(f, "# {h}"),
Self::Heading2(h) => writeln!(f, "## {h}"), Self::Heading2(h) => writeln!(f, "## {h}"),
Self::Heading3(h) => writeln!(f, "### {h}"), Self::Heading3(h) => writeln!(f, "### {h}"),
Self::Quote(q) => { Self::Quote(q) => writeln!(f, "> {q}"),
for l in q.lines() {
writeln!(f, "> {l}")?;
}
Ok(())
}
Self::Preformatted(p) => writeln!(f, "```\n{p}\n```"), Self::Preformatted(p) => writeln!(f, "```\n{p}\n```"),
Self::Link(l) => writeln!(f, "=> {l}"),
} }
} }
} }
@ -57,6 +57,7 @@ impl fmt::Display for Lines {
#[cfg_attr(feature = "serde", derive(Deserialize, Serialize))] #[cfg_attr(feature = "serde", derive(Deserialize, Serialize))]
pub struct Message { pub struct Message {
pub id: String, pub id: String,
pub from: Mailbox,
pub senders: Vec<Mailbox>, pub senders: Vec<Mailbox>,
pub recipients: Vec<Mailbox>, pub recipients: Vec<Mailbox>,
pub timstamp: Option<String>, pub timstamp: Option<String>,

View file

@ -0,0 +1,2 @@
#[derive(Debug)]
pub struct Parser;