forked from getlantern/systray
-
Notifications
You must be signed in to change notification settings - Fork 0
/
systray_windows.go
219 lines (199 loc) · 5.74 KB
/
systray_windows.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
// +build windows
package systray
import (
"crypto/md5"
"fmt"
"io/ioutil"
"path/filepath"
"os"
"sync/atomic"
"github.com/lxn/walk"
)
var (
tmpDir string
mainWindow *walk.MainWindow
webView *walk.WebView
notifyIcon *walk.NotifyIcon
actions = make(map[int32]*walk.Action)
nextActionId int32
okayToClose int32
)
func nativeLoop(title string, width int, height int) {
var err error
mainWindow, err = walk.NewMainWindow()
if err != nil {
fail("Unable to create main window", err)
}
mainWindow.Closing().Attach(func(canceled *bool, reason walk.CloseReason) {
// don't close app unless we're actually finished
actuallyClose := atomic.LoadInt32(&okayToClose) == 1
*canceled = !actuallyClose
if !actuallyClose {
mainWindow.SetVisible(false)
}
})
layout := walk.NewVBoxLayout()
if err := mainWindow.SetLayout(layout); err != nil {
fail("Unable to set main layout", err)
}
notifyIcon, err = walk.NewNotifyIcon(mainWindow)
if err != nil {
fail("Unable to create notify icon", err)
}
if title != "" {
webView, err = walk.NewWebView(mainWindow)
if err != nil {
fail("Unable to create web view", err)
}
if err := mainWindow.SetTitle(title); err != nil {
fail("Unable to set main title", err)
}
if err := mainWindow.SetWidth(width); err != nil {
fail("Unable to set width", err)
}
if err := mainWindow.SetHeight(height); err != nil {
fail("Unable to set height", err)
}
}
systrayReady()
mainWindow.Run()
}
func quit() {
atomic.StoreInt32(&okayToClose, 1)
mainWindow.Close()
notifyIcon.Dispose()
systrayExit()
}
// SetIcon sets the systray icon.
// iconBytes should be the content of .ico for windows and .ico/.jpg/.png
// for other platforms.
func SetIcon(iconBytes []byte) {
md5 := md5.Sum(iconBytes)
filename := fmt.Sprintf("systray.%x.ico", md5)
iconpath := filepath.Join(walk.Resources.RootDirPath(), filename)
// First, try to find a previously loaded icon in walk cache
icon, err := walk.Resources.Icon(filename)
if err != nil {
// Cache miss, load the icon
err := ioutil.WriteFile(iconpath, iconBytes, 0644)
if err != nil {
fail("Unable to save icon to disk", err)
}
defer os.Remove(iconpath)
icon, err = walk.Resources.Icon(filename)
if err != nil {
fail("Unable to load icon", err)
}
}
err = notifyIcon.SetIcon(icon)
if err != nil {
fail("Unable to set systray icon", err)
}
err = notifyIcon.SetVisible(true)
if err != nil {
fail("Unable to make systray icon visible", err)
}
}
// SetTemplateIcon sets the systray icon as a template icon (on macOS), falling back
// to a regular icon on other platforms.
// templateIconBytes and iconBytes should be the content of .ico for windows and
// .ico/.jpg/.png for other platforms.
func SetTemplateIcon(templateIconBytes []byte, regularIconBytes []byte) {
SetIcon(regularIconBytes)
}
// SetTitle sets the systray title, only available on Mac.
func SetTitle(title string) {
// not supported on Windows
}
// SetTooltip sets the systray tooltip to display on mouse hover of the tray icon,
// only available on Mac and Windows.
func SetTooltip(tooltip string) {
if err := notifyIcon.SetToolTip(tooltip); err != nil {
fail("Unable to set tooltip", err)
}
}
// ShowAppWindow shows the given URL in the application window. Only works if
// configureAppWindow has been called first.
func ShowAppWindow(url string) {
if webView == nil {
return
}
webView.SetURL(url)
mainWindow.SetVisible(true)
}
func addOrUpdateMenuItem(item *MenuItem) {
action := actions[item.id]
if action == nil {
item.id = nextActionId
action = walk.NewAction()
action.Triggered().Attach(func() {
select {
case item.ClickedCh <- struct{}{}:
// okay
default:
// no listener, ignore
}
})
if err := notifyIcon.ContextMenu().Actions().Add(action); err != nil {
fail("Unable to add menu item to systray", err)
}
actions[item.id] = action
atomic.AddInt32(&nextActionId, 1)
}
err := action.SetText(item.title)
if err != nil {
fail("Unable to set menu item text", err)
}
err = action.SetChecked(item.checked)
if err != nil {
fail("Unable to set menu item checked", err)
}
err = action.SetEnabled(!item.Disabled())
if err != nil {
fail("Unable to set menu item enabled", err)
}
}
// SetIcon sets the icon of a menu item. Only works on macOS and Windows.
// iconBytes should be the content of .ico/.jpg/.png
func (item *MenuItem) SetIcon(iconBytes []byte) {
md5 := md5.Sum(iconBytes)
filename := fmt.Sprintf("systray.%x.ico", md5)
iconpath := filepath.Join(walk.Resources.RootDirPath(), filename)
// First, try to find a previously loaded icon in walk cache
icon, err := walk.Resources.Image(filename)
if err != nil {
// Cache miss, load the icon
err := ioutil.WriteFile(iconpath, iconBytes, 0644)
if err != nil {
fail("Unable to save icon to disk", err)
}
defer os.Remove(iconpath)
icon, err = walk.Resources.Image(filename)
if err != nil {
fail("Unable to load icon", err)
}
}
actions[item.id].SetImage(icon)
}
// SetTemplateIcon sets the icon of a menu item as a template icon (on macOS). On Windows, it
// falls back to the regular icon bytes and on Linux it does nothing.
// templateIconBytes and regularIconBytes should be the content of .ico for windows and
// .ico/.jpg/.png for other platforms.
func (item *MenuItem) SetTemplateIcon(templateIconBytes []byte, regularIconBytes []byte) {
item.SetIcon(regularIconBytes)
}
func addSeparator(id int32) {
action := walk.NewSeparatorAction()
if err := notifyIcon.ContextMenu().Actions().Add(action); err != nil {
fail("Unable to add separator", err)
}
}
func hideMenuItem(item *MenuItem) {
actions[item.id].SetVisible(false)
}
func showMenuItem(item *MenuItem) {
actions[item.id].SetVisible(true)
}
func fail(msg string, err error) {
panic(fmt.Errorf("%v: %v", msg, err))
}