| - |
#[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_ansi_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_ansi_colors() -> bool {
|
| - |
unsafe { enable_ansi_colors_raw().unwrap_or(false) }
|
| - |
}
|
| - |
}
|