use anyhow::{bail, Result}; use std::sync::Arc; use rusoto_signature::Region; use crate::config::*; use crate::login::{ldap_provider::*, static_provider::*, *}; use crate::mailbox::Mailbox; use boitalettres::proto::{Request, Response}; use boitalettres::server::accept::addr::{AddrIncoming, AddrStream}; use boitalettres::server::Server as ImapServer; use std::pin::Pin; use std::task::{Context, Poll}; use tower::Service; pub struct Server { pub login_provider: Box, } struct Connection; impl Service for Connection { type Response = Response; type Error = anyhow::Error; type Future = Pin> + Send>>; fn poll_ready(&mut self, cx: &mut Context<'_>) -> Poll> { Poll::Ready(Ok(())) } fn call(&mut self, req: Request) -> Self::Future { tracing::debug!("Got request: {:#?}", req); Box::pin(async move { use imap_codec::types::{ command::CommandBody, response::{Capability, Data}, }; let r = match req.body { CommandBody::Capability => { let capabilities = vec![Capability::Imap4Rev1, Capability::Idle]; let body = vec![Data::Capability(capabilities)]; Response::ok( "Pre-login capabilities listed, post-login capabilities have more.", )? .with_body(body) } CommandBody::Login { username: _, password: _, } => Response::ok("Logged in")?, _ => Response::bad("Error in IMAP command received by server.")?, }; Ok(r) }) } } struct Instance; impl<'a> Service<&'a AddrStream> for Instance { type Response = Connection; type Error = anyhow::Error; type Future = Pin> + Send>>; fn poll_ready(&mut self, cx: &mut Context<'_>) -> Poll> { Poll::Ready(Ok(())) } fn call(&mut self, addr: &'a AddrStream) -> Self::Future { tracing::info!(remote_addr = %addr.remote_addr, local_addr = %addr.local_addr, "accept"); Box::pin(async { Ok(Connection) }) } } impl Server { pub fn new(config: Config) -> Result> { let s3_region = Region::Custom { name: config.aws_region.clone(), endpoint: config.s3_endpoint, }; let k2v_region = Region::Custom { name: config.aws_region, endpoint: config.k2v_endpoint, }; let login_provider: Box = match (config.login_static, config.login_ldap) { (Some(st), None) => Box::new(StaticLoginProvider::new(st, k2v_region, s3_region)?), (None, Some(ld)) => Box::new(LdapLoginProvider::new(ld, k2v_region, s3_region)?), (Some(_), Some(_)) => bail!("A single login provider must be set up in config file"), (None, None) => bail!("No login provider is set up in config file"), }; Ok(Arc::new(Self { login_provider })) } pub async fn run(self: &Arc) -> Result<()> { // tracing_subscriber::fmt::init(); let incoming = AddrIncoming::new("127.0.0.1:4567").await?; let server = ImapServer::new(incoming).serve(Instance); let _ = server.await?; /*let creds = self.login_provider.login("quentin", "poupou").await?; let mut mailbox = Mailbox::new(&creds, "TestMailbox".to_string()).await?; mailbox.test().await?;*/ Ok(()) } }