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
|
/*
Package systray is a cross platfrom Go library to place an icon and menu in the notification area.
Supports Windows, Mac OSX and Linux currently.
Methods can be called from any goroutine except Run(), which should be called at the very beginning of main() to lock at main thread.
*/
package systray
/*
#cgo linux pkg-config: gtk+-3.0 appindicator3-0.1
#cgo windows CFLAGS: -DWIN32 -DUNICODE -D_UNICODE
#cgo darwin CFLAGS: -DDARWIN -x objective-c -fobjc-arc
#cgo darwin LDFLAGS: -framework Cocoa
#include "systray.h"
*/
import "C"
import (
"code.google.com/p/go-uuid/uuid"
"runtime"
"sync"
"unsafe"
)
// MenuItem is used to keep track each menu item of systray
// Don't create it directly, use the one systray.AddMenuItem() returned
type MenuItem struct {
// ClickedCh is the channel which will be notified when the menu item is clicked
ClickedCh chan interface{}
// id uniquely identify a menu item, not supposed to be modified
id string
// title is the text shown on menu item
title string
// tooltip is the text shown when pointing to menu item
tooltip string
// disabled menu item is grayed out and has no effect when clicked
disabled bool
// checked menu item has a tick before the title
checked bool
}
var (
readyCh = make(chan interface{})
clickedCh = make(chan interface{})
menuItems = make(map[string]*MenuItem)
menuItemsLock sync.RWMutex
)
// Run initializes GUI and starts the event loop, then invokes the onReady
// callback.
// It blocks until systray.Quit() is called.
// Should be called at the very beginning of main() to lock at main thread.
func Run(onReady func()) {
runtime.LockOSThread()
go func() {
<-readyCh
onReady()
}()
C.nativeLoop()
}
// Quit the systray and whole app
func Quit() {
C.quit()
}
// 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) {
cstr := (*C.char)(unsafe.Pointer(&iconBytes[0]))
C.setIcon(cstr, (C.int)(len(iconBytes)))
}
// SetTitle sets the systray title, only available on Mac.
func SetTitle(title string) {
C.setTitle(C.CString(title))
}
// SetTitle sets the systray tooltip to display on mouse hover of the tray icon,
// only available on Mac.
func SetTooltip(tooltip string) {
C.setTooltip(C.CString(tooltip))
}
// Add menu item with designated title and tooltip, returning a channel that
// notifies whenever that menu item has been clicked.
//
// Menu items are keyed to an id. If the same id is added twice, the 2nd one
// overwrites the first.
//
// AddMenuItem can be safely invoked from different goroutines.
func AddMenuItem(title string, tooltip string) *MenuItem {
id := uuid.New()
item := &MenuItem{nil, id, title, tooltip, false, false}
item.ClickedCh = make(chan interface{})
item.update()
return item
}
func (item *MenuItem) SetTitle(title string) {
item.title = title
item.update()
}
func (item *MenuItem) SetTooltip(tooltip string) {
item.tooltip = tooltip
item.update()
}
func (item *MenuItem) Disabled() bool {
return item.disabled
}
func (item *MenuItem) Enable() {
item.disabled = false
item.update()
}
func (item *MenuItem) Disable() {
item.disabled = true
item.update()
}
func (item *MenuItem) Checked() bool {
return item.checked
}
func (item *MenuItem) Check() {
item.checked = true
item.update()
}
func (item *MenuItem) Uncheck() {
item.checked = false
item.update()
}
// update propogates changes on a menu item to systray
func (item *MenuItem) update() {
menuItemsLock.Lock()
defer menuItemsLock.Unlock()
menuItems[item.id] = item
var disabled C.short = 0
if item.disabled {
disabled = 1
}
var checked C.short = 0
if item.checked {
checked = 1
}
C.add_or_update_menu_item(
C.CString(item.id),
C.CString(item.title),
C.CString(item.tooltip),
disabled,
checked,
)
}
//export systray_ready
func systray_ready() {
readyCh <- nil
}
//export systray_menu_item_selected
func systray_menu_item_selected(cId *C.char) {
id := C.GoString(cId)
menuItemsLock.RLock()
item := menuItems[id]
menuItemsLock.RUnlock()
select {
case item.ClickedCh <- nil:
// in case no one waiting for the channel
default:
}
}
|
