k3s/vendor/github.com/sirupsen/logrus/terminal_check_windows.go

35 lines
572 B
Go
Raw Normal View History

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