Add bin other torrent when torrent is completed
This commit is contained in:
@@ -3,7 +3,6 @@ package torrent
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
"sync"
|
||||
@@ -152,7 +151,7 @@ func (t *TorrentManager) refreshTorrents() []string {
|
||||
existingIDs := mapset.NewSet[string]()
|
||||
t.getInfoFiles().Each(func(path string) bool {
|
||||
path = filepath.Base(path)
|
||||
torrentID := strings.TrimSuffix(path, ".info_zurg")
|
||||
torrentID := strings.TrimSuffix(path, ".zurginfo")
|
||||
if !t.binOnceDone(torrentID) {
|
||||
existingIDs.Add(torrentID)
|
||||
}
|
||||
@@ -407,108 +406,3 @@ func (t *TorrentManager) IsPlayable(filePath string) bool {
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
// initializeBins reads from bins.json and assigns values to t.trashBin and t.repairBin
|
||||
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]()
|
||||
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]()
|
||||
return
|
||||
}
|
||||
|
||||
data := map[string][]string{}
|
||||
|
||||
err = json.Unmarshal(fileData, &data)
|
||||
if err != nil {
|
||||
t.log.Errorf("Failed to unmarshal bin data: %v", err)
|
||||
return
|
||||
}
|
||||
|
||||
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())
|
||||
}
|
||||
func (t *TorrentManager) setToBinImmediately(torrentId string) {
|
||||
t.log.Debugf("Set to delete immediately: %s", 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.persistBins()
|
||||
}
|
||||
|
||||
func (t *TorrentManager) binImmediately(torrentId string) bool {
|
||||
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)
|
||||
return true
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
func (t *TorrentManager) binOnceDone(torrentId string) bool {
|
||||
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)
|
||||
return true
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
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]
|
||||
}
|
||||
|
||||
jsonData, err := json.Marshal(data)
|
||||
if err != nil {
|
||||
t.log.Errorf("Failed to marshal bin data: %v", err)
|
||||
return
|
||||
}
|
||||
|
||||
file, err := os.Create("data/bins.json")
|
||||
if err != nil {
|
||||
t.log.Errorf("Failed to create bins.json file: %v", err)
|
||||
return
|
||||
}
|
||||
defer file.Close()
|
||||
|
||||
_, err = file.Write(jsonData)
|
||||
if err != nil {
|
||||
t.log.Errorf("Failed to write to bins.json file: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
func (t *TorrentManager) cleanupBins(freshIDs mapset.Set[string]) {
|
||||
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)
|
||||
return false
|
||||
})
|
||||
t.persistBins()
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user