301 lines
9.5 KiB
Go
301 lines
9.5 KiB
Go
package main
|
||
|
||
import (
|
||
"log"
|
||
"unsafe"
|
||
|
||
"github.com/jchv/go-webview2"
|
||
"golang.org/x/sys/windows"
|
||
)
|
||
|
||
var (
|
||
user32 = windows.NewLazySystemDLL("user32.dll")
|
||
procFindWindowW = user32.NewProc("FindWindowW")
|
||
procFindWindowExW = user32.NewProc("FindWindowExW")
|
||
procSendMessageTimeoutW = user32.NewProc("SendMessageTimeoutW")
|
||
procSetParent = user32.NewProc("SetParent")
|
||
procGetForegroundWindow = user32.NewProc("GetForegroundWindow")
|
||
procGetWindowRect = user32.NewProc("GetWindowRect")
|
||
procGetSystemMetrics = user32.NewProc("GetSystemMetrics")
|
||
procMoveWindow = user32.NewProc("MoveWindow")
|
||
procShowWindow = user32.NewProc("ShowWindow")
|
||
procSetProcessDPIAware = user32.NewProc("SetProcessDPIAware")
|
||
procSetWindowLongPtrW = user32.NewProc("SetWindowLongPtrW")
|
||
procGetWindowLongPtrW = user32.NewProc("GetWindowLongPtrW")
|
||
procGetDpiForWindow = user32.NewProc("GetDpiForWindow")
|
||
procGetClientRect = user32.NewProc("GetClientRect")
|
||
procMessageBoxW = user32.NewProc("MessageBoxW")
|
||
procGetMessageW = user32.NewProc("GetMessageW")
|
||
procPostMessageW = user32.NewProc("PostMessageW")
|
||
procTranslateMessage = user32.NewProc("TranslateMessage")
|
||
procDispatchMessageW = user32.NewProc("DispatchMessageW")
|
||
procGetParent = user32.NewProc("GetParent")
|
||
procCallWindowProcW = user32.NewProc("CallWindowProcW")
|
||
procSetForegroundWindow = user32.NewProc("SetForegroundWindow")
|
||
procScreenToClient = user32.NewProc("ScreenToClient")
|
||
procEnumChildWindows = user32.NewProc("EnumChildWindows")
|
||
procGetWindow = user32.NewProc("GetWindow")
|
||
procSetWindowsHookExW = user32.NewProc("SetWindowsHookExW")
|
||
procCallNextHookEx = user32.NewProc("CallNextHookEx")
|
||
procWindowFromPoint = user32.NewProc("WindowFromPoint")
|
||
)
|
||
|
||
var wv webview2.WebView
|
||
var wvHwnd uintptr
|
||
var jsQueue = make(chan string, 64)
|
||
var paused int32
|
||
|
||
const (
|
||
wmEvalJS = 0x0401
|
||
wmSetHtml = 0x0402
|
||
)
|
||
|
||
const (
|
||
gwlStyle = uintptr(0xFFFFFFF0) // GWL_STYLE = -16
|
||
gwlExStyle = uintptr(0xFFFFFFFEC) // GWL_EXSTYLE = -20
|
||
wsPopup = uintptr(0x80000000)
|
||
wsVisible = uintptr(0x10000000)
|
||
wsChild = uintptr(0x02000000)
|
||
wsSizebox = uintptr(0x00040000)
|
||
wsMaxbox = uintptr(0x00010000)
|
||
)
|
||
|
||
const gwlpWndProc = ^uintptr(3) // GWLP_WNDPROC = -4
|
||
|
||
const (
|
||
wsExTransparent = uintptr(0x00000020)
|
||
wsExLayered = uintptr(0x00080000)
|
||
wsExToolwindow = uintptr(0x00000080)
|
||
)
|
||
|
||
const (
|
||
gwChild = 5 // GW_CHILD
|
||
gwHwndNext = 2 // GW_HWNDNEXT
|
||
)
|
||
|
||
var htmlQueue = make(chan string, 1)
|
||
|
||
// overlay 窗口子类化：动态 hit test
|
||
var (
|
||
overlayOldWndProc uintptr
|
||
overlayWndProcCb uintptr
|
||
overlayHitRect struct{ Left, Top, Right, Bottom int32 }
|
||
overlayHasHitRect bool
|
||
childOldWndProcs = make(map[uintptr]uintptr) // 子窗口 hwnd -> 旧 wndproc
|
||
)
|
||
|
||
// overlayWndProc 拦截 WM_NCHITTEST：交互区域返回 HTCLIENT，其余 HTTRANSPARENT（穿透）
|
||
// 注意：由于 overlay 在桌面图标层之下，此 wndproc 主要作为备用
|
||
func overlayWndProc(hwnd, msg, wp, lp uintptr) uintptr {
|
||
if msg == 0x0084 && overlayHasHitRect { // WM_NCHITTEST
|
||
x := int32(int16(lp & 0xFFFF))
|
||
y := int32(int16(lp >> 16))
|
||
pt := struct{ X, Y int32 }{x, y}
|
||
procScreenToClient.Call(wvHwnd, uintptr(unsafe.Pointer(&pt)))
|
||
if pt.X >= overlayHitRect.Left && pt.X <= overlayHitRect.Right &&
|
||
pt.Y >= overlayHitRect.Top && pt.Y <= overlayHitRect.Bottom {
|
||
return uintptr(1) // HTCLIENT
|
||
}
|
||
return 0xFFFFFFFF // HTTRANSPARENT
|
||
}
|
||
// 查找该窗口的旧 wndproc
|
||
oldProc := overlayOldWndProc
|
||
if v, ok := childOldWndProcs[hwnd]; ok {
|
||
oldProc = v
|
||
}
|
||
r, _, _ := procCallWindowProcW.Call(oldProc, hwnd, msg, wp, lp)
|
||
return r
|
||
}
|
||
|
||
// installMouseHook 安装 WH_MOUSE_LL 全局钩子拦截刷新按钮点击
|
||
// overlay 在 WorkerW 中，位于桌面图标层之下，WM_NCHITTEST 无法到达
|
||
// 因此用低级鼠标钩子直接在 Go 层处理点击
|
||
var llMouseHook uintptr
|
||
var llMouseHookCb uintptr // 防止 GC 回收回调函数
|
||
var progmanHwnd uintptr // Progman 句柄，用于判断点击是否在桌面
|
||
var btnHoverActive bool // 按钮悬停状态，避免重复推送 JS
|
||
|
||
// isMouseInBtnRect 检查鼠标是否在按钮 hit rect 内（屏幕坐标转 wvHwnd 客户端坐标）
|
||
func isMouseInBtnRect(screenX, screenY int32) bool {
|
||
if !overlayHasHitRect {
|
||
return false
|
||
}
|
||
pt := struct{ X, Y int32 }{screenX, screenY}
|
||
procScreenToClient.Call(wvHwnd, uintptr(unsafe.Pointer(&pt)))
|
||
return pt.X >= overlayHitRect.Left && pt.X <= overlayHitRect.Right &&
|
||
pt.Y >= overlayHitRect.Top && pt.Y <= overlayHitRect.Bottom
|
||
}
|
||
|
||
// isDesktopAtPoint 检查屏幕坐标处是否为桌面背景（无其他窗口遮挡）
|
||
func isDesktopAtPoint(screenX, screenY int32) bool {
|
||
ptPacked := uintptr(screenX) | (uintptr(uint32(screenY)) << 32)
|
||
hwndAt, _, _ := procWindowFromPoint.Call(ptPacked)
|
||
for h := hwndAt; h != 0; h, _, _ = procGetParent.Call(h) {
|
||
if h == progmanHwnd {
|
||
return true
|
||
}
|
||
}
|
||
return false
|
||
}
|
||
|
||
func installMouseHook() {
|
||
llMouseHookCb = windows.NewCallback(func(nCode, wp, lp uintptr) uintptr {
|
||
if int32(nCode) >= 0 {
|
||
type msll struct {
|
||
X, Y int32
|
||
MouseData uint32
|
||
Flags uint32
|
||
Time uint32
|
||
DwExtraInfo uintptr
|
||
}
|
||
s := (*msll)(unsafe.Pointer(lp))
|
||
|
||
switch wp {
|
||
case 0x0200: // WM_MOUSEMOVE — 悬停检测
|
||
inBtn := isMouseInBtnRect(s.X, s.Y)
|
||
if inBtn && !btnHoverActive {
|
||
btnHoverActive = true
|
||
evalJS("document.querySelector('.knowledge-refresh-btn')?.classList.add('hover-active')")
|
||
} else if !inBtn && btnHoverActive {
|
||
btnHoverActive = false
|
||
evalJS("document.querySelector('.knowledge-refresh-btn')?.classList.remove('hover-active')")
|
||
}
|
||
|
||
case 0x0201: // WM_LBUTTONDOWN — 点击
|
||
if !isDesktopAtPoint(s.X, s.Y) {
|
||
break
|
||
}
|
||
if isMouseInBtnRect(s.X, s.Y) {
|
||
pushKnowledgeLoading("")
|
||
go fetchAndPushKnowledge()
|
||
return 1
|
||
}
|
||
}
|
||
}
|
||
r, _, _ := procCallNextHookEx.Call(llMouseHook, nCode, wp, lp)
|
||
return r
|
||
})
|
||
var llMouseHookVal uintptr
|
||
llMouseHookVal, _, _ = procSetWindowsHookExW.Call(
|
||
14, // WH_MOUSE_LL
|
||
llMouseHookCb, 0, 0,
|
||
)
|
||
llMouseHook = llMouseHookVal
|
||
log.Printf("mouse hook installed, progman=0x%x, hook=0x%x", progmanHwnd, llMouseHookVal)
|
||
}
|
||
|
||
// subclassOverlayWindows 子类化 WebView2 父窗口及其所有子窗口
|
||
func subclassOverlayWindows() {
|
||
overlayWndProcCb = windows.NewCallback(overlayWndProc)
|
||
oldProc, _, _ := procSetWindowLongPtrW.Call(wvHwnd, gwlpWndProc, overlayWndProcCb)
|
||
overlayOldWndProc = oldProc
|
||
subclassChildTree(wvHwnd)
|
||
}
|
||
|
||
// subclassChildTree 递归子类化 parent 下的所有子窗口
|
||
func subclassChildTree(parent uintptr) {
|
||
child, _, _ := procGetWindow.Call(parent, uintptr(gwChild))
|
||
for child != 0 {
|
||
old, _, _ := procSetWindowLongPtrW.Call(child, gwlpWndProc, overlayWndProcCb)
|
||
childOldWndProcs[child] = old
|
||
subclassChildTree(child)
|
||
child, _, _ = procGetWindow.Call(child, uintptr(gwHwndNext))
|
||
}
|
||
}
|
||
|
||
var (
|
||
classProgman = uintptr(unsafe.Pointer(windows.StringToUTF16Ptr("Progman")))
|
||
classShellDefView = uintptr(unsafe.Pointer(windows.StringToUTF16Ptr("SHELLDLL_DefView")))
|
||
classWorkerW = uintptr(unsafe.Pointer(windows.StringToUTF16Ptr("WorkerW")))
|
||
)
|
||
|
||
func evalJS(js string) {
|
||
select {
|
||
case jsQueue <- js:
|
||
log.Println("evalJS queued:", js[:min(len(js), 60)])
|
||
default:
|
||
log.Println("jsQueue full, dropping eval")
|
||
}
|
||
if wvHwnd != 0 {
|
||
procPostMessageW.Call(wvHwnd, wmEvalJS, 0, 0)
|
||
}
|
||
}
|
||
|
||
func findWorkerW() uintptr {
|
||
progman, _, _ := procFindWindowW.Call(
|
||
classProgman, 0)
|
||
if progman == 0 {
|
||
log.Println("findWorkerW: Progman not found")
|
||
return 0
|
||
}
|
||
progmanHwnd = progman // 保存 Progman 句柄供鼠标钩子判断
|
||
// 发送 0x052C 触发 Progman 创建 WorkerW
|
||
var result uintptr
|
||
procSendMessageTimeoutW.Call(progman, 0x052C, 0, 0, 0x0000, 1000, uintptr(unsafe.Pointer(&result)))
|
||
|
||
// 方法1: Progman 下找 SHELLDLL_DefView，再找其后的 WorkerW
|
||
shellDefView, _, _ := procFindWindowExW.Call(progman, 0, classShellDefView, 0)
|
||
if shellDefView != 0 {
|
||
ww, _, _ := procFindWindowExW.Call(progman, shellDefView, classWorkerW, 0)
|
||
if ww != 0 {
|
||
log.Printf("findWorkerW: 方法1成功, WorkerW=0x%x", ww)
|
||
return ww
|
||
}
|
||
}
|
||
|
||
// 方法2: 遍历顶层 WorkerW，找到含 SHELLDLL_DefView 的那个，取它后面的 WorkerW
|
||
var prev uintptr
|
||
for {
|
||
ww, _, _ := procFindWindowExW.Call(0, prev, classWorkerW, 0)
|
||
if ww == 0 {
|
||
break
|
||
}
|
||
child, _, _ := procFindWindowExW.Call(ww, 0, classShellDefView, 0)
|
||
if child != 0 {
|
||
// 这个 WorkerW 包含 SHELLDLL_DefView，找它后面的 WorkerW
|
||
next, _, _ := procFindWindowExW.Call(0, ww, classWorkerW, 0)
|
||
if next != 0 {
|
||
log.Printf("findWorkerW: 方法2成功, WorkerW=0x%x (after 0x%x)", next, ww)
|
||
return next
|
||
}
|
||
}
|
||
prev = ww
|
||
}
|
||
|
||
// 方法3: 遍历顶层 WorkerW，找任意不含 SHELLDLL_DefView 的可见 WorkerW
|
||
prev = 0
|
||
for {
|
||
ww, _, _ := procFindWindowExW.Call(0, prev, classWorkerW, 0)
|
||
if ww == 0 {
|
||
break
|
||
}
|
||
child, _, _ := procFindWindowExW.Call(ww, 0, classShellDefView, 0)
|
||
if child == 0 {
|
||
log.Printf("findWorkerW: 方法3成功, WorkerW=0x%x", ww)
|
||
return ww
|
||
}
|
||
prev = ww
|
||
}
|
||
|
||
// 方法4: 兜底，Progman 下的第一个 WorkerW
|
||
ww, _, _ := procFindWindowExW.Call(progman, 0, classWorkerW, 0)
|
||
if ww != 0 {
|
||
log.Printf("findWorkerW: 方法4(兜底), WorkerW=0x%x", ww)
|
||
}
|
||
return ww
|
||
}
|
||
|
||
func getDPI(hwnd uintptr) int {
|
||
dpi, _, _ := procGetDpiForWindow.Call(hwnd)
|
||
if dpi == 0 {
|
||
return 96
|
||
}
|
||
return int(dpi)
|
||
}
|
||
|
||
func getScreenSize() (int32, int32) {
|
||
w, _, _ := procGetSystemMetrics.Call(0)
|
||
h, _, _ := procGetSystemMetrics.Call(1)
|
||
return int32(w), int32(h)
|
||
}
|