ableos_time/ableos/src/kmain.rs

#![allow(clippy::empty_loop)]

use alloc::{fmt, format};
use shadeable::rhai_test;
use vga::{colors::Color16, writers::GraphicsWriter};

use crate::{
    logger,
    relib::network::socket::SocketReturns,
    unicode_utils,
    vga_e::{self, VGAE},
};

use log::*;

use {
    crate::{
        arch::{drivers::graphics::GraphicsBuffer, init, sloop},
        driver_traits::graphics::Graphics,
        experiments::{
            info::master,
            systeminfo::{KERNEL_VERSION, RELEASE_TYPE},
        },
        // log::LOG_STATE,
        relib::math::rand::RAND_HANDLE,
        relib::network::socket::Socket,
        scheduler::{test_fn, Thread, ThreadList},
    },
    alloc::{
        string::{String, ToString},
        vec,
    },
    lazy_static::lazy_static,
};

#[no_mangle]
#[allow(unconditional_recursion)]
pub extern "C" fn stack_overflow() -> u8 {
    stack_overflow();
    // meme number
    69 // NOTE: Any specific reason for this number aside from memes?
}

lazy_static! {
    pub static ref KEY_BUFFER: spin::Mutex<String> = spin::Mutex::new("".to_string());
    pub static ref THREAD_LIST: spin::Mutex<ThreadList> = spin::Mutex::new(vec![]);
    pub static ref TICK: spin::Mutex<u64> = spin::Mutex::new(0);
}

#[no_mangle]
pub fn kernel_main() -> ! {
    init::init();

    log::set_max_level(LevelFilter::Info);

    {
        let mut a_thread = Thread::new();

        a_thread.new_task(test_fn);
        a_thread.new_task(test_fn);
        THREAD_LIST.lock().push(a_thread);
        GraphicsBuffer::draw();
        GraphicsBuffer::hide_cursor();
    }

    crate::wasm::evaluate();
    {
        info!("{} v{}", RELEASE_TYPE, KERNEL_VERSION);
        info!(
            "Brand String: {:?}",
            master().unwrap().brand_string().unwrap()
        );
        if false {
            println!("$PINK$Hi$RED$ from$GREEN$ able!$RESET$");
            println!("$RED$hi$RESET$");
        }
    }

    {
        use crate::relib::network::socket::SimpleSock;

        let mut xyz = SimpleSock::new();

        xyz.peek();

        xyz.write(vec![0, 1, 2, 3]);

        let x = "simple 🧦".to_string().into_bytes();
        xyz.write(x);

        info!("{:?}", &xyz.read(4).unwrap());

        println!("{:?}", &xyz.peek().unwrap());

        match &xyz.peek() {
            SocketReturns::ReadOk(strr) => {
                let out = String::from_utf8_lossy(strr);

                info!("{}", out);
            }
            SocketReturns::ReadIndexOutOfBounds => todo!(),
            SocketReturns::WriteOk => todo!(),
        }
    }

    println!("$GREEN$able$RESET$@$LIGHTBLUE$stAble$ $RED$->$RESET$");

    if false {
        VGAE.lock().clear_screen(Color16::Black);

        let xyz = format!("\u{e100}ableOS");

        for x in xyz.chars().enumerate() {
            vga_e::draw_char(x.1, x.0);
        }
    }
    for y in 0..480 {
        for x in 0..640 {
            // info!("{x}");
            VGAE.lock().set_pixel(x, y, Color16::Blue)
        }
    }

    let _ = rhai_test();

    println!("hi");
    // vga_e::test_it_fucko();
    // stack_overflow();

    sloop()
}

/// called by arch specific timers to tick up all kernel related functions
pub fn tick() {
    let mut data = TICK.lock();
    *data += 1;
    RAND_HANDLE.lock().seed_entropy_timer(*data);

    // println!("{}", *data);
    crate::kernel_state::KERNEL_STATE.lock().update_state();
}

pub fn key_entropy(key: u8) {
    RAND_HANDLE.lock().seed_entropy_keyboard(key);
}