220 lines
5.2 KiB
Go
Raw Normal View History

2018-04-12 11:39:58 +00:00
package main
import (
"fmt"
2018-04-17 13:28:57 +00:00
"github.com/aerogo/http/client"
2018-04-12 11:39:58 +00:00
"github.com/animenotifier/arn"
"github.com/animenotifier/mal"
"github.com/fatih/color"
)
2018-04-17 13:33:19 +00:00
const imageWidthThreshold = 225
2018-04-17 13:28:57 +00:00
2018-04-12 11:39:58 +00:00
var (
malDB = arn.Node.Namespace("mal").RegisterTypes((*mal.Anime)(nil))
characterFinder = arn.NewCharacterFinder("myanimelist/character")
)
func main() {
color.Yellow("Syncing with MAL DB")
defer color.Green("Finished.")
defer arn.Node.Close()
2018-04-12 17:05:05 +00:00
// Invoke via parameters
if InvokeShellArgs() {
return
}
2018-04-12 11:39:58 +00:00
// Sync the most important ones first
allAnime := arn.AllAnime()
arn.SortAnimeByQuality(allAnime)
for _, anime := range allAnime {
malID := anime.GetMapping("myanimelist/anime")
if malID == "" {
continue
}
sync(anime, malID)
}
}
func sync(anime *arn.Anime, malID string) {
obj, err := malDB.Get("Anime", malID)
if err != nil {
fmt.Println(err)
return
}
malAnime := obj.(*mal.Anime)
// Log title
fmt.Printf("%s %s\n", color.CyanString(anime.Title.Canonical), malID)
if len(anime.Genres) == 0 && len(malAnime.Genres) > 0 {
fmt.Println("Genres:", malAnime.Genres)
anime.Genres = malAnime.Genres
}
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
}
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
}
if anime.Source == "" && malAnime.Source != "" {
fmt.Println("Source:", malAnime.Source)
anime.Source = malAnime.Source
}
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
}
2018-04-17 13:33:19 +00:00
if (!anime.HasImage() || anime.Image.Width <= imageWidthThreshold) && malAnime.Image != "" {
2018-04-17 13:28:57 +00:00
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")
}
}
2018-04-26 01:35:08 +00:00
// Characters
syncCharacters(anime, malAnime)
// Save in database
anime.Save()
}
func syncCharacters(anime *arn.Anime, malAnime *mal.Anime) {
2018-04-12 11:39:58 +00:00
// 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.ImagePath == "" {
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()
}
}
func importCharacter(malCharacter *mal.Character) *arn.Character {
fmt.Println("Importing MAL Character:", malCharacter.ID, malCharacter.Name, malCharacter.ImagePath)
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()
2018-04-22 15:43:20 +00:00
// Add to character finder so we don't create duplicates of this character
characterFinder.Add(character)
2018-04-12 11:39:58 +00:00
return character
}