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

tmdb/drop.go

@@ -0,0 +1,115 @@
+package tmdb
+
+import (
+	"context"
+	"fmt"
+	"math/rand/v2"
+	"time"
+
+	"codeberg.org/danjones000/my-log/files"
+	"codeberg.org/danjones000/my-log/models"
+	sdk "github.com/cyruzin/golang-tmdb"
+)
+
+func Drop(ctx context.Context, url string, now time.Time) (models.Log, error) {
+	var log models.Log
+	select {
+	case <-ctx.Done():
+		return log, context.Cause(ctx)
+	default:
+	}
+
+	inf, err := Parse(url)
+	if err != nil {
+		return log, err
+	}
+
+	var entry models.Entry
+
+	if inf.Type == Movie {
+		det, err := FetchMovieDetails(inf)
+		if err != nil {
+			return log, err
+		}
+		entry, err = GetLogEntryForMovie(det)
+		if err != nil {
+			return log, err
+		}
+	} else if inf.Type == Series {
+		det, err := FetchEpisodeDetails(inf)
+		if err != nil {
+			return log, err
+		}
+		entry, err = GetLogEntryForEpisode(det)
+		if err != nil {
+			return log, err
+		}
+	} else {
+		return log, fmt.Errorf("unknown type: %s", inf.Type)
+	}
+
+	log.Name = "watched"
+	log.Entries = []models.Entry{entry}
+
+	files.Append(log)
+
+	return log, nil
+}
+
+func GetLogEntryForEpisode(det TVDetails) (ent models.Entry, err error) {
+	ent.Title = fmt.Sprintf("%s - %s %dx%02d", det.Episode.Name, det.Series.Name, det.Episode.SeasonNumber, det.Episode.EpisodeNumber)
+	ent.Date = time.Now().Local()
+
+	fields := &models.Metas{}
+	fields.AppendTo("tmdb", det.Series.ID)
+	fields.AppendTo("show", det.Series.Name)
+	fields.AppendTo("episode", det.Episode.Name)
+	fields.AppendTo("air_date", det.Episode.AirDate)
+	fields.AppendTo("id", fmt.Sprintf("tag:themoviedborg,%s:Episode/%d/season/%d/episode/%d/%d", det.Episode.AirDate, det.Series.ID, det.Episode.SeasonNumber, det.Episode.EpisodeNumber, rand.Int()))
+	fields.AppendTo("url", fmt.Sprintf("https://themoviedb.org/tv/%d/season/%d/episode/%d", det.Series.ID, det.Episode.SeasonNumber, det.Episode.EpisodeNumber))
+	fields.AppendTo("episode_num", det.Episode.EpisodeNumber)
+	fields.AppendTo("season_num", det.Episode.SeasonNumber)
+	fields.AppendTo("via", "drop-tmdb")
+	if det.Episode.Overview != "" {
+		fields.AppendTo("note", det.Episode.Overview)
+	}
+
+	ent.Fields = *fields
+	return
+}
+
+func GetLogEntryForMovie(det *sdk.MovieDetails) (ent models.Entry, err error) {
+	ent.Title = fmt.Sprintf("%s (%s)", det.Title, det.ReleaseDate)
+	ent.Date = time.Now().Local()
+
+	fields := &models.Metas{}
+	fields.AppendTo("tmdb", det.ID)
+	fields.AppendTo("id", fmt.Sprintf("tag:themoviedborg,%s:Movie/%d/%d", det.ReleaseDate, det.ID, rand.Int()))
+	fields.AppendTo("url", fmt.Sprintf("https://themoviedb.org/movie/%d", det.ID))
+	// url https://www.themoviedb.org/movie/776503
+	fields.AppendTo("release_date", det.ReleaseDate)
+	cts := det.OriginCountry
+	if len(cts) == 1 {
+		fields.AppendTo("country", cts[0])
+	} else if len(cts) > 1 {
+		fields.AppendTo("country", cts)
+	}
+
+	gens := make([]string, len(det.Genres))
+	for idx, g := range det.Genres {
+		gens[idx] = g.Name
+	}
+	if len(gens) == 1 {
+		fields.AppendTo("genre", gens[0])
+	} else if len(gens) > 1 {
+		fields.AppendTo("genre", gens)
+	}
+
+	// fields.AppendTo("mpaa")
+	if det.Overview != "" {
+		fields.AppendTo("note", det.Overview)
+	}
+
+	ent.Fields = *fields
+	return
+}