zenity/notify_windows.go

128 lines
2.7 KiB
Go
Raw Normal View History

2020-01-26 11:04:49 -05:00
package zenity
2020-01-27 08:44:47 -05:00
import (
2021-08-13 21:50:04 -04:00
"math/rand"
2020-01-27 13:11:38 -05:00
"runtime"
2020-01-27 08:44:47 -05:00
"syscall"
2021-08-13 21:50:04 -04:00
"time"
2020-01-27 08:44:47 -05:00
"unsafe"
2020-01-30 21:18:43 -05:00
2022-06-17 22:45:49 -04:00
"github.com/ncruces/zenity/internal/win"
2020-01-30 21:18:43 -05:00
"github.com/ncruces/zenity/internal/zenutil"
2020-01-27 08:44:47 -05:00
)
var (
2021-08-13 21:50:04 -04:00
rtlGetNtVersionNumbers = ntdll.NewProc("RtlGetNtVersionNumbers")
wtsSendMessage = wtsapi32.NewProc("WTSSendMessageW")
2020-01-27 08:44:47 -05:00
)
2021-03-04 07:42:30 -05:00
func notify(text string, opts options) error {
2020-01-30 20:31:54 -05:00
if opts.ctx != nil && opts.ctx.Err() != nil {
return opts.ctx.Err()
}
2022-06-17 22:45:49 -04:00
var args win.NOTIFYICONDATA
2020-01-27 08:44:47 -05:00
args.StructSize = uint32(unsafe.Sizeof(args))
2021-08-13 21:50:04 -04:00
args.ID = rand.Uint32()
2020-01-27 08:44:47 -05:00
args.Flags = 0x00000010 // NIF_INFO
args.State = 0x00000001 // NIS_HIDDEN
info := syscall.StringToUTF16(text)
copy(args.Info[:len(args.Info)-1], info)
2021-03-03 22:25:45 -05:00
if opts.title != nil {
title := syscall.StringToUTF16(*opts.title)
copy(args.InfoTitle[:len(args.InfoTitle)-1], title)
}
2020-01-27 08:44:47 -05:00
switch opts.icon {
2021-03-04 08:06:49 -05:00
case InfoIcon, QuestionIcon:
2020-01-27 08:44:47 -05:00
args.InfoFlags |= 0x1 // NIIF_INFO
case WarningIcon:
args.InfoFlags |= 0x2 // NIIF_WARNING
case ErrorIcon:
args.InfoFlags |= 0x3 // NIIF_ERROR
2022-05-23 19:12:38 -04:00
default:
icon := getIcon(opts.icon)
if icon.handle != 0 {
defer icon.delete()
args.Icon = icon.handle
args.Flags |= 0x00000002 // NIF_ICON
args.InfoFlags |= 0x4 // NIIF_USER
}
2020-01-27 08:44:47 -05:00
}
2020-01-27 13:11:38 -05:00
runtime.LockOSThread()
defer runtime.UnlockOSThread()
2022-06-17 22:45:49 -04:00
s, err := win.ShellNotifyIcon(win.NIM_ADD, &args)
2020-01-29 09:15:21 -05:00
if s == 0 {
2020-01-27 13:11:38 -05:00
if errno, ok := err.(syscall.Errno); ok && errno == 0 {
2020-01-30 21:18:43 -05:00
return wtsMessage(text, opts)
2020-01-27 13:11:38 -05:00
}
2020-01-27 08:44:47 -05:00
return err
}
2021-08-13 21:50:04 -04:00
var major, minor, build uint32
rtlGetNtVersionNumbers.Call(
uintptr(unsafe.Pointer(&major)),
uintptr(unsafe.Pointer(&minor)),
uintptr(unsafe.Pointer(&build)))
// On Windows 7 (6.1) and lower, wait up to 10 seconds to clean up.
if major < 6 || major == 6 && minor < 2 {
if opts.ctx != nil {
select {
case <-opts.ctx.Done():
case <-time.After(10 * time.Second):
}
} else {
time.Sleep(10 * time.Second)
}
}
2022-06-17 22:45:49 -04:00
win.ShellNotifyIcon(win.NIM_DELETE, &args)
2020-01-27 08:44:47 -05:00
return nil
}
2020-01-30 21:18:43 -05:00
func wtsMessage(text string, opts options) error {
var flags uintptr
switch opts.icon {
case ErrorIcon:
2022-03-28 22:05:46 -04:00
flags |= _MB_ICONERROR
2020-01-30 21:18:43 -05:00
case QuestionIcon:
2022-03-28 22:05:46 -04:00
flags |= _MB_ICONQUESTION
2020-01-30 21:18:43 -05:00
case WarningIcon:
2022-03-28 22:05:46 -04:00
flags |= _MB_ICONWARNING
2020-01-30 21:18:43 -05:00
case InfoIcon:
2022-03-28 22:05:46 -04:00
flags |= _MB_ICONINFORMATION
2020-01-30 21:18:43 -05:00
}
title := opts.title
2021-03-03 22:25:45 -05:00
if title == nil {
title = stringPtr("Notification")
2020-01-30 21:18:43 -05:00
}
timeout := zenutil.Timeout
if timeout == 0 {
timeout = 10
}
ptext := syscall.StringToUTF16(text)
2021-03-03 22:25:45 -05:00
ptitle := syscall.StringToUTF16(*title)
2020-01-30 21:18:43 -05:00
var res uint32
s, _, err := wtsSendMessage.Call(
0, // WTS_CURRENT_SERVER_HANDLE
0xffffffff, // WTS_CURRENT_SESSION
uintptr(unsafe.Pointer(&ptitle[0])), uintptr(2*len(ptitle)),
uintptr(unsafe.Pointer(&ptext[0])), uintptr(2*len(ptext)),
flags, uintptr(timeout), uintptr(unsafe.Pointer(&res)), 0)
if s == 0 {
return err
}
return nil
}