2018-03-21 05:15:03 +01:00

180 lines
4.4 KiB
Go

package main
import (
"fmt"
"github.com/animenotifier/arn"
"github.com/animenotifier/kitsu"
"github.com/fatih/color"
)
func main() {
color.Yellow("Importing Kitsu anime")
defer color.Green("Finished.")
defer arn.Node.Close()
// In case we refresh only one anime
if InvokeShellArgs() {
return
}
// Get a stream of all anime
allAnime := kitsu.StreamAnime()
// Iterate over the stream
for anime := range allAnime {
sync(anime)
}
}
func sync(anime *kitsu.Anime) {
fmt.Println(anime.ID, anime.Attributes.CanonicalTitle)
arn.Kitsu.Set("Anime", anime.ID, 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
// }