2b2a682218
- Refined download uncached options - Deprecate qbittorent log level - Skip Repair for specified arr
95 lines
2.8 KiB
Go
95 lines
2.8 KiB
Go
package debrid
|
|
|
|
import (
|
|
"cmp"
|
|
"fmt"
|
|
"github.com/sirrobot01/debrid-blackhole/internal/cache"
|
|
"github.com/sirrobot01/debrid-blackhole/internal/config"
|
|
"github.com/sirrobot01/debrid-blackhole/internal/utils"
|
|
"github.com/sirrobot01/debrid-blackhole/pkg/arr"
|
|
"github.com/sirrobot01/debrid-blackhole/pkg/debrid/alldebrid"
|
|
"github.com/sirrobot01/debrid-blackhole/pkg/debrid/debrid_link"
|
|
"github.com/sirrobot01/debrid-blackhole/pkg/debrid/engine"
|
|
"github.com/sirrobot01/debrid-blackhole/pkg/debrid/realdebrid"
|
|
"github.com/sirrobot01/debrid-blackhole/pkg/debrid/torbox"
|
|
"github.com/sirrobot01/debrid-blackhole/pkg/debrid/torrent"
|
|
)
|
|
|
|
func New() *engine.Engine {
|
|
cfg := config.GetConfig()
|
|
maxCachedSize := cmp.Or(cfg.MaxCacheSize, 1000)
|
|
debrids := make([]engine.Service, 0)
|
|
// Divide the cache size by the number of debrids
|
|
maxCacheSize := maxCachedSize / len(cfg.Debrids)
|
|
|
|
for _, dc := range cfg.Debrids {
|
|
d := createDebrid(dc, cache.New(maxCacheSize))
|
|
logger := d.GetLogger()
|
|
logger.Info().Msg("Debrid Service started")
|
|
debrids = append(debrids, d)
|
|
}
|
|
d := &engine.Engine{Debrids: debrids, LastUsed: 0}
|
|
return d
|
|
}
|
|
|
|
func createDebrid(dc config.Debrid, cache *cache.Cache) engine.Service {
|
|
switch dc.Name {
|
|
case "realdebrid":
|
|
return realdebrid.New(dc, cache)
|
|
case "torbox":
|
|
return torbox.New(dc, cache)
|
|
case "debridlink":
|
|
return debrid_link.New(dc, cache)
|
|
case "alldebrid":
|
|
return alldebrid.New(dc, cache)
|
|
default:
|
|
return realdebrid.New(dc, cache)
|
|
}
|
|
}
|
|
|
|
func ProcessTorrent(d *engine.Engine, magnet *utils.Magnet, a *arr.Arr, isSymlink, downloadUncached bool) (*torrent.Torrent, error) {
|
|
debridTorrent := &torrent.Torrent{
|
|
InfoHash: magnet.InfoHash,
|
|
Magnet: magnet,
|
|
Name: magnet.Name,
|
|
Arr: a,
|
|
Size: magnet.Size,
|
|
DownloadUncached: cmp.Or(downloadUncached, a.DownloadUncached),
|
|
}
|
|
|
|
errs := make([]error, 0)
|
|
|
|
for index, db := range d.Debrids {
|
|
logger := db.GetLogger()
|
|
logger.Info().Msgf("Processing debrid: %s", db.GetName())
|
|
|
|
logger.Info().Msgf("Torrent Hash: %s", debridTorrent.InfoHash)
|
|
if db.GetCheckCached() {
|
|
hash, exists := db.IsAvailable([]string{debridTorrent.InfoHash})[debridTorrent.InfoHash]
|
|
if !exists || !hash {
|
|
logger.Info().Msgf("Torrent: %s is not cached", debridTorrent.Name)
|
|
continue
|
|
} else {
|
|
logger.Info().Msgf("Torrent: %s is cached(or downloading)", debridTorrent.Name)
|
|
}
|
|
}
|
|
|
|
dbt, err := db.SubmitMagnet(debridTorrent)
|
|
if dbt != nil {
|
|
dbt.Arr = a
|
|
}
|
|
if err != nil || dbt == nil || dbt.Id == "" {
|
|
errs = append(errs, err)
|
|
continue
|
|
}
|
|
logger.Info().Msgf("Torrent: %s(id=%s) submitted to %s", dbt.Name, dbt.Id, db.GetName())
|
|
d.LastUsed = index
|
|
return db.CheckStatus(dbt, isSymlink)
|
|
}
|
|
err := fmt.Errorf("failed to process torrent")
|
|
for _, e := range errs {
|
|
err = fmt.Errorf("%w\n%w", err, e)
|
|
}
|
|
return nil, err
|
|
}
|