190 lines
4.7 KiB
Go
Raw Normal View History

2018-04-26 19:36:26 +00:00
package main
import (
"fmt"
"github.com/aerogo/http/client"
2019-04-23 05:52:55 +00:00
"github.com/akyoto/color"
2019-06-03 09:32:43 +00:00
"github.com/animenotifier/notify.moe/arn"
2018-04-26 19:36:26 +00:00
"github.com/animenotifier/mal"
)
// Sync titles
func syncTitles(anime *arn.Anime, malAnime *mal.Anime) {
if anime.Title.Japanese == "" && malAnime.JapaneseTitle != "" {
fmt.Println("JapaneseTitle:", malAnime.JapaneseTitle)
anime.Title.Japanese = malAnime.JapaneseTitle
}
if anime.Title.English == "" && malAnime.EnglishTitle != "" {
fmt.Println("EnglishTitle:", malAnime.EnglishTitle)
anime.Title.English = malAnime.EnglishTitle
}
}
// Sync dates
func syncDates(anime *arn.Anime, malAnime *mal.Anime) {
if anime.StartDate == "" && malAnime.StartDate != "" {
fmt.Println("StartDate:", malAnime.StartDate)
anime.StartDate = malAnime.StartDate
}
if anime.EndDate == "" && malAnime.EndDate != "" {
fmt.Println("EndDate:", malAnime.EndDate)
anime.EndDate = malAnime.EndDate
}
}
// Sync episodes
func syncEpisodes(anime *arn.Anime, malAnime *mal.Anime) {
if anime.EpisodeCount == 0 && malAnime.EpisodeCount != 0 {
fmt.Println("EpisodeCount:", malAnime.EpisodeCount)
anime.EpisodeCount = malAnime.EpisodeCount
}
if anime.EpisodeLength == 0 && malAnime.EpisodeLength != 0 {
fmt.Println("EpisodeLength:", malAnime.EpisodeLength)
anime.EpisodeLength = malAnime.EpisodeLength
}
}
// Sync others
func syncOthers(anime *arn.Anime, malAnime *mal.Anime) {
if len(anime.Genres) == 0 && len(malAnime.Genres) > 0 {
fmt.Println("Genres:", malAnime.Genres)
anime.Genres = malAnime.Genres
}
if anime.Source == "" && malAnime.Source != "" {
fmt.Println("Source:", malAnime.Source)
anime.Source = malAnime.Source
}
}
// Sync image
func syncImage(anime *arn.Anime, malAnime *mal.Anime) {
if (anime.HasImage() && anime.Image.Width > imageWidthThreshold) || malAnime.Image == "" {
return
}
fmt.Println("Downloading image:", malAnime.Image)
response, err := client.Get(malAnime.Image).End()
if err == nil && response.StatusCode() == 200 {
2019-04-28 10:51:06 +00:00
err := anime.SetImageBytes(response.Bytes())
if err != nil {
color.Red("Error setting image: %s", err.Error())
}
2018-04-26 19:36:26 +00:00
} else {
color.Red("Error downloading image")
}
}
// Sync characters
func syncCharacters(anime *arn.Anime, malAnime *mal.Anime) {
// Check for existence of characters
animeCharacters := anime.Characters()
modifiedCharacters := false
for _, malAnimeCharacter := range malAnime.Characters {
// Make sure we have no invalid entries
if malAnimeCharacter.ID == "" || malAnimeCharacter.Role == "" {
fmt.Println("Skip:", malAnimeCharacter)
continue
}
animeCharacter := animeCharacters.FindByMapping("myanimelist/character", malAnimeCharacter.ID)
if animeCharacter != nil {
continue
}
obj, err := malDB.Get("Character", malAnimeCharacter.ID)
// If we don't have the MAL character in the DB,
// we can't import anything here.
if err != nil {
continue
}
malCharacter := obj.(*mal.Character)
2018-10-30 03:01:42 +00:00
if malCharacter.ID == "" || malCharacter.Name == "" || malCharacter.Image == "" {
2018-04-26 19:36:26 +00:00
fmt.Println("Skip character:", malAnimeCharacter.ID)
continue
}
fmt.Println("Importing MAL AnimeCharacter:", malAnimeCharacter.ID, "as", malAnimeCharacter.Role)
// Import character if needed
character := characterFinder.GetCharacter(malAnimeCharacter.ID)
if character == nil {
character = importCharacter(malCharacter)
} else {
fmt.Println("Found existing character:", character)
// Download image if missing
if !character.HasImage() {
fmt.Println("Downloading missing image for character:", character)
2019-04-28 10:51:06 +00:00
err := character.DownloadImage(malCharacter.ImageLink())
2018-04-26 19:36:26 +00:00
// Cancel import if that character has no image
if err != nil {
color.Red(err.Error())
continue
}
}
}
// If import failed, continue
if character == nil {
continue
}
// Add to anime characters
err = animeCharacters.Add(&arn.AnimeCharacter{
CharacterID: character.ID,
Role: malAnimeCharacter.Role,
})
if err != nil {
color.Red(err.Error())
}
modifiedCharacters = true
}
if modifiedCharacters {
animeCharacters.Save()
}
}
// import character
func importCharacter(malCharacter *mal.Character) *arn.Character {
2018-10-30 03:01:42 +00:00
fmt.Println("Importing MAL Character:", malCharacter.ID, malCharacter.Name, malCharacter.Image)
2018-04-26 19:36:26 +00:00
character := arn.NewCharacter()
character.Name.Canonical = malCharacter.Name
// Cancel the import if image could not be fetched
err := character.DownloadImage(malCharacter.ImageLink())
if err != nil {
color.Red(err.Error())
return nil
}
// Add mapping
character.SetMapping("myanimelist/character", malCharacter.ID)
// Save character in DB
character.Save()
// Add to character finder so we don't create duplicates of this character
characterFinder.Add(character)
return character
}