68 lines
1.8 KiB
Go
68 lines
1.8 KiB
Go
package handlers
|
|
|
|
import (
|
|
"log/slog"
|
|
"net/http"
|
|
|
|
"github.com/gofiber/fiber/v2"
|
|
|
|
"github.com/zepyrshut/rating-orama/internal/scraper"
|
|
"github.com/zepyrshut/rating-orama/internal/sqlc"
|
|
)
|
|
|
|
func (hq *Handlers) GetTVShow(c *fiber.Ctx) error {
|
|
ttShowID := c.Query("ttid")
|
|
|
|
if ttShowID == "" {
|
|
return c.SendStatus(http.StatusBadRequest)
|
|
}
|
|
|
|
var title string
|
|
var scraperEpisodes []scraper.Episode
|
|
var sqlcEpisodes []sqlc.Episode
|
|
|
|
tvShow, err := hq.queries.CheckTVShowExists(c.Context(), ttShowID)
|
|
if err != nil {
|
|
title, scraperEpisodes = scraper.ScrapeEpisodes(ttShowID)
|
|
//TODO: make transactional
|
|
ttShow, err := hq.queries.CreateTVShow(c.Context(), sqlc.CreateTVShowParams{
|
|
TtImdb: ttShowID,
|
|
Name: title,
|
|
})
|
|
if err != nil {
|
|
slog.Error("failed to create tv show", "ttid", ttShowID, "error", err)
|
|
return c.SendStatus(http.StatusInternalServerError)
|
|
}
|
|
|
|
slog.Info("ttshowid", "id", ttShow.ID)
|
|
for _, episode := range scraperEpisodes {
|
|
sqlcEpisodesParams := episode.ToEpisodeParams(ttShow.ID)
|
|
sqlcEpisode, err := hq.queries.CreateEpisodes(c.Context(), sqlcEpisodesParams)
|
|
if err != nil {
|
|
slog.Error("failed to create episodes", "ttid", ttShowID, "error", err)
|
|
return c.SendStatus(http.StatusInternalServerError)
|
|
}
|
|
|
|
sqlcEpisodes = append(sqlcEpisodes, sqlcEpisode)
|
|
}
|
|
|
|
slog.Info("scraped seasons", "ttid", ttShowID, "title", title)
|
|
} else {
|
|
title = tvShow.Name
|
|
sqlcEpisodes, err = hq.queries.GetEpisodes(c.Context(), tvShow.ID)
|
|
if err != nil {
|
|
slog.Error("failed to get episodes", "ttid", ttShowID, "error", err)
|
|
return c.SendStatus(http.StatusInternalServerError)
|
|
}
|
|
|
|
hq.queries.IncreasePopularity(c.Context(), ttShowID)
|
|
slog.Info("tv show exists", "ttid", ttShowID, "title", tvShow.Name)
|
|
}
|
|
|
|
return c.JSON(fiber.Map{
|
|
"popularity": tvShow.Popularity,
|
|
"title": title,
|
|
"seasons": sqlcEpisodes,
|
|
})
|
|
}
|