Add try_init

This commit is contained in:
nbouteme 2019-02-21 11:09:21 +01:00
parent 6973b1cf0f
commit 078d9a24a7

View File

@ -23,7 +23,7 @@
//! you can leave this line in your Unix builds too, it will simply do nothing. //! you can leave this line in your Unix builds too, it will simply do nothing.
#[cfg(windows)] #[cfg(windows)]
pub fn init() { pub fn try_init() -> Result<(), ()> {
use winapi::shared::minwindef::DWORD; use winapi::shared::minwindef::DWORD;
use winapi::um::consoleapi::{GetConsoleMode, SetConsoleMode}; use winapi::um::consoleapi::{GetConsoleMode, SetConsoleMode};
use winapi::um::processenv::GetStdHandle; use winapi::um::processenv::GetStdHandle;
@ -33,11 +33,33 @@ pub fn init() {
let console_out = unsafe { GetStdHandle(STD_OUTPUT_HANDLE) }; let console_out = unsafe { GetStdHandle(STD_OUTPUT_HANDLE) };
let mut state: DWORD = 0; let mut state: DWORD = 0;
assert_ne!(unsafe { GetConsoleMode(console_out, &mut state) }, 0); let mut ret: Result<(), _> = Ok(());
state |= ENABLE_VIRTUAL_TERMINAL_PROCESSING; unsafe {
state &= !DISABLE_NEWLINE_AUTO_RETURN; let mut e: DWORD = 0;
assert_ne!(unsafe { SetConsoleMode(console_out, state) }, 0); if GetConsoleMode(console_out, &mut state) == 0 {
ret = Err(());
}
if ret.is_ok() {
state |= ENABLE_VIRTUAL_TERMINAL_PROCESSING;
state &= !DISABLE_NEWLINE_AUTO_RETURN;
if SetConsoleMode(console_out, state) == 0 {
ret = Err(());
}
}
}
return ret;
} }
#[cfg(windows)]
pub fn init() {
assert_eq!(try_init().is_ok(), true);
}
#[cfg(not(windows))]
pub fn try_init() -> Result<(), ()> {
Ok(())
}
#[cfg(not(windows))] #[cfg(not(windows))]
pub fn init() { pub fn init() {
; ;