186 lines
4.6 KiB
Go

package main
import (
"fmt"
"github.com/aerogo/http/client"
"github.com/akyoto/color"
"github.com/animenotifier/arn"
"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 {
anime.SetImageBytes(response.Bytes())
} 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)
if malCharacter.ID == "" || malCharacter.Name == "" || malCharacter.Image == "" {
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)
character.DownloadImage(malCharacter.ImageLink())
// 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 {
fmt.Println("Importing MAL Character:", malCharacter.ID, malCharacter.Name, malCharacter.Image)
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
}