Cleanup

jobs/sync-anime/sync-anime.go

@@ -0,0 +1,120 @@
+package main
+
+import (
+	"encoding/json"
+	"fmt"
+	"strings"
+
+	"github.com/animenotifier/arn"
+	"github.com/animenotifier/kitsu"
+	"github.com/fatih/color"
+)
+
+func main() {
+	color.Yellow("Syncing Anime")
+
+	// Get a stream of all anime
+	allAnime := kitsu.StreamAnime()
+
+	// Iterate over the stream
+	for anime := range allAnime {
+		sync(anime)
+	}
+
+	color.Green("Finished.")
+}
+
+func sync(data *kitsu.Anime) {
+	anime, err := arn.GetAnime(data.ID)
+
+	if err != nil {
+		if strings.Contains(err.Error(), "not found") {
+			anime = &arn.Anime{}
+		} else {
+			panic(err)
+		}
+	}
+
+	attr := data.Attributes
+
+	// General data
+	anime.ID = data.ID
+	anime.Type = strings.ToLower(attr.ShowType)
+	anime.Title.Canonical = attr.CanonicalTitle
+	anime.Title.English = attr.Titles.En
+	anime.Title.Romaji = attr.Titles.EnJp
+	anime.Title.Synonyms = attr.AbbreviatedTitles
+	anime.Image.Tiny = kitsu.FixImageURL(attr.PosterImage.Tiny)
+	anime.Image.Small = kitsu.FixImageURL(attr.PosterImage.Small)
+	anime.Image.Large = kitsu.FixImageURL(attr.PosterImage.Large)
+	anime.Image.Original = kitsu.FixImageURL(attr.PosterImage.Original)
+	anime.StartDate = attr.StartDate
+	anime.EndDate = attr.EndDate
+	anime.EpisodeCount = attr.EpisodeCount
+	anime.EpisodeLength = attr.EpisodeLength
+	anime.Status = attr.Status
+	anime.Summary = arn.FixAnimeDescription(attr.Synopsis)
+
+	if anime.Mappings == nil {
+		anime.Mappings = []*arn.Mapping{}
+	}
+
+	if anime.Episodes == nil {
+		anime.Episodes = []*arn.AnimeEpisode{}
+	}
+
+	// Prefer Shoboi Japanese titles over Kitsu JP titles
+	if anime.GetMapping("shoboi/anime") != "" {
+		// Only take Kitsu title when our JP title is empty
+		if anime.Title.Japanese == "" {
+			anime.Title.Japanese = attr.Titles.JaJp
+		}
+	} else {
+		// Update JP title with Kitsu JP title
+		anime.Title.Japanese = attr.Titles.JaJp
+	}
+
+	// NSFW
+	if attr.Nsfw {
+		anime.NSFW = 1
+	} else {
+		anime.NSFW = 0
+	}
+
+	// Rating
+	if anime.Rating == nil {
+		anime.Rating = &arn.AnimeRating{}
+	}
+
+	if anime.Rating.IsNotRated() {
+		anime.Rating.Reset()
+	}
+
+	// Trailers
+	anime.Trailers = []*arn.ExternalMedia{}
+
+	if attr.YoutubeVideoID != "" {
+		anime.Trailers = append(anime.Trailers, &arn.ExternalMedia{
+			Service:   "Youtube",
+			ServiceID: attr.YoutubeVideoID,
+		})
+	}
+
+	// Save in database
+	err = anime.Save()
+	status := ""
+
+	if err == nil {
+		status = color.GreenString("✔")
+	} else {
+		color.Red(err.Error())
+
+		data, _ := json.MarshalIndent(anime, "", "\t")
+		fmt.Println(string(data))
+
+		status = color.RedString("✘")
+	}
+
+	// Log
+	fmt.Println(status, anime.ID, anime.Title.Canonical)
+}