packager/pkgr/src/logging.rs

use colored::Colorize;
use fern::Dispatch;
use log::SetLoggerError;
use std::env;

pub fn setup_logger() -> Result<(), SetLoggerError> {
    Dispatch::new()
        .format(|out, message, record| {
            match record.metadata().target() {
                "command:stdout" => {
                    out.finish(format_args!("{} {}", "::".cyan(), message.to_string()));
                    return;
                }
                "command:stderr" => {
                    out.finish(format_args!("{} {}", "!!".red(), message.to_string()));
                    return;
                }
                _ => {
                    out.finish(format_args!(
                        "{} {}",
                        // Some logic so messages look nice
                        if message.to_string().len() > 0 {
                            match record.level().to_string().chars().nth(0).unwrap_or('T') {
                                'T' => "[TRACE]".bright_blue(),
                                'D' => "??".green(),
                                'I' => "=>".blue(),
                                'W' => "##".yellow(),
                                'E' => "!!".red(),
                                _ => "**".blue(),
                            }
                            .to_string()
                        } else {
                            "".to_string()
                        },
                        message.to_string().bright_white()
                    ))
                }
            }
        })
        .level(
            env::var("PKGR_LOG_LEVEL")
                .unwrap_or_else(|_| "info".to_string())
                .parse()
                .unwrap_or(log::LevelFilter::Info),
        )
        .chain(std::io::stdout())
        .apply()
}