use crate::syntax::SyntaxDetails;
use crate::util::ser_with_display::ser_with_display;
use async_std_resolver::{lookup::MxLookup, resolver_from_system_conf, ResolveError};
use serde::{ser::SerializeMap, Serialize, Serializer};
use std::io::Error;
#[derive(Debug)]
pub struct MxDetails {
pub lookup: Result<MxLookup, ResolveError>,
}
impl Default for MxDetails {
fn default() -> Self {
MxDetails {
lookup: Err(ResolveError::from("Skipped")),
}
}
}
impl From<MxLookup> for MxDetails {
fn from(lookup: MxLookup) -> Self {
MxDetails { lookup: Ok(lookup) }
}
}
impl Serialize for MxDetails {
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
where
S: Serializer,
{
let records = self
.lookup
.as_ref()
.map(|lookup| {
lookup
.iter()
.map(|host| host.exchange().to_string())
.collect::<Vec<_>>()
})
.unwrap_or_else(|_| Vec::new());
let mut map = serializer.serialize_map(Some(2))?;
map.serialize_entry("accepts_mail", &!records.is_empty())?;
map.serialize_entry("records", &records)?;
map.end()
}
}
#[derive(Debug, Serialize)]
#[serde(tag = "type", content = "message")]
pub enum MxError {
#[serde(serialize_with = "ser_with_display")]
IoError(Error),
#[serde(serialize_with = "ser_with_display")]
ResolveError(Box<ResolveError>),
}
impl From<ResolveError> for MxError {
fn from(error: ResolveError) -> Self {
MxError::ResolveError(Box::new(error))
}
}
pub async fn check_mx(syntax: &SyntaxDetails) -> Result<MxDetails, MxError> {
let resolver = resolver_from_system_conf().await?;
match resolver.mx_lookup(syntax.domain.as_ref()).await {
Ok(lookup) => Ok(MxDetails::from(lookup)),
Err(err) => Ok(MxDetails { lookup: Err(err) }),
}
}