From b9699c8a3eb6868d64b40cc9d017c221b4266907 Mon Sep 17 00:00:00 2001
From: Ben Adrian Sarmiento <me@bensarmiento.com>
Date: Thu, 11 Jul 2024 16:50:22 +0200
Subject: [PATCH] Clear download and unrestrict maps

---
 internal/torrent/manager.go | 6 ++++--
 pkg/realdebrid/api.go       | 4 ----
 2 files changed, 4 insertions(+), 6 deletions(-)

diff --git a/internal/torrent/manager.go b/internal/torrent/manager.go
index f57a7fe..8e25461 100644
--- a/internal/torrent/manager.go
+++ b/internal/torrent/manager.go
@@ -381,6 +381,10 @@ func (t *TorrentManager) mountNewDownloads() {
 	token := t.Config.GetToken()
 	tokenMap, _ := t.rd.UnrestrictMap.Get(token)
 
+	// clear maps
+	tokenMap.Clear()
+	t.DownloadMap.Clear()
+
 	downloads := t.rd.GetDownloads()
 	mountedCount := 0
 	for i := range downloads {
@@ -426,10 +430,8 @@ func (t *TorrentManager) StartDownloadsJob() {
 		for {
 			select {
 			case <-remountTicker.C:
-				t.DownloadMap.Clear()
 				t.mountNewDownloads()
 			case <-t.RemountTrigger:
-				t.DownloadMap.Clear()
 				t.mountNewDownloads()
 			}
 		}
diff --git a/pkg/realdebrid/api.go b/pkg/realdebrid/api.go
index af13d27..504449e 100644
--- a/pkg/realdebrid/api.go
+++ b/pkg/realdebrid/api.go
@@ -82,8 +82,6 @@ func (rd *RealDebrid) UnrestrictAndVerify(link string) (*Download, error) {
 			if utils.IsBytesLimitReached(err) {
 				rd.TokenManager.SetTokenAsExpired(token, "bandwidth limit exceeded")
 				continue
-			} else if utils.IsInvalidDownloadCode(err) {
-				continue
 			} else if err == nil {
 				rd.verifiedLinks.Set(download.ID, time.Now().Unix()+DOWNLOAD_LINK_EXPIRY)
 				return download, nil
@@ -103,8 +101,6 @@ func (rd *RealDebrid) UnrestrictAndVerify(link string) (*Download, error) {
 		if utils.IsBytesLimitReached(err) {
 			rd.TokenManager.SetTokenAsExpired(token, "bandwidth limit exceeded")
 			continue
-		} else if utils.IsInvalidDownloadCode(err) {
-			continue
 		} else if err != nil {
 			return nil, err
 		}