okansi 0.3.0

An ok library for creating ANSI terminal styles
Documentation 
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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52

use std::iter::once;
use std::{ffi::OsStr, os::windows::ffi::OsStrExt, ptr::null_mut};
use winapi::um::consoleapi::{GetConsoleMode, SetConsoleMode};
use winapi::um::errhandlingapi::GetLastError;
use winapi::um::handleapi::INVALID_HANDLE_VALUE;
use winapi::um::wincon::ENABLE_VIRTUAL_TERMINAL_PROCESSING;
use winapi::um::{
	fileapi::{CreateFileW, OPEN_EXISTING},
	winnt::{FILE_SHARE_WRITE, GENERIC_READ, GENERIC_WRITE},
};

// I HATE WINDOWS

unsafe fn get_out_handle() -> *mut winapi::ctypes::c_void {
	unsafe {
		let out_name: Vec<u16> = OsStr::new("CONOUT$").encode_wide().chain(once(0)).collect();

		CreateFileW(
			out_name.as_ptr(),
			GENERIC_READ | GENERIC_WRITE,
			FILE_SHARE_WRITE,
			null_mut(),
			OPEN_EXISTING,
			0,
			null_mut(),
		)
	}
}

/// Enables support for ANSI escape codes in Windows terminals (Powershell, Command Prompt)
#[cfg(windows)]
pub fn enable_ansi_support() -> Result<(), u32> {
	unsafe {
		let h_stdout = get_out_handle();
		if h_stdout == INVALID_HANDLE_VALUE {
			return Err(GetLastError());
		}

		let mut console_mode: u32 = 0;
		if 0 == GetConsoleMode(h_stdout, &mut console_mode) {
			return Err(GetLastError());
		}

		if console_mode & ENABLE_VIRTUAL_TERMINAL_PROCESSING == 0
			&& 0 == SetConsoleMode(h_stdout, console_mode | ENABLE_VIRTUAL_TERMINAL_PROCESSING)
		{
			return Err(GetLastError());
		}
	}

	Ok(())
}