From cfe722c1483933b812575b066f06931d57fa5fdb Mon Sep 17 00:00:00 2001
From: TheBeastLT <pauliox@beyond.lt>
Date: Fri, 27 Mar 2020 08:29:27 +0100
Subject: [PATCH] [addon] fixes id splitting

---
 addon/addon.js | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

diff --git a/addon/addon.js b/addon/addon.js
index 76783b5..cce0743 100644
--- a/addon/addon.js
+++ b/addon/addon.js
@@ -51,13 +51,13 @@ async function seriesRecordsHandler(args) {
   if (args.id.match(/tt\d+/)) {
     const parts = args.id.split(':');
     const imdbId = parts[0];
-    const season = parseInt(parts[1], 10);
-    const episode = parseInt(parts[2], 10);
+    const season = parts[1] ? parseInt(parts[1], 10) : 1;
+    const episode = parts[2] ? parseInt(parts[2], 10) : 1;
     return repository.getImdbIdSeriesEntries(imdbId, season, episode);
   } else if (args.id.match(/kitsu:\d+/i)) {
     const parts = args.id.split(':');
     const kitsuId = parts[1];
-    const episode = parseInt(parts[2], 10);
+    const episode = parts[2] ? parseInt(parts[2], 10) : 1;
     return repository.getKitsuIdSeriesEntries(kitsuId, episode);
   }
   return Promise.reject(`Unsupported id type: ${args.id}`);
@@ -65,9 +65,13 @@ async function seriesRecordsHandler(args) {
 
 async function movieRecordsHandler(args) {
   if (args.id.match(/tt\d+/)) {
-    return repository.getImdbIdMovieEntries(args.id);
+    const parts = args.id.split(':');
+    const imdbId = parts[0];
+    return repository.getImdbIdMovieEntries(imdbId);
   } else if (args.id.match(/kitsu:\d+/i)) {
-    return repository.getKitsuIdMovieEntries(args.id.replace('kitsu:', ''));
+    const parts = args.id.split(':');
+    const kitsuId = parts[1];
+    return repository.getKitsuIdMovieEntries(kitsuId);
   }
   return Promise.reject(`Unsupported id type: ${args.id}`);
 }