Add obfuscated logger.

unshell-lib/src/logger/macros.rs

@@ -0,0 +1,95 @@
+#[macro_export]
+macro_rules! debug {
+    ($fmt:tt) => {{
+        let log_result = unshell_obfuscate::format_obs!($fmt);
+
+        $crate::logger::add_record(
+            $crate::logger::LogLevel::Debug,
+            String::from(unshell_obfuscate::file_symbol!()),
+            std::time::SystemTime::now(),
+            log_result
+        );
+    }};
+    ($fmt:tt, $($arg:expr),*) => {{
+        let log_result = unshell_obfuscate::format_obs!($fmt, $($arg),*);
+
+        $crate::logger::add_record(
+            $crate::logger::LogLevel::Debug,
+            String::from(unshell_obfuscate::file_symbol!()),
+            std::time::SystemTime::now(),
+            log_result
+        );
+    }};
+}
+
+#[macro_export]
+macro_rules! info {
+    ($fmt:tt) => {{
+        let log_result = unshell_obfuscate::format_obs!($fmt);
+
+        $crate::logger::add_record(
+            $crate::logger::LogLevel::Info,
+            String::from(unshell_obfuscate::file_symbol!()),
+            std::time::SystemTime::now(),
+            log_result
+        );
+    }};
+    ($fmt:tt, $($arg:expr),*) => {{
+        let log_result = unshell_obfuscate::format_obs!($fmt, $($arg),*);
+
+        $crate::logger::add_record(
+            $crate::logger::LogLevel::Info,
+            String::from(unshell_obfuscate::file_symbol!()),
+            std::time::SystemTime::now(),
+            log_result
+        );
+    }};
+}
+
+#[macro_export]
+macro_rules! warn {
+    ($fmt:tt) => {{
+        let log_result = unshell_obfuscate::format_obs!($fmt);
+
+        $crate::logger::add_record(
+            $crate::logger::LogLevel::Warn,
+            String::from(unshell_obfuscate::file_symbol!()),
+            std::time::SystemTime::now(),
+            log_result
+        );
+    }};
+    ($fmt:tt, $($arg:expr),*) => {{
+        let log_result = unshell_obfuscate::format_obs!($fmt, $($arg),*);
+
+        $crate::logger::add_record(
+            $crate::logger::LogLevel::Warn,
+            String::from(unshell_obfuscate::file_symbol!()),
+            std::time::SystemTime::now(),
+            log_result
+        );
+    }};
+}
+
+#[macro_export]
+macro_rules! error {
+    ($fmt:tt) => {{
+        let log_result = unshell_obfuscate::format_obs!($fmt);
+
+        $crate::logger::add_record(
+            $crate::logger::LogLevel::Error,
+            String::from(unshell_obfuscate::file_symbol!()),
+            std::time::SystemTime::now(),
+            log_result
+        );
+    }};
+    ($fmt:tt, $($arg:expr),*) => {{
+        let log_result = unshell_obfuscate::format_obs!($fmt, $($arg),*);
+
+        $crate::logger::add_record(
+            $crate::logger::LogLevel::Error,
+            String::from(unshell_obfuscate::file_symbol!()),
+            std::time::SystemTime::now(),
+            log_result
+        );
+    }};
+}

unshell-lib/src/logger/mod.rs

@@ -0,0 +1,69 @@
+pub mod macros;
+mod pretty_logger;
+
+use std::time::SystemTime;
+
+pub use pretty_logger::PrettyLogger;
+
+static mut LOGGER: &dyn Logger = &DefaultLogger;
+
+#[derive(Debug)]
+pub enum LogLevel {
+    Debug,
+    Info,
+    Warn,
+    Error,
+}
+
+#[derive(Debug)]
+pub struct Record {
+    log_level: LogLevel,
+    location: String,
+    // line: u32,
+    time: SystemTime,
+    message: String,
+}
+
+pub trait Logger {
+    fn log(&self, log: Record);
+}
+
+struct DefaultLogger;
+
+impl Logger for DefaultLogger {
+    fn log(&self, _: Record) {}
+}
+
+pub fn set_logger_box(logger: Box<dyn Logger>) {
+    unsafe {
+        LOGGER = Box::leak(logger);
+    }
+}
+
+pub fn set_logger(logger: &'static dyn Logger) {
+    unsafe {
+        LOGGER = logger;
+    }
+}
+
+pub fn add_record(log_level: LogLevel, location: String, time: SystemTime, message: String) {
+    logger().log(Record {
+        log_level,
+        location,
+        time,
+        message,
+    });
+}
+
+pub fn logger<'a>() -> &'static dyn Logger {
+    unsafe { LOGGER }
+}
+
+#[allow(dead_code, improper_ctypes_definitions)]
+pub type SetupLogger = extern "C" fn(logger: &'static dyn Logger);
+
+#[unsafe(no_mangle)]
+#[allow(improper_ctypes_definitions)]
+pub extern "C" fn setup_logger(logger: &'static dyn Logger) {
+    set_logger(logger);
+}

unshell-lib/src/logger/pretty_logger.rs

@@ -0,0 +1,41 @@
+use chrono::{DateTime, Utc};
+
+use crate::logger::{LogLevel, Logger, Record};
+
+pub struct PrettyLogger;
+
+// static TRACE_COLOR: &str = "\x1b[34m";
+static DEBUG_COLOR: &str = "\x1b[36m";
+static INFO_COLOR: &str = "\x1b[32m";
+static WARN_COLOR: &str = "\x1b[33m";
+static ERROR_COLOR: &str = "\x1b[31m";
+
+static WHITE: &str = "\x1b[97m";
+static OFF_WHITE: &str = "\x1b[37m";
+static TIME_COLOR: &str = "\x1b[36m";
+static GREY: &str = "\x1b[90m";
+
+impl Logger for PrettyLogger {
+    fn log(&self, message: Record) {
+        let log_level = match message.log_level {
+            LogLevel::Debug => format!("{DEBUG_COLOR}DEBUG"),
+            LogLevel::Info => format!("{INFO_COLOR}INFO"),
+            LogLevel::Warn => format!("{WARN_COLOR}WARN"),
+            LogLevel::Error => format!("{ERROR_COLOR}ERROR"),
+        };
+
+        let date: DateTime<Utc> = message.time.into();
+        let date = date.to_rfc2822().to_string();
+
+        println!(
+            "{OFF_WHITE}[{TIME_COLOR}{}{OFF_WHITE}] {} {WHITE}{} {GREY}{}{WHITE}",
+            date, log_level, message.message, message.location
+        );
+    }
+}
+
+impl PrettyLogger {
+    pub fn init() {
+        crate::logger::set_logger_box(Box::new(PrettyLogger));
+    }
+}