use std::io;
use tracing::{level_filters::LevelFilter, Subscriber};
use tracing_appender::rolling::{RollingFileAppender, Rotation};
use tracing_subscriber::{fmt, layer::SubscriberExt, layer::Filter, prelude::*};

use super::util::VAR_LOG;

// A layer filter to prevent polling timeout errors from clogging logs
struct TeloxideNoiseFilter {}
impl<S: Subscriber> Filter<S> for TeloxideNoiseFilter {
    fn enabled(&self, meta: &tracing::Metadata<'_>, _: &tracing_subscriber::layer::Context<'_,S>) -> bool {
        if let Some(module_path) = meta.module_path() {
            if module_path == "teloxide::error_handlers" {
                false
            } else {
                true
            }
        } else {
            true
        }
    }
}

pub fn log_init() {
    let log_appender = RollingFileAppender::builder()
        .rotation(Rotation::DAILY)
        .filename_prefix("mk-dl-bot.log")
        .max_log_files(7)
        .build(VAR_LOG)
        .unwrap();

    let stderr_layer = fmt::layer()
        .with_writer(io::stderr)
        .with_filter(TeloxideNoiseFilter {})
        .with_filter(LevelFilter::ERROR);

    let file_layer = fmt::layer()
        .with_writer(log_appender)
        .with_filter(TeloxideNoiseFilter {})
        .with_filter(LevelFilter::INFO);

    tracing_subscriber::registry()
        .with(stderr_layer)
        .with(file_layer)
        .init();
}