oscarmlage/hugo-medialog
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added series

Browse Source
This commit is contained in:
Óscar M. Lage 2024-10-09 21:56:40 +02:00
parent 48fd96763d
commit 38a946f946
6 changed files with 446 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

107
internal/series/api_fanart.go Normal file
View File

@ -0,0 +1,107 @@
// internal/series/api_trakt.go
package series
import (
"encoding/json"
"fmt"
"hugo-medialog/utils"
"io/ioutil"
"net/http"
"os"
"path/filepath"
"strconv"
)
// FanartResponse Struct
type FanartResponse struct {
SeriePosters []FanartImage `json:"tvposter"`
SerieBackgrounds []FanartImage `json:"showbackground"`
SerieLogos []FanartImage `json:"hdtvlogo"`
}
// FanartImage struct
type FanartImage struct {
URL string `json:"url"`
Likes string `json:"likes"`
}
func FetchImagesFromFanart(ID int) (posterURL, backgroundURL string, logoURL string, err error) {
fanartAPIKey := os.Getenv("FANART_API_KEY")
url := fmt.Sprintf("https://webservice.fanart.tv/v3/tv/%s?api_key=%s", strconv.Itoa(ID), fanartAPIKey)
resp, err := http.Get(url)
if err != nil {
return "", "", "", fmt.Errorf("error fetching data from fanart.tv: %w", err)
}
defer resp.Body.Close()
if resp.StatusCode != http.StatusOK {
return "", "", "", fmt.Errorf("fanart.tv returned non-200 status: %d", resp.StatusCode)
}
body, err := ioutil.ReadAll(resp.Body)
if err != nil {
return "", "", "", fmt.Errorf("error reading response body: %w", err)
}
var fanartResp FanartResponse
err = json.Unmarshal(body, &fanartResp)
if err != nil {
return "", "", "", fmt.Errorf("error unmarshalling response: %w", err)
}
// Get the most voted poster
if len(fanartResp.SeriePosters) > 0 {
posterURL = fanartResp.SeriePosters[0].URL
}
// Get the most voted background
if len(fanartResp.SerieBackgrounds) > 0 {
backgroundURL = fanartResp.SerieBackgrounds[0].URL
}
// Get the most voted logo
if len(fanartResp.SerieLogos) > 0 {
logoURL = fanartResp.SerieLogos[0].URL
}
return posterURL, backgroundURL, logoURL, nil
}
func DownloadImage(url, slug, imageType string) error {
imageDir := filepath.Join(os.Getenv("MARKDOWN_OUTPUT_SERIES_DIR"), os.Getenv("IMAGES_OUTPUT_DIR"))
if err := utils.CreateDirIfNotExists(imageDir); err != nil {
return err
}
filename := fmt.Sprintf("%s-%s.jpg", slug, imageType)
filePath := filepath.Join(imageDir, filename)
resp, err := http.Get(url)
if err != nil {
return fmt.Errorf("error downloading image: %w", err)
}
defer resp.Body.Close()
if resp.StatusCode != http.StatusOK {
return fmt.Errorf("non-200 response while downloading image: %d", resp.StatusCode)
}
file, err := os.Create(filePath)
if err != nil {
return fmt.Errorf("error creating image file: %w", err)
}
defer file.Close()
body, err := ioutil.ReadAll(resp.Body)
if err != nil {
return fmt.Errorf("error reading image data: %w", err)
}
_, err = file.Write(body)
if err != nil {
return fmt.Errorf("error writing image data: %w", err)
}
fmt.Printf(" - Image saved successfully at: %s\n", filePath)
return nil
}

200
internal/series/api_trakt.go Normal file
View File

@ -0,0 +1,200 @@
// internal/series/api_trakt.go
package series
import (
"context"
"encoding/json"
"fmt"
"hugo-medialog/utils"
"io/ioutil"
"log"
"net/http"
"net/url"
"os"
"strconv"
"golang.org/x/oauth2"
)
var (
traktAPIBaseURL = "https://api.trakt.tv"
traktClientID string
traktClientSecret string
traktRedirectURI string
)
var traktOAuthConfig = &oauth2.Config{
ClientID: os.Getenv("TRAKT_CLIENT_ID"),
ClientSecret: os.Getenv("TRAKT_CLIENT_SECRET"),
Endpoint: oauth2.Endpoint{
AuthURL: "https://api.trakt.tv/oauth/authorize",
TokenURL: "https://api.trakt.tv/oauth/token",
},
RedirectURL: os.Getenv("TRAKT_REDIRECT_URI"),
}
func init() {
// Load runtime variables
utils.LoadConfig()
traktClientID = os.Getenv("TRAKT_CLIENT_ID")
traktClientSecret = os.Getenv("TRAKT_CLIENT_SECRET")
traktRedirectURI = os.Getenv("TRAKT_REDIRECT_URI")
if traktClientID == "" || traktClientSecret == "" || traktRedirectURI == "" {
log.Fatal("Missing Trakt API credentials in environment variables")
}
}
// Function to fetch a token and save it for reuse
func GetTraktToken() (*oauth2.Token, error) {
// Check if the token already exists (from a previous session)
tokenFile := "trakt_token.json"
if _, err := os.Stat(tokenFile); err == nil {
// If token file exists, read and return it
file, err := os.Open(tokenFile)
if err != nil {
return nil, err
}
defer file.Close()
token := &oauth2.Token{}
if err = json.NewDecoder(file).Decode(token); err != nil {
return nil, err
}
return token, nil
}
// If no token exists, we need to go through the OAuth flow
authURL := traktOAuthConfig.AuthCodeURL("state-token", oauth2.AccessTypeOffline)
fmt.Printf("Visit the URL to authorize: %v\n", authURL)
// After user authorization, you will receive an authorization code.
// Once you have the code, get a token by calling the token endpoint.
var authCode string
fmt.Print("Enter authorization code: ")
fmt.Scan(&authCode)
token, err := traktOAuthConfig.Exchange(context.Background(), authCode)
if err != nil {
return nil, err
}
// Save the token to file for future use
file, err := os.Create(tokenFile)
if err != nil {
return nil, err
}
defer file.Close()
if err := json.NewEncoder(file).Encode(token); err != nil {
return nil, err
}
return token, nil
}
// Function to get serie details by Trakt ID
func GetSerieByID(traktID int) (*Serie, error) {
token, err := GetTraktToken()
if err != nil {
return nil, err
}
client := traktOAuthConfig.Client(context.Background(), token)
// Call the Trakt API to get serie details by ID
apiURL := fmt.Sprintf("%s/shows/%s", traktAPIBaseURL, strconv.Itoa(traktID))
req, _ := http.NewRequest("GET", apiURL, nil)
req.Header.Set("Content-Type", "application/json")
req.Header.Set("Authorization", "Bearer "+token.AccessToken)
req.Header.Set("trakt-api-version", "2")
req.Header.Set("trakt-api-key", traktClientID)
resp, err := client.Do(req)
if err != nil {
return nil, err
}
defer resp.Body.Close()
if resp.StatusCode != http.StatusOK {
return nil, fmt.Errorf("failed to fetch serie: %v", resp.Status)
}
body, _ := ioutil.ReadAll(resp.Body)
var traktSerie Serie
if err := json.Unmarshal(body, &traktSerie); err != nil {
return nil, err
}
return &traktSerie, nil
}
// Function to search for a serie by title
func SearchSerieByTitle(title string, serie *Serie) error {
token, err := GetTraktToken()
if err != nil {
return err
}
client := traktOAuthConfig.Client(context.Background(), token)
// URL encode the title
query := url.QueryEscape(title)
apiURL := fmt.Sprintf("%s/search/show?query=%s", traktAPIBaseURL, query)
req, _ := http.NewRequest("GET", apiURL, nil)
req.Header.Set("Content-Type", "application/json")
req.Header.Set("Authorization", "Bearer "+token.AccessToken)
req.Header.Set("trakt-api-version", "2")
req.Header.Set("trakt-api-key", traktClientID)
resp, err := client.Do(req)
if err != nil {
return err
}
defer resp.Body.Close()
if resp.StatusCode != http.StatusOK {
return fmt.Errorf("failed to search serie: %v", resp.Status)
}
body, _ := ioutil.ReadAll(resp.Body)
// utils.DebugBody(body)
// Parse trakt.tv response to a struct
var traktResponse []struct {
Serie struct {
Title string `json:"title"`
Year int `json:"year"`
IDs struct {
Slug string `json:"slug"`
IMDB string `json:"imdb"`
TMDB int `json:"tmdb"`
Trakt int `json:"trakt"`
TVDB int `json:"tvdb"`
} `json:"ids"`
} `json:"show"`
}
if err := json.Unmarshal(body, &traktResponse); err != nil {
return err
}
if len(traktResponse) == 0 {
return fmt.Errorf("no serie found with title %s", title)
}
// Using 1st result to fill our Serie structure
foundSerie := traktResponse[0].Serie
serie.Title = foundSerie.Title
serie.IDs.Slug = foundSerie.IDs.Slug
serie.IDs.IMDB = foundSerie.IDs.IMDB
serie.IDs.TMDB = foundSerie.IDs.TMDB
serie.IDs.Trakt = foundSerie.IDs.Trakt
serie.IDs.TVDB = foundSerie.IDs.TVDB
serie.Subtitle = fmt.Sprintf("%d", foundSerie.Year)
serie.Year = foundSerie.Year
serie.Link = fmt.Sprintf("https://trakt.tv/shows/%d", foundSerie.IDs.Trakt)
return nil
}

116
internal/series/controller.go
View File

@ -1,2 +1,118 @@
// internal/series/controller.go
package series
import (
"fmt"
"hugo-medialog/utils"
"os"
"path/filepath"
"gopkg.in/yaml.v3"
)
func LoadSeries() ([]Serie, error) {
seriesFile := os.Getenv("OBSIDIAN_SERIES_FILE")
fileData, err := os.ReadFile(seriesFile)
if err != nil {
return nil, err
}
var series []Serie
err = yaml.Unmarshal(fileData, &series)
if err != nil {
return nil, err
}
return series, nil
}
func ProcessSeries(series []Serie) error {
utils.Sep()
for _, serie := range series {
// Debug print
fmt.Printf("Título: %s, Puntuación: %.1f, Fecha: %s\n",
serie.Title, serie.Rate, serie.Date)
// If we dont have IDs, search serie by Title and get the IDs
if serie.IDs.Trakt == 0 {
err := SearchSerieByTitle(serie.Title, &serie)
if err != nil {
fmt.Printf("Error searching serie by title %s: %s\n", serie.Title, err)
continue
}
}
// Now we need to get the images from fanart
posterURL, backgroundURL, logoURL, err := FetchImagesFromFanart(serie.IDs.TVDB)
if err != nil {
fmt.Printf("Error fetching images from Fanart.tv for %s: %s\n", serie.Title, err)
continue
}
imageTypes := []struct {
URL string
ImageType string
SetField func(imagePath string)
}{
{
URL: posterURL,
ImageType: "poster",
SetField: func(imagePath string) { serie.Poster = imagePath },
},
{
URL: backgroundURL,
ImageType: "background",
SetField: func(imagePath string) { serie.Background = imagePath },
},
{
URL: logoURL,
ImageType: "logo",
SetField: func(imagePath string) { serie.Image = imagePath },
},
}
for _, image := range imageTypes {
if image.URL != "" {
err := DownloadImage(image.URL, serie.IDs.Slug, image.ImageType)
if err != nil {
fmt.Printf("Error downloading %s for %s: %s\n", image.ImageType, serie.Title, err)
} else {
image.SetField(fmt.Sprintf("%s-%s.jpg", serie.IDs.Slug, image.ImageType))
}
}
}
// utils.Debug(serie)
err = generateSerieMarkdown(serie)
if err != nil {
return err
}
utils.Sep()
}
return nil
}
func generateSerieMarkdown(serie Serie) error {
templatePath := filepath.Join(os.Getenv("TEMPLATES_DIR"), "serie.md.tpl")
outputDir := os.Getenv("MARKDOWN_OUTPUT_SERIES_DIR")
if err := utils.CreateDirIfNotExists(outputDir); err != nil {
return err
}
outputPath := filepath.Join(outputDir, fmt.Sprintf("%s.md", serie.IDs.Slug))
data := map[string]interface{}{
"Title": serie.Title,
"Link": serie.Link,
"Subtitle": serie.Year,
"Year": serie.Year,
"Rate": serie.Rate,
"Progress": serie.Progress,
"Image": serie.Image,
"Poster": serie.Poster,
"Background": serie.Background,
"Date": serie.Date,
"Tags": "watching",
}
return utils.GenerateMarkdown(templatePath, outputPath, data)
}

11
internal/series/model.go
View File

@ -2,9 +2,16 @@
package series
type Serie struct {
Title string `yaml:"title"`
Link string `yaml:"link"`
Title string `yaml:"title"`
IDs struct {
Slug string `json:"slug"`
IMDB string `json:"imdb"`
TMDB int `json:"tmdb"`
Trakt int `json:"trakt"`
TVDB int `json:"tvdb"`
} `json:"ids"`
Subtitle string `yaml:"subtitle"`
Link string `yaml:"link"`
Year int `yaml:"year"`
Rate float64 `yaml:"rate"`
Progress string `yaml:"progress"`

12
main.go
View File

@ -3,6 +3,7 @@ package main
import (
"fmt"
"hugo-medialog/internal/movies"
"hugo-medialog/internal/series"
"hugo-medialog/utils"
)
@ -21,4 +22,15 @@ func main() {
fmt.Printf("Error processing movies: %v\n", err)
}
// Series
seriesList, err := series.LoadSeries()
if err != nil {
fmt.Printf("Error reading series file: %v\n", err)
return
}
err = series.ProcessSeries(seriesList)
if err != nil {
fmt.Printf("Error processing series: %v\n", err)
}
}

4
templates/serie.md.tpl
View File

@ -6,8 +6,8 @@ year: {{ .Year }}
rate: {{ .Rate }}
progress: {{ .Progress }}
image: {{ .Image }}
poster-image: {{ .PosterImage }}
background-image: {{ .BackgroundImage }}
poster-image: {{ .Poster }}
background-image: {{ .Background }}
date: {{ .Date }}
draft: false
tags: {{ .Tags }}