golang 托盤菜單應用及打開系統預設浏覽器

之前看到一個應用，用go語言編寫，說是某某程式的windows圖形化用戶端，體驗一下發現隻是一個托盤，然後托盤菜單的控制台功能直接打開本地浏覽器通路程式啟動的web server網頁完成gui相關功能。頓時感覺，嗯，是個曲線繞開類似electron等架構的方法。

這種方式的好處是，可以把擅長寫web服務的應用桌面化，當需要gui的時候，直接托盤菜單啟動浏覽器，完成相關功能後，直接關閉浏覽器省記憶體。

我的gost-ui-3程式用electron編寫，内部內建浏覽器，安裝包60-~85M，啟動後記憶體占用超過100M，是以後面考慮節省資源的方式，可以使用托盤圖示+預設浏覽器的方式解決。

當然如果對 cgo 敏感的話，就不能用了，三個平台都依賴 cgo

1. golang 托盤圖示的使用

因為最近常用的是windows，家裡的mac已經吃灰很久了。是以暫時預設适配的是windows環境。理論上mac和linux上也是可以适配的。

最核心的是應用了 github.com/getlantern/systray 這個庫。

一個最簡單的示例如下：

注意提前準備好相應的圖示檔案

package main

import (
	_ "embed"
	"fmt"

	"github.com/getlantern/systray"
)

// embed 指令直接讀取icon檔案并在編譯時嵌入程式中
//
//go:embed icon.ico
var iconWin []byte

// 托盤菜單描述，自定義，為了友善定義和事件管理
type Menu struct {
	Title   string
	Tips    string
	Icon    []byte
	OnClick func(m *systray.MenuItem)
}

// 添加菜單
func AddMenu(menu *Menu) *systray.MenuItem {
	m := systray.AddMenuItem(menu.Title, menu.Tips)
	if len(menu.Icon) > 0 {
		m.SetIcon(menu.Icon)
	}
	go func() {
		for range m.ClickedCh {
			menu.OnClick(m)
		}
	}()

	return m
}

// 添加checkbox菜單
func AddCheckboxMenu(menu *Menu, checked bool) *systray.MenuItem {
	m := systray.AddMenuItemCheckbox(menu.Title, menu.Tips, checked)
	if len(menu.Icon) > 0 {
		m.SetIcon(menu.Icon)
	}
	go func() {
		for range m.ClickedCh {
			menu.OnClick(m)
		}
	}()

	return m
}

func main() {
	systray.Run(onReady, onExit)
}

func onReady() {
	systray.SetIcon(iconWin)
	systray.SetTitle("托盤圖示示例")
	systray.SetTooltip("托盤圖示示例提示")

	// 選擇框和動态菜單綜合示例
    AddCheckboxMenu(&Menu{
		Title: "啟動",
		OnClick: func(m *systray.MenuItem) {
			if m.Checked() {
				m.SetTitle("啟動")
				m.Uncheck()
			} else {
				m.SetTitle("停止")
				m.Check()
			}
		},
	}, false)

	// 添加退出菜單
	AddMenu(&Menu{
		Title: "退出",
		Tips:  "退出程式",
		OnClick: func(m *systray.MenuItem) {
			systray.Quit()
		},
	})
}

func onExit() {
	fmt.Printf("退出喽")
}
           

同時看到一些可能在應用中需要用到的api方法如下

// 設定主托盤圖示， 比如一個服務分别在啟動狀态和關閉狀态使用不同的圖示
systray.SetTemplateIcon(_icon, _icon)

// 菜單可以顯示和隐藏
systray.MenuItem.Show(); systray.MenuItem.Hide()
// 菜單可禁止和啟用
systray.MenuItem.Disable(); systray.MenuItem.Enable()
// 添加一組菜單的方式
func AddMenuGroup(title string, sub []*Menu) {
	boot := systray.AddMenuItem(title, "")
	for _, v := range sub {
		mi := boot.AddSubMenuItem(v.Title, v.Title)
		_v := v
		go func() {
			for {
				select {
				case <-mi.ClickedCh:
					_v.OnClick(mi)
				}
			}
		}()
	}
}
           

2. 用 golang 打開預設浏覽器

打開預設浏覽器其實就是執行對應平台的系統指令。

結合我的托盤示例，一個完整的例子程式如下。主要看Open(uri string)方法。

//go:generate goversioninfo
package main

import (
    _ "embed"
    "fmt"
    "os/exec"
    "runtime"

    "github.com/getlantern/systray"
)

//go:embed icon/icon.png
var icon []byte

//go:embed icon/icon_off.png
var iconOff []byte

//go:embed icon/icon.ico
var iconWin []byte

//go:embed icon/icon_off.ico
var iconOffWin []byte

//go:embed icon/logo.png
var logo []byte

// 不同平台打開浏覽器對應的指令
var commands = map[string]string{
    "windows": "cmd",
    "darwin":  "open",
    "linux":   "xdg-open",
}

// 托盤菜單自定義資料結構
type Menu struct {
    Title   string
    Tips    string
    Icon    []byte
    OnClick func(m *systray.MenuItem)
}

func main() {
    systray.Run(onReady, onExit)
}

// 執行打開預設浏覽器并通路指定uri的指令
func Open(uri string) error {
    run, ok := commands[runtime.GOOS]
    if !ok {
        return fmt.Errorf("don't know how to open things on %s platform", runtime.GOOS)
    }

    var cmd *exec.Cmd
    if runtime.GOOS == "windows" {
        cmd = exec.Command(run, `/c`, `start`, uri)
		// 無console調用
        cmd.SysProcAttr = &syscall.SysProcAttr{HideWindow: true}
    } else {
        // linux和mac下暫未測試
        cmd = exec.Command(run, uri)
    }

    return cmd.Start()
}

// 添加一個正常菜單
func AddMenu(menu *Menu) *systray.MenuItem {
    m := systray.AddMenuItem(menu.Title, menu.Tips)
    if len(menu.Icon) > 0 {
        m.SetIcon(menu.Icon)
    }
    go func() {
        for range m.ClickedCh {
            menu.OnClick(m)
        }
    }()

    return m
}

// 托盤啟動時
func onReady() {
    systray.SetIcon(iconWin)
    systray.SetTitle("托盤圖示示例")
    systray.SetTooltip("托盤圖示示例提示")

    // 打開一個浏覽器網址
    AddMenu(&Menu{
        Title: "我的部落格",
        Tips: "blog.wavesxa.com",
        OnClick: func(m *systray.MenuItem) {
            Open("https://blog.wavesxa.com")
        },
    })

    // 退出菜單
    AddMenu(&Menu{
        Title: "退出",
        Tips: "退出程式",
        OnClick: func(m *systray.MenuItem) {
            systray.Quit()
        },
    })
}

// 托盤退出時
func onExit() {
    fmt.Printf("退出喽")
}