Changelog: 0.1.3

debrid/debrid.go

@@ -18,14 +18,16 @@ type Debrid struct {
 	Host             string `json:"host"`
 	APIKey           string
 	DownloadUncached bool
+	client           *common.RLHTTPClient
+	cache            *common.Cache
 }
 
-func NewDebrid(dc common.DebridConfig) Service {
+func NewDebrid(dc common.DebridConfig, cache *common.Cache) Service {
 	switch dc.Name {
 	case "realdebrid":
-		return NewRealDebrid(dc)
+		return NewRealDebrid(dc, cache)
 	default:
-		return NewRealDebrid(dc)
+		return NewRealDebrid(dc, cache)
 	}
 }
 
@@ -81,3 +83,31 @@ func getTorrentInfo(filePath string) (*Torrent, error) {
 	}
 	return torrent, nil
 }
+
+func GetLocalCache(infohashes []string, cache *common.Cache) (string, map[string]bool) {
+	result := make(map[string]bool)
+
+	if len(infohashes) == 0 {
+		return "", result
+	}
+	if len(infohashes) == 1 {
+		if cache.Exists(infohashes[0]) {
+			return "", map[string]bool{infohashes[0]: true}
+		}
+		return infohashes[0], result
+	}
+
+	cachedHashes := cache.GetMultiple(infohashes)
+
+	hashes := ""
+	for _, h := range infohashes {
+		_, exists := cachedHashes[h]
+		if !exists {
+			hashes += h + "/"
+		} else {
+			result[h] = true
+		}
+	}
+
+	return hashes, result
+}

debrid/realdebrid.go

@@ -18,6 +18,7 @@ type RealDebrid struct {
 	APIKey           string
 	DownloadUncached bool
 	client           *common.RLHTTPClient
+	cache            *common.Cache
 }
 
 func (r *RealDebrid) Process(arr *Arr, magnet string) (*Torrent, error) {
@@ -28,7 +29,8 @@ func (r *RealDebrid) Process(arr *Arr, magnet string) (*Torrent, error) {
 	}
 	log.Printf("Torrent Name: %s", torrent.Name)
 	if !r.DownloadUncached {
-		if !r.IsAvailable([]string{torrent.InfoHash})[torrent.InfoHash] {
+		hash, exists := r.IsAvailable([]string{torrent.InfoHash})[torrent.InfoHash]
+		if !exists || !hash {
 			return torrent, fmt.Errorf("torrent is not cached")
 		}
 		log.Printf("Torrent: %s is cached", torrent.Name)
@@ -42,12 +44,18 @@ func (r *RealDebrid) Process(arr *Arr, magnet string) (*Torrent, error) {
 }
 
 func (r *RealDebrid) IsAvailable(infohashes []string) map[string]bool {
-	hashes := strings.Join(infohashes, "/")
-	result := make(map[string]bool)
+	// Check if the infohashes are available in the local cache
+	hashes, result := GetLocalCache(infohashes, r.cache)
+
+	if hashes == "" {
+		// Either all the infohashes are locally cached or none are
+		r.cache.AddMultiple(result)
+		return result
+	}
+
 	url := fmt.Sprintf("%s/torrents/instantAvailability/%s", r.Host, hashes)
 	resp, err := r.client.MakeRequest(http.MethodGet, url, nil)
 	if err != nil {
-		log.Println(url)
 		log.Println("Error checking availability:", err)
 		return result
 	}
@@ -59,12 +67,11 @@ func (r *RealDebrid) IsAvailable(infohashes []string) map[string]bool {
 	}
 	for _, h := range infohashes {
 		hosters, exists := data[strings.ToLower(h)]
-		if !exists || len(hosters.Rd) < 1 {
-			result[h] = false
-		} else {
+		if exists && len(hosters.Rd) > 0 {
 			result[h] = true
 		}
 	}
+	r.cache.AddMultiple(result) // Add the results to the cache
 	return result
 }
 
@@ -149,7 +156,7 @@ func (r *RealDebrid) DownloadLink(torrent *Torrent) error {
 	return nil
 }
 
-func NewRealDebrid(dc common.DebridConfig) *RealDebrid {
+func NewRealDebrid(dc common.DebridConfig, cache *common.Cache) *RealDebrid {
 	rl := common.ParseRateLimit(dc.RateLimit)
 	headers := map[string]string{
 		"Authorization": fmt.Sprintf("Bearer %s", dc.APIKey),
@@ -160,5 +167,6 @@ func NewRealDebrid(dc common.DebridConfig) *RealDebrid {
 		APIKey:           dc.APIKey,
 		DownloadUncached: dc.DownloadUncached,
 		client:           client,
+		cache:            cache,
 	}
 }