Add premium status monitoring and it wont start unless you are premium

This commit is contained in:
Ben Sarmiento
2024-01-09 00:00:23 +01:00
parent 14cb593b81
commit 6c4c73a5b0
2 changed files with 38 additions and 0 deletions

View File

@@ -14,6 +14,7 @@ import (
"github.com/debridmediamanager/zurg/internal/universal"
"github.com/debridmediamanager/zurg/pkg/http"
"github.com/debridmediamanager/zurg/pkg/logutil"
"github.com/debridmediamanager/zurg/pkg/premium"
"github.com/debridmediamanager/zurg/pkg/realdebrid"
"github.com/debridmediamanager/zurg/pkg/utils"
"github.com/go-chi/chi/v5"
@@ -38,6 +39,8 @@ func MainApp(configPath string) {
rd := realdebrid.NewRealDebrid(apiClient, log.Named("realdebrid"))
premium.MonitorPremiumStatus(rd, zurglog)
p, err := ants.NewPool(config.GetNumOfWorkers() + 1)
if err != nil {
zurglog.Errorf("Failed to create worker pool: %v", err)

35
pkg/premium/monitor.go Normal file
View File

@@ -0,0 +1,35 @@
package premium
import (
"time"
"github.com/debridmediamanager/zurg/pkg/logutil"
"github.com/debridmediamanager/zurg/pkg/realdebrid"
)
const PREMIUM_THRESHOLD = 172800
func MonitorPremiumStatus(rd *realdebrid.RealDebrid, zurglog *logutil.Logger) {
go func() {
for {
userInfo, err := rd.GetUserInformation()
if err != nil {
zurglog.Errorf("Failed to get user information: %v", err)
time.Sleep(5 * time.Minute)
continue
}
if userInfo.Premium <= 0 {
zurglog.Fatal("Your account is no longer premium, exiting...")
} else {
if userInfo.Premium <= PREMIUM_THRESHOLD {
zurglog.Warnf("Your account will expire in %d hours", userInfo.Premium/3600)
} else {
zurglog.Infof("Your account will expire in %d days", userInfo.Premium/86400)
}
}
remaining := userInfo.Premium - PREMIUM_THRESHOLD
sleepDuration := time.Duration(remaining) * time.Second
time.Sleep(sleepDuration)
}
}()
}