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

Add bins to home page

Browse Source
This commit is contained in:
Ben Sarmiento
2024-05-25 01:54:35 +02:00
parent abb80b4257
commit a681912fc1
3 changed files with 53 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
internal/torrent/refresh.go
View File

@@ -412,16 +412,16 @@ func (t *TorrentManager) IsPlayable(filePath string) bool {
func (t *TorrentManager) initializeBins() {
if _, err := os.Stat("data/bins.json"); os.IsNotExist(err) {
t.log.Warn("data/bins.json does not exist. Initializing empty bins.")
t.immediateBin = mapset.NewSet[string]()
t.onceDoneBin = mapset.NewSet[string]()
t.ImmediateBin = mapset.NewSet[string]()
t.OnceDoneBin = mapset.NewSet[string]()
return
}
fileData, err := os.ReadFile("data/bins.json")
if err != nil {
t.log.Errorf("Failed to read bins.json file: %v", err)
t.immediateBin = mapset.NewSet[string]()
t.onceDoneBin = mapset.NewSet[string]()
t.ImmediateBin = mapset.NewSet[string]()
t.OnceDoneBin = mapset.NewSet[string]()
return
}
@@ -433,44 +433,44 @@ func (t *TorrentManager) initializeBins() {
return
}
t.immediateBin = mapset.NewSet[string](data["trash_bin"]...)
t.onceDoneBin = mapset.NewSet[string](data["repair_bin"]...)
t.ImmediateBin = mapset.NewSet[string](data["trash_bin"]...)
t.OnceDoneBin = mapset.NewSet[string](data["repair_bin"]...)
t.log.Debug("Successfully read bins from bins.json")
t.log.Debugf("Bin immediately: %v", t.immediateBin.ToSlice())
t.log.Debugf("Bin once done: %v", t.onceDoneBin.ToSlice())
t.log.Debugf("Bin immediately: %v", t.ImmediateBin.ToSlice())
t.log.Debugf("Bin once done: %v", t.OnceDoneBin.ToSlice())
}
func (t *TorrentManager) setToBinImmediately(torrentId string) {
t.log.Debugf("Set to delete immediately: %s", torrentId)
t.immediateBin.Add(torrentId)
t.ImmediateBin.Add(torrentId)
t.persistBins()
}
func (t *TorrentManager) setToBinOnceDone(torrentId string) {
t.log.Debugf("Set to delete once completed: %s", torrentId)
t.onceDoneBin.Add(torrentId)
t.OnceDoneBin.Add(torrentId)
t.persistBins()
}
func (t *TorrentManager) binImmediately(torrentId string) bool {
if t.immediateBin.Contains(torrentId) {
if t.ImmediateBin.Contains(torrentId) {
if err := t.api.DeleteTorrent(torrentId); err != nil {
t.log.Errorf("Failed to delete torrent %s: %v", torrentId, err)
return false
}
t.immediateBin.Remove(torrentId)
t.ImmediateBin.Remove(torrentId)
return true
}
return false
}
func (t *TorrentManager) binOnceDone(torrentId string) bool {
if t.onceDoneBin.Contains(torrentId) {
if t.OnceDoneBin.Contains(torrentId) {
if err := t.api.DeleteTorrent(torrentId); err != nil {
t.log.Errorf("Failed to delete torrent %s: %v", torrentId, err)
return false
}
t.onceDoneBin.Remove(torrentId)
t.OnceDoneBin.Remove(torrentId)
return true
}
return false
@@ -478,8 +478,8 @@ func (t *TorrentManager) binOnceDone(torrentId string) bool {
func (t *TorrentManager) persistBins() {
data := map[string]interface{}{
"trash_bin": t.immediateBin.ToSlice(), // Assuming trashBin is a mapset.Set[string]
"repair_bin": t.onceDoneBin.ToSlice(), // Assuming repairBin is a mapset.Set[string]
"trash_bin": t.ImmediateBin.ToSlice(), // Assuming trashBin is a mapset.Set[string]
"repair_bin": t.OnceDoneBin.ToSlice(), // Assuming repairBin is a mapset.Set[string]
}
jsonData, err := json.Marshal(data)
@@ -502,12 +502,12 @@ func (t *TorrentManager) persistBins() {
}
func (t *TorrentManager) cleanupBins(freshIDs mapset.Set[string]) {
t.immediateBin.Difference(freshIDs).Each(func(id string) bool {
t.immediateBin.Remove(id)
t.ImmediateBin.Difference(freshIDs).Each(func(id string) bool {
t.ImmediateBin.Remove(id)
return false
})
t.onceDoneBin.Difference(freshIDs).Each(func(id string) bool {
t.onceDoneBin.Remove(id)
t.OnceDoneBin.Difference(freshIDs).Each(func(id string) bool {
t.OnceDoneBin.Remove(id)
return false
})
t.persistBins()