✨ Add drop:tmdb command for adding shows/movies to watched log

tmdb/fetch.go

@@ -0,0 +1,79 @@
+package tmdb
+
+import (
+	"errors"
+	"fmt"
+	"math/rand/v2"
+	"os"
+	"strconv"
+
+	sdk "github.com/cyruzin/golang-tmdb"
+)
+
+var ErrMissingKey = errors.New("missing TMDB key")
+
+func FetchMovieDetails(inf Info) (*sdk.MovieDetails, error) {
+	if inf.Type != Movie {
+		return nil, fmt.Errorf("%s is not a movie", inf.Type)
+	}
+	if inf.ID < 1 {
+		return nil, fmt.Errorf("%d is not a valid TMDB id", inf.ID)
+	}
+
+	sdk, err := GetClient()
+	if err != nil {
+		return nil, err
+	}
+
+	return sdk.GetMovieDetails(inf.ID, map[string]string{
+		"cache": strconv.Itoa(rand.Int()),
+	})
+}
+
+func FetchEpisodeDetails(inf Info) (TVDetails, error) {
+	var det TVDetails
+	if inf.Type != Series {
+		return det, fmt.Errorf("%s is not an episode", inf.Type)
+	}
+	if inf.ID < 1 {
+		return det, fmt.Errorf("%d is not a valid TMDB id", inf.ID)
+	}
+	if inf.Season < 1 {
+		return det, fmt.Errorf("%d is not a valid season", inf.Season)
+	}
+	if inf.Episode < 1 {
+		return det, fmt.Errorf("%d is not a valid episode", inf.Episode)
+	}
+
+	sdk, err := GetClient()
+	if err != nil {
+		return det, err
+	}
+
+	epDet, err := sdk.GetTVEpisodeDetails(inf.ID, inf.Season, inf.Episode, map[string]string{
+		"cache": strconv.Itoa(rand.Int()),
+	})
+	if err != nil {
+		return det, err
+	}
+
+	showDet, err := sdk.GetTVDetails(inf.ID, map[string]string{
+		"cache": strconv.Itoa(rand.Int()),
+	})
+	if err != nil {
+		return det, err
+	}
+
+	det.Episode = epDet
+	det.Series = showDet
+
+	return det, nil
+}
+
+func GetClient() (*sdk.Client, error) {
+	key := os.Getenv("TMDB_KEY")
+	if key == "" {
+		return nil, ErrMissingKey
+	}
+	return sdk.Init(key)
+}