A lot of rewrite here

internal/torrent/cache.go

@@ -0,0 +1 @@
+package torrent

internal/torrent/manager.go

@@ -0,0 +1,93 @@
+package torrent
+
+import (
+	"log"
+
+	"github.com/debridmediamanager.com/zurg/pkg/realdebrid"
+	"github.com/dgraph-io/ristretto"
+)
+
+type TorrentManager struct {
+	token      string
+	cache      *ristretto.Cache
+	workerPool chan bool
+}
+
+// NewTorrentManager creates a new torrent manager
+// it will fetch all torrents and their info in the background
+// and cache them
+func NewTorrentManager(token string) *TorrentManager {
+	cache, err := ristretto.NewCache(&ristretto.Config{
+		NumCounters: 1e7,     // number of keys to track frequency of (10M).
+		MaxCost:     1 << 30, // maximum cost of cache (1GB).
+		BufferItems: 64,      // number of keys per Get buffer.
+	})
+	if err != nil {
+		panic(err)
+	}
+
+	handler := &TorrentManager{
+		token:      token,
+		cache:      cache,
+		workerPool: make(chan bool, 10),
+	}
+
+	torrents := handler.getAll()
+
+	for _, torrent := range torrents {
+		go func(id string) {
+			handler.workerPool <- true
+			handler.getInfo(id)
+			// sleep for 1 second to avoid rate limiting
+			<-handler.workerPool
+		}(torrent.ID)
+	}
+
+	return handler
+}
+
+func (t *TorrentManager) getAll() []realdebrid.Torrent {
+	cacheKey := "t:all"
+	torrents, err := realdebrid.GetTorrents(t.token)
+	if err != nil {
+		log.Printf("Cannot get torrents: %v\n", err.Error())
+		return nil
+	}
+	t.cache.Set(cacheKey, torrents, 0)
+	return torrents
+}
+
+func (t *TorrentManager) GetAll() []realdebrid.Torrent {
+	cacheKey := "t:all"
+	if data, found := t.cache.Get(cacheKey); found {
+		if cachedTorrents, ok := data.([]realdebrid.Torrent); ok {
+			return cachedTorrents
+		} else {
+			t.cache.Del(cacheKey)
+		}
+	}
+	return t.getAll()
+}
+
+func (t *TorrentManager) getInfo(torrentID string) *realdebrid.Torrent {
+	cacheKey := "t:" + torrentID
+	info, err := realdebrid.GetTorrentInfo(t.token, torrentID)
+	if err != nil {
+		log.Printf("Cannot get info: %v\n", err.Error())
+		return nil
+	}
+	t.cache.Set(cacheKey, info, 0)
+	return info
+}
+
+func (t *TorrentManager) GetInfo(torrentID string) *realdebrid.Torrent {
+	cacheKey := "t:" + torrentID
+	if data, found := t.cache.Get(cacheKey); found {
+		if torrent, ok := data.(*realdebrid.Torrent); ok {
+			return torrent
+		} else {
+			t.cache.Del(cacheKey)
+		}
+	}
+	return t.getInfo(torrentID)
+}