mirror of https://github.com/k3s-io/k3s
37 lines
715 B
Go
37 lines
715 B
Go
|
// +build windows
|
||
|
|
|
||
|
|
package sequences
|
||
|
|
|
||
|
|
import (
|
||
|
|
"syscall"
|
||
|
|
"unsafe"
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
kernel32Dll *syscall.LazyDLL = syscall.NewLazyDLL("Kernel32.dll")
|
||
|
|
setConsoleMode *syscall.LazyProc = kernel32Dll.NewProc("SetConsoleMode")
|
||
|
|
)
|
||
|
|
|
||
|
|
func EnableVirtualTerminalProcessing(stream syscall.Handle, enable bool) error {
|
||
|
|
const ENABLE_VIRTUAL_TERMINAL_PROCESSING uint32 = 0x4
|
||
|
|
|
||
|
|
var mode uint32
|
||
|
|
err := syscall.GetConsoleMode(syscall.Stdout, &mode)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
if enable {
|
||
|
|
mode |= ENABLE_VIRTUAL_TERMINAL_PROCESSING
|
||
|
|
} else {
|
||
|
|
mode &^= ENABLE_VIRTUAL_TERMINAL_PROCESSING
|
||
|
|
}
|
||
|
|
|
||
|
|
ret, _, err := setConsoleMode.Call(uintptr(unsafe.Pointer(stream)), uintptr(mode))
|
||
|
|
if ret == 0 {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|