From b5aaa4e519e88f8cd029a0f144f86c1fc02ba231 Mon Sep 17 00:00:00 2001 From: Ruben Pollan <meskio@sindominio.net> Date: Fri, 16 Feb 2018 00:07:03 +0100 Subject: [PATCH] [feat] add a command line flag to enable gateway selection - Resolves: #12 --- config.go | 8 ++++++++ systray.go | 4 ++++ 2 files changed, 12 insertions(+) diff --git a/config.go b/config.go index 01cf6d5f..d751a7ba 100644 --- a/config.go +++ b/config.go @@ -17,6 +17,7 @@ package main import ( "encoding/json" + "flag" "os" "path" "time" @@ -36,6 +37,7 @@ var ( type systrayConfig struct { LastNotification time.Time Donated time.Time + SelectWateway bool } func parseConfig() (*systrayConfig, error) { @@ -52,9 +54,15 @@ func parseConfig() (*systrayConfig, error) { dec := json.NewDecoder(f) err = dec.Decode(&conf) + conf.parseFlags() return &conf, err } +func (c *systrayConfig) parseFlags() { + flag.BoolVar(&c.SelectWateway, "select-gateway", false, "Enable gateway selection") + flag.Parse() +} + func (c *systrayConfig) hasDonated() bool { return c.Donated.Add(oneMonth).After(time.Now()) } diff --git a/systray.go b/systray.go index 0d37e037..bc1cab79 100644 --- a/systray.go +++ b/systray.go @@ -65,6 +65,10 @@ func (bt *bmTray) onReady() { go bt.mCancel.Hide() systray.AddSeparator() + if bt.conf.SelectWateway { + bt.addGateways() + } + mHelp := systray.AddMenuItem("Help ...", "") bt.mDonate = systray.AddMenuItem("Donate ...", "") bt.mHaveDonated = systray.AddMenuItem("... I have donated", "") -- GitLab