#![allow(clippy::needless_doctest_main)]
#[cfg(windows)]
pub fn enable_ansi_support() -> Result<(), std::io::Error> {
use std::{ffi::OsStr, iter::once, os::windows::ffi::OsStrExt};
use windows_sys::Win32::{
Foundation::INVALID_HANDLE_VALUE,
Storage::FileSystem::{
CreateFileW, FILE_GENERIC_READ, FILE_GENERIC_WRITE, FILE_SHARE_WRITE, OPEN_EXISTING,
},
System::Console::{GetConsoleMode, SetConsoleMode, ENABLE_VIRTUAL_TERMINAL_PROCESSING},
};
unsafe {
let console_out_name: Vec<u16> =
OsStr::new("CONOUT$").encode_wide().chain(once(0)).collect();
let console_handle = CreateFileW(
console_out_name.as_ptr(),
FILE_GENERIC_READ | FILE_GENERIC_WRITE,
FILE_SHARE_WRITE,
std::ptr::null(),
OPEN_EXISTING,
0,
0,
);
if console_handle == INVALID_HANDLE_VALUE {
return Err(std::io::Error::last_os_error());
}
let mut console_mode = 0;
if 0 == GetConsoleMode(console_handle, &mut console_mode) {
return Err(std::io::Error::last_os_error());
}
if console_mode & ENABLE_VIRTUAL_TERMINAL_PROCESSING == 0 {
if 0 == SetConsoleMode(
console_handle,
console_mode | ENABLE_VIRTUAL_TERMINAL_PROCESSING,
) {
return Err(std::io::Error::last_os_error());
}
}
}
Ok(())
}
#[cfg(not(windows))]
#[inline]
pub fn enable_ansi_support() -> Result<(), std::io::Error> {
Ok(())
}