decypharr/pkg/debrid/torbox.go

package debrid

import (
	"bytes"
	"encoding/json"
	"fmt"
	"github.com/rs/zerolog"
	"github.com/sirrobot01/debrid-blackhole/common"
	"github.com/sirrobot01/debrid-blackhole/internal/config"
	"github.com/sirrobot01/debrid-blackhole/internal/logger"
	"github.com/sirrobot01/debrid-blackhole/internal/request"
	"github.com/sirrobot01/debrid-blackhole/pkg/debrid/types"
	"log"
	"mime/multipart"
	"net/http"
	gourl "net/url"
	"os"
	"path"
	"path/filepath"
	"slices"
	"strconv"
	"strings"
)

type Torbox struct {
	BaseDebrid
}

func (r *Torbox) GetMountPath() string {
	return r.MountPath
}

func (r *Torbox) GetName() string {
	return r.Name
}

func (r *Torbox) GetLogger() zerolog.Logger {
	return r.logger
}

func (r *Torbox) IsAvailable(infohashes []string) map[string]bool {
	// Check if the infohashes are available in the local cache
	hashes, result := GetLocalCache(infohashes, r.cache)

	if len(hashes) == 0 {
		// Either all the infohashes are locally cached or none are
		r.cache.AddMultiple(result)
		return result
	}

	// Divide hashes into groups of 100
	for i := 0; i < len(hashes); i += 100 {
		end := i + 100
		if end > len(hashes) {
			end = len(hashes)
		}

		// Filter out empty strings
		validHashes := make([]string, 0, end-i)
		for _, hash := range hashes[i:end] {
			if hash != "" {
				validHashes = append(validHashes, hash)
			}
		}

		// If no valid hashes in this batch, continue to the next batch
		if len(validHashes) == 0 {
			continue
		}

		hashStr := strings.Join(validHashes, ",")
		url := fmt.Sprintf("%s/api/torrents/checkcached?hash=%s", r.Host, hashStr)
		req, _ := http.NewRequest(http.MethodGet, url, nil)
		resp, err := r.client.MakeRequest(req)
		if err != nil {
			r.logger.Info().Msgf("Error checking availability: %v", err)
			return result
		}
		var res types.TorBoxAvailableResponse
		err = json.Unmarshal(resp, &res)
		if err != nil {
			r.logger.Info().Msgf("Error marshalling availability: %v", err)
			return result
		}
		if res.Data == nil {
			return result
		}

		for h, cache := range *res.Data {
			if cache.Size > 0 {
				result[strings.ToUpper(h)] = true
			}
		}
	}
	r.cache.AddMultiple(result) // Add the results to the cache
	return result
}

func (r *Torbox) SubmitMagnet(torrent *Torrent) (*Torrent, error) {
	url := fmt.Sprintf("%s/api/torrents/createtorrent", r.Host)
	payload := &bytes.Buffer{}
	writer := multipart.NewWriter(payload)
	_ = writer.WriteField("magnet", torrent.Magnet.Link)
	err := writer.Close()
	if err != nil {
		return nil, err
	}
	req, _ := http.NewRequest(http.MethodPost, url, payload)
	req.Header.Set("Content-Type", writer.FormDataContentType())
	resp, err := r.client.MakeRequest(req)
	if err != nil {
		return nil, err
	}
	var data types.TorBoxAddMagnetResponse
	err = json.Unmarshal(resp, &data)
	if err != nil {
		return nil, err
	}
	if data.Data == nil {
		return nil, fmt.Errorf("error adding torrent")
	}
	dt := *data.Data
	torrentId := strconv.Itoa(dt.Id)
	log.Printf("Torrent: %s added with id: %s", torrent.Name, torrentId)
	torrent.Id = torrentId

	return torrent, nil
}

func getTorboxStatus(status string, finished bool) string {
	if finished {
		return "downloaded"
	}
	downloading := []string{"completed", "cached", "paused", "downloading", "uploading",
		"checkingResumeData", "metaDL", "pausedUP", "queuedUP", "checkingUP",
		"forcedUP", "allocating", "downloading", "metaDL", "pausedDL",
		"queuedDL", "checkingDL", "forcedDL", "checkingResumeData", "moving"}
	switch {
	case slices.Contains(downloading, status):
		return "downloading"
	default:
		return "error"
	}
}

func (r *Torbox) GetTorrent(id string) (*Torrent, error) {
	torrent := &Torrent{}
	url := fmt.Sprintf("%s/api/torrents/mylist/?id=%s", r.Host, id)
	req, _ := http.NewRequest(http.MethodGet, url, nil)
	resp, err := r.client.MakeRequest(req)
	if err != nil {
		return torrent, err
	}
	var res types.TorboxInfoResponse
	err = json.Unmarshal(resp, &res)
	if err != nil {
		return torrent, err
	}
	data := res.Data
	name := data.Name
	torrent.Id = id
	torrent.Name = name
	torrent.Bytes = data.Size
	torrent.Folder = name
	torrent.Progress = data.Progress * 100
	torrent.Status = getTorboxStatus(data.DownloadState, data.DownloadFinished)
	torrent.Speed = data.DownloadSpeed
	torrent.Seeders = data.Seeds
	torrent.Filename = name
	torrent.OriginalFilename = name
	files := make([]TorrentFile, 0)
	cfg := config.GetConfig()
	for _, f := range data.Files {
		fileName := filepath.Base(f.Name)
		if common.RegexMatch(common.SAMPLEMATCH, fileName) {
			// Skip sample files
			continue
		}
		if !cfg.IsAllowedFile(fileName) {
			continue
		}

		if !cfg.IsSizeAllowed(f.Size) {
			continue
		}
		file := TorrentFile{
			Id:   strconv.Itoa(f.Id),
			Name: fileName,
			Size: f.Size,
			Path: fileName,
		}
		files = append(files, file)
	}
	var cleanPath string
	if len(files) > 0 {
		cleanPath = path.Clean(data.Files[0].Name)
	} else {
		cleanPath = path.Clean(data.Name)
	}

	torrent.OriginalFilename = strings.Split(cleanPath, "/")[0]
	torrent.Files = files
	torrent.Debrid = r
	return torrent, nil
}

func (r *Torbox) CheckStatus(torrent *Torrent, isSymlink bool) (*Torrent, error) {
	for {
		tb, err := r.GetTorrent(torrent.Id)

		torrent = tb

		if err != nil || tb == nil {
			return tb, err
		}
		status := torrent.Status
		if status == "downloaded" {
			r.logger.Info().Msgf("Torrent: %s downloaded", torrent.Name)
			if !isSymlink {
				err = r.GetDownloadLinks(torrent)
				if err != nil {
					return torrent, err
				}
			}
			break
		} else if status == "downloading" {
			if !r.DownloadUncached {
				go torrent.Delete()
				return torrent, fmt.Errorf("torrent: %s not cached", torrent.Name)
			}
			// Break out of the loop if the torrent is downloading.
			// This is necessary to prevent infinite loop since we moved to sync downloading and async processing
			break
		} else {
			return torrent, fmt.Errorf("torrent: %s has error", torrent.Name)
		}

	}
	return torrent, nil
}

func (r *Torbox) DeleteTorrent(torrent *Torrent) {
	url := fmt.Sprintf("%s/api/torrents/controltorrent/%s", r.Host, torrent.Id)
	payload := map[string]string{"torrent_id": torrent.Id, "action": "Delete"}
	jsonPayload, _ := json.Marshal(payload)
	req, _ := http.NewRequest(http.MethodDelete, url, bytes.NewBuffer(jsonPayload))
	_, err := r.client.MakeRequest(req)
	if err == nil {
		r.logger.Info().Msgf("Torrent: %s deleted", torrent.Name)
	} else {
		r.logger.Info().Msgf("Error deleting torrent: %s", err)
	}
}

func (r *Torbox) GetDownloadLinks(torrent *Torrent) error {
	downloadLinks := make([]TorrentDownloadLinks, 0)
	for _, file := range torrent.Files {
		url := fmt.Sprintf("%s/api/torrents/requestdl/", r.Host)
		query := gourl.Values{}
		query.Add("torrent_id", torrent.Id)
		query.Add("token", r.APIKey)
		query.Add("file_id", file.Id)
		url += "?" + query.Encode()
		req, _ := http.NewRequest(http.MethodGet, url, nil)
		resp, err := r.client.MakeRequest(req)
		if err != nil {
			return err
		}
		var data types.TorBoxDownloadLinksResponse
		if err = json.Unmarshal(resp, &data); err != nil {
			return err
		}
		if data.Data == nil {
			return fmt.Errorf("error getting download links")
		}
		idx := 0
		link := *data.Data

		dl := TorrentDownloadLinks{
			Link:         link,
			Filename:     torrent.Files[idx].Name,
			DownloadLink: link,
		}
		downloadLinks = append(downloadLinks, dl)
	}
	torrent.DownloadLinks = downloadLinks
	return nil
}

func (r *Torbox) GetCheckCached() bool {
	return r.CheckCached
}

func NewTorbox(dc config.Debrid, cache *common.Cache) *Torbox {
	rl := request.ParseRateLimit(dc.RateLimit)
	headers := map[string]string{
		"Authorization": fmt.Sprintf("Bearer %s", dc.APIKey),
	}
	client := request.NewRLHTTPClient(rl, headers)
	return &Torbox{
		BaseDebrid: BaseDebrid{
			Name:             "torbox",
			Host:             dc.Host,
			APIKey:           dc.APIKey,
			DownloadUncached: dc.DownloadUncached,
			client:           client,
			cache:            cache,
			MountPath:        dc.Folder,
			logger:           logger.NewLogger(dc.Name, config.GetConfig().LogLevel, os.Stdout),
			CheckCached:      dc.CheckCached,
		},
	}
}