zenity/dlgs_windows.go

594 lines
16 KiB
Go
Raw Normal View History

2018-01-25 10:37:06 -05:00
// +build windows,!linux,!darwin,!js
package dlgs
import (
"strings"
"syscall"
"unicode/utf16"
"unsafe"
)
var (
user32 = syscall.NewLazyDLL("user32.dll")
gdi32 = syscall.NewLazyDLL("gdi32.dll")
comdlg32 = syscall.NewLazyDLL("comdlg32.dll")
shell32 = syscall.NewLazyDLL("shell32.dll")
kernel32 = syscall.NewLazyDLL("kernel32.dll")
messageBoxW = user32.NewProc("MessageBoxW")
createWindowExW = user32.NewProc("CreateWindowExW")
defWindowProcW = user32.NewProc("DefWindowProcW")
destroyWindowW = user32.NewProc("DestroyWindow")
dispatchMessageW = user32.NewProc("DispatchMessageW")
getMessageW = user32.NewProc("GetMessageW")
sendMessageW = user32.NewProc("SendMessageW")
postQuitMessageW = user32.NewProc("PostQuitMessage")
registerClassExW = user32.NewProc("RegisterClassExW")
2018-03-10 16:29:31 -05:00
unregisterClassW = user32.NewProc("UnregisterClassW")
2018-01-25 10:37:06 -05:00
translateMessageW = user32.NewProc("TranslateMessage")
setWindowTextW = user32.NewProc("SetWindowTextW")
getWindowTextLengthW = user32.NewProc("GetWindowTextLengthW")
getWindowTextW = user32.NewProc("GetWindowTextW")
getWindowLongW = user32.NewProc("GetWindowLongW")
setWindowLongW = user32.NewProc("SetWindowLongW")
getWindowRectW = user32.NewProc("GetWindowRect")
setWindowPosW = user32.NewProc("SetWindowPos")
showWindowW = user32.NewProc("ShowWindow")
updateWindowW = user32.NewProc("UpdateWindow")
isDialogMessageW = user32.NewProc("IsDialogMessageW")
getSystemMetricsW = user32.NewProc("GetSystemMetrics")
systemParametersInfoW = user32.NewProc("SystemParametersInfoW")
createFontIndirectW = gdi32.NewProc("CreateFontIndirectW")
getOpenFileNameW = comdlg32.NewProc("GetOpenFileNameW")
chooseColorW = comdlg32.NewProc("ChooseColorW")
shBrowseForFolderW = shell32.NewProc("SHBrowseForFolderW")
shGetPathFromIDListW = shell32.NewProc("SHGetPathFromIDListW")
getModuleHandleW = kernel32.NewProc("GetModuleHandleW")
)
const (
mbOk = 0x00000000
mbYesNo = 0x00000004
mbDefaultIcon1 = 0x00000000
mbDefaultIcon2 = 0x00000100
mbIconInfo = 0x00000040
mbIconWarning = 0x00000030
mbIconError = 0x00000010
mbIconQuestion = 0x00000020
idOk = 1
idYes = 6
swShow = 5
swShowNormal = 1
swUseDefault = 0x80000000
swpNoZOrder = 0x0004
swpNoSize = 0x0001
smCxScreen = 0
smCyScreen = 1
wsThickFrame = 0x00040000
wsSysMenu = 0x00080000
wsBorder = 0x00800000
wsCaption = 0x00C00000
wsChild = 0x40000000
wsVisible = 0x10000000
wsMaximizeBox = 0x00010000
wsMinimizeBox = 0x00020000
wsTabStop = 0x00010000
wsGroup = 0x00020000
wsOverlappedWindow = 0x00CF0000
wsExClientEdge = 0x00000200
wmCreate = 0x0001
wmDestroy = 0x0002
wmClose = 0x0010
wmCommand = 0x0111
wmSetFont = 0x0030
wmKeydown = 0x0100
wmInitDialog = 0x0110
ofnAllowMultiSelect = 0x00000200
ofnExplorer = 0x00080000
ofnFileMustExist = 0x00001000
ofnHideReadOnly = 0x00000004
ofnOverwriteprompt = 0x00000002
esPassword = 0x0020
esAutoVScroll = 0x0040
esAutoHScroll = 0x0080
bifEditBox = 0x00000010
bifNewDialogStyle = 0x00000040
ccRgbInit = 0x00000001
ccFullOpen = 0x00000002
lbAddString = 0x0180
lbGetCurSel = 0x0188
lbGetSelCount = 0x0190
lbGetSelItems = 0x0191
lbGetItemData = 0x0199
lbSetItemData = 0x019A
lbSeparator = "LB_SEP"
lbsExtendedsel = 0x0800
dtsUpdown = 0x0001
dtsShowNone = 0x0002
dtsShortDateFormat = 0x0000
dtsLongDateFormat = 0x0004
dtmFirst = 0x1000
dtmGetSystemTime = dtmFirst + 1
dtmSetSystemTime = dtmFirst + 2
gdtError = -1
gdtValid = 0
gdtNone = 1
vkEscape = 0x1B
enUpdate = 0x0400
bsPushButton = 0
colorWindow = 5
spiGetNonClientMetrics = 0x0029
gwlStyle = -16
maxPath = 260
)
// wndClassExW https://msdn.microsoft.com/en-us/library/windows/desktop/ms633577.aspx
type wndClassExW struct {
size uint32
style uint32
wndProc uintptr
clsExtra int32
wndExtra int32
instance syscall.Handle
icon syscall.Handle
cursor syscall.Handle
background syscall.Handle
menuName *uint16
className *uint16
iconSm syscall.Handle
}
// msgW https://msdn.microsoft.com/en-us/library/windows/desktop/ms644958.aspx
type msgW struct {
hwnd syscall.Handle
message uint32
wParam uintptr
lParam uintptr
time uint32
pt pointW
}
// nonClientMetricsW https://msdn.microsoft.com/en-us/library/windows/desktop/ff729175.aspx
type nonClientMetricsW struct {
cbSize uint32
iBorderWidth int32
iScrollWidth int32
iScrollHeight int32
iCaptionWidth int32
iCaptionHeight int32
lfCaptionFont logfontW
iSmCaptionWidth int32
iSmCaptionHeight int32
lfSmCaptionFont logfontW
iMenuWidth int32
iMenuHeight int32
lfMenuFont logfontW
lfStatusFont logfontW
lfMessageFont logfontW
}
// logfontW https://msdn.microsoft.com/en-us/library/windows/desktop/dd145037.aspx
type logfontW struct {
lfHeight int32
lfWidth int32
lfEscapement int32
lfOrientation int32
lfWeight int32
lfItalic byte
lfUnderline byte
lfStrikeOut byte
lfCharSet byte
lfOutPrecision byte
lfClipPrecision byte
lfQuality byte
lfPitchAndFamily byte
lfFaceName [32]uint16
}
type pointW struct {
x, y int32
}
type rectW struct {
left int32
top int32
right int32
bottom int32
}
// openfilenameW https://msdn.microsoft.com/en-us/library/windows/desktop/ms646839.aspx
type openfilenameW struct {
lStructSize uint32
hwndOwner syscall.Handle
hInstance syscall.Handle
lpstrFilter *uint16
lpstrCustomFilter *uint16
nMaxCustFilter uint32
nFilterIndex uint32
lpstrFile *uint16
nMaxFile uint32
lpstrFileTitle *uint16
nMaxFileTitle uint32
lpstrInitialDir *uint16
lpstrTitle *uint16
flags uint32
nFileOffset uint16
nFileExtension uint16
lpstrDefExt *uint16
lCustData uintptr
lpfnHook syscall.Handle
lpTemplateName *uint16
pvReserved unsafe.Pointer
dwReserved uint32
flagsEx uint32
}
// browseinfoW http://msdn.microsoft.com/en-us/library/windows/desktop/bb773205.aspx
type browseinfoW struct {
owner syscall.Handle
root *uint16
displayName *uint16
title *uint16
flags uint32
callbackFunc uintptr
lParam uintptr
image int32
}
// choosecolorW https://msdn.microsoft.com/en-us/library/windows/desktop/ms646830.aspx
type choosecolorW struct {
lStructSize uint32
hwndOwner syscall.Handle
hInstance syscall.Handle
rgbResult uint32
lpCustColors *uint32
flags uint32
lCustData uintptr
lpfnHook uintptr
lpTemplateName *uint16
}
// systemtimeW https://msdn.microsoft.com/en-us/library/windows/desktop/ms724950.aspx
type systemtimeW struct {
wYear uint16
wMonth uint16
wDayOfWeek uint16
wDay uint16
wHour uint16
wMinute uint16
wSecond uint16
wMilliseconds uint16
}
func messageBox(title, text string, flags int) int {
ret, _, _ := messageBoxW.Call(0, uintptr(unsafe.Pointer(syscall.StringToUTF16Ptr(text))),
uintptr(unsafe.Pointer(syscall.StringToUTF16Ptr(title))), uintptr(uint(flags)))
return int(ret)
}
func getModuleHandle() (syscall.Handle, error) {
ret, _, err := getModuleHandleW.Call(uintptr(0))
if ret == 0 {
return 0, err
}
return syscall.Handle(ret), nil
}
func createWindow(exStyle uint64, className, windowName string, style uint64, x, y, width, height int64,
parent, menu, instance syscall.Handle) (syscall.Handle, error) {
ret, _, err := createWindowExW.Call(uintptr(exStyle), uintptr(unsafe.Pointer(syscall.StringToUTF16Ptr(className))),
uintptr(unsafe.Pointer(syscall.StringToUTF16Ptr(windowName))), uintptr(style), uintptr(x), uintptr(y),
uintptr(width), uintptr(height), uintptr(parent), uintptr(menu), uintptr(instance), uintptr(0))
if ret == 0 {
return 0, err
}
return syscall.Handle(ret), nil
}
func destroyWindow(hwnd syscall.Handle) error {
ret, _, err := destroyWindowW.Call(uintptr(hwnd))
if ret == 0 {
return err
}
return nil
}
func defWindowProc(hwnd syscall.Handle, msg uint32, wparam, lparam uintptr) uintptr {
ret, _, _ := defWindowProcW.Call(uintptr(hwnd), uintptr(msg), uintptr(wparam), uintptr(lparam))
return uintptr(ret)
}
func registerClassEx(wcx *wndClassExW) (uint16, error) {
ret, _, err := registerClassExW.Call(uintptr(unsafe.Pointer(wcx)))
if ret == 0 {
return 0, err
}
return uint16(ret), nil
}
2018-03-10 16:29:31 -05:00
func unregisterClass(className string, instance syscall.Handle) bool {
ret, _, _ := unregisterClassW.Call(uintptr(unsafe.Pointer(syscall.StringToUTF16Ptr(className))), uintptr(instance))
return ret != 0
}
2018-01-25 10:37:06 -05:00
func getMessage(msg *msgW, hwnd syscall.Handle, msgFilterMin, msgFilterMax uint32) (bool, error) {
ret, _, err := getMessageW.Call(uintptr(unsafe.Pointer(msg)), uintptr(hwnd), uintptr(msgFilterMin), uintptr(msgFilterMax))
if int32(ret) == -1 {
return false, err
}
return int32(ret) != 0, nil
}
func sendMessage(hwnd syscall.Handle, msg uint32, wparam, lparam uintptr) uintptr {
ret, _, _ := sendMessageW.Call(uintptr(hwnd), uintptr(msg), wparam, lparam, 0, 0)
return ret
}
func dispatchMessage(msg *msgW) {
dispatchMessageW.Call(uintptr(unsafe.Pointer(msg)))
}
func postQuitMessage(exitCode int32) {
postQuitMessageW.Call(uintptr(exitCode))
}
func translateMessage(msg *msgW) {
translateMessageW.Call(uintptr(unsafe.Pointer(msg)))
}
func setWindowText(hwnd syscall.Handle, text string) {
setWindowTextW.Call(uintptr(hwnd), uintptr(unsafe.Pointer(syscall.StringToUTF16Ptr(text))))
}
func getWindowTextLength(hwnd syscall.Handle) int {
ret, _, _ := getWindowTextLengthW.Call(uintptr(hwnd))
return int(ret)
}
func getWindowText(hwnd syscall.Handle) string {
textLen := getWindowTextLength(hwnd) + 1
buf := make([]uint16, textLen)
getWindowTextW.Call(uintptr(hwnd), uintptr(unsafe.Pointer(&buf[0])), uintptr(textLen))
return syscall.UTF16ToString(buf)
}
func systemParametersInfo(uiAction, uiParam uint32, pvParam unsafe.Pointer, fWinIni uint32) bool {
ret, _, _ := systemParametersInfoW.Call(uintptr(uiAction), uintptr(uiParam), uintptr(pvParam), uintptr(fWinIni), 0, 0)
return int32(ret) != 0
}
func createFontIndirect(lplf *logfontW) uintptr {
ret, _, _ := createFontIndirectW.Call(uintptr(unsafe.Pointer(lplf)), 0, 0)
return uintptr(ret)
}
func getWindowLong(hwnd syscall.Handle, index int32) int32 {
ret, _, _ := getWindowLongW.Call(uintptr(hwnd), uintptr(index), 0)
return int32(ret)
}
func setWindowLong(hwnd syscall.Handle, index, value int32) int32 {
ret, _, _ := setWindowLongW.Call(uintptr(hwnd), uintptr(index), uintptr(value))
return int32(ret)
}
func getWindowRect(hwnd syscall.Handle, rect *rectW) bool {
ret, _, _ := getWindowRectW.Call(uintptr(hwnd), uintptr(unsafe.Pointer(rect)), 0)
return ret != 0
}
func setWindowPos(hwnd, hwndInsertAfter syscall.Handle, x, y, width, height int32, flags uint32) bool {
ret, _, _ := setWindowPosW.Call(uintptr(hwnd), uintptr(hwndInsertAfter),
uintptr(x), uintptr(y), uintptr(width), uintptr(height), uintptr(flags), 0, 0)
return ret != 0
}
func showWindow(hwnd syscall.Handle, nCmdShow int32) bool {
ret, _, _ := showWindowW.Call(uintptr(hwnd), uintptr(nCmdShow), 0)
return ret != 0
}
func updateWindow(hwnd syscall.Handle) bool {
ret, _, _ := updateWindowW.Call(uintptr(hwnd), 0, 0)
return ret != 0
}
func isDialogMessage(hwnd syscall.Handle, msg *msgW) bool {
ret, _, _ := isDialogMessageW.Call(uintptr(hwnd), uintptr(unsafe.Pointer(msg)), 0)
return ret != 0
}
func getSystemMetrics(nindex int32) int32 {
ret, _, _ := getSystemMetricsW.Call(uintptr(nindex), 0, 0)
return int32(ret)
}
func getOpenFileName(lpofn *openfilenameW) bool {
ret, _, _ := getOpenFileNameW.Call(uintptr(unsafe.Pointer(lpofn)), 0, 0)
return ret != 0
}
func shBrowseForFolder(lpbi *browseinfoW) uintptr {
ret, _, _ := shBrowseForFolderW.Call(uintptr(unsafe.Pointer(lpbi)), 0, 0)
return ret
}
func shGetPathFromIDList(pidl uintptr, pszPath *uint16) bool {
ret, _, _ := shGetPathFromIDListW.Call(pidl, uintptr(unsafe.Pointer(pszPath)), 0)
return ret != 0
}
func chooseColor(lpcc *choosecolorW) bool {
ret, _, _ := chooseColorW.Call(uintptr(unsafe.Pointer(lpcc)), 0, 0)
return ret != 0
}
func centerWindow(hwnd syscall.Handle) {
var rc rectW
getWindowRect(hwnd, &rc)
xPos := (getSystemMetrics(smCxScreen) - (rc.right - rc.left)) / 2
yPos := (getSystemMetrics(smCyScreen) - (rc.bottom - rc.top)) / 2
setWindowPos(hwnd, 0, xPos, yPos, 0, 0, swpNoZOrder|swpNoSize)
}
func getMessageFont() uintptr {
var metrics nonClientMetricsW
metrics.cbSize = uint32(unsafe.Sizeof(metrics))
systemParametersInfo(spiGetNonClientMetrics, uint32(unsafe.Sizeof(metrics)), unsafe.Pointer(&metrics), 0)
return createFontIndirect(&metrics.lfMessageFont)
}
func registerClass(className string, instance syscall.Handle, fn interface{}) error {
var wcx wndClassExW
wcx.size = uint32(unsafe.Sizeof(wcx))
wcx.wndProc = syscall.NewCallback(fn)
wcx.instance = instance
wcx.background = colorWindow + 1
wcx.className = syscall.StringToUTF16Ptr(className)
_, err := registerClassEx(&wcx)
return err
}
func messageLoop(hwnd syscall.Handle) error {
for {
msg := msgW{}
gotMessage, err := getMessage(&msg, 0, 0, 0)
if err != nil {
return err
}
if gotMessage {
if !isDialogMessage(hwnd, &msg) {
translateMessage(&msg)
dispatchMessage(&msg)
}
} else {
break
}
}
return nil
}
func utf16PtrFromString(s string) *uint16 {
b := utf16.Encode([]rune(s))
return &b[0]
}
func stringFromUtf16Ptr(p *uint16) string {
b := *(*[maxPath]uint16)(unsafe.Pointer(p))
r := utf16.Decode(b[:])
return strings.Trim(string(r), "\x00")
}
// editBox displays textedit/inputbox dialog.
func editBox(title, text, defaultText, className string, password bool) (string, bool, error) {
var out string
var hwndEdit syscall.Handle
instance, err := getModuleHandle()
if err != nil {
return out, false, err
}
fn := func(hwnd syscall.Handle, msg uint32, wparam, lparam uintptr) uintptr {
switch msg {
case wmClose:
destroyWindow(hwnd)
case wmDestroy:
postQuitMessage(0)
case wmKeydown:
if wparam == vkEscape {
destroyWindow(hwnd)
}
case wmCommand:
if wparam == 100 {
out = getWindowText(hwndEdit)
destroyWindow(hwnd)
} else if wparam == 110 {
destroyWindow(hwnd)
}
default:
ret := defWindowProc(hwnd, msg, wparam, lparam)
return ret
}
return 0
}
err = registerClass(className, instance, fn)
if err != nil {
return out, false, err
}
2018-03-10 16:29:31 -05:00
defer unregisterClass(className, instance)
2018-01-25 10:37:06 -05:00
hwnd, _ := createWindow(0, className, title, wsOverlappedWindow, swUseDefault, swUseDefault, 235, 140, 0, 0, instance)
hwndText, _ := createWindow(0, "STATIC", text, wsChild|wsVisible, 10, 10, 200, 16, hwnd, 0, instance)
flags := wsBorder | wsChild | wsVisible | wsGroup | wsTabStop | esAutoHScroll
if password {
flags |= esPassword
}
hwndEdit, _ = createWindow(wsExClientEdge, "EDIT", defaultText, uint64(flags), 10, 30, 200, 24, hwnd, 0, instance)
hwndOK, _ := createWindow(wsExClientEdge, "BUTTON", "OK", wsChild|wsVisible|bsPushButton|wsGroup|wsTabStop, 10, 65, 90, 24, hwnd, 100, instance)
hwndCancel, _ := createWindow(wsExClientEdge, "BUTTON", "Cancel", wsChild|wsVisible|bsPushButton|wsGroup|wsTabStop, 120, 65, 90, 24, hwnd, 110, instance)
setWindowLong(hwnd, gwlStyle, getWindowLong(hwnd, gwlStyle)^wsMinimizeBox)
setWindowLong(hwnd, gwlStyle, getWindowLong(hwnd, gwlStyle)^wsMaximizeBox)
font := getMessageFont()
sendMessage(hwndText, wmSetFont, font, 0)
sendMessage(hwndEdit, wmSetFont, font, 0)
sendMessage(hwndOK, wmSetFont, font, 0)
sendMessage(hwndCancel, wmSetFont, font, 0)
centerWindow(hwnd)
showWindow(hwnd, swShowNormal)
updateWindow(hwnd)
err = messageLoop(hwnd)
if err != nil {
return out, false, err
}
2021-02-10 18:07:18 -05:00
return out, true, nil
2018-01-25 10:37:06 -05:00
}