zeitfetch 0.1.16

Instantaneous snapshots of cross-platform system information
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

// Windows doesn't always have ANSI color processing enabled by default.
// This function attempts to safely force-enable it.
pub fn enforce_ansi_windows() -> bool {
    #[cfg(windows)]
    {
        if cfg!(target_os = "windows") {
            use winapi::um::consoleapi::{GetConsoleMode, SetConsoleMode};
            use winapi::um::processenv::GetStdHandle;
            use winapi::um::winbase::STD_OUTPUT_HANDLE;
            use winapi::um::wincon::ENABLE_VIRTUAL_TERMINAL_PROCESSING;

            unsafe {
                let console_handle = GetStdHandle(STD_OUTPUT_HANDLE);
                let mut current_mode: u32 = 0;

                if console_handle == std::ptr::null_mut() {
                    return false;
                }

                if 0 == GetConsoleMode(console_handle, &mut current_mode) {
                    return false;
                }

                if 0 == SetConsoleMode(
                    console_handle,
                    current_mode | ENABLE_VIRTUAL_TERMINAL_PROCESSING,
                ) {
                    return false;
                }
            }
        }
    }

    true
}