1 files changed, 108 insertions, 0 deletions
diff --git a/pkg/systray2/config.go b/pkg/systray2/config.go
new file mode 100644
index 0000000..75a4a98
--- /dev/null
+++ b/pkg/systray2/config.go
@@ -0,0 +1,108 @@
+// Copyright (C) 2018 LEAP
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU General Public License for more details.
+//
+// You should have received a copy of the GNU General Public License
+// along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+package systray
+
+import (
+	"encoding/json"
+	"os"
+	"path"
+	"time"
+
+	"0xacab.org/leap/bitmask-vpn/pkg/config"
+	"golang.org/x/text/message"
+)
+
+const (
+	oneDay   = time.Hour * 24
+	oneMonth = oneDay * 30
+)
+
+var (
+	configPath = path.Join(config.Path, "systray.json")
+)
+
+// Config holds the configuration of the systray
+type Config struct {
+	file struct {
+		LastNotification  time.Time
+		Donated           time.Time
+		SelectGateway     bool
+		Obfs4             bool
+		UserStoppedVPN    bool
+		DisableAustostart bool
+	}
+	SelectGateway     bool
+	Obfs4             bool
+	DisableAustostart bool
+	StartVPN          bool
+	Version           string
+	Printer           *message.Printer
+}
+
+// ParseConfig reads the configuration from the configuration file
+func ParseConfig() *Config {
+	var conf Config
+
+	f, err := os.Open(configPath)
+	if err != nil {
+		conf.save()
+	} else {
+		defer f.Close()
+		dec := json.NewDecoder(f)
+		err = dec.Decode(&conf.file)
+	}
+
+	conf.SelectGateway = conf.file.SelectGateway
+	conf.Obfs4 = conf.file.Obfs4
+	conf.DisableAustostart = conf.file.DisableAustostart
+	conf.StartVPN = !conf.file.UserStoppedVPN
+	return &conf
+}
+
+func (c *Config) setUserStoppedVPN(vpnStopped bool) error {
+	c.file.UserStoppedVPN = vpnStopped
+	return c.save()
+}
+
+func (c *Config) hasDonated() bool {
+	return c.file.Donated.Add(oneMonth).After(time.Now())
+}
+
+func (c *Config) needsNotification() bool {
+	return !c.hasDonated() && c.file.LastNotification.Add(oneDay).Before(time.Now())
+}
+
+func (c *Config) setNotification() error {
+	c.file.LastNotification = time.Now()
+	return c.save()
+}
+
+func (c *Config) setDonated() error {
+	c.file.Donated = time.Now()
+	return c.save()
+}
+
+func (c *Config) save() error {
+	f, err := os.Create(configPath)
+	if err != nil {
+		return err
+	}
+	defer f.Close()
+
+	enc := json.NewEncoder(f)
+	enc.SetIndent("", "  ")
+	return enc.Encode(c.file)
+}