Refactor workers

internal/torrent/refresh.go

@@ -9,11 +9,12 @@ import (
 
 	"github.com/debridmediamanager/zurg/internal/config"
 	"github.com/debridmediamanager/zurg/pkg/realdebrid"
+	"github.com/debridmediamanager/zurg/pkg/utils"
 	mapset "github.com/deckarep/golang-set/v2"
 	cmap "github.com/orcaman/concurrent-map/v2"
 )
 
-func (t *TorrentManager) RefreshTorrents() []string {
+func (t *TorrentManager) refreshTorrents() []string {
 	instances, _, err := t.Api.GetTorrents(0, false)
 	if err != nil {
 		t.log.Warnf("Cannot get torrents: %v", err)
@@ -120,28 +121,24 @@ func (t *TorrentManager) RefreshTorrents() []string {
 
 // startRefreshJob periodically refreshes the torrents
 func (t *TorrentManager) startRefreshJob() {
-	t.log.Info("Starting periodic refresh")
-	for {
-		<-time.After(time.Duration(t.Config.GetRefreshEverySeconds()) * time.Second)
+	_ = t.workerPool.Submit(func() {
+		t.log.Info("Starting periodic refresh")
+		for {
+			<-time.After(time.Duration(t.Config.GetRefreshEverySeconds()) * time.Second)
 
-		checksum := t.getCurrentState()
-		if t.latestState.equal(checksum) {
-			continue
+			checksum := t.getCurrentState()
+			if t.latestState.equal(checksum) {
+				continue
+			}
+			t.SetNewLatestState(checksum)
+			t.log.Infof("Detected changes! Refreshing %d torrents", checksum.TotalCount)
+
+			updatedPaths := t.refreshTorrents()
+			t.log.Info("Finished refreshing torrents")
+
+			t.TriggerHookOnLibraryUpdate(updatedPaths)
 		}
-		t.SetNewLatestState(checksum)
-		t.log.Infof("Detected changes! Refreshing %d torrents", checksum.TotalCount)
-
-		updatedPaths := t.RefreshTorrents()
-		t.log.Info("Finished refreshing torrents")
-
-		t.TriggerHookOnLibraryUpdate(updatedPaths)
-
-		if t.Config.EnableRepair() {
-			t.RepairAll()
-		} else {
-			t.log.Info("Repair is disabled, skipping repair check")
-		}
-	}
+	})
 }
 
 // getMoreInfo gets original name, size and files for a torrent
@@ -313,3 +310,35 @@ func (t *TorrentManager) mergeToMain(existing, toMerge *Torrent) Torrent {
 
 	return mainTorrent
 }
+
+func (t *TorrentManager) assignedDirectoryCb(tor *Torrent, cb func(string)) {
+	torrentIDs := tor.DownloadedIDs.Union(tor.InProgressIDs).ToSlice()
+	// get filenames needed for directory conditions
+	var filenames []string
+	var fileSizes []int64
+	unplayable := true
+	tor.SelectedFiles.IterCb(func(key string, file *File) {
+		filenames = append(filenames, filepath.Base(file.Path))
+		fileSizes = append(fileSizes, file.Bytes)
+		if unplayable && utils.IsStreamable(file.Path) {
+			unplayable = false
+		}
+	})
+	// Map torrents to directories
+	switch t.Config.GetVersion() {
+	case "v1":
+		configV1 := t.Config.(*config.ZurgConfigV1)
+		for _, directories := range configV1.GetGroupMap() {
+			for _, directory := range directories {
+				if unplayable {
+					cb(config.UNPLAYABLE_TORRENTS)
+					break
+				}
+				if t.Config.MeetsConditions(directory, t.GetKey(tor), tor.ComputeTotalSize(), torrentIDs, filenames, fileSizes) {
+					cb(directory)
+					break
+				}
+			}
+		}
+	}
+}