use commonware_runtime::{
    telemetry::metrics::{histogram, status, Gauge, MetricsExt as _},
    Metrics as RuntimeMetrics,
};

/// Metrics for the peer actor.
pub struct Metrics {
    /// Current number of pending fetch requests
    pub fetch_pending: Gauge,
    /// Current number of active fetch requests
    pub fetch_active: Gauge,
    /// Current number of serves currently in flight
    pub serve_processing: Gauge,
    /// Current number of blocked peers
    pub peers_blocked: Gauge,
    /// Number of fetches by status
    pub fetch: status::Counter,
    /// Number of canceled fetches by status
    pub cancel: status::Counter,
    /// Number of serves by status
    pub serve: status::Counter,
    /// Histogram of successful serves
    pub serve_duration: histogram::Timed,
    /// Histogram of successful fetches
    pub fetch_duration: histogram::Timed,
}

impl Metrics {
    /// Create and return a new set of metrics, registered with the given context.
    pub fn init<E: RuntimeMetrics>(context: &E) -> Self {
        let fetch_pending =
            context.gauge("fetch_pending", "Current number of pending fetch requests");
        let fetch_active = context.gauge("fetch_active", "Current number of active fetch requests");
        let serve_processing = context.gauge(
            "serve_processing",
            "Current number of serves currently processing",
        );
        let peers_blocked = context.gauge("peers_blocked", "Current number of blocked peers");
        let fetch = context.family("fetch", "Number of fetches by status");
        let cancel = context.family("cancel", "Number of canceled fetches by status");
        let serve = context.family("serve", "Number of serves by status");
        let serve_duration_registered = context.histogram(
            "serve_duration",
            "Histogram of successful serves",
            histogram::Buckets::LOCAL,
        );
        let fetch_duration_registered = context.histogram(
            "fetch_duration",
            "Histogram of successful fetches",
            histogram::Buckets::NETWORK,
        );

        Self {
            fetch_pending,
            fetch_active,
            serve_processing,
            peers_blocked,
            fetch,
            cancel,
            serve,
            fetch_duration: histogram::Timed::new(fetch_duration_registered),
            serve_duration: histogram::Timed::new(serve_duration_registered),
        }
    }
}