forked from AbleOS/ableos
71 lines
2.2 KiB
Rust
71 lines
2.2 KiB
Rust
use core::sync::atomic::Ordering;
|
|
|
|
use crate::network::socket::{SimpleSock, Socket};
|
|
use crate::serial_println;
|
|
|
|
use kernel::TICK;
|
|
use lliw::{Fg, Reset};
|
|
use log::{Level, Metadata, Record};
|
|
|
|
struct SimpleLogger;
|
|
// TODO: Rebuild this to take advantage of sockets
|
|
// DETAIL: Log to a socket instead of the screen
|
|
// So that we can log in the kernel and display it in userland
|
|
impl log::Log for SimpleLogger {
|
|
fn enabled(&self, metadata: &Metadata) -> bool {
|
|
metadata.level() <= Level::Trace
|
|
}
|
|
|
|
fn log(&self, record: &Record) {
|
|
if self.enabled(record.metadata()) {
|
|
let color;
|
|
let time = TICK.load(Ordering::Relaxed) as f64;
|
|
|
|
let time_float = time;
|
|
|
|
match record.level() {
|
|
log::Level::Error => color = (Fg::Red, "$RED$"),
|
|
log::Level::Warn => color = (Fg::LightYellow, "$LIGHTYELLOW$"),
|
|
log::Level::Info => color = (Fg::LightWhite, "$LIGHTGRAY$"),
|
|
log::Level::Debug => color = (Fg::Blue, "$BLUE$"),
|
|
log::Level::Trace => color = (Fg::Yellow, "$YELLOW$"),
|
|
}
|
|
let msg = format!(
|
|
"[{}{}$RESET$][$GREEN${}$RESET$]{}\n",
|
|
color.1,
|
|
record.level(),
|
|
time_float,
|
|
record.args()
|
|
);
|
|
serial_println!(
|
|
"[{}{}{}][{}{}{}] {}",
|
|
color.0,
|
|
record.level(),
|
|
Fg::Reset,
|
|
Fg::Green,
|
|
time_float,
|
|
Reset,
|
|
record.args(),
|
|
);
|
|
|
|
let log_socket_id = SimpleSock::grab_socket("Logger".to_string());
|
|
match log_socket_id {
|
|
Some(mut log_socket_id) => {
|
|
log_socket_id.write(msg.as_bytes().to_vec());
|
|
}
|
|
None => warn!("No socket found for Logger"),
|
|
}
|
|
}
|
|
}
|
|
/// Clear the log buffer
|
|
fn flush(&self) {}
|
|
}
|
|
|
|
use log::{LevelFilter, SetLoggerError};
|
|
|
|
static LOGGER: SimpleLogger = SimpleLogger;
|
|
|
|
pub fn init() -> Result<(), SetLoggerError> {
|
|
log::set_logger(&LOGGER).map(|()| log::set_max_level(LevelFilter::Trace))
|
|
}
|