From c35a78336e51f1806c5266ceb1b25e4adb76400e Mon Sep 17 00:00:00 2001
From: marco <m.toma99@gmail.com>
Date: Thu, 30 Jul 2020 20:54:05 +0200
Subject: [PATCH] HDmario: supporto a registrazione/login

---
 core/support.py      |  5 ++--
 servers/hdmario.json | 18 +++++++++++++-
 servers/hdmario.py   | 57 +++++++++++++++++++++++++++++++++++++++++++-
 3 files changed, 76 insertions(+), 4 deletions(-)

diff --git a/core/support.py b/core/support.py
index 61214289..dde717a3 100755
--- a/core/support.py
+++ b/core/support.py
@@ -64,10 +64,11 @@ def hdpass_get_servers(item):
 
     data = httptools.downloadpage(url, CF=False).data
     patron_res = '<div class="buttons-bar resolutions-bar">(.*?)<div class="buttons-bar'
-    patron_mir = '<div class="buttons-bar hosts-bar">(.*?)<div id="fake'
-    patron_option = r'<a href="([^"]+?)".*?>([^<]+?)</a>'
+    patron_mir = '<div class="buttons-bar hosts-bar">(.*?)<div id="main-player'
+    patron_option = r'<a href="([^"]+?)"[^>]+>([^<]+?)</a'
 
     res = scrapertools.find_single_match(data, patron_res)
+    # dbg()
 
     with futures.ThreadPoolExecutor() as executor:
         thL = []
diff --git a/servers/hdmario.json b/servers/hdmario.json
index 3fb9ec50..6165244b 100644
--- a/servers/hdmario.json
+++ b/servers/hdmario.json
@@ -4,7 +4,7 @@
     "ignore_urls": [],
     "patterns": [
       {
-        "pattern": "https?://hdmario.live/embed/([0-9]+)",
+        "pattern": "https?://hdmario.live/\\w+/([0-9]+)",
         "url": "https://hdmario.live/embed/\\1"
       }
     ]
@@ -36,6 +36,22 @@
       ],
       "type": "list",
       "visible": false
+    },
+    {
+      "default": "",
+      "enabled": true,
+      "id": "email",
+      "label": "email",
+      "type": "text",
+      "visible": true
+    },
+    {
+      "default": "",
+      "enabled": true,
+      "id": "password",
+      "label": "password",
+      "type": "text",
+      "visible": true
     }
   ]
 }
diff --git a/servers/hdmario.py b/servers/hdmario.py
index 1a692b75..253f04ad 100644
--- a/servers/hdmario.py
+++ b/servers/hdmario.py
@@ -2,7 +2,7 @@
 import xbmc
 
 from core import httptools, scrapertools, filetools
-from platformcode import logger, config
+from platformcode import logger, config, platformtools
 
 baseUrl = 'https://hdmario.live'
 
@@ -19,10 +19,58 @@ def test_video_exists(page_url):
     return True, ""
 
 
+def login():
+    httptools.downloadpage(page.url.replace('/unauthorized', '/login'),
+                           post={'email': config.get_setting('username', server='hdmario'),
+                                 'password': config.get_setting('password', server='hdmario')})
+
+
+def registerOrLogin(page_url, forced=False):
+    if not forced and config.get_setting('username', server='hdmario') and config.get_setting('password', server='hdmario'):
+        login()
+    else:
+        if platformtools.dialog_yesno('HDmario',
+                                      'Questo server necessita di un account, ne hai già uno oppure vuoi tentare una registrazione automatica?',
+                                      yeslabel='Accedi', nolabel='Tenta registrazione'):
+            from specials import setting
+            from core.item import Item
+            setting.server_config(Item(config='hdmario'))
+            login()
+        else:
+            logger.info('Registrazione automatica in corso')
+            import random
+            import string
+            randEmail = ''.join(random.choice(string.ascii_letters + string.digits) for i in range(random.randint(9, 14))) + '@gmail.com'
+            randPsw = ''.join(random.choice(string.ascii_letters + string.digits) for i in range(10))
+            logger.info('email: ' + randEmail)
+            logger.info('pass: ' + randPsw)
+            nTry = 0
+            while nTry < 5:
+                nTry += 1
+                rq = 'loggedin' in httptools.downloadpage(baseUrl + '/register/',
+                                                          post={'email': randEmail, 'email_confirmation': randEmail,
+                                                                'password': randPsw,
+                                                                'password_confirmation': randPsw}).url
+                if rq:
+                    from core import support
+                    support.dbg()
+                    config.set_setting('username', randEmail, server='hdmario')
+                    config.set_setting('password', randPsw, server='hdmario')
+                    platformtools.dialog_ok('HDmario',
+                                            'Registrato automaticamente con queste credenziali:\nemail:' + randEmail + '\npass: ' + randPsw)
+                    break
+            else:
+                platformtools.dialog_ok('HDmario', 'Impossibile registrarsi automaticamente')
+            logger.info('Registrazione completata')
+    global page, data
+    page = httptools.downloadpage(page_url)
+    data = page.data
+
 def get_video_url(page_url, premium=False, user="", password="", video_password=""):
     global page, data
     page_url = page_url.replace('?', '')
     logger.info("url=" + page_url)
+
     if 'unconfirmed' in page.url:
         from lib import onesecmail
         id = page_url.split('/')[-1]
@@ -38,6 +86,13 @@ def get_video_url(page_url, premium=False, user="", password="", video_password=
             code = jsonMail['subject'].split(' - ')[0]
             page = httptools.downloadpage(page_url + '?code=' + code)
             data = page.data
+
+    if '/unauthorized' in page.url:
+        registerOrLogin(page_url)
+
+    if 'Registrati' in data:
+        platformtools.dialog_ok('HDmario', 'Username/password non validi')
+        registerOrLogin(page_url, True)
     logger.info(data)
     from lib import jsunpack_js2py
     unpacked = jsunpack_js2py.unpack(scrapertools.find_single_match(data, '<script type="text/javascript">\n*\s*\n*(eval.*)'))