packer-cn/vendor/github.com/sirupsen/logrus/terminal_check_windows.go

35 lines
572 B
Go
Raw Normal View History

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