use core::{fmt::Write, sync::atomic::Ordering};

use crate::{
    arch::MMArch, driver::tty::serial::serial8250::send_to_default_serial8250_port,
    filesystem::procfs::kmsg::kmsg_init, libs::printk::PrintkWriter, mm::mmio_buddy::mmio_init,
};

use super::MemoryManagementArch;

#[atomic_enum]
#[derive(PartialEq, Eq)]
pub enum MMInitStatus {
    NotInit,
    Initializing,
    Initialized,
}
/// 内存管理的初始化状态
static MM_INIT: AtomicMMInitStatus = AtomicMMInitStatus::new(MMInitStatus::NotInit);

pub unsafe fn mm_init() {
    send_to_default_serial8250_port("mm_init\n\0".as_bytes());
    PrintkWriter
        .write_fmt(format_args!("mm_init() called\n"))
        .unwrap();
    // printk_color!(GREEN, BLACK, "mm_init() called\n");

    if MM_INIT
        .compare_exchange(
            MMInitStatus::NotInit,
            MMInitStatus::Initializing,
            Ordering::SeqCst,
            Ordering::SeqCst,
        )
        .is_err()
    {
        send_to_default_serial8250_port("mm_init err\n\0".as_bytes());
        panic!("mm_init() can only be called once");
    }

    MMArch::init();

    // enable mmio
    mmio_init();
    // enable KMSG
    kmsg_init();

    MM_INIT
        .compare_exchange(
            MMInitStatus::Initializing,
            MMInitStatus::Initialized,
            Ordering::SeqCst,
            Ordering::SeqCst,
        )
        .unwrap();
    kinfo!("mm init done.");
}

/// 获取内存管理的初始化状态
pub fn mm_init_status() -> MMInitStatus {
    MM_INIT.load(Ordering::SeqCst)
}