package main

import (
	"fmt"
	"net/http"
	"strings"

	"github.com/aerogo/http/client"
	"github.com/animenotifier/arn"
	"github.com/animenotifier/kitsu"
	"github.com/fatih/color"
)

func main() {
	color.Yellow("Importing Kitsu anime")

	defer arn.Node.Close()
	defer color.Green("Finished.")

	// In case we refresh only one anime
	if InvokeShellArgs() {
		return
	}

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

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

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

	// This stops overwriting existing data
	// if anime != nil {
	// 	return anime
	// }

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

	// Status "unreleased" means the same as "upcoming" so we should normalize it
	if anime.Status == "unreleased" {
		anime.Status = "upcoming"
	}

	// Normalize image extension to .jpg if .jpeg is used
	if anime.Image.Extension == ".jpeg" {
		anime.Image.Extension = ".jpg"
	}

	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.SetMapping("myanimelist/anime", mapping.Attributes.ExternalID, "")
		case "anidb":
			anime.SetMapping("anidb/anime", mapping.Attributes.ExternalID, "")
		case "thetvdb", "thetvdb/series":
			fmt.Println(mapping.Attributes.ExternalSite, mapping.Attributes.ExternalID)
			anime.SetMapping("thetvdb/anime", mapping.Attributes.ExternalID, "")
		case "thetvdb/season":
			// Ignore
		default:
			color.Yellow("Unknown mapping: %s %s", mapping.Attributes.ExternalSite, mapping.Attributes.ExternalID)
		}
	}

	return anime

	// Download image
	response, err := client.Get(attr.PosterImage.Original).End()

	if err == nil && response.StatusCode() == http.StatusOK {
		anime.SetImageBytes(response.Bytes())
	} else {
		color.Red("No image for [%s] %s (%d)", anime.ID, anime, response.StatusCode())
	}

	// 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 {
		episodes := &arn.AnimeEpisodes{
			AnimeID: anime.ID,
			Items:   []*arn.AnimeEpisode{},
		}

		arn.DB.Set("AnimeEpisodes", anime.ID, episodes)
	}

	// Relations
	relations, _ := arn.GetAnimeRelations(anime.ID)

	if relations == nil {
		relations := &arn.AnimeRelations{
			AnimeID: anime.ID,
			Items:   []*arn.AnimeRelation{},
		}

		arn.DB.Set("AnimeRelations", anime.ID, relations)
	}

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

	return anime
}