package shared

import (
	"cmp"
	"github.com/rs/zerolog"
	"github.com/sirrobot01/debrid-blackhole/common"
	"github.com/sirrobot01/debrid-blackhole/pkg/arr"
	"github.com/sirrobot01/debrid-blackhole/pkg/debrid"
	"os"
)

type QBit struct {
	Username        string   `json:"username"`
	Password        string   `json:"password"`
	Port            string   `json:"port"`
	DownloadFolder  string   `json:"download_folder"`
	Categories      []string `json:"categories"`
	Debrid          *debrid.DebridService
	Storage         *TorrentStorage
	debug           bool
	logger          zerolog.Logger
	Arrs            *arr.Storage
	Tags            []string
	RefreshInterval int
}

func NewQBit(config *common.Config, deb *debrid.DebridService, logger zerolog.Logger, arrs *arr.Storage) *QBit {
	cfg := config.QBitTorrent
	port := cmp.Or(cfg.Port, os.Getenv("QBIT_PORT"), "8182")
	refreshInterval := cmp.Or(cfg.RefreshInterval, 10)
	return &QBit{
		Username:        cfg.Username,
		Password:        cfg.Password,
		Port:            port,
		DownloadFolder:  cfg.DownloadFolder,
		Categories:      cfg.Categories,
		Debrid:          deb,
		Storage:         NewTorrentStorage("torrents.json"),
		logger:          logger,
		Arrs:            arrs,
		RefreshInterval: refreshInterval,
	}
}