logrus/terminal_check_windows.go

35 lines
572 B
Go
Raw Normal View History

2018-09-16 11:14:59 +03:00
// +build !appengine,!js,windows
2018-04-03 05:40:58 +03:00
package logrus
import (
"io"
"os"
"syscall"
2019-05-18 12:44:18 +03:00
sequences "github.com/konsorten/go-windows-terminal-sequences"
2018-04-03 05:40:58 +03:00
)
2019-05-18 12:44:18 +03:00
func initTerminal(w io.Writer) {
switch v := w.(type) {
case *os.File:
sequences.EnableVirtualTerminalProcessing(syscall.Handle(v.Fd()), true)
}
}
2018-04-03 05:40:58 +03:00
func checkIfTerminal(w io.Writer) bool {
2019-05-18 12:44:18 +03:00
var ret bool
2018-04-03 05:40:58 +03:00
switch v := w.(type) {
case *os.File:
var mode uint32
err := syscall.GetConsoleMode(syscall.Handle(v.Fd()), &mode)
2019-05-18 12:44:18 +03:00
ret = (err == nil)
2018-04-03 05:40:58 +03:00
default:
2019-05-18 12:44:18 +03:00
ret = false
}
if ret {
initTerminal(w)
2018-04-03 05:40:58 +03:00
}
2019-05-18 12:44:18 +03:00
return ret
2018-04-03 05:40:58 +03:00
}