all repos

moviefeed @ 6d7ce55

rss feed server for tracking new tv show episodes

moviefeed/main.go(view raw) 

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
 
package main

import (
	"flag"
	"fmt"
	"log/slog"
	"net/http"
	"os"
	"time"

	"github.com/gorilla/feeds"
)

func main() {
	configFile := flag.String("config", "config.yaml", "Path to config file")
	flag.Parse()

	config, err := loadConfig(*configFile)
	if err != nil {
		slog.Error("failed to load config", "err", err)
		os.Exit(1)
	}

	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
		episodes, err := fetchNewEpisodes(config)
		if err != nil {
			slog.Error("failed to fetch episodes", "err", err)
			os.Exit(1)
		}

		rssFeed := generateRSS(episodes)
		w.Header().Set("Content-Type", "application/rss+xml")
		w.Write([]byte(rssFeed))
	})

	slog.Info("server starting", "port", config.Port)
	http.ListenAndServe(":"+config.Port, nil)
}

func generateRSS(episodes []TMDBEpisode) string {
	feed := &feeds.Feed{
		Title:       "moviefeed",
		Link:        &feeds.Link{Href: "http://localhost"},
		Description: "Latest episodes from followed shows",
		Created:     time.Now(),
	}

	for _, ep := range episodes {
		airDate, _ := time.Parse("2006-01-02", ep.AirDate)
		feed.Items = append(feed.Items, &feeds.Item{
			Id: fmt.Sprintf("%s-%d-%d", ep.ShowID, ep.SeasonNumber, ep.EpisodeNumber),
			Title: fmt.Sprintf(
				"%s S%dE%d: %s",
				ep.ShowName,
				ep.SeasonNumber,
				ep.EpisodeNumber,
				ep.Name,
			),
			Link: &feeds.Link{
				Href: fmt.Sprintf("https://www.themoviedb.org/tv/episode/%d", ep.ID),
			},
			Description: ep.Overview,
			Created:     airDate,
		})
	}

	rss, err := feed.ToRss()
	if err != nil {
		slog.Error("error generating RSS", "err", err)
		return ""
	}

	return rss
}