Initial commit 🌈

pkg/realdebrid/api.go

@@ -0,0 +1,172 @@
+package realdebrid
+
+import (
+	"bytes"
+	"encoding/json"
+	"fmt"
+	"io"
+	"net/http"
+	"net/url"
+	"strconv"
+)
+
+func UnrestrictCheck(accessToken, link string) (UnrestrictResponse, error) {
+	data := url.Values{}
+	data.Set("link", link)
+
+	req, err := http.NewRequest("POST", "https://api.real-debrid.com/rest/1.0/unrestrict/check", bytes.NewBufferString(data.Encode()))
+	if err != nil {
+		return UnrestrictResponse{}, err
+	}
+
+	req.Header.Set("Authorization", "Bearer "+accessToken)
+	req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
+
+	client := &http.Client{}
+	resp, err := client.Do(req)
+	if err != nil {
+		return UnrestrictResponse{}, err
+	}
+	defer resp.Body.Close()
+
+	body, err := io.ReadAll(resp.Body)
+	if err != nil {
+		return UnrestrictResponse{}, err
+	}
+
+	if resp.StatusCode != http.StatusOK {
+		return UnrestrictResponse{}, fmt.Errorf("HTTP error: %s", resp.Status)
+	}
+
+	var response UnrestrictResponse
+	err = json.Unmarshal(body, &response)
+	if err != nil {
+		return UnrestrictResponse{}, err
+	}
+
+	return response, nil
+}
+
+func UnrestrictLink(accessToken, link string) (UnrestrictResponse, error) {
+	data := url.Values{}
+	data.Set("link", link)
+
+	req, err := http.NewRequest("POST", "https://api.real-debrid.com/rest/1.0/unrestrict/link", bytes.NewBufferString(data.Encode()))
+	if err != nil {
+		return UnrestrictResponse{}, err
+	}
+
+	req.Header.Set("Authorization", "Bearer "+accessToken)
+	req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
+
+	client := &http.Client{}
+	resp, err := client.Do(req)
+	if err != nil {
+		return UnrestrictResponse{}, err
+	}
+	defer resp.Body.Close()
+
+	body, err := io.ReadAll(resp.Body)
+	if err != nil {
+		return UnrestrictResponse{}, err
+	}
+
+	if resp.StatusCode != http.StatusOK {
+		return UnrestrictResponse{}, fmt.Errorf("HTTP error: %s", resp.Status)
+	}
+
+	var response UnrestrictResponse
+	err = json.Unmarshal(body, &response)
+	if err != nil {
+		return UnrestrictResponse{}, err
+	}
+
+	return response, nil
+}
+
+func GetTorrents(accessToken string) ([]Torrent, error) {
+	baseURL := "https://api.real-debrid.com/rest/1.0/torrents"
+	var allTorrents []Torrent
+	page := 1
+	limit := 10
+
+	for {
+		params := url.Values{}
+		params.Set("page", fmt.Sprintf("%d", page))
+		params.Set("limit", fmt.Sprintf("%d", limit))
+
+		reqURL := baseURL + "?" + params.Encode()
+
+		req, err := http.NewRequest("GET", reqURL, nil)
+		if err != nil {
+			return nil, err
+		}
+
+		req.Header.Set("Authorization", "Bearer "+accessToken)
+
+		client := &http.Client{}
+		resp, err := client.Do(req)
+		if err != nil {
+			return nil, err
+		}
+		defer resp.Body.Close()
+
+		if resp.StatusCode != http.StatusOK {
+			return nil, fmt.Errorf("HTTP error: %s", resp.Status)
+		}
+
+		var torrents []Torrent
+		decoder := json.NewDecoder(resp.Body)
+		err = decoder.Decode(&torrents)
+		if err != nil {
+			return nil, err
+		}
+
+		allTorrents = append(allTorrents, torrents...)
+
+		totalCountHeader := "10" // resp.Header.Get("x-total-count")
+		totalCount, err := strconv.Atoi(totalCountHeader)
+		if err != nil {
+			break
+		}
+
+		if len(torrents) < limit || len(allTorrents) >= totalCount {
+			break
+		}
+
+		page++
+	}
+
+	return deduplicateTorrents(allTorrents), nil
+}
+
+func deduplicateTorrents(torrents []Torrent) []Torrent {
+	mappedTorrents := make(map[string]Torrent)
+
+	for _, t := range torrents {
+		if existing, ok := mappedTorrents[t.Filename]; ok {
+			if existing.Hash == t.Hash {
+				// If hash is the same, combine the links
+				existing.Links = append(existing.Links, t.Links...)
+				mappedTorrents[t.Filename] = existing
+			} else {
+				// If hash is different, delete old entry and create two new entries
+				delete(mappedTorrents, t.Filename)
+				newKey1 := t.Filename + " - " + t.Hash[:4]
+				newKey2 := existing.Filename + " - " + existing.Hash[:4]
+				mappedTorrents[newKey1] = t
+				mappedTorrents[newKey2] = existing
+			}
+		} else {
+			mappedTorrents[t.Filename] = t
+		}
+	}
+
+	// Convert the map back to a slice
+	deduplicated := make([]Torrent, 0, len(mappedTorrents))
+	for _, value := range mappedTorrents {
+		deduplicated = append(deduplicated, value)
+	}
+
+	return deduplicated
+}