Use correct output when enabling ANSI colors on Windows (see #39)

This commit is contained in:
Christian Rocha
2020-12-30 22:08:38 -05:00
parent c4aeadd762
commit 64da3bcf7a
3 changed files with 12 additions and 4 deletions

View File

@@ -3,6 +3,7 @@
package tea
import (
"io"
"os"
"golang.org/x/sys/windows"
@@ -10,8 +11,13 @@ import (
// enableAnsiColors enables support for ANSI color sequences in Windows
// default console. Note that this only works with Windows 10.
func enableAnsiColors() {
stdout := windows.Handle(os.Stdout.Fd())
func enableAnsiColors(w io.Writer) {
f, ok := w.(*os.File)
if !ok {
return
}
stdout := windows.Handle(f.Fd())
var originalMode uint32
windows.GetConsoleMode(stdout, &originalMode)