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

Add accidental removal fix

Browse Source
This commit is contained in:
Ben Adrian Sarmiento
2024-07-19 04:42:09 +02:00
parent eaabd20a83
commit d2c1e12e2c
6 changed files with 17 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
internal/torrent/delete.go
View File

@@ -1,6 +1,7 @@
package torrent
import (
"github.com/debridmediamanager/zurg/internal/config"
cmap "github.com/orcaman/concurrent-map/v2"
)
@@ -32,8 +33,12 @@ func (t *TorrentManager) Delete(accessKey string, deleteInRD bool) {
}
// t.log.Infof("Removing torrent %s from zurg database (not real-debrid)", accessKey)
t.DirectoryMap.IterCb(func(directory string, torrents cmap.ConcurrentMap[string, *Torrent]) {
if directory == config.DOWNLOADS || directory == config.DUMPED_TORRENTS {
return
}
torrents.Remove(accessKey)
})
allTorrents.Remove(accessKey)
}

6
internal/torrent/manager.go
View File

@@ -535,8 +535,12 @@ func (t *TorrentManager) initializeDirectoryMaps() {
// create directory maps
for _, directory := range t.Config.GetDirectories() {
t.DirectoryMap.Set(directory, cmap.New[*Torrent]())
// t.RootNode.AddChild(fs.NewFileNode(directory, true))
}
// create special directories
t.DirectoryMap.Set(config.ALL_TORRENTS, cmap.New[*Torrent]())
t.DirectoryMap.Set(config.DOWNLOADS, cmap.New[*Torrent]())
t.DirectoryMap.Set(config.DUMPED_TORRENTS, cmap.New[*Torrent]())
t.DirectoryMap.Set(config.UNPLAYABLE_TORRENTS, cmap.New[*Torrent]())
}
func (t *TorrentManager) getTorrentFilename(torrent *Torrent) string {

2
internal/torrent/refresh.go
View File

@@ -357,7 +357,7 @@ func (t *TorrentManager) assignDirectory(tor *Torrent, triggerHook bool, outputL
accessKey := t.GetKey(tor)
t.DirectoryMap.IterCb(func(directory string, torrents cmap.ConcurrentMap[string, *Torrent]) {
if strings.HasPrefix(directory, "int__") {
if directory == config.DOWNLOADS || directory == config.DUMPED_TORRENTS {
return
}
torrents.Remove(accessKey)

9
internal/torrent/repair.go
View File

@@ -590,13 +590,10 @@ func (t *TorrentManager) canCapacityHandle() bool {
}
func (t *TorrentManager) markAsUnplayable(torrent *Torrent) {
// reassign to unplayable torrents directory
t.DirectoryMap.IterCb(func(directory string, torrents cmap.ConcurrentMap[string, *Torrent]) {
if strings.HasPrefix(directory, "int__") {
return
}
for _, directory := range t.Config.GetDirectories() {
torrents, _ := t.DirectoryMap.Get(directory)
torrents.Remove(t.GetKey(torrent))
})
}
torrents, _ := t.DirectoryMap.Get(config.UNPLAYABLE_TORRENTS)
torrents.Set(t.GetKey(torrent), torrent)
}