Initial commit

dlgs_windows.go

@@ -0,0 +1,590 @@
+// +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")
+	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
+}
+
+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
+	}
+
+	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
+	}
+
+	ret := false
+	if out != "" {
+		ret = true
+	}
+
+	return out, ret, nil
+}