Merge pull request #23 from thiagokokada/add-pause-to-systray

Add 1 hour pause to systray

main.go

@@ -13,8 +13,8 @@ import (
 
 var (
 	version           = "development"
-	ctx               context.Context
-	ctxCancel         context.CancelFunc
+	mainCtx           context.Context
+	mainCtxCancel     context.CancelFunc
 	duration          = new(time.Duration)
 	frequency         = new(time.Duration)
 	notificationSound = new(bool)
@@ -144,8 +144,8 @@ func runTwentyTwentyTwenty() {
 		)
 	}
 
-	ctx, ctxCancel = context.WithCancel(context.Background())
-	go twentyTwentyTwenty(ctx, notifier, duration, frequency, notificationSound)
+	mainCtx, mainCtxCancel = context.WithCancel(context.Background())
+	go twentyTwentyTwenty(mainCtx, notifier, duration, frequency, notificationSound)
 }
 
 func main() {

systray.go

@@ -4,8 +4,10 @@
 package main
 
 import (
+	"context"
 	_ "embed"
 	"log"
+	"time"
 
 	"fyne.io/systray"
 )
@@ -17,22 +19,52 @@ func onReady() {
 	systray.SetIcon(data)
 	systray.SetTooltip("TwentyTwentyTwenty")
 	mEnabled := systray.AddMenuItemCheckbox("Enabled", "Enable twenty-twenty-twenty", true)
+	mPause := systray.AddMenuItemCheckbox("Pause for 1 hour", "Pause twenty-twenty-twenty for 1 hour", false)
 	mSound := new(systray.MenuItem)
 	if notificationSoundEnabled {
 		mSound = systray.AddMenuItemCheckbox("Sound", "Enable notification sound", *notificationSound)
 	}
 	systray.AddSeparator()
 	mQuit := systray.AddMenuItem("Quit", "Quit the whole app")
+	var ctx context.Context
+	var cancel context.CancelFunc
 
 	for {
 		select {
 		case <-mEnabled.ClickedCh:
 			if mEnabled.Checked() {
-				ctxCancel()
+				mainCtxCancel()
 				mEnabled.Uncheck()
+				mPause.Disable()
 			} else {
 				runTwentyTwentyTwenty()
 				mEnabled.Check()
+				mPause.Enable()
+			}
+		case <-mPause.ClickedCh:
+			if mPause.Checked() {
+				mainCtxCancel() // make sure the loop stopped
+				cancel()
+				runTwentyTwentyTwenty()
+				mEnabled.Enable()
+				mPause.Uncheck()
+			} else {
+				log.Println("Pausing twenty-twenty-twenty for 1 hour...")
+				ctx, cancel = context.WithCancel(context.Background())
+				go func(ctx context.Context) {
+					mainCtxCancel()
+					timer := time.NewTimer(time.Hour)
+					select {
+					case <-timer.C:
+						runTwentyTwentyTwenty()
+						mEnabled.Enable()
+						mPause.Uncheck()
+					case <-ctx.Done():
+					}
+					cancel()
+				}(ctx)
+				mEnabled.Disable()
+				mPause.Check()
 			}
 		case <-mSound.ClickedCh:
 			if mSound.Checked() {