Prevent double repairs

internal/torrent/delete.go

@@ -12,7 +12,9 @@ func (t *TorrentManager) CheckDeletedStatus(torrent *Torrent) bool {
 	if len(unselectedIDs) == torrent.SelectedFiles.Count() && len(unselectedIDs) > 0 {
 		return true
 	} else if len(unselectedIDs) > 0 {
+		infoCache, _ := t.DirectoryMap.Get(INT_INFO_CACHE)
 		torrent.DownloadedIDs.Each(func(id string) bool {
+			infoCache.Set(id, torrent)
 			t.writeTorrentToFile(id, torrent)
 			return true
 		})

internal/torrent/refresh.go

@@ -23,12 +23,12 @@ func (t *TorrentManager) RefreshTorrents() []string {
 	var wg sync.WaitGroup
 
 	for i := range instances {
-		wg.Add(1)
-		idx := i // capture the loop variable
+		idx := i
 		_ = t.workerPool.Submit(func() {
 			defer wg.Done()
 			infoChan <- t.getMoreInfo(instances[idx])
 		})
+		wg.Add(1)
 	}
 
 	wg.Wait()
@@ -43,7 +43,7 @@ func (t *TorrentManager) RefreshTorrents() []string {
 			noInfoCount++
 			continue
 		}
-		if !info.AllInProgress() {
+		if !info.AnyInProgress() {
 			freshKeys.Add(info.AccessKey)
 		}
 		if torrent, exists := allTorrents.Get(info.AccessKey); !exists {

internal/torrent/repair.go

@@ -36,6 +36,12 @@ func (t *TorrentManager) repairAll() {
 }
 
 func (t *TorrentManager) Repair(torrent *Torrent) {
+	infoCache, _ := t.DirectoryMap.Get(INT_INFO_CACHE)
+	torrent.DownloadedIDs.Each(func(id string) bool {
+		infoCache.Set(id, torrent)
+		t.writeTorrentToFile(id, torrent)
+		return true
+	})
 	_ = t.repairWorker.Submit(func() {
 		t.log.Infof("Repairing torrent %s", torrent.AccessKey)
 		t.repair(torrent)
@@ -73,8 +79,9 @@ func (t *TorrentManager) repair(torrent *Torrent) {
 	// second solution: add only the missing files
 	var missingFiles []File
 	torrent.SelectedFiles.IterCb(func(_ string, file *File) {
-		if !strings.HasPrefix(file.Link, "http") {
+		if file.Link == "repair" || file.Link == "" {
 			missingFiles = append(missingFiles, *file)
+			file.Link = "repairing"
 		}
 	})
 	t.log.Debugf("During repair, zurg found %d broken files for torrent %s", len(missingFiles), torrent.AccessKey)