use tako::Method;
use tako::middleware::IntoMiddleware;
use tako::middleware::access_log::AccessLog;
use tako::middleware::healthcheck::Healthcheck;
use tako::middleware::request_id::RequestId;
use tako::middleware::traceparent::Traceparent;
use tako::plugins::metrics::PrometheusMetricsConfig;
use tako::responder::Responder;
use tako::router::Router;

async fn hello() -> impl Responder { "ok" }

#[tokio::main]
async fn main() -> anyhow::Result<()> {
  let listener = tokio::net::TcpListener::bind("127.0.0.1:8080").await?;
  let mut router = Router::new();

  // Request IDs: generate X-Request-ID for inbound, propagate for outbound.
  let request_id = RequestId::new().into_middleware();

  // Access log: one structured line per request via the `tracing` macros.
  let access_log = AccessLog::new().into_middleware();

  // W3C Trace Context: parse `traceparent` / `tracestate`, expose them as
  // a TraceContext extension so handlers can propagate the span.
  let traceparent = Traceparent::new().into_middleware();

  // Health endpoints at /live, /ready, /__drain (and a SIGTERM-friendly handle).
  let healthcheck = Healthcheck::new();
  let _drain = healthcheck.handle();
  let healthcheck_mw = healthcheck.into_middleware();

  router.middleware(request_id);
  router.middleware(access_log);
  router.middleware(traceparent);
  router.middleware(healthcheck_mw);

  // Prometheus scrape endpoint at /metrics (set via PrometheusMetricsConfig).
  let _registry = PrometheusMetricsConfig::default().install(&mut router);

  router.route(Method::GET, "/", hello);

  tako::serve(listener, router).await;
  Ok(())
}