new: feat: add comando torrents indexer

api/indexers/comando_torrents.go

@@ -0,0 +1,164 @@
+package indexers
+
+import (
+	"encoding/json"
+	"fmt"
+	"net/http"
+	"net/url"
+	"regexp"
+
+	"github.com/PuerkitoBio/goquery"
+)
+
+const (
+	URL         = "https://comando.la/"
+	queryFilter = "?s="
+)
+
+type Audio string
+
+const (
+	AudioPortuguese = "Português"
+	AudioEnglish    = "Inglês"
+	AudioSpanish    = "Espanhol"
+)
+
+type IndexedTorrent struct {
+	Title         string  `json:"title"`
+	OriginalTitle string  `json:"original_title"`
+	Details       string  `json:"details"`
+	Year          string  `json:"year"`
+	Audio         []Audio `json:"audio"`
+	MagnetLink    string  `json:"magnet_link"`
+}
+
+func HandlerComandoIndexer(w http.ResponseWriter, r *http.Request) {
+	// supported query params: q, season, episode
+	q := r.URL.Query().Get("q")
+	if q == "" {
+		w.WriteHeader(http.StatusBadRequest)
+		json.NewEncoder(w).Encode(map[string]string{"error": "param q is required"})
+		return
+	}
+
+	// URL encode query param
+	q = url.QueryEscape(q)
+	url := URL + queryFilter + q
+
+	fmt.Println("URL:>", url)
+	resp, err := http.Get(url)
+	if err != nil {
+		w.WriteHeader(http.StatusInternalServerError)
+		json.NewEncoder(w).Encode(map[string]string{"error": err.Error()})
+		return
+	}
+	defer resp.Body.Close()
+
+	doc, err := goquery.NewDocumentFromReader(resp.Body)
+	if err != nil {
+		w.WriteHeader(http.StatusInternalServerError)
+		json.NewEncoder(w).Encode(map[string]string{"error": err.Error()})
+		return
+	}
+
+	var links []string
+	doc.Find("article").Each(func(i int, s *goquery.Selection) {
+		// get link from h2.entry-title > a
+		link, _ := s.Find("h2.entry-title > a").Attr("href")
+		links = append(links, link)
+	})
+	fmt.Println(links)
+	fmt.Println(doc.Text())
+
+	var indexedTorrents []IndexedTorrent
+	for _, link := range links {
+		torrents, err := getTorrents(link)
+		if err != nil {
+			fmt.Println(err)
+			continue
+		}
+		indexedTorrents = append(indexedTorrents, torrents...)
+	}
+
+	w.Header().Set("Content-Type", "application/json")
+	json.NewEncoder(w).Encode(indexedTorrents)
+}
+
+func getTorrents(link string) ([]IndexedTorrent, error) {
+	var indexedTorrents []IndexedTorrent
+	resp, err := http.Get(link)
+	if err != nil {
+		return nil, err
+	}
+	defer resp.Body.Close()
+
+	doc, err := goquery.NewDocumentFromReader(resp.Body)
+	if err != nil {
+		return nil, err
+	}
+
+	article := doc.Find("article")
+	title := article.Find("h2.entry-title > a").Text()
+	textContent := article.Find("div.entry-content")
+	magnets := textContent.Find("a[href^=\"magnet\"]")
+	var magnetLinks []string
+	magnets.Each(func(i int, s *goquery.Selection) {
+		magnetLink, _ := s.Attr("href")
+		magnetLinks = append(magnetLinks, magnetLink)
+	})
+
+	var audio []Audio
+	var year string
+	article.Find("div.entry-content > p").Each(func(i int, s *goquery.Selection) {
+		// pattern:
+		// Título Traduzido: Fundação
+		// Título Original: Foundation
+		// IMDb: 7,5
+		// Ano de Lançamento: 2023
+		// Gênero: Ação | Aventura | Ficção
+		// Formato: MKV
+		// Qualidade: WEB-DL
+		// Áudio: Português | Inglês
+		// Legenda: Português
+		// Tamanho: –
+		// Qualidade de Áudio: 10
+		// Qualidade de Vídeo: 10
+		// Duração: 59 Min.
+		// Servidor: Torrent
+
+		re := regexp.MustCompile(`Áudio: (.*)`)
+		audioMatch := re.FindStringSubmatch(s.Text())
+		if len(audioMatch) > 0 {
+			audio = append(audio, Audio(audioMatch[1]))
+		}
+
+		re = regexp.MustCompile(`Ano de Lançamento: (.*)`)
+		yearMatch := re.FindStringSubmatch(s.Text())
+		if len(yearMatch) > 0 {
+			year = yearMatch[1]
+		}
+	})
+
+	// for each magnet link, create a new indexed torrent
+	for _, magnetLink := range magnetLinks {
+		indexedTorrents = append(indexedTorrents, IndexedTorrent{
+			Title:         extractReleaseName(magnetLink),
+			OriginalTitle: title,
+			Details:       link,
+			Year:          year,
+			Audio:         audio,
+			MagnetLink:    magnetLink,
+		})
+	}
+
+	return indexedTorrents, nil
+}
+
+func extractReleaseName(magnetLink string) string {
+	re := regexp.MustCompile(`dn=(.*)&`)
+	matches := re.FindStringSubmatch(magnetLink)
+	if len(matches) > 0 {
+		return matches[1]
+	}
+	return ""
+}