use std::net::SocketAddr;

use axum::{routing::get, Router};
use lazy_static;
use log::warn;
use prometheus::{register_int_counter, IntCounter, Registry};

lazy_static! {
    static ref REGISTRY: Registry = Registry::new();
    pub static ref PLAY_COUNTER: IntCounter =
        register_int_counter!("play_cmd", "Number of calls to /play").unwrap();
    pub static ref UPLOAD_COUNTER: IntCounter =
        register_int_counter!("upload_cmd", "Number of calls to /upload").unwrap();
    pub static ref DELETE_COUNTER: IntCounter =
        register_int_counter!("delete_cmd", "Number of calls to /delete").unwrap();
}

pub fn init_metrics() {
    REGISTRY.register(Box::new(PLAY_COUNTER.clone())).unwrap();
}

pub async fn serve() {
    let app = Router::new().route("/metrics", get(metrics));
    let addr = SocketAddr::from(([127, 0, 0, 1], 31755));

    axum::Server::bind(&addr)
        .serve(app.into_make_service())
        .await
        .unwrap();
}

async fn metrics() -> String {
    let encoder = prometheus::TextEncoder::new();
    let res_custom = encoder.encode_to_string(&REGISTRY.gather());

    match res_custom {
        Ok(s) => s,
        Err(e) => {
            warn!("Error encoding metrics: {:?}", e);

            String::new()
        }
    }
}