blob: ef18d8f978762602a46a3d90f5ce9bd6c1bb4fff [file] [log] [blame]
khenaidooab1f7bd2019-11-14 14:00:27 -05001// +build windows
2
3package sequences
4
5import (
6 "syscall"
7 "unsafe"
8)
9
10var (
11 kernel32Dll *syscall.LazyDLL = syscall.NewLazyDLL("Kernel32.dll")
12 setConsoleMode *syscall.LazyProc = kernel32Dll.NewProc("SetConsoleMode")
13)
14
15func EnableVirtualTerminalProcessing(stream syscall.Handle, enable bool) error {
16 const ENABLE_VIRTUAL_TERMINAL_PROCESSING uint32 = 0x4
17
18 var mode uint32
19 err := syscall.GetConsoleMode(syscall.Stdout, &mode)
20 if err != nil {
21 return err
22 }
23
24 if enable {
25 mode |= ENABLE_VIRTUAL_TERMINAL_PROCESSING
26 } else {
27 mode &^= ENABLE_VIRTUAL_TERMINAL_PROCESSING
28 }
29
30 ret, _, err := setConsoleMode.Call(uintptr(unsafe.Pointer(stream)), uintptr(mode))
31 if ret == 0 {
32 return err
33 }
34
35 return nil
36}