杭州明婳科技
52 lines
1.4 KiB
Go
52 lines
1.4 KiB
Go
//go:build windows
|
|
|
|
package console
|
|
|
|
import (
|
|
"fmt"
|
|
"syscall"
|
|
"unsafe"
|
|
)
|
|
|
|
var kernel32 = syscall.NewLazyDLL("kernel32.dll")
|
|
|
|
// 颜色映射：ANSI -> Windows控制台颜色
|
|
// 0 黑色, 1 蓝色, 2 绿色, 3 青色, 4 红色, 5 紫色, 6 黄色, 7 淡灰色
|
|
// 8 灰色, 9 亮蓝色, 10 亮绿色, 11 亮青色, 12 亮红色, 13 亮紫色, 14 亮黄色, 15 白色
|
|
var colorMap = map[string]uintptr{
|
|
"0;36": 11, // 青色 -> 亮青色
|
|
"0;37": 15, // 白色 -> 白色
|
|
"0;92": 10, // 亮绿色
|
|
"1;93": 14, // 亮黄色
|
|
"1;31": 12, // 亮红色
|
|
}
|
|
|
|
// printColor 彩色打印
|
|
func (c *Console) printColor(code, msg string) {
|
|
color := colorMap[code]
|
|
if color == 0 {
|
|
color = 7 // 默认淡灰色
|
|
}
|
|
|
|
proc := kernel32.NewProc("SetConsoleTextAttribute")
|
|
_, _, _ = proc.Call(uintptr(syscall.Stdout), color)
|
|
fmt.Fprintln(c.output, msg)
|
|
_, _, _ = proc.Call(uintptr(syscall.Stdout), 7) // 恢复默认颜色
|
|
}
|
|
|
|
// EnableVirtualTerminal 启用虚拟终端支持（Windows 10+）
|
|
func EnableVirtualTerminal() error {
|
|
// 尝试启用 ANSI 转义序列支持
|
|
proc := kernel32.NewProc("GetConsoleMode")
|
|
var mode uint32
|
|
_, _, err := proc.Call(uintptr(syscall.Stdout), uintptr(unsafe.Pointer(&mode)))
|
|
if err != syscall.Errno(0) {
|
|
return err
|
|
}
|
|
|
|
mode |= 0x0004 // ENABLE_VIRTUAL_TERMINAL_PROCESSING
|
|
proc = kernel32.NewProc("SetConsoleMode")
|
|
_, _, err = proc.Call(uintptr(syscall.Stdout), uintptr(mode))
|
|
return err
|
|
}
|