forked from KokaKiwi/boitalettres
59 lines
1.8 KiB
Rust
59 lines
1.8 KiB
Rust
use miette::{IntoDiagnostic, Result};
|
|
|
|
use boitalettres::proto::{Request, Response};
|
|
use boitalettres::server::accept::addr::{AddrIncoming, AddrStream};
|
|
use boitalettres::server::Server;
|
|
|
|
async fn handle_req(req: Request) -> Result<Response> {
|
|
use imap_codec::types::response::{Capability, Data};
|
|
|
|
tracing::debug!("Got request: {:#?}", req);
|
|
|
|
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<()> {
|
|
setup_logging();
|
|
|
|
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");
|
|
|
|
let service = tower::ServiceBuilder::new().service_fn(handle_req);
|
|
|
|
futures::future::ok::<_, std::convert::Infallible>(service)
|
|
});
|
|
|
|
let server = Server::new(incoming).serve(make_service);
|
|
|
|
server.await.map_err(Into::into)
|
|
}
|
|
|
|
// Don't mind this, this is just for debugging.
|
|
fn setup_logging() {
|
|
use tracing_subscriber::prelude::*;
|
|
|
|
let registry = tracing_subscriber::registry().with(
|
|
tracing_subscriber::fmt::layer().with_filter(
|
|
tracing_subscriber::filter::Targets::new()
|
|
.with_default(tracing::Level::DEBUG)
|
|
.with_target("boitalettres", tracing::Level::TRACE)
|
|
.with_target("simple", tracing::Level::TRACE)
|
|
.with_target("tower", tracing::Level::TRACE)
|
|
.with_target("tokio_tower", tracing::Level::TRACE)
|
|
.with_target("mio", tracing::Level::TRACE),
|
|
),
|
|
);
|
|
|
|
#[cfg(tokio_unstable)]
|
|
let registry = registry.with(console_subscriber::spawn());
|
|
|
|
registry.init();
|
|
}
|