From 85944f89877264c3cb808e3bf68610e6b2762dea Mon Sep 17 00:00:00 2001
From: newt-sc <47229722+newt-sc@users.noreply.github.com>
Date: Mon, 10 May 2021 00:49:49 +0300
Subject: [PATCH] release: v2.6.0

---
 CHANGELOG.md              |  3 +++
 a4kSubtitles/api.py       |  2 ++
 a4kSubtitles/lib/video.py |  5 +++++
 addon.xml                 |  5 ++++-
 packages/addons.xml       |  5 ++++-
 packages/addons.xml.crc   |  2 +-
 tests/test_suite.py       | 16 ++++++++++++++++
 7 files changed, 35 insertions(+), 3 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index ec9267a..71e664b 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -1,3 +1,6 @@
+* [v2.6.0](https://github.com/newt-sc/a4kSubtitles/releases/tag/service.subtitles.a4ksubtitles%2Fservice.subtitles.a4ksubtitles-2.6.0):
+  * Support imdb id passed from video file url.
+
 * [v2.5.0](https://github.com/newt-sc/a4kSubtitles/releases/tag/service.subtitles.a4ksubtitles%2Fservice.subtitles.a4ksubtitles-2.5.0):
   * Add TLS adapter fallback
 
diff --git a/a4kSubtitles/api.py b/a4kSubtitles/api.py
index 153bb75..57bcf28 100644
--- a/a4kSubtitles/api.py
+++ b/a4kSubtitles/api.py
@@ -40,6 +40,8 @@ def get_info_label(label):
                 return meta.get('_title', '')
             if label == 'VideoPlayer.IMDBNumber':
                 return meta.get('imdb_id', '')
+            if label == 'Player.FilenameAndPath':
+                return meta.get('url', '')
         default = self.core.kodi.xbmc.getInfoLabel
         self.core.kodi.xbmc.getInfoLabel = get_info_label
 
diff --git a/a4kSubtitles/lib/video.py b/a4kSubtitles/lib/video.py
index f41f51f..46212aa 100644
--- a/a4kSubtitles/lib/video.py
+++ b/a4kSubtitles/lib/video.py
@@ -301,6 +301,11 @@ def __get_basic_info():
     meta.filename_without_ext = meta.filename
     meta.imdb_id = xbmc.getInfoLabel('VideoPlayer.IMDBNumber')
 
+    if meta.imdb_id == '':
+        regex_result = re.search(r'.*(tt\d{7,}).*', xbmc.getInfoLabel('Player.FilenameAndPath'), re.IGNORECASE)
+        if regex_result:
+            meta.imdb_id = regex_result.group(1)
+
     return meta
 
 def get_meta(core):
diff --git a/addon.xml b/addon.xml
index 07e3b68..f5669c4 100644
--- a/addon.xml
+++ b/addon.xml
@@ -1,7 +1,7 @@
 <?xml version="1.0" encoding="UTF-8" standalone="yes"?>
 <addon id="service.subtitles.a4ksubtitles"
        name="a4kSubtitles"
-       version="2.5.0"
+       version="2.6.0"
        provider-name="Unknown">
     <requires>
       <import addon="script.module.requests"/>
@@ -27,6 +27,9 @@ Supports: OpenSubtitles, BSPlayer, Podnadpisi.NET, Subscene, Addic7ed
             <screenshot>screenshot-03.png</screenshot>
         </assets>
         <news>
+[v2.6.0]:
+  * Support imdb id passed from video file url.
+
 [v2.5.0]:
   * Add TLS adapter fallback
 
diff --git a/packages/addons.xml b/packages/addons.xml
index 6cf0779..e8d14ee 100644
--- a/packages/addons.xml
+++ b/packages/addons.xml
@@ -4,7 +4,7 @@
 <addons>
 <addon id="service.subtitles.a4ksubtitles"
        name="a4kSubtitles"
-       version="2.5.0"
+       version="2.6.0"
        provider-name="Unknown">
     <requires>
       <import addon="script.module.requests"/>
@@ -30,6 +30,9 @@ Supports: OpenSubtitles, BSPlayer, Podnadpisi.NET, Subscene, Addic7ed
             <screenshot>screenshot-03.png</screenshot>
         </assets>
         <news>
+[v2.6.0]:
+  * Support imdb id passed from video file url.
+
 [v2.5.0]:
   * Add TLS adapter fallback
 
diff --git a/packages/addons.xml.crc b/packages/addons.xml.crc
index f535665..b10373a 100644
--- a/packages/addons.xml.crc
+++ b/packages/addons.xml.crc
@@ -1 +1 @@
-ffd194a94b96801a0bf4512fcd5d65ac00836de4
\ No newline at end of file
+3e49ed72de384e85712a5c6e9074690f8c7a8520
\ No newline at end of file
diff --git a/tests/test_suite.py b/tests/test_suite.py
index a8b98b2..490fe6a 100644
--- a/tests/test_suite.py
+++ b/tests/test_suite.py
@@ -347,6 +347,22 @@ def test_opensubtitles_missing_imdb_id():
 
     assert len(search.results) > 0
 
+def test_opensubtitles_missing_imdb_id_but_in_url():
+    a4ksubtitles_api = api.A4kSubtitlesApi({'kodi': True})
+    __remove_all_cache(a4ksubtitles_api)
+
+    # search
+    settings = {
+        'opensubtitles.enabled': 'true',
+    }
+    video_meta = {
+        'imdb_id': '',
+        'url': 'https://example.com/example.mkv?imdb_id=tt3183660'
+    }
+    search = __search_movie(a4ksubtitles_api, settings, video_meta)
+
+    assert len(search.results) > 0
+
 def test_opensubtitles_tvshow_missing_imdb_id():
     a4ksubtitles_api = api.A4kSubtitlesApi({'kodi': True})
     __remove_all_cache(a4ksubtitles_api)