PluralKit/lib/libpk/src/lib.rs

#![feature(let_chains)]
use metrics_exporter_prometheus::PrometheusBuilder;
use tracing_subscriber::EnvFilter;

pub mod db;
pub mod proto;
pub mod util;

pub mod _config;
pub use crate::_config::CONFIG as config;

pub fn init_logging(component: &str) -> anyhow::Result<()> {
    // todo: fix component
    if config.json_log {
        tracing_subscriber::fmt()
            .json()
            .with_env_filter(EnvFilter::from_default_env())
            .init();
    } else {
        tracing_subscriber::fmt()
            .with_env_filter(EnvFilter::from_default_env())
            .init();
    }

    Ok(())
}

pub fn init_metrics() -> anyhow::Result<()> {
    if config.run_metrics_server {
        // automatically spawns a http listener at :9000
        let builder = PrometheusBuilder::new();
        builder.install()?;
    }
    Ok(())
}