use alloc::boxed::Box;
use alloc::vec::Vec;
use core::{
    mem::MaybeUninit,
    sync::atomic::{AtomicPtr, AtomicUsize, Ordering},
};
use protocol::info::{Info, Severity};

pub const LOG_LEVEL: u8 = Severity::Warning as u8;
// pub const LOG_LEVEL: u8 = Severity::Trace as u8;

pub const MAX_LOG_LINE_SIZE: usize = 150;
const SIZE_OF_LOG_LINE_BUFFER: usize = 1024;
static mut LOG_LINES: [AtomicPtr<Info>; SIZE_OF_LOG_LINE_BUFFER] =
    unsafe { MaybeUninit::zeroed().assume_init() };
static START_INDEX: AtomicUsize = unsafe { MaybeUninit::zeroed().assume_init() };
static END_INDEX: AtomicUsize = unsafe { MaybeUninit::zeroed().assume_init() };

pub fn add_line(log_line: Info) {
    let mut index = END_INDEX.fetch_add(1, Ordering::Acquire);
    unsafe {
        index %= SIZE_OF_LOG_LINE_BUFFER;
        let ptr = &mut LOG_LINES[index];
        let line = Box::new(log_line);
        let old = ptr.swap(Box::into_raw(line), Ordering::SeqCst);
        if !old.is_null() {
            _ = Box::from_raw(old);
        }
    }
}

pub fn flush() -> Vec<Info> {
    let mut vec = Vec::new();
    let end_index = END_INDEX.load(Ordering::Acquire);
    let start_index = START_INDEX.load(Ordering::Acquire);
    if end_index <= start_index {
        return vec;
    }
    unsafe {
        let count = end_index - start_index;
        for i in start_index..start_index + count {
            let index = i % SIZE_OF_LOG_LINE_BUFFER;
            let ptr = LOG_LINES[index].swap(core::ptr::null_mut(), Ordering::SeqCst);
            if !ptr.is_null() {
                vec.push(*Box::from_raw(ptr));
            }
        }
    }

    START_INDEX.store(end_index, Ordering::Release);
    vec
}

#[macro_export]
macro_rules! log_internal {
    ($log_line:expr, $($arg:tt)*) => ({
        use core::fmt::Write;
        _ = write!($log_line, "{}:{} ", file!(), line!());
        _ = write!($log_line, $($arg)*);
        $crate::logger::add_line($log_line);
    });
}

#[macro_export]
macro_rules! crit {
    ($($arg:tt)*) => ({
        if protocol::info::Severity::Critical as u8 >= $crate::logger::LOG_LEVEL {
            let message = alloc::format!($($arg)*);
            $crate::logger::add_line(protocol::info::Severity::Critical, alloc::format!("{}:{} ", file!(), line!()), message)
        }
    });
}

#[macro_export]
macro_rules! err {
    ($($arg:tt)*) => ({
        if protocol::info::Severity::Error as u8 >= $crate::logger::LOG_LEVEL {
            let mut log_line = protocol::info::log_line(protocol::info::Severity::Error, $crate::logger::MAX_LOG_LINE_SIZE);
            $crate::log_internal!(log_line, $($arg)*);
        }
    });
}

#[macro_export]
macro_rules! warn {
    ($($arg:tt)*) => ({
        if protocol::info::Severity::Warning as u8 >= $crate::logger::LOG_LEVEL {
            let mut log_line = protocol::info::log_line(protocol::info::Severity::Warning, $crate::logger::MAX_LOG_LINE_SIZE);
            $crate::log_internal!(log_line, $($arg)*);
        }
    });
}

#[macro_export]
macro_rules! dbg {
    ($($arg:tt)*) => ({
        if protocol::info::Severity::Debug as u8 >= $crate::logger::LOG_LEVEL {
            let mut log_line = protocol::info::log_line(protocol::info::Severity::Debug, $crate::logger::MAX_LOG_LINE_SIZE);
            $crate::log_internal!(log_line, $($arg)*);
        }
    });
}

#[macro_export]
macro_rules! info {
    ($($arg:tt)*) => ({
        if protocol::info::Severity::Info as u8 >= $crate::logger::LOG_LEVEL {
            let mut log_line = protocol::info::log_line(protocol::info::Severity::Info, $crate::logger::MAX_LOG_LINE_SIZE);
            $crate::log_internal!(log_line, $($arg)*);
        }
    });
}