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

Save when its complete

Browse Source
This commit is contained in:
Ben Sarmiento
2024-02-17 02:23:38 +01:00
parent 95882ace3f
commit 586466ce80
2 changed files with 17 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
internal/config/v1.go
View File

@@ -220,17 +220,24 @@ func (z *ZurgConfigV1) matchFilter(torrentName string, torrentSize int64, torren
return false return false
} }
if filter.HasEpisodes { if filter.HasEpisodes {
regex := regexp.MustCompile(`(?i)s\d{2,3}.?e\d{2,3}`) regexes := []*regexp.Regexp{
if regex.MatchString(torrentName) { regexp.MustCompile(`(?i)s\d\d\d?.?e\d\d\d?`),
return true regexp.MustCompile(`(?i)seasons?\s?\d+`),
regexp.MustCompile(`(?i)episodes?\s?\d+`),
regexp.MustCompile(`(?i)[a-fA-F0-9]{8}`),
} }
for _, filename := range fileNames { for _, regex := range regexes {
if regex.MatchString(filename) { if regex.MatchString(torrentName) {
return true return true
} }
for _, filename := range fileNames {
if regex.MatchString(filename) {
return true
}
}
} }
//remove resolution from filenames //remove resolution from filenames
regex = regexp.MustCompile(`(?i)((720|1080|2160|480|360|240|144)[pi]|\d{3,4}x\d{3,4})`) regex := regexp.MustCompile(`(?i)((720|1080|2160|480|360|240|144)[pi]|\d{3,4}x\d{3,4})`)
for i, filename := range fileNames { for i, filename := range fileNames {
fileNames[i] = regex.ReplaceAllString(filename, "") fileNames[i] = regex.ReplaceAllString(filename, "")
} }

11
internal/torrent/refresh.go
View File

@@ -220,16 +220,13 @@ func (t *TorrentManager) getMoreInfo(rdTorrent realdebrid.Torrent) *Torrent {
} }
torrent.DownloadedIDs = mapset.NewSet[string]() torrent.DownloadedIDs = mapset.NewSet[string]()
torrent.InProgressIDs = mapset.NewSet[string]() torrent.InProgressIDs = mapset.NewSet[string]()
if info.Progress == 100 { if rdTorrent.Progress == 100 {
torrent.DownloadedIDs.Add(info.ID) torrent.DownloadedIDs.Add(info.ID)
} else { // save to cache if it's not in progress anymore
torrent.InProgressIDs.Add(info.ID)
}
// save to cache if it's not in progress anymore
if info.Progress == 100 {
infoCache.Set(rdTorrent.ID, &torrent) infoCache.Set(rdTorrent.ID, &torrent)
t.saveTorrentChangesToDisk(&torrent, nil) t.saveTorrentChangesToDisk(&torrent, nil)
} else {
torrent.InProgressIDs.Add(info.ID)
} }
return &torrent return &torrent