1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35

//! A Rust logger. Provides macro `debugln!()` and `println!()`.
//!
//! # Example
//! ```edition2021
//! rog::reg("main");
//! rog::debugln!("debug");
//! rog::println!("print");
//! ```

use std::collections::HashSet;
pub use std::println;

static mut C: Option<HashSet<&'static str>> = None;

/// Returns a hashset indicating which modules have been registered.
pub fn cfg() -> &'static mut HashSet<&'static str> {
    unsafe { C.get_or_insert(HashSet::new()) }
}

/// Debugs to the standard output, with a newline.
#[macro_export]
macro_rules! debugln {
    ($($arg:tt)*) => ({
        if rog::cfg().contains(module_path!()) {
            print!($($arg)*);
            print!("\n");
        }
    })
}

/// Register a series of module names, the logs in this module will be printed.
pub fn reg(module: &'static str) {
    let c = cfg();
    c.insert(module);
}