first commit

term/ansi_windows.go

@@ -0,0 +1,35 @@
+//go:build windows
+
+package term
+
+import (
+	"os"
+	"syscall"
+	"unsafe"
+)
+
+const enableVirtualTerminalProcessing = 0x0004
+
+var (
+	kernel32           = syscall.NewLazyDLL("kernel32.dll")
+	procGetConsoleMode = kernel32.NewProc("GetConsoleMode")
+	procSetConsoleMode = kernel32.NewProc("SetConsoleMode")
+)
+
+func EnableConsoleANSI() {
+	enableHandleANSI(os.Stdout)
+	enableHandleANSI(os.Stderr)
+}
+
+func enableHandleANSI(f *os.File) {
+	if f == nil {
+		return
+	}
+	var mode uint32
+	fd := f.Fd()
+	r1, _, _ := procGetConsoleMode.Call(fd, uintptr(unsafe.Pointer(&mode)))
+	if r1 == 0 {
+		return
+	}
+	_, _, _ = procSetConsoleMode.Call(fd, uintptr(mode|enableVirtualTerminalProcessing))
+}