Add bin other torrent when torrent is completed

internal/torrent/bins.go

@@ -0,0 +1,132 @@
+package torrent
+
+import (
+	"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.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(BINS_FILE)
+	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.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
+	}
+
+	// special case: xxx-yyy means if xxx is done, delete yyy
+	found := false
+	specialCases := t.OnceDoneBin.ToSlice()
+	for _, specialCase := range specialCases {
+		if strings.Contains(specialCase, "-") {
+			parts := strings.Split(specialCase, "-")
+			if parts[0] == torrentId {
+				if err := t.api.DeleteTorrent(parts[1]); err != nil {
+					t.log.Errorf("Failed to delete torrent %s: %v", parts[1], err)
+					continue
+				}
+				t.OnceDoneBin.Remove(specialCase)
+				found = true
+			}
+		}
+	}
+
+	return found
+}
+
+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(BINS_FILE)
+	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()
+}