Proactive repair

internal/torrent/repair.go

@@ -8,6 +8,7 @@ import (
 
 	"github.com/debridmediamanager/zurg/internal/config"
 	"github.com/debridmediamanager/zurg/pkg/realdebrid"
+	mapset "github.com/deckarep/golang-set/v2"
 	cmap "github.com/orcaman/concurrent-map/v2"
 )
 
@@ -23,14 +24,40 @@ func (t *TorrentManager) RepairAll() {
 func (t *TorrentManager) repairAll() {
 	allTorrents, _ := t.DirectoryMap.Get(INT_ALL)
 
-	hashes := make([]string, 0)
+	var hashGroups []mapset.Set[string]
+	const maxGroupSize = 399
+
+	currentGroup := mapset.NewSet[string]()
+	hashGroups = append(hashGroups, currentGroup)
+
 	allTorrents.IterCb(func(_ string, torrent *Torrent) {
 		if torrent.AnyInProgress() || torrent.Unfixable {
 			return
 		}
-		hashes = append(hashes, torrent.Hash)
+		// Check if the current group is full
+		if currentGroup.Cardinality() >= maxGroupSize {
+			// Create a new group and add it to the hashGroups
+			currentGroup = mapset.NewSet[string]()
+			hashGroups = append(hashGroups, currentGroup)
+		}
+		// Add the hash to the current group
+		currentGroup.Add(torrent.Hash)
 	})
 
+	var availabilityChecks = make(map[string]bool)
+	for i := range hashGroups {
+		resp, err := t.Api.AvailabilityCheck(hashGroups[i].ToSlice())
+		if err != nil {
+			t.log.Warnf("Cannot check availability: %v", err)
+			continue
+		}
+
+		for hash, hosterHash := range resp {
+			// Check if HosterHash is a map (Variants field is used)
+			availabilityChecks[hash] = len(hosterHash.Variants) > 0
+		}
+	}
+
 	var toRepair []*Torrent
 	allTorrents.IterCb(func(_ string, torrent *Torrent) {
 		if torrent.AnyInProgress() || torrent.Unfixable {
@@ -38,6 +65,10 @@ func (t *TorrentManager) repairAll() {
 		}
 
 		// check 1: for cached status
+		isCached := true
+		if _, ok := availabilityChecks[torrent.Hash]; !ok || !availabilityChecks[torrent.Hash] {
+			isCached = false
+		}
 
 		// check 2: for broken files
 		hasBrokenFiles := false
@@ -46,7 +77,8 @@ func (t *TorrentManager) repairAll() {
 				hasBrokenFiles = true
 			}
 		})
-		if hasBrokenFiles {
+
+		if !isCached || hasBrokenFiles {
 			toRepair = append(toRepair, torrent)
 		}
 	})