94 lines
2.1 KiB
Go
94 lines
2.1 KiB
Go
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)
|
|
}
|