初次提交

2023-03-22 11:30:21 +08:00
parent 0c34cff145
commit 30bf0dc9d0
95 changed files with 14609 additions and 0 deletions
@@ -0,0 +1,75 @@
+//go:build windows
+// +build windows
+
+package common
+
+import (
+	"github.com/creack/pty"
+	"syscall"
+	"unsafe"
+)
+
+func SetConsoleVT() {
+	if kernel32, err := syscall.LoadDLL("kernel32.dll"); err == nil {
+		if GetStdHandle, err := kernel32.FindProc("GetStdHandle"); err == nil {
+			if GetConsoleMode, err := kernel32.FindProc("GetConsoleMode"); err == nil {
+				if SetConsoleMode, err := kernel32.FindProc("SetConsoleMode"); err == nil {
+					//仅限win10
+					v := int32(-11)
+					hand, _, _ := GetStdHandle.Call(uintptr(v))
+					if hand == ^uintptr(0) {
+						return
+					}
+
+					var dwMode int32
+					if res, _, _ := GetConsoleMode.Call(hand, uintptr(unsafe.Pointer(&dwMode))); res == 1 {
+						res, _, _ = SetConsoleMode.Call(hand, uintptr(dwMode|0x0004))
+						EnableTermVt = res == 1
+					}
+				}
+			}
+		}
+	}
+}
+
+type COORD struct {
+	X uint16
+	Y uint16
+}
+type SMALL_RECT struct {
+	Left   uint16
+	Top    uint16
+	Right  uint16
+	Bottom uint16
+}
+type CONSOLE_SCREEN_BUFFER_INFO struct {
+	Size              COORD
+	CursorPosition    COORD
+	Attributes        uint16
+	Window            SMALL_RECT
+	MaximumWindowSize COORD
+}
+
+func GetSize() (size *pty.Winsize) {
+	var csbi CONSOLE_SCREEN_BUFFER_INFO
+
+	if kernel32, err := syscall.LoadDLL("kernel32.dll"); err == nil {
+		if GetStdHandle, err := kernel32.FindProc("GetStdHandle"); err == nil {
+			if GetConsoleScreenBufferInfo, err := kernel32.FindProc("GetConsoleScreenBufferInfo"); err == nil {
+
+				//仅限win10
+				v := int32(-11)
+				hand, _, _ := GetStdHandle.Call(uintptr(v))
+				if hand == ^uintptr(0) {
+					return nil
+				}
+				if res, _, _ := GetConsoleScreenBufferInfo.Call(hand, uintptr(unsafe.Pointer(&csbi))); res == 1 {
+					size = &pty.Winsize{}
+					size.Cols = csbi.Window.Right - csbi.Window.Left + 1
+					size.Rows = csbi.Window.Bottom - csbi.Window.Top + 1
+				}
+			}
+		}
+	}
+	return
+}