package main

import (
	"fmt"
	"strings"

	"github.com/animenotifier/arn"
	"github.com/animenotifier/kitsu"
	"github.com/fatih/color"
)

func main() {
	color.Yellow("Syncing Anime")
	defer arn.Node.Close()

	// In case we refresh only one anime
	if InvokeShellArgs() {
		color.Green("Finished.")
		return
	}

	// Get a stream of all anime
	allAnime := kitsu.StreamAnimeWithMappings()

	// Iterate over the stream
	for anime := range allAnime {
		sync(anime)
	}

	color.Green("Finished.")
}

func sync(data *kitsu.Anime) *arn.Anime {
	anime, err := arn.GetAnime(data.ID)

	if err != nil {
		if strings.Contains(err.Error(), "not found") {
			anime = &arn.Anime{
				Title: &arn.AnimeTitle{},
				Image: &arn.AnimeImageTypes{},
			}
		} 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{}
	}

	// 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
	}

	// Import mappings
	for _, mapping := range data.Mappings {
		switch mapping.Attributes.ExternalSite {
		case "myanimelist/anime":
			anime.AddMapping("myanimelist/anime", mapping.Attributes.ExternalID, "")
		case "anidb":
			anime.AddMapping("anidb/anime", mapping.Attributes.ExternalID, "")
		case "thetvdb/series":
			anime.AddMapping("thetvdb/anime", mapping.Attributes.ExternalID, "")
		case "thetvdb/season":
			// Ignore
		default:
			color.Yellow("Unknown mapping: %s %s", mapping.Attributes.ExternalSite, mapping.Attributes.ExternalID)
		}
	}

	// 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()
	}

	// Popularity
	if anime.Popularity == nil {
		anime.Popularity = &arn.AnimePopularity{}
	}

	// Trailers
	anime.Trailers = []*arn.ExternalMedia{}

	if attr.YoutubeVideoID != "" {
		anime.Trailers = append(anime.Trailers, &arn.ExternalMedia{
			Service:   "Youtube",
			ServiceID: attr.YoutubeVideoID,
		})
	}

	// Save in database
	anime.Save()

	// Episodes
	episodes, err := arn.GetAnimeEpisodes(anime.ID)

	if err != nil || episodes == nil {
		anime.RefreshEpisodes()
	}

	// Log
	fmt.Println(color.GreenString("✔"), anime.ID, anime.Title.Canonical)

	return anime
}