Add premium status monitoring and it wont start unless you are premium
This commit is contained in:
@@ -14,6 +14,7 @@ import (
|
|||||||
"github.com/debridmediamanager/zurg/internal/universal"
|
"github.com/debridmediamanager/zurg/internal/universal"
|
||||||
"github.com/debridmediamanager/zurg/pkg/http"
|
"github.com/debridmediamanager/zurg/pkg/http"
|
||||||
"github.com/debridmediamanager/zurg/pkg/logutil"
|
"github.com/debridmediamanager/zurg/pkg/logutil"
|
||||||
|
"github.com/debridmediamanager/zurg/pkg/premium"
|
||||||
"github.com/debridmediamanager/zurg/pkg/realdebrid"
|
"github.com/debridmediamanager/zurg/pkg/realdebrid"
|
||||||
"github.com/debridmediamanager/zurg/pkg/utils"
|
"github.com/debridmediamanager/zurg/pkg/utils"
|
||||||
"github.com/go-chi/chi/v5"
|
"github.com/go-chi/chi/v5"
|
||||||
@@ -38,6 +39,8 @@ func MainApp(configPath string) {
|
|||||||
|
|
||||||
rd := realdebrid.NewRealDebrid(apiClient, log.Named("realdebrid"))
|
rd := realdebrid.NewRealDebrid(apiClient, log.Named("realdebrid"))
|
||||||
|
|
||||||
|
premium.MonitorPremiumStatus(rd, zurglog)
|
||||||
|
|
||||||
p, err := ants.NewPool(config.GetNumOfWorkers() + 1)
|
p, err := ants.NewPool(config.GetNumOfWorkers() + 1)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
zurglog.Errorf("Failed to create worker pool: %v", err)
|
zurglog.Errorf("Failed to create worker pool: %v", err)
|
||||||
|
|||||||
35
pkg/premium/monitor.go
Normal file
35
pkg/premium/monitor.go
Normal 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)
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user