1 files changed, 69 insertions, 0 deletions
diff --git a/vendor/yansi/src/windows.rs b/vendor/yansi/src/windows.rs
new file mode 100644
index 000000000..324cfc39a
--- /dev/null
+++ b/vendor/yansi/src/windows.rs
@@ -0,0 +1,69 @@
+#[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;