refactor: don't change activity state directly

sonjek · Apr 9, 2024 · 652505f · 652505f
1 parent 63fea9b
commit 652505f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/internal/config/config.go b/internal/config/config.go
@@ -34,6 +34,10 @@ func NewConfig() *Config {
 	}
 }
 
+func (c *Config) Toggle() {
+	c.Enabled = !c.Enabled
+}
+
 func (c *Config) SetWorkingHoursInterval(interval string) {
 	c.WorkingHoursInterval = interval
 }
diff --git a/internal/tray/tray.go b/internal/tray/tray.go
@@ -86,13 +86,13 @@ func (t *Tray) onReady() {
 		for {
 			select {
 			case <-mEnable.ClickedCh:
-				t.conf.Enabled = true
+				t.conf.Toggle()
 				mEnable.Hide()
 				mDisable.Show()
 				mWorkingHours.Enable()
 				go t.mouseController.MoveMouse()
 			case <-mDisable.ClickedCh:
-				t.conf.Enabled = false
+				t.conf.Toggle()
 				mDisable.Hide()
 				mEnable.Show()
 				mWorkingHours.Disable()