From f9bbf7e6db2cd84cd0f499b9ce1e1416ddd36e8f Mon Sep 17 00:00:00 2001
From: Eduard Urbach <e.urbach@gmail.com>
Date: Sun, 2 Jul 2017 01:44:10 +0200
Subject: [PATCH] Improved import

---
 .../import-anilist-user.go                    | 42 ++++++++++
 .../import-old/import-old-matches.go          |  0
 patches/import-anilist/import-anilist.go      | 83 +++----------------
 3 files changed, 52 insertions(+), 73 deletions(-)
 create mode 100644 patches/import-anilist-user/import-anilist-user.go
 rename patches/{import-anilist => import-anilist-user}/import-old/import-old-matches.go (100%)

diff --git a/patches/import-anilist-user/import-anilist-user.go b/patches/import-anilist-user/import-anilist-user.go
new file mode 100644
index 00000000..80a13763
--- /dev/null
+++ b/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))
+}
diff --git a/patches/import-anilist/import-old/import-old-matches.go b/patches/import-anilist-user/import-old/import-old-matches.go
similarity index 100%
rename from patches/import-anilist/import-old/import-old-matches.go
rename to patches/import-anilist-user/import-old/import-old-matches.go
diff --git a/patches/import-anilist/import-anilist.go b/patches/import-anilist/import-anilist.go
index e2f90083..da1a1276 100644
--- a/patches/import-anilist/import-anilist.go
+++ b/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++
+	}
 }