Basic Anilist importer

patches/import-anilist/import-anilist.go

@@ -1,9 +1,85 @@
 package main
 
 import (
+	"fmt"
+	"strings"
+
 	"github.com/animenotifier/arn"
+	"github.com/fatih/color"
 )
 
+var allAnime []*arn.Anime
+
+func init() {
+	allAnime, _ = arn.AllAnime()
+}
+
 func main() {
 	arn.PanicOnError(arn.AniList.Authorize())
+	println(arn.AniList.AccessToken)
+
+	user, _ := arn.GetUserByNick("Boltasar")
+	animeList, err := arn.AniList.GetAnimeList(user)
+	arn.PanicOnError(err)
+
+	importList(animeList.Lists.Watching)
+	importList(animeList.Lists.Completed)
+}
+
+func importList(animeListItems []*arn.AniListAnimeListItem) {
+	imported := []*arn.Anime{}
+
+	for _, item := range animeListItems {
+		anime := findAnimeByName(item.Anime)
+		if anime != nil {
+			// fmt.Println(item.Anime.TitleRomaji, "=>", anime.Title.Romaji)
+			imported = append(imported, anime)
+		}
+	}
+
+	color.Green("%d / %d", len(imported), len(animeListItems))
+}
+
+func findAnimeByName(search *arn.AniListAnime) *arn.Anime {
+	var mostSimilar *arn.Anime
+	var similarity float64
+
+	for _, anime := range allAnime {
+		anime.Title.Japanese = strings.Replace(anime.Title.Japanese, "2ndシーズン", "2", 1)
+		anime.Title.Romaji = strings.Replace(anime.Title.Romaji, " 2nd Season", " 2", 1)
+		search.TitleJapanese = strings.TrimSpace(strings.Replace(search.TitleJapanese, "2ndシーズン", "2", 1))
+		search.TitleRomaji = strings.TrimSpace(strings.Replace(search.TitleRomaji, " 2nd Season", " 2", 1))
+
+		titleSimilarity := arn.StringSimilarity(anime.Title.Romaji, search.TitleRomaji)
+
+		if strings.ToLower(anime.Title.Japanese) == strings.ToLower(search.TitleJapanese) {
+			titleSimilarity += 1.0
+		}
+
+		if strings.ToLower(anime.Title.Romaji) == strings.ToLower(search.TitleRomaji) {
+			titleSimilarity += 1.0
+		}
+
+		if strings.ToLower(anime.Title.English) == strings.ToLower(search.TitleEnglish) {
+			titleSimilarity += 1.0
+		}
+
+		if titleSimilarity > similarity {
+			mostSimilar = anime
+			similarity = titleSimilarity
+		}
+	}
+
+	if mostSimilar.EpisodeCount != search.TotalEpisodes {
+		similarity -= 0.02
+	}
+
+	if similarity >= 0.92 {
+		fmt.Printf("MATCH:    %s => %s (%.2f)\n", search.TitleRomaji, mostSimilar.Title.Romaji, similarity)
+		return mostSimilar
+	}
+
+	color.Red("MISMATCH: %s => %s (%.2f)", search.TitleRomaji, mostSimilar.Title.Romaji, similarity)
+
+	return nil
 }