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
+}