Merge pull request #96 from DesSolo/hdrezka_auth

feat: Add auth

Author: dandygithub

addons/plugin.video.hdrezka.tv/addon.xml

@@ -1,5 +1,5 @@
 <?xml version="1.0" encoding="UTF-8" standalone="yes"?>
-<addon id="plugin.video.hdrezka.tv" name="Hdrezka.tv" version="3.2.1" provider-name="MrStealth, dandy, DesSolo">
+<addon id="plugin.video.hdrezka.tv" name="Hdrezka.tv" version="3.2.2" provider-name="MrStealth, dandy, DesSolo">
     <requires>
         <import addon="xbmc.python" version="3.0.0"/>
         <import addon="script.module.xbmc.helpers" version="3.0.0"/>

addons/plugin.video.hdrezka.tv/default.py

@@ -2,7 +2,6 @@
 # -*- coding: utf-8 -*-
 #
 # Writer (c) 2012-2021, MrStealth, dandy
-
 import os
 import re
 import sys
@@ -59,6 +58,13 @@ def _load_session(self):
                 'Referer': self.domain,
                 'User-Agent': USER_AGENT,
             }
+
+        saved_cookies = self.addon.getSetting('cookies')
+        if saved_cookies:
+            session.cookies = helpers.load_cookies(
+                self.addon.getSetting('cookies')
+            )
+
         return session
 
     def _load_proxy_settings(self):
@@ -577,6 +583,28 @@ def _normalize_url(self, item):
             item = self.url + item
         return item
 
+def authorize(plugin):
+    log('*** authorize')
+
+    login_response = plugin.make_response('POST', '/ajax/login/', data={
+        'login_name': plugin.addon.getSetting('username'),
+        'login_password': plugin.addon.getSetting('password'),
+        'login_not_save': '0'
+    })
+
+    data = login_response.json()
+    if not data.get('success'):
+        raise Exception('Authorization failed status: %s text: %s' % (login_response.status_code, login_response.text))
+
+    plugin.addon.setSetting('cookies', helpers.dump_cookies(login_response.cookies))
+
+def main():
+    plugin = HdrezkaTV()
+
+    if sys.argv[2] == 'authorize':
+        authorize(plugin)
+    else:
+        plugin.main()
 
-plugin = HdrezkaTV()
-plugin.main()
+if __name__ == '__main__':
+    main()

addons/plugin.video.hdrezka.tv/helpers.py

@@ -1,3 +1,6 @@
+import json
+
+from requests.cookies import cookiejar_from_dict
 import xbmc
 
 
@@ -28,6 +31,11 @@ def built_title(name, country_years, **kwargs):
     colored_info = f'[COLOR=55FFFFFF]{kwargs["age_limit"]} ({country_years})[/COLOR]'
     return f'{name} {colored_rating} {colored_info}'
 
+def dump_cookies(cookies):
+    return json.dumps(dict(cookies))
+
+def load_cookies(src):
+    return cookiejar_from_dict(json.loads(src))
 
 def log(msg, level=xbmc.LOGINFO):
     xbmc.log(f'hdrezka: {msg}', level)

addons/plugin.video.hdrezka.tv/resources/settings.xml

@@ -10,6 +10,12 @@
         <setting type="bool" id="united_search" visible="false" default="true" />
         <setting type="text" id="us_command" visible="false" default="mode=search&keyword=" />
     </category>
+    <category label="Authorization">
+        <setting id="username" type="text" label="Username" />
+        <setting id="password" type="text" label="Password" />
+        <setting id="cookies" type="text" visible="false" label="Authorization cookies" />
+        <setting type="action" label="Authorize" action="RunScript(plugin.video.hdrezka.tv, 0, authorize)" />
+    </category>
     <category label="Proxy settings">
         <setting id="use_proxy" type="bool" label="Use proxy settings" default="false" />
         <setting id="protocol" type="labelenum" label="Protocol" values="http|https|socks5" default="http"/>