add gui/notifyicon example

hallazzang · hallazzang · commit 36688f6d0ee3 · 2019-05-26T14:30:26.000+09:00
also update win package
diff --git a/example/gui/notifyicon/icon.ico b/example/gui/notifyicon/icon.ico
diff --git a/example/gui/notifyicon/main.go b/example/gui/notifyicon/main.go
@@ -0,0 +1,98 @@
+package main
+
+import (
+	"log"
+	"unsafe"
+
+	"github.com/hallazzang/go-windows-programming/pkg/win"
+	"golang.org/x/sys/windows"
+)
+
+func wndProc(hWnd uintptr, msg uint32, wParam, lParam uintptr) uintptr {
+	switch msg {
+	case notifyIconMsg:
+		switch nmsg := win.LOWORD(uint32(lParam)); nmsg {
+		case win.NIN_BALLOONUSERCLICK:
+			log.Print("User has clicked the balloon message")
+		case win.WM_LBUTTONDOWN:
+			clickHandler()
+		}
+	case win.WM_DESTROY:
+		win.PostQuitMessage(0)
+	default:
+		return win.DefWindowProc(hWnd, msg, wParam, lParam)
+	}
+	return 0
+}
+
+func createMainWindow() (uintptr, error) {
+	hInstance := win.GetModuleHandle(nil)
+
+	wndClass := windows.StringToUTF16Ptr("MyWindow")
+
+	var wcex win.WNDCLASSEX
+	wcex.CbSize = uint32(unsafe.Sizeof(wcex))
+	wcex.Style = win.CS_HREDRAW | win.CS_VREDRAW
+	wcex.LpfnWndProc = windows.NewCallback(wndProc)
+	wcex.HInstance = hInstance
+	wcex.HbrBackground = win.COLOR_WINDOW + 1
+	wcex.LpszClassName = wndClass
+	if win.RegisterClassEx(&wcex) == 0 {
+		return 0, win.GetLastError()
+	}
+
+	hwnd := win.CreateWindowEx(0, wndClass, windows.StringToUTF16Ptr("NotifyIcon Example"), win.WS_OVERLAPPEDWINDOW, win.CW_USEDEFAULT, win.CW_USEDEFAULT, 400, 300, 0, 0, hInstance, nil)
+	if hwnd == win.NULL {
+		return 0, win.GetLastError()
+	}
+	win.ShowWindow(hwnd, win.SW_SHOW)
+
+	return hwnd, nil
+}
+
+func loadIcon(name string) (uintptr, error) {
+	pname, err := windows.UTF16PtrFromString(name)
+	if err != nil {
+		return 0, err
+	}
+
+	hIcon := win.LoadImage(win.NULL, pname, win.IMAGE_ICON, 0, 0, win.LR_DEFAULTSIZE|win.LR_LOADFROMFILE)
+	if hIcon == win.NULL {
+		return 0, win.GetLastError()
+	}
+
+	return hIcon, nil
+}
+
+func clickHandler() {
+	log.Print("User has clicked the notify icon")
+}
+
+func main() {
+	hwnd, err := createMainWindow()
+	if err != nil {
+		panic(err)
+	}
+
+	hIcon, err := loadIcon("icon.ico")
+	defer win.DestroyIcon(hIcon)
+
+	win.SendMessage(hwnd, win.WM_SETICON, win.ICON_BIG, hIcon)
+	win.SendMessage(hwnd, win.WM_SETICON, win.ICON_SMALL, hIcon)
+
+	ni, err := newNotifyIcon(hwnd)
+	if err != nil {
+		panic(err)
+	}
+	defer ni.Dispose()
+
+	ni.SetIcon(hIcon)
+	ni.SetTooltip("NotifyIcon Example")
+	ni.ShowNotification("Hello", "NotifyIcon!")
+
+	var msg win.MSG
+	for win.GetMessage(&msg, 0, 0, 0) != 0 {
+		win.TranslateMessage(&msg)
+		win.DispatchMessage(&msg)
+	}
+}
diff --git a/example/gui/notifyicon/notifyicon.go b/example/gui/notifyicon/notifyicon.go
@@ -0,0 +1,88 @@
+package main
+
+import (
+	"errors"
+	"math/rand"
+	"time"
+	"unsafe"
+
+	"github.com/hallazzang/go-windows-programming/pkg/win"
+	"golang.org/x/sys/windows"
+)
+
+const notifyIconMsg = win.WM_APP + 1
+
+var errShellNotifyIcon = errors.New("Shell_NotifyIcon error")
+
+func init() {
+	rand.Seed(time.Now().UnixNano())
+}
+
+type notifyIcon struct {
+	hwnd uintptr
+	guid win.GUID
+}
+
+func newNotifyIcon(hwnd uintptr) (*notifyIcon, error) {
+	ni := &notifyIcon{
+		hwnd: hwnd,
+		guid: newGUID(),
+	}
+	data := ni.newData()
+	data.UFlags |= win.NIF_MESSAGE
+	data.UCallbackMessage = notifyIconMsg
+	if win.Shell_NotifyIcon(win.NIM_ADD, data) == win.FALSE {
+		return nil, errShellNotifyIcon
+	}
+	return ni, nil
+}
+
+func (ni *notifyIcon) Dispose() {
+	win.Shell_NotifyIcon(win.NIM_DELETE, ni.newData())
+}
+
+func (ni *notifyIcon) SetTooltip(tooltip string) error {
+	data := ni.newData()
+	data.UFlags |= win.NIF_TIP
+	copy(data.SzTip[:], windows.StringToUTF16(tooltip))
+	if win.Shell_NotifyIcon(win.NIM_MODIFY, data) == win.FALSE {
+		return errShellNotifyIcon
+	}
+	return nil
+}
+
+func (ni *notifyIcon) SetIcon(hIcon uintptr) error {
+	data := ni.newData()
+	data.UFlags |= win.NIF_ICON
+	data.HIcon = hIcon
+	if win.Shell_NotifyIcon(win.NIM_MODIFY, data) == win.FALSE {
+		return errShellNotifyIcon
+	}
+	return nil
+}
+
+func (ni *notifyIcon) ShowNotification(title, text string) error {
+	data := ni.newData()
+	data.UFlags |= win.NIF_INFO
+	copy(data.SzInfoTitle[:], windows.StringToUTF16(title))
+	copy(data.SzInfo[:], windows.StringToUTF16(text))
+	if win.Shell_NotifyIcon(win.NIM_MODIFY, data) == win.FALSE {
+		return errShellNotifyIcon
+	}
+	return nil
+}
+
+func (ni *notifyIcon) newData() *win.NOTIFYICONDATA {
+	var nid win.NOTIFYICONDATA
+	nid.CbSize = uint32(unsafe.Sizeof(nid))
+	nid.UFlags = win.NIF_GUID
+	nid.HWnd = ni.hwnd
+	nid.GuidItem = ni.guid
+	return &nid
+}
+
+func newGUID() win.GUID {
+	var buf [16]byte
+	rand.Read(buf[:])
+	return *(*win.GUID)(unsafe.Pointer(&buf[0]))
+}
diff --git a/pkg/win/constant.go b/pkg/win/constant.go
@@ -1,5 +1,12 @@
 package win
 
+const (
+	NULL = 0
+
+	TRUE  = BOOL(1)
+	FALSE = BOOL(0)
+)
+
 // GetSysColor constants
 const (
 	COLOR_WINDOW = 5
@@ -15,6 +22,12 @@ const (
 	CW_USEDEFAULT = ^0x7fffffff
 )
 
+// WM_SETICON constants
+const (
+	ICON_BIG   = 1
+	ICON_SMALL = 0
+)
+
 // LoadCursor constants
 const (
 	IDC_ARROW = 32512
@@ -25,6 +38,58 @@ const (
 	IDI_APPLICATION = 32512
 )
 
+// LoadImage constants
+const (
+	IMAGE_BITMAP = 0
+	IMAGE_CURSOR = 2
+	IMAGE_ICON   = 1
+
+	LR_CREATEDIBSECTION = 0x00002000
+	LR_DEFAULTCOLOR     = 0x00000000
+	LR_DEFAULTSIZE      = 0x00000040
+	LR_LOADFROMFILE     = 0x00000010
+	LR_LOADMAP3DCOLORS  = 0x00001000
+	LR_LOADTRANSPARENT  = 0x00000020
+	LR_MONOCHROME       = 0x00000001
+	LR_SHARED           = 0x00008000
+	LR_VGACOLOR         = 0x00000080
+)
+
+// NotifyIcon constants
+const (
+	NIM_ADD        = 0x00000000
+	NIM_MODIFY     = 0x00000001
+	NIM_DELETE     = 0x00000002
+	NIM_SETFOCUS   = 0x00000003
+	NIM_SETVERSION = 0x00000004
+
+	NIF_MESSAGE  = 0x00000001
+	NIF_ICON     = 0x00000002
+	NIF_TIP      = 0x00000004
+	NIF_STATE    = 0x00000008
+	NIF_INFO     = 0x00000010
+	NIF_GUID     = 0x00000020
+	NIF_REALTIME = 0x00000040
+	NIF_SHOWTIP  = 0x00000080
+
+	NIS_HIDDEN     = 0x00000001
+	NIS_SHAREDICON = 0x00000002
+
+	NIIF_NONE               = 0x00000000
+	NIIF_INFO               = 0x00000001
+	NIIF_WARNING            = 0x00000002
+	NIIF_ERROR              = 0x00000003
+	NIIF_USER               = 0x00000004
+	NIIF_NOSOUND            = 0x00000010
+	NIIF_LARGE_ICON         = 0x00000020
+	NIIF_RESPECT_QUIET_TIME = 0x00000080
+	NIIF_ICON_MASK          = 0x0000000F
+
+	NIN_BALLOONSHOW      = 0x0402
+	NIN_BALLOONTIMEOUT   = 0x0404
+	NIN_BALLOONUSERCLICK = 0x0405
+)
+
 // ShowWindow constants
 const (
 	SW_HIDE        = 0
@@ -35,7 +100,16 @@ const (
 
 // Window messages
 const (
-	WM_DESTROY = 0x0002
+	WM_DESTROY       = 0x0002
+	WM_SETICON       = 0x0080
+	WM_MOUSEMOVE     = 0x0200
+	WM_LBUTTONDOWN   = 0x0201
+	WM_LBUTTONUP     = 0x0202
+	WM_LBUTTONDBLCLK = 0x0203
+	WM_RBUTTONDOWN   = 0x0204
+	WM_RBUTTONUP     = 0x0205
+	WM_RBUTTONDBLCLK = 0x0206
+	WM_APP           = 0x8000
 )
 
 // Window styles
diff --git a/pkg/win/macro.go b/pkg/win/macro.go
@@ -5,3 +5,11 @@ import "unsafe"
 func MAKEINTRESOURCE(i uintptr) *uint16 {
 	return (*uint16)(unsafe.Pointer(i))
 }
+
+func LOWORD(dwValue uint32) uint16 {
+	return uint16(dwValue)
+}
+
+func HIWORD(dwValue uint32) uint16 {
+	return uint16((dwValue >> 16) & 0xffff)
+}
diff --git a/pkg/win/type.go b/pkg/win/type.go
@@ -1,5 +1,14 @@
 package win
 
+type BOOL int32
+
+type GUID struct {
+	Data1 uint32
+	Data2 uint16
+	Data3 uint16
+	Data4 [8]byte
+}
+
 type MSG struct {
 	Hwnd     uintptr
 	Message  uint32
@@ -10,6 +19,24 @@ type MSG struct {
 	LPrivate uint32
 }
 
+type NOTIFYICONDATA struct {
+	CbSize           uint32
+	HWnd             uintptr
+	UID              uint32
+	UFlags           uint32
+	UCallbackMessage uint32
+	HIcon            uintptr
+	SzTip            [128]uint16
+	DwState          uint32
+	DwStateMask      uint32
+	SzInfo           [256]uint16
+	UVersion         uint32
+	SzInfoTitle      [64]uint16
+	DwInfoFlags      uint32
+	GuidItem         GUID
+	HBalloonIcon     uintptr
+}
+
 type POINT struct {
 	X int32
 	Y int32
diff --git a/pkg/win/win.go b/pkg/win/win.go
