Improved import

patches/import-anilist-user/import-anilist-user.go

@@ -0,0 +1,42 @@
+package main
+
+import (
+	"fmt"
+
+	"github.com/animenotifier/arn"
+	"github.com/fatih/color"
+)
+
+var userName = "Akyoto"
+var allAnime []*arn.Anime
+
+func init() {
+	allAnime, _ = arn.AllAnime()
+}
+
+func main() {
+	arn.PanicOnError(arn.AniList.Authorize())
+	println(arn.AniList.AccessToken)
+
+	user, _ := arn.GetUserByNick(userName)
+	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 := arn.FindAniListAnime(item.Anime, allAnime)
+
+		if anime != nil {
+			fmt.Println(item.Anime.TitleRomaji, "=>", anime.Title.Romaji)
+			imported = append(imported, anime)
+		}
+	}
+
+	color.Green("%d / %d", len(imported), len(animeListItems))
+}

patches/import-anilist/import-anilist.go

@@ -1,92 +1,29 @@
 package main
 
 import (
-	"strconv"
-	"strings"
+	"fmt"
 
 	"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)
+	color.Green(arn.AniList.AccessToken)
 
-	user, _ := arn.GetUserByNick("Boltasar")
-	animeList, err := arn.AniList.GetAnimeList(user)
+	allAnime, err := arn.AllAnime()
 	arn.PanicOnError(err)
 
-	importList(animeList.Lists.Watching)
-	importList(animeList.Lists.Completed)
-}
+	count := 0
+	stream := arn.AniList.StreamAnime()
 
-func importList(animeListItems []*arn.AniListAnimeListItem) {
-	imported := []*arn.Anime{}
+	for aniListAnime := range stream {
+		anime := arn.FindAniListAnime(aniListAnime, allAnime)
 
-	for _, item := range animeListItems {
-		anime := findAniListAnime(item.Anime)
 		if anime != nil {
-			// fmt.Println(item.Anime.TitleRomaji, "=>", anime.Title.Romaji)
-			imported = append(imported, anime)
+			fmt.Println(aniListAnime.TitleRomaji, "=>", anime.Title.Canonical)
 		}
-	}
 
-	color.Green("%d / %d", len(imported), len(animeListItems))
-}
-
-func findAniListAnime(search *arn.AniListAnime) *arn.Anime {
-	match, err := arn.GetAniListToAnime(strconv.Itoa(search.ID))
-
-	if err == nil {
-		anime, _ := arn.GetAnime(match.AnimeID)
-		return 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
+		count++
+	}
 }