From 0a0979a8c507db00ff7131f75f36a98207b1278a Mon Sep 17 00:00:00 2001 From: Ben Sarmiento Date: Mon, 5 Feb 2024 20:29:25 +0100 Subject: [PATCH] Fix logging of premium status --- pkg/premium/monitor.go | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) diff --git a/pkg/premium/monitor.go b/pkg/premium/monitor.go index 019e405..7c7de7c 100644 --- a/pkg/premium/monitor.go +++ b/pkg/premium/monitor.go @@ -7,7 +7,10 @@ import ( "github.com/debridmediamanager/zurg/pkg/realdebrid" ) -const PREMIUM_THRESHOLD = 172800 +const ( + PREMIUM_THRESHOLD = 172800 // 2 days + MINIMUM_SLEEP = 60 // 60 seconds +) func MonitorPremiumStatus(rd *realdebrid.RealDebrid, zurglog *logutil.Logger) { go func() { @@ -18,7 +21,7 @@ func MonitorPremiumStatus(rd *realdebrid.RealDebrid, zurglog *logutil.Logger) { time.Sleep(5 * time.Minute) continue } - if userInfo.Premium <= 0 { + if userInfo.Premium <= MINIMUM_SLEEP { zurglog.Fatal("Your account is no longer premium, exiting...") } else { if userInfo.Premium <= PREMIUM_THRESHOLD { @@ -28,6 +31,13 @@ func MonitorPremiumStatus(rd *realdebrid.RealDebrid, zurglog *logutil.Logger) { } } remaining := userInfo.Premium - PREMIUM_THRESHOLD + if remaining < MINIMUM_SLEEP { + // Ensure minimum sleep duration is 60 seconds + remaining = MINIMUM_SLEEP + } else { + // Round up to the nearest multiple of 60 seconds + remaining = ((remaining + MINIMUM_SLEEP - 1) / MINIMUM_SLEEP) * MINIMUM_SLEEP + } sleepDuration := time.Duration(remaining) * time.Second time.Sleep(sleepDuration) }