69 lines
2 KiB
Rust
69 lines
2 KiB
Rust
#[cfg(windows)]
|
|
mod windows_console {
|
|
use std::os::raw::c_void;
|
|
|
|
#[allow(non_camel_case_types)] type c_ulong = u32;
|
|
#[allow(non_camel_case_types)] type c_int = i32;
|
|
type DWORD = c_ulong;
|
|
type LPDWORD = *mut DWORD;
|
|
type HANDLE = *mut c_void;
|
|
type BOOL = c_int;
|
|
|
|
const ENABLE_VIRTUAL_TERMINAL_PROCESSING: DWORD = 0x0004;
|
|
const STD_OUTPUT_HANDLE: DWORD = 0xFFFFFFF5;
|
|
const STD_ERROR_HANDLE: DWORD = 0xFFFFFFF4;
|
|
const INVALID_HANDLE_VALUE: HANDLE = -1isize as HANDLE;
|
|
const FALSE: BOOL = 0;
|
|
const TRUE: BOOL = 1;
|
|
|
|
// This is the win32 console API, taken from the 'winapi' crate.
|
|
extern "system" {
|
|
fn GetStdHandle(nStdHandle: DWORD) -> HANDLE;
|
|
fn GetConsoleMode(hConsoleHandle: HANDLE, lpMode: LPDWORD) -> BOOL;
|
|
fn SetConsoleMode(hConsoleHandle: HANDLE, dwMode: DWORD) -> BOOL;
|
|
}
|
|
|
|
unsafe fn get_handle(handle_num: DWORD) -> Result<HANDLE, ()> {
|
|
match GetStdHandle(handle_num) {
|
|
handle if handle == INVALID_HANDLE_VALUE => Err(()),
|
|
handle => Ok(handle)
|
|
}
|
|
}
|
|
|
|
unsafe fn enable_vt(handle: HANDLE) -> Result<(), ()> {
|
|
let mut dw_mode: DWORD = 0;
|
|
if GetConsoleMode(handle, &mut dw_mode) == FALSE {
|
|
return Err(());
|
|
}
|
|
|
|
dw_mode |= ENABLE_VIRTUAL_TERMINAL_PROCESSING;
|
|
match SetConsoleMode(handle, dw_mode) {
|
|
result if result == TRUE => Ok(()),
|
|
_ => Err(())
|
|
}
|
|
}
|
|
|
|
unsafe fn enable_ascii_colors_raw() -> Result<bool, ()> {
|
|
let stdout_handle = get_handle(STD_OUTPUT_HANDLE)?;
|
|
let stderr_handle = get_handle(STD_ERROR_HANDLE)?;
|
|
|
|
enable_vt(stdout_handle)?;
|
|
if stdout_handle != stderr_handle {
|
|
enable_vt(stderr_handle)?;
|
|
}
|
|
|
|
Ok(true)
|
|
}
|
|
|
|
#[inline]
|
|
pub fn enable_ascii_colors() -> bool {
|
|
unsafe { enable_ascii_colors_raw().unwrap_or(false) }
|
|
}
|
|
}
|
|
|
|
#[cfg(not(windows))]
|
|
mod windows_console {
|
|
pub fn enable_ascii_colors() -> bool { true }
|
|
}
|
|
|
|
pub use self::windows_console::enable_ascii_colors;
|