mangofetch-core 0.5.5

use std::path::PathBuf;
use tracing_appender::non_blocking;
use tracing_subscriber::{fmt, prelude::*, Registry};

pub fn init_logging(verbose: bool) {
    init_logging_ext(verbose, true);
}

pub fn init_logging_ext(verbose: bool, use_stdout: bool) {
    let log_dir = crate::core::paths::app_data_dir()
        .map(|d| d.join("logs"))
        .unwrap_or_else(|| PathBuf::from("logs"));

    let _ = std::fs::create_dir_all(&log_dir);

    let file_appender = tracing_appender::rolling::daily(&log_dir, "mangofetch.log");
    let (non_blocking_appender, _guard) = non_blocking(file_appender);

    // Keep the guard alive as long as the program runs.
    Box::leak(Box::new(_guard));

    let file_layer = fmt::layer()
        .with_ansi(false)
        .with_writer(non_blocking_appender);

    if use_stdout {
        let stdout_layer = fmt::layer()
            .with_ansi(true)
            .with_target(verbose)
            .with_filter(if verbose {
                tracing_subscriber::filter::LevelFilter::DEBUG
            } else {
                tracing_subscriber::filter::LevelFilter::INFO
            });

        let subscriber = Registry::default().with(file_layer).with(stdout_layer);
        tracing::subscriber::set_global_default(subscriber)
            .expect("setting default subscriber failed");
    } else {
        let subscriber = Registry::default().with(file_layer);
        tracing::subscriber::set_global_default(subscriber)
            .expect("setting default subscriber failed");
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use std::env;
    use std::process::Command;
    use uuid::Uuid;

    #[test]
    fn test_init_logging_ext_subprocess() {
        if env::var("TEST_INIT_LOGGING_EXT").is_ok() {
            let temp_dir = env::temp_dir().join(Uuid::new_v4().to_string());
            env::set_var("MANGOFETCH_DATA_DIR", temp_dir.to_str().unwrap());

            init_logging_ext(true, true);

            let log_dir = temp_dir.join("logs");
            assert!(log_dir.exists());

            std::process::exit(0);
        }

        let exe = env::current_exe().unwrap();
        let output = Command::new(exe)
            .env("TEST_INIT_LOGGING_EXT", "1")
            .arg("core::logger::tests::test_init_logging_ext_subprocess")
            .arg("--exact")
            .output()
            .unwrap();

        assert!(
            output.status.success(),
            "init_logging_ext failed: {:?}",
            output
        );
    }

    #[test]
    fn test_init_logging_ext_no_stdout_subprocess() {
        if env::var("TEST_INIT_LOGGING_EXT_NO_STDOUT").is_ok() {
            let temp_dir = env::temp_dir().join(Uuid::new_v4().to_string());
            env::set_var("MANGOFETCH_DATA_DIR", temp_dir.to_str().unwrap());

            init_logging_ext(false, false);

            let log_dir = temp_dir.join("logs");
            assert!(log_dir.exists());

            std::process::exit(0);
        }

        let exe = env::current_exe().unwrap();
        let output = Command::new(exe)
            .env("TEST_INIT_LOGGING_EXT_NO_STDOUT", "1")
            .arg("core::logger::tests::test_init_logging_ext_no_stdout_subprocess")
            .arg("--exact")
            .output()
            .unwrap();

        assert!(
            output.status.success(),
            "init_logging_ext failed: {:?}",
            output
        );
    }

    #[test]
    fn test_init_logging() {
        if env::var("TEST_INIT_LOGGING").is_ok() {
            let temp_dir = env::temp_dir().join(Uuid::new_v4().to_string());
            env::set_var("MANGOFETCH_DATA_DIR", temp_dir.to_str().unwrap());

            init_logging(false);

            let log_dir = temp_dir.join("logs");
            assert!(log_dir.exists());

            std::process::exit(0);
        }

        let exe = env::current_exe().unwrap();
        let output = Command::new(exe)
            .env("TEST_INIT_LOGGING", "1")
            .arg("core::logger::tests::test_init_logging")
            .arg("--exact")
            .output()
            .unwrap();

        assert!(output.status.success(), "init_logging failed: {:?}", output);
    }
}