94 lines
2.5 KiB
Go
94 lines
2.5 KiB
Go
package torrent
|
|
|
|
import (
|
|
"os"
|
|
|
|
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.Info("data/bins.json does not exist. Initializing empty bins.")
|
|
return
|
|
}
|
|
|
|
fileData, err := os.ReadFile(BINS_FILE)
|
|
if err != nil {
|
|
t.repairLog.Errorf("Failed to read bins.json file: %v Initializing empty bins.", err)
|
|
return
|
|
}
|
|
|
|
data := map[string][]string{}
|
|
|
|
err = json.Unmarshal(fileData, &data)
|
|
if err != nil {
|
|
t.repairLog.Errorf("Failed to unmarshal bin data: %v Initializing empty bins.", err)
|
|
return
|
|
}
|
|
|
|
t.OnceDoneBin = mapset.NewSet[string](data["repair_bin"]...)
|
|
|
|
t.repairLog.Debugf("These IDs will be deleted after completion: %v", t.OnceDoneBin.ToSlice())
|
|
}
|
|
|
|
func (t *TorrentManager) persistBins() {
|
|
data := map[string]interface{}{
|
|
"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) willDeleteOnceDone(torrentId string) {
|
|
t.repairLog.Debugf("Torrent id=%s will be deleted once it is completed", torrentId)
|
|
t.OnceDoneBin.Add(torrentId)
|
|
t.persistBins()
|
|
}
|
|
|
|
func (t *TorrentManager) cleanupBins(freshIDs mapset.Set[string]) {
|
|
t.OnceDoneBin.Clone().Each(func(entry string) bool {
|
|
// check for: delete once done cases
|
|
if !freshIDs.ContainsOne(entry) {
|
|
t.OnceDoneBin.Remove(entry)
|
|
}
|
|
return false
|
|
})
|
|
t.persistBins()
|
|
}
|
|
|
|
// deleteOnceDone checks if the torrent is in the OnceDoneBin and deletes it if it is.
|
|
// returns true if the torrent was in the bin and was deleted, false otherwise
|
|
func (t *TorrentManager) deleteOnceDone(completedTorrentId string, failed bool) {
|
|
if !t.OnceDoneBin.ContainsOne(completedTorrentId) {
|
|
return
|
|
}
|
|
|
|
t.DeleteByID(completedTorrentId)
|
|
t.OnceDoneBin.Remove(completedTorrentId)
|
|
if failed {
|
|
t.repairLog.Infof("Deleting torrent id=%s early because it has encountered an error", completedTorrentId)
|
|
} else {
|
|
t.repairLog.Debugf("Deleting unnecessary torrent id=%s", completedTorrentId)
|
|
}
|
|
t.persistBins()
|
|
}
|