Fix enabling ANSI support when stdout is redirected. (crossterm-rs#497)

quininer · Oct 10, 2020 · 4b1c857 · 4b1c857
1 parent efb06b8
commit 4b1c857
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/ansi_support.rs b/src/ansi_support.rs
@@ -1,4 +1,4 @@
-use crossterm_winapi::ConsoleMode;
+use crossterm_winapi::{ConsoleMode, Handle};
 use winapi::um::wincon::ENABLE_VIRTUAL_TERMINAL_PROCESSING;
 
 use lazy_static::lazy_static;
@@ -17,7 +17,7 @@ use crate::Result;
 pub(crate) fn set_virtual_terminal_processing(yes: bool) -> Result<()> {
     let mask = ENABLE_VIRTUAL_TERMINAL_PROCESSING;
 
-    let console_mode = ConsoleMode::new()?;
+    let console_mode = ConsoleMode::from(Handle::current_out_handle()?);
     let old_mode = console_mode.mode()?;
 
     let new_mode = if yes {