180 lines
4.4 KiB
Go
180 lines
4.4 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/akyoto/color"
|
|
"github.com/animenotifier/kitsu"
|
|
"github.com/animenotifier/notify.moe/arn"
|
|
)
|
|
|
|
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.MediaTitle{},
|
|
// }
|
|
// } 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
|
|
// }
|