From fbbf15173b804bfaa810b8405982ae14c3bf7f37 Mon Sep 17 00:00:00 2001
From: TheBeastLT <pauliox@beyond.lt>
Date: Tue, 10 Mar 2020 19:47:06 +0100
Subject: [PATCH] assigns kitsu episodes for anime type when no metadata
 available

---
 scraper/lib/torrentFiles.js | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/scraper/lib/torrentFiles.js b/scraper/lib/torrentFiles.js
index 0800d2c..201fddc 100644
--- a/scraper/lib/torrentFiles.js
+++ b/scraper/lib/torrentFiles.js
@@ -49,7 +49,7 @@ async function parseTorrentFiles(torrent) {
           .filter((file) => file.size > MIN_SIZE)
           .map((file) => parseSeriesFile(file, parsedTorrentName)))
       .then((files) => decomposeEpisodes(torrent, files, metadata))
-      .then((files) => assignKitsuOrImdbEpisodes(files, metadata))
+      .then((files) => assignKitsuOrImdbEpisodes(torrent, files, metadata))
       .then((files) => Promise.all(files.map(file => file.isMovie
           ? mapSeriesMovie(file, torrent)
           : mapSeriesEpisode(file, torrent, files))))
@@ -236,8 +236,18 @@ function getTimeZoneOffset(country) {
   }
 }
 
-function assignKitsuOrImdbEpisodes(files, metadata) {
+function assignKitsuOrImdbEpisodes(torrent, files, metadata) {
   if (!metadata || !metadata.videos || !metadata.videos.length) {
+    if (torrent.type === Type.ANIME) {
+      // assign episodes as kitsu episodes for anime when no metadata available for imdb mapping
+      files
+          .filter(file => file.season && file.episodes)
+          .forEach(file => {
+            file.kitsuEpisodes = file.episodes;
+            file.season = undefined;
+            file.episodes = undefined;
+          })
+    }
     return files;
   }