boitalettres/examples/simple.rs

use miette::{IntoDiagnostic, Result};

use boitalettres::proto::{Request, Response};
use boitalettres::server::accept::addr::{AddrIncoming, AddrStream};
use boitalettres::server::Server;
use tracing_subscriber;


use tokio::time::{sleep, Duration};

async fn handle_req(req: Request) -> Result<Response> {
    use imap_codec::types::response::{Capability, Data};

    tracing::debug!("Got request: {:#?}", req);

    println!("WILL SLEEP");
    sleep(Duration::from_millis(100)).await;
    println!("DONE SLEEP");

    let capabilities = vec![Capability::Imap4Rev1, Capability::Idle];
    let body = vec![Data::Capability(capabilities)];

    Ok(Response::ok("Done")?.with_body(body))
}

#[tokio::main]
async fn main() -> Result<()> {
    tracing_subscriber::fmt::init();

    let incoming = AddrIncoming::new("127.0.0.1:4567")
        .await
        .into_diagnostic()?;

    let make_service = tower::service_fn(|addr: &AddrStream| {
        tracing::debug!(remote_addr = %addr.remote_addr, local_addr = %addr.local_addr, "accept");
        async {

          println!("222 WILL SLEEP");
          sleep(Duration::from_millis(100)).await;
          println!("222 DONE SLEEP");

          let service = tower::ServiceBuilder::new().service_fn(handle_req);
        
          let r: Result<_, boitalettres::errors::Error> = Ok(service);
          r
        }
    });

    let server = Server::new(incoming).serve(make_service);

    server.await.map_err(Into::into)
}