137 lines
3.8 KiB
Go
137 lines
3.8 KiB
Go
package torrent
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
|
|
mapset "github.com/deckarep/golang-set/v2"
|
|
)
|
|
|
|
const BINS_FILE = "data/bins.json"
|
|
|
|
// initializeBins reads from bins.json and assigns values to t.trashBin and t.repairBin
|
|
func (t *TorrentManager) initializeBins() {
|
|
if _, err := os.Stat(BINS_FILE); os.IsNotExist(err) {
|
|
t.repairLog.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(BINS_FILE)
|
|
if err != nil {
|
|
t.repairLog.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.repairLog.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.repairLog.Debugf("Bin immediately: %v", t.ImmediateBin.ToSlice())
|
|
t.repairLog.Debugf("Bin once done: %v", t.OnceDoneBin.ToSlice())
|
|
}
|
|
|
|
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.repairLog.Errorf("Failed to marshal bin data: %v", err)
|
|
return
|
|
}
|
|
|
|
file, err := os.Create(BINS_FILE)
|
|
if err != nil {
|
|
t.repairLog.Errorf("Failed to create bins.json file: %v", err)
|
|
return
|
|
}
|
|
defer file.Close()
|
|
|
|
_, err = file.Write(jsonData)
|
|
if err != nil {
|
|
t.repairLog.Errorf("Failed to write to bins.json file: %v", err)
|
|
}
|
|
}
|
|
|
|
func (t *TorrentManager) setToBinImmediately(torrentId string) {
|
|
t.repairLog.Debugf("id=%s set to delete immediately", torrentId)
|
|
t.ImmediateBin.Add(torrentId)
|
|
t.persistBins()
|
|
}
|
|
|
|
func (t *TorrentManager) setToBinOnceDone(torrentId string) {
|
|
t.repairLog.Debugf("id=%s set to delete once it completes", torrentId)
|
|
t.OnceDoneBin.Add(torrentId)
|
|
t.persistBins()
|
|
}
|
|
|
|
func (t *TorrentManager) setXToBinOnceYDone(deleteId, completeId string) {
|
|
t.repairLog.Debugf("id=%s set to delete once id=%s completes", deleteId, completeId)
|
|
t.OnceDoneBin.Add(fmt.Sprintf("%s-", completeId))
|
|
t.OnceDoneBin.Add(fmt.Sprintf("%s-%s", completeId, deleteId))
|
|
t.persistBins()
|
|
}
|
|
|
|
func (t *TorrentManager) binImmediately(torrentId string) bool {
|
|
if t.ImmediateBin.Contains(torrentId) {
|
|
if err := t.api.DeleteTorrent(torrentId); err != nil {
|
|
t.repairLog.Warnf("Failed to delete torrent %s: %v", torrentId, err)
|
|
}
|
|
t.ImmediateBin.Remove(torrentId)
|
|
t.persistBins()
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (t *TorrentManager) binOnceDoneErrorCheck(torrentId, status string) bool {
|
|
if status == "downloading" || status == "downloaded" || status == "uploading" || status == "queued" || status == "compressing" {
|
|
return false
|
|
}
|
|
return t.binOnceDone(torrentId)
|
|
}
|
|
|
|
func (t *TorrentManager) binOnceDone(torrentId string) bool {
|
|
if t.OnceDoneBin.Contains(torrentId) {
|
|
if err := t.api.DeleteTorrent(torrentId); err != nil {
|
|
t.repairLog.Warnf("Failed to delete torrent %s: %v", torrentId, err)
|
|
}
|
|
t.OnceDoneBin.Remove(torrentId)
|
|
t.persistBins()
|
|
return true
|
|
}
|
|
|
|
// special case: yyy-xxx means if yyy is done, delete xxx
|
|
specialCase := fmt.Sprintf("%s-", torrentId)
|
|
if !t.OnceDoneBin.Contains(specialCase) {
|
|
return false
|
|
}
|
|
t.OnceDoneBin.Clone().Each(func(entry string) bool {
|
|
if strings.Contains(entry, specialCase) {
|
|
idToDelete := strings.Split(entry, "-")[1]
|
|
if err := t.api.DeleteTorrent(idToDelete); err != nil {
|
|
t.repairLog.Warnf("Failed to delete torrent %s: %v", idToDelete, err)
|
|
}
|
|
t.OnceDoneBin.Remove(entry)
|
|
}
|
|
return false
|
|
})
|
|
t.OnceDoneBin.Remove(specialCase)
|
|
t.persistBins()
|
|
return true
|
|
}
|