zenity/msg_windows.go

package zenity

import (
	"bytes"
	"context"
	"os"
	"syscall"
	"unsafe"
)

var (
	messageBox             = user32.NewProc("MessageBoxW")
	getDlgCtrlID           = user32.NewProc("GetDlgCtrlID")
	loadImage              = user32.NewProc("LoadImageW")
	destroyIcon            = user32.NewProc("DestroyIcon")
	createIconFromResource = user32.NewProc("CreateIconFromResource")
)

func message(kind messageKind, text string, opts options) error {
	var flags uintptr

	switch {
	case kind == questionKind && opts.extraButton != nil:
		flags |= _MB_YESNOCANCEL
	case kind == questionKind:
		flags |= _MB_OKCANCEL
	case opts.extraButton != nil:
		flags |= _MB_YESNO
	}

	switch opts.icon {
	case ErrorIcon:
		flags |= _MB_ICONERROR
	case QuestionIcon:
		flags |= _MB_ICONQUESTION
	case WarningIcon:
		flags |= _MB_ICONWARNING
	case InfoIcon:
		flags |= _MB_ICONINFORMATION
	case nil:
		switch kind {
		case errorKind:
			flags |= _MB_ICONERROR
		case questionKind:
			flags |= _MB_ICONQUESTION
		case warningKind:
			flags |= _MB_ICONWARNING
		case infoKind:
			flags |= _MB_ICONINFORMATION
		}
	}

	if kind == questionKind && opts.defaultCancel {
		if opts.extraButton == nil {
			flags |= _MB_DEFBUTTON2
		} else {
			flags |= _MB_DEFBUTTON3
		}
	}

	defer setup()()

	if opts.ctx != nil || opts.okLabel != nil || opts.cancelLabel != nil || opts.extraButton != nil || opts.icon != nil {
		unhook, err := hookMessageDialog(kind, opts)
		if err != nil {
			return err
		}
		defer unhook()
	}

	var title *uint16
	if opts.title != nil {
		title = syscall.StringToUTF16Ptr(*opts.title)
	}

	s, _, err := messageBox.Call(0, strptr(text), uintptr(unsafe.Pointer(title)), flags)

	if opts.ctx != nil && opts.ctx.Err() != nil {
		return opts.ctx.Err()
	}
	switch s {
	case _IDOK, _IDYES:
		return nil
	case _IDCANCEL:
		return ErrCanceled
	case _IDNO:
		return ErrExtraButton
	default:
		return err
	}
}

func hookMessageDialog(kind messageKind, opts options) (unhook context.CancelFunc, err error) {
	return hookDialog(opts.ctx, func(wnd uintptr) {
		enumChildWindows.Call(wnd,
			syscall.NewCallback(hookMessageDialogCallback),
			uintptr(unsafe.Pointer(&opts)))
	})
}

func hookMessageDialogCallback(wnd uintptr, lparam *options) uintptr {
	ctl, _, _ := getDlgCtrlID.Call(wnd)

	var text *string
	switch ctl {
	case _IDOK, _IDYES:
		text = lparam.okLabel
	case _IDCANCEL:
		text = lparam.cancelLabel
	case _IDNO:
		text = lparam.extraButton
	}
	if text != nil {
		setWindowText.Call(wnd, strptr(*text))
	}

	if i, ok := lparam.icon.(string); ok && ctl == 20 /*IDC_STATIC_OK*/ {
		var icon uintptr
		data, _ := os.ReadFile(i)
		switch {
		case bytes.HasPrefix(data, []byte("\x00\x00\x01\x00")):
			icon, _, _ = loadImage.Call(0,
				strptr(i),
				1, /*IMAGE_ICON*/
				0, 0,
				0x00008050 /*LR_LOADFROMFILE|LR_DEFAULTSIZE|LR_SHARED*/)
		case bytes.HasPrefix(data, []byte("\x89PNG\r\n\x1a\n")):
			icon, _, _ = createIconFromResource.Call(
				uintptr(unsafe.Pointer(&data[0])),
				uintptr(len(data)),
				1, 0x00030000)
			defer destroyIcon.Call(icon)
		}

		if icon != 0 {
			sendMessage.Call(wnd, 0x0170 /*STM_SETICON*/, icon, 0)
		}
	}
	return 1
}