Features:

- Make Blackhole and Proxy optional
- Query all Hashes at once, speeeding up the process
- refractor the code a little bit

debrid/debrid.go

@@ -11,7 +11,7 @@ type Service interface {
 	CheckStatus(torrent *Torrent) (*Torrent, error)
 	DownloadLink(torrent *Torrent) error
 	Process(arr *Arr, magnet string) (*Torrent, error)
-	IsAvailable(magnet *common.Magnet) bool
+	IsAvailable(infohashes []string) map[string]bool
 }
 
 type Debrid struct {

debrid/realdebrid.go

@@ -28,7 +28,7 @@ func (r *RealDebrid) Process(arr *Arr, magnet string) (*Torrent, error) {
 	}
 	log.Printf("Torrent Name: %s", torrent.Name)
 	if !r.DownloadUncached {
-		if !r.IsAvailable(torrent.Magnet) {
+		if !r.IsAvailable([]string{torrent.InfoHash})[torrent.InfoHash] {
 			return torrent, fmt.Errorf("torrent is not cached")
 		}
 		log.Printf("Torrent: %s is cached", torrent.Name)
@@ -41,22 +41,31 @@ func (r *RealDebrid) Process(arr *Arr, magnet string) (*Torrent, error) {
 	return r.CheckStatus(torrent)
 }
 
-func (r *RealDebrid) IsAvailable(magnet *common.Magnet) bool {
-	url := fmt.Sprintf("%s/torrents/instantAvailability/%s", r.Host, magnet.InfoHash)
+func (r *RealDebrid) IsAvailable(infohashes []string) map[string]bool {
+	hashes := strings.Join(infohashes, "/")
+	result := make(map[string]bool)
+	url := fmt.Sprintf("%s/torrents/instantAvailability/%s", r.Host, hashes)
 	resp, err := r.client.MakeRequest(http.MethodGet, url, nil)
 	if err != nil {
-		return false
+		log.Println(url)
+		log.Println("Error checking availability:", err)
+		return result
 	}
 	var data structs.RealDebridAvailabilityResponse
 	err = json.Unmarshal(resp, &data)
 	if err != nil {
-		return false
+		log.Println("Error marshalling availability:", err)
+		return result
 	}
-	hosters, exists := data[strings.ToLower(magnet.InfoHash)]
-	if !exists || len(hosters) < 1 {
-		return false
+	for _, h := range infohashes {
+		hosters, exists := data[strings.ToLower(h)]
+		if !exists || len(hosters.Rd) < 1 {
+			result[h] = false
+		} else {
+			result[h] = true
+		}
 	}
-	return true
+	return result
 }
 
 func (r *RealDebrid) SubmitMagnet(torrent *Torrent) (*Torrent, error) {

debrid/structs/realdebrid.go

@@ -1,10 +1,36 @@
 package structs
 
-type RealDebridAvailabilityResponse map[string]Hosters
+import (
+	"encoding/json"
+	"fmt"
+)
 
-type Hosters map[string][]FileIDs
+type RealDebridAvailabilityResponse map[string]Hoster
 
-type FileIDs map[string]FileVariant
+type Hoster struct {
+	Rd []map[string]FileVariant `json:"rd"`
+}
+
+func (h *Hoster) UnmarshalJSON(data []byte) error {
+	// Attempt to unmarshal into the expected structure (an object with an "rd" key)
+	type Alias Hoster
+	var obj Alias
+	if err := json.Unmarshal(data, &obj); err == nil {
+		*h = Hoster(obj)
+		return nil
+	}
+
+	// If unmarshalling into an object fails, check if it's an empty array
+	var arr []interface{}
+	if err := json.Unmarshal(data, &arr); err == nil && len(arr) == 0 {
+		// It's an empty array; initialize with no entries
+		*h = Hoster{Rd: nil}
+		return nil
+	}
+
+	// If both attempts fail, return an error
+	return fmt.Errorf("hoster: cannot unmarshal JSON data: %s", string(data))
+}
 
 type FileVariant struct {
 	Filename string `json:"filename"`
@@ -39,3 +65,5 @@ type RealDebridTorrentInfo struct {
 	Speed   int      `json:"speed,omitempty"`
 	Seeders int      `json:"seeders,omitempty"`
 }
+
+// 5e6e2e77fd3921a7903a41336c844cc409bf8788/14527C07BDFDDFC642963238BB6E7507B9742947/66A1CD1A5C7F4014877A51AC2620E857E3BB4D16