hugo-server 0.3.0

use std::path::Path;
use std::{env, io};

use anyhow::Result;
use clap_verbosity_flag::Verbosity;
use logroller::{LogRollerBuilder, Rotation, RotationAge, TimeZone};
use supports_color::Stream;
use tracing_appender::non_blocking::WorkerGuard;
use tracing_subscriber::EnvFilter;
use tracing_subscriber::fmt::Layer;
use tracing_subscriber::fmt::time::ChronoLocal;
use tracing_subscriber::layer::SubscriberExt;
use tracing_subscriber::util::SubscriberInitExt;

pub fn init_log<T>(verbose: &Verbosity, log_directory: T) -> Result<WorkerGuard>
where
    T: AsRef<Path>,
{
    let filter_layer = EnvFilter::try_from_default_env().unwrap_or_else(|_| {
        if verbose.is_silent() {
            "none".into()
        } else {
            format!(
                "{}={1},tower_http={1},axum::rejection=trace",
                env!("CARGO_CRATE_NAME"),
                verbose.filter(),
            )
            .into()
        }
    });

    let log_file_name = env::current_exe()?
        .with_extension("log")
        .file_name()
        .unwrap()
        .to_str()
        .expect("the file name is not in valid UTF-8")
        .to_string();

    let file_appender = LogRollerBuilder::new(log_directory.as_ref(), Path::new(&log_file_name))
        .rotation(Rotation::AgeBased(RotationAge::Daily))
        .time_zone(TimeZone::Local)
        .max_keep_files(7)
        .graceful_shutdown(true)
        .build()?;

    let (file_writer, guard) = tracing_appender::non_blocking(file_appender);
    let file_layer = Layer::new()
        .with_writer(file_writer)
        .with_timer(ChronoLocal::rfc_3339())
        .with_ansi(false);

    let stdout_layer = Layer::new()
        .with_writer(io::stdout)
        .with_timer(ChronoLocal::rfc_3339())
        .with_ansi(supports_color::on(Stream::Stdout).is_some());

    tracing_subscriber::registry()
        .with(filter_layer)
        .with(file_layer)
        .with(stdout_layer)
        .init();

    Ok(guard)
}

#[cfg(test)]
mod tests {
    use testresult::TestResult;

    use super::*;

    #[test]
    fn test_init_log() -> TestResult {
        let _ = init_log(&clap_verbosity_flag::Verbosity::new(4, 0), "log")?;
        tracing::trace!("test trace");

        Ok(())
    }
}