feat: Add more series

2 files changed, 66 insertions, 13 deletions

diff --git a/Makefile b/Makefile
index e498b35..0bb3b31 100644
--- a/Makefile
+++ b/Makefile
@@ -23,3 +23,6 @@ compress_into_oblivion: build
 
 run:
 	$(GO_RUN) .
+
+series:
+	$(GO_RUN) . series
diff --git a/main.go b/main.go
index b9bed96..22d1489 100644
--- a/main.go
+++ b/main.go
@@ -7,6 +7,7 @@ import (
 	"fmt"
 	"io"
 	"net/http"
+	"os"
 	"regexp"
 	"strconv"
 	"strings"
@@ -18,22 +19,34 @@ import (
 	"github.com/prometheus/client_golang/prometheus/promhttp"
 )
 
-type ErrorRequestHandler func(w http.ResponseWriter, r *http.Request) error
-
 const (
 	FeedUrl = "https://api.jovemnerd.com.br/feed-nerdcast/"
 )
 
+type (
+	ErrorRequestHandler func(w http.ResponseWriter, r *http.Request) error
+)
+
+var (
+	SerieRegex = regexp.MustCompile(`(?P<serie>.+) (?P<number>[0-9abc]+) \- (?P<title>.+)`)
+)
+
 var (
-	RegexCollection = map[string]string{
-		"nerdcast":     "NerdCast [0-9]+[a-c]* -",
-		"empreendedor": "Empreendedor [0-9]+ -",
-		"mamicas":      "Caneca de Mamicas [0-9]+ -",
-		"english":      "Speak English [0-9]+ -",
-		"nerdcash":     "NerdCash [0-9]+ -",
-		"bunker":       "Lá do Bunker [0-9]+ -",
-		"tech":         "NerdTech [0-9]+ -",
-		"genera":       "Generacast [0-9]+ -",
+	regexCollection = map[string]string{
+		"nerdcast":     "NerdCast [0-9]+[a-c]*",
+		"empreendedor": "Empreendedor [0-9]+",
+		"mamicas":      "Caneca de Mamicas [0-9]+",
+		"english":      "Speak English [0-9]+",
+		"nerdcash":     "NerdCash [0-9]+",
+		"bunker":       "Lá do Bunker( LDB especial Oscar|) [0-9]+",
+		"tech":         "NerdTech [0-9]+",
+		"genera":       "Generacast [0-9]+",
+		"rpg":          "NerdCast RPG [0-9]+[a-c]*",
+		"catar":        "Vai te Catar [0-9]+",
+		"cloud":        "Nerd na Cloud [0-9]+",
+		"contar":       "Vou (T|t)e Contar [0-9]+",
+		"parceiro":     "Papo de Parceiro [0-9]+",
+		"cash":         "NerdCash [0-9]+",
 	}
 
 	feedRequest = promauto.NewHistogramVec(prometheus.HistogramOpts{
@@ -60,7 +73,7 @@ func getSeries(r *http.Request) []string {
 	var series []string
 
 	for _, q := range strings.Split(query, ",") {
-		if _, ok := RegexCollection[q]; ok {
+		if _, ok := regexCollection[q]; ok {
 			series = append(series, q)
 		}
 	}
@@ -74,7 +87,7 @@ func getSeries(r *http.Request) []string {
 
 func match(title string, series []string) bool {
 	for _, s := range series {
-		if ok, err := regexp.MatchString(RegexCollection[s], title); err == nil && ok {
+		if ok, err := regexp.MatchString(regexCollection[s], title); err == nil && ok {
 			return true
 		}
 	}
@@ -224,7 +237,44 @@ func podcast(w http.ResponseWriter, r *http.Request) error {
 	return nil
 }
 
+func genSeries() error {
+	xml, err := fetchXML(context.Background())
+	if err != nil {
+		return err
+	}
+
+	doc := etree.NewDocument()
+	err = doc.ReadFromBytes(xml)
+	if err != nil {
+		return err
+	}
+
+	unique := make(map[string]any)
+	els := doc.FindElements("//channel/item")
+	for _, e := range els {
+		txt := e.FindElement("title").Text()
+		res := SerieRegex.FindStringSubmatch(txt)
+		if len(res) > 1 {
+			unique[res[1]] = nil
+		}
+	}
+
+	for k := range unique {
+		fmt.Println(k)
+	}
+
+	return nil
+}
+
 func main() {
+	if len(os.Args) > 1 && os.Args[1] == "series" {
+		err := genSeries()
+		if err != nil {
+			panic(err.Error())
+		}
+		return
+	}
+
 	var (
 		addr = flag.String("addr", ":8080", "Server address")
 	)