hpz937/zurg
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Handle repairs

Browse Source
This commit is contained in:
Ben Sarmiento
2024-01-17 06:17:36 +01:00
parent b67c73dc41
commit c12e2cc467
2 changed files with 41 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
internal/torrent/repair.go
View File

@@ -216,7 +216,7 @@ func (t *TorrentManager) repair(torrent *Torrent) {
if len(brokenFiles) > 0 {
t.log.Infof("Redownloading %dof%d files for torrent %s", len(brokenFiles), torrent.SelectedFiles.Count(), t.GetKey(torrent))
brokenFileIDs := strings.Join(getFileIDs(brokenFiles), ",")
if t.reinsertTorrent(torrent, brokenFileIDs) {
if t.redownloadTorrent(torrent, brokenFileIDs) {
t.log.Infof("Successfully downloaded torrent %s to repair it", t.GetKey(torrent))
} else {
t.log.Warnf("Cannot repair torrent %s", t.GetKey(torrent))
@@ -226,14 +226,15 @@ func (t *TorrentManager) repair(torrent *Torrent) {
}
}
func (t *TorrentManager) reinsertTorrent(torrent *Torrent, brokenFiles string) bool {
// broken files means broken links
func (t *TorrentManager) redownloadTorrent(torrent *Torrent, brokenFiles string) bool {
t.log.Debugf("Redownloading torrent %s, broken files=%s (all if empty)", t.GetKey(torrent), brokenFiles)
oldTorrentIDs := make([]string, 0)
// broken files means broken links
// if brokenFiles is not provided
if brokenFiles == "" {
// only replace the torrent if we are reinserting all files
// only replace the torrent if we are redownloading all files
oldTorrentIDs = torrent.DownloadedIDs.ToSlice()
tmpSelection := ""
torrent.SelectedFiles.IterCb(func(_ string, file *File) {
@@ -310,7 +311,7 @@ func (t *TorrentManager) reinsertTorrent(torrent *Torrent, brokenFiles string) b
return false
}
t.log.Infof("Repair successful id=%s", newTorrentID)
t.log.Infof("Redownload successful id=%s", newTorrentID)
t.onlyForRepair.Add(newTorrentID)
if len(oldTorrentIDs) > 0 {
for _, id := range oldTorrentIDs {
@@ -385,3 +386,30 @@ func (t *TorrentManager) markAsUnfixable(torrent *Torrent) {
return false
})
}
func (t *TorrentManager) handleRepairTorrents(info *Torrent) bool {
allTorrents, _ := t.DirectoryMap.Get(INT_ALL)
accessKey := t.GetKey(info)
torrentIDs := info.DownloadedIDs.ToSlice()
inRepairList := false
for _, torrentID := range torrentIDs {
if t.onlyForRepair.Contains(torrentID) {
inRepairList = true
break
}
}
if !info.AnyInProgress() && inRepairList {
t.log.Debugf("Newly downloaded %s (id=%s) is in repair list", info.Name, torrentIDs[0])
torrent, stillExists := allTorrents.Get(accessKey)
if stillExists && t.redownloadTorrent(torrent, "") {
t.log.Debugf("Deleting repair temp id=%s because it has served its purpose", torrentIDs[0])
// if it's 100% and it's a temp repair, remove it
for _, torrentID := range torrentIDs {
t.Api.DeleteTorrent(torrentID)
t.onlyForRepair.Remove(torrentID)
}
return true
}
}
return false
}