diff options
author | kali <kali@leap.se> | 2018-09-26 20:36:18 +0200 |
---|---|---|
committer | Kali Kaneko (leap communications) <kali@leap.se> | 2018-09-26 20:43:51 +0200 |
commit | b26241ce8ec2b992b4209677ea75df608964cdbf (patch) | |
tree | 237b93bf1173cd4207e960cad6ff0c3c2104a75f /vendor/github.com/getlantern/systray/systray_windows.go | |
parent | f1e820ba1664cc1d3f69e15d0356552fecaf6fcd (diff) |
[pkg] vendor getlantern/systray
this is a workaround for riseup_vpn#28
Diffstat (limited to 'vendor/github.com/getlantern/systray/systray_windows.go')
-rw-r--r-- | vendor/github.com/getlantern/systray/systray_windows.go | 176 |
1 files changed, 176 insertions, 0 deletions
diff --git a/vendor/github.com/getlantern/systray/systray_windows.go b/vendor/github.com/getlantern/systray/systray_windows.go new file mode 100644 index 0000000..9ee0297 --- /dev/null +++ b/vendor/github.com/getlantern/systray/systray_windows.go @@ -0,0 +1,176 @@ +package systray + +import ( + "fmt" + "io/ioutil" + "os" + "path/filepath" + "runtime" + "syscall" + "unsafe" + + "github.com/getlantern/filepersist" +) + +var ( + iconFiles = make([]*os.File, 0) + dllDir = filepath.Join(os.Getenv("APPDATA"), "systray") + dllFileName = "systray" + runtime.GOARCH + ".dll" + dllFile = filepath.Join(dllDir, dllFileName) + + mod = syscall.NewLazyDLL(dllFile) + _nativeLoop = mod.NewProc("nativeLoop") + _quit = mod.NewProc("quit") + _setIcon = mod.NewProc("setIcon") + _setTitle = mod.NewProc("setTitle") + _setTooltip = mod.NewProc("setTooltip") + _add_or_update_menu_item = mod.NewProc("add_or_update_menu_item") + _add_separator = mod.NewProc("add_separator") + _hide_menu_item = mod.NewProc("hide_menu_item") +) + +func init() { + // Write DLL to file + b, err := Asset(dllFileName) + if err != nil { + panic(fmt.Errorf("Unable to read "+dllFileName+": %v", err)) + } + + err = os.MkdirAll(dllDir, 0755) + if err != nil { + panic(fmt.Errorf("Unable to create directory %v to hold "+dllFileName+": %v", dllDir, err)) + } + + err = filepersist.Save(dllFile, b, 0644) + if err != nil { + panic(fmt.Errorf("Unable to save "+dllFileName+" to %v: %v", dllFile, err)) + } +} + +func nativeLoop() { + _nativeLoop.Call( + syscall.NewCallbackCDecl(systray_ready), + syscall.NewCallbackCDecl(systray_on_exit), + syscall.NewCallbackCDecl(systray_menu_item_selected)) +} + +func quit() { + _quit.Call() + for _, f := range iconFiles { + err := os.Remove(f.Name()) + if err != nil { + log.Debugf("Unable to delete temporary icon file %v: %v", f.Name(), err) + } + } +} + +// 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) { + f, err := ioutil.TempFile("", "systray_temp_icon") + if err != nil { + log.Errorf("Unable to create temp icon: %v", err) + return + } + defer f.Close() + _, err = f.Write(iconBytes) + if err != nil { + log.Errorf("Unable to write icon to temp file %v: %v", f.Name(), f) + return + } + // Need to close file before we load it to make sure contents is flushed. + f.Close() + name, err := strUTF16(f.Name()) + if err != nil { + log.Errorf("Unable to convert name to string pointer: %v", err) + return + } + _setIcon.Call(name.Raw()) +} + +// SetTitle sets the systray title, only available on Mac. +func SetTitle(title string) { + // do nothing +} + +// SetTooltip sets the systray tooltip to display on mouse hover of the tray icon, +// only available on Mac and Windows. +func SetTooltip(tooltip string) { + t, err := strUTF16(tooltip) + if err != nil { + log.Errorf("Unable to convert tooltip to string pointer: %v", err) + return + } + _setTooltip.Call(t.Raw()) +} + +func addOrUpdateMenuItem(item *MenuItem) { + var disabled = 0 + if item.disabled { + disabled = 1 + } + var checked = 0 + if item.checked { + checked = 1 + } + title, err := strUTF16(item.title) + if err != nil { + log.Errorf("Unable to convert title to string pointer: %v", err) + return + } + tooltip, err := strUTF16(item.tooltip) + if err != nil { + log.Errorf("Unable to convert tooltip to string pointer: %v", err) + return + } + _add_or_update_menu_item.Call( + uintptr(item.id), + title.Raw(), + tooltip.Raw(), + uintptr(disabled), + uintptr(checked), + ) +} + +func addSeparator(id int32) { + _add_separator.Call(uintptr(id)) +} + +func hideMenuItem(item *MenuItem) { + _hide_menu_item.Call(uintptr(item.id)) +} + +func showMenuItem(item *MenuItem) { + addOrUpdateMenuItem(item) +} + +type utf16 []uint16 + +// Raw returns the underlying *wchar_t of an utf16 so we can pass to DLL +func (u utf16) Raw() uintptr { + return uintptr(unsafe.Pointer(&u[0])) +} + +// strUTF16 converts a Go string into a utf16 byte sequence +func strUTF16(s string) (utf16, error) { + return syscall.UTF16FromString(s) +} + +// systray_ready takes an ignored parameter just so we can compile a callback +// (for some reason in Go 1.4.x, syscall.NewCallback panics if there's no +// parameter to the function). +func systray_ready(ignore uintptr) uintptr { + systrayReady() + return 0 +} + +func systray_on_exit(ignore uintptr) uintptr { + systrayExit() + return 0 +} + +func systray_menu_item_selected(id uintptr) uintptr { + systrayMenuItemSelected(int32(id)) + return 0 +} |