handsome_logger 0.9.2

use std::io::Write;
use std::sync::Mutex;

use log::{set_boxed_logger, set_max_level, LevelFilter, Log, Metadata, Record, SetLoggerError};

use crate::common::get_env_log;
use crate::{Config, SharedLogger};

use super::logging::try_log;

pub struct WriteLogger<W: Write + Send + 'static> {
    level: LevelFilter,
    config: Config,
    writable: Mutex<W>,
}

impl<W: Write + Send + 'static> WriteLogger<W> {
    pub fn init(config: Config, writable: W) -> Result<(), SetLoggerError> {
        let log_level = get_env_log().unwrap_or(config.level);
        set_max_level(log_level);
        let logger = Self::new(config, writable);
        set_boxed_logger(logger)
    }

    #[must_use]
    pub fn new(mut config: Config, writable: W) -> Box<Self> {
        config.calculate_data();

        let log_level = get_env_log().unwrap_or(config.level);
        Box::new(Self {
            level: log_level,
            config,
            writable: Mutex::new(writable),
        })
    }
}

impl<W: Write + Send + 'static> Log for WriteLogger<W> {
    fn enabled(&self, metadata: &Metadata) -> bool {
        metadata.level() <= self.level
    }

    fn log(&self, record: &Record) {
        if let Some(message_filtering) = &self.config.message_filtering {
            if !message_filtering(record) {
                return;
            }
        }

        if self.enabled(record.metadata()) {
            let mut write_lock = self.writable.lock().unwrap();

            if let Some(write_formatter) = &self.config.write_formatter {
                if self.config.write_once {
                    let mut buffer: Vec<u8> = Vec::new();
                    let _ = write_formatter(record, &mut buffer);
                    let _ = write_lock.write_all(buffer.as_slice());
                } else {
                    let _ = write_formatter(record, &mut *write_lock);
                }
            } else if self.config.write_once {
                let mut buffer: Vec<u8> = Vec::new();
                let _ = try_log(&self.config, record, &mut buffer);
                let _ = write_lock.write_all(buffer.as_slice());
            } else {
                let _ = try_log(&self.config, record, &mut *write_lock);
            }
        }
    }

    fn flush(&self) {
        let _ = self.writable.lock().unwrap().flush();
    }
}

impl<W: Write + Send + 'static> SharedLogger for WriteLogger<W> {
    fn level(&self) -> LevelFilter {
        self.level
    }

    fn config(&self) -> Option<&Config> {
        Some(&self.config)
    }

    fn as_log(self: Box<Self>) -> Box<dyn Log> {
        Box::new(*self)
    }
}