extern crate imap;
extern crate native_tls;
use imap::Client;
use native_tls::TlsConnector;
use native_tls::TlsStream;
use std::env;
use std::error::Error;
use std::fmt;
use std::net::{SocketAddr, TcpStream, ToSocketAddrs};
use std::time::Duration;
fn main() -> Result<(), Box<dyn Error>> {
let server = env::var("IMAP_SERVER")?;
let port = env::var("IMAP_PORT").unwrap_or_else(|_| String::from("993"));
let port = port.parse()?;
let username = env::var("IMAP_USER")?;
let password = env::var("IMAP_PASSWORD")?;
let timeout = env::var("IMAP_TIMEOUT").unwrap_or_else(|_| String::from("1"));
let timeout = timeout.parse()?;
let timeout = Duration::from_secs(timeout);
let tls = TlsConnector::builder().build()?;
let client = connect_all_timeout((server.as_str(), port), server.as_str(), &tls, timeout)?;
let mut session = client.login(&username, &password).map_err(|e| e.0)?;
session.logout()?;
Ok(())
}
fn connect_timeout<S: AsRef<str>>(
addr: &SocketAddr,
domain: S,
ssl_connector: &TlsConnector,
timeout: Duration,
) -> Result<Client<TlsStream<TcpStream>>, Box<dyn Error>> {
let tcp_stream = TcpStream::connect_timeout(addr, timeout)?;
let tls_stream = TlsConnector::connect(ssl_connector, domain.as_ref(), tcp_stream)?;
let mut client = Client::new(tls_stream);
client.read_greeting()?;
Ok(client)
}
fn connect_all_timeout<A: ToSocketAddrs, S: AsRef<str>>(
addr: A,
domain: S,
ssl_connector: &TlsConnector,
timeout: Duration,
) -> Result<Client<TlsStream<TcpStream>>, Box<dyn Error>> {
let addrs = addr.to_socket_addrs()?;
for addr in addrs {
match connect_timeout(&addr, &domain, ssl_connector, timeout) {
Ok(client) => return Ok(client),
Err(error) => eprintln!("couldn't connect to {}: {}", addr, error),
}
}
Err(Box::new(TimeoutError))
}
#[derive(Debug)]
struct TimeoutError;
impl fmt::Display for TimeoutError {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "all addresses failed to connect")
}
}
impl Error for TimeoutError {}