From 82db99b5ce0e27464472ecbbaa60b8678c61e481 Mon Sep 17 00:00:00 2001
From: Piter <revelation@o2.pl>
Date: Wed, 23 Sep 2015 21:03:03 +0200
Subject: [Openload.io] Rewrite the plugin using official API

---
 module/plugins/hoster/OpenloadIo.py | 55 ++++++++++++++++++++++++++++++++-----
 1 file changed, 48 insertions(+), 7 deletions(-)

(limited to 'module/plugins/hoster/OpenloadIo.py')

diff --git a/module/plugins/hoster/OpenloadIo.py b/module/plugins/hoster/OpenloadIo.py
index 372ce28f9..f5d677bb2 100644
--- a/module/plugins/hoster/OpenloadIo.py
+++ b/module/plugins/hoster/OpenloadIo.py
@@ -1,31 +1,72 @@
 # -*- coding: utf-8 -*-
+import json
+import re
+from time import sleep
 
 from module.plugins.internal.SimpleHoster import SimpleHoster, create_getInfo
+from module.network.RequestFactory import getURL
 
 
 class OpenloadIo(SimpleHoster):
     __name__    = "OpenloadIo"
     __type__    = "hoster"
-    __version__ = "0.05"
+    __version__ = "0.06"
     __status__  = "testing"
 
-    __pattern__ = r'https?://(?:www\.)?openload\.(co|io)/f/[\w-]+'
+    _FILE_ID_PATTERN = '/f/([\w\-_]+)/?'
+    __pattern__ = r'https?://(?:www\.)?openload\.(?:co|io)' + _FILE_ID_PATTERN
 
     __description__ = """Openload.co hoster plugin"""
     __license__     = "GPLv3"
     __authors__     = [(None, None)]
 
 
-    NAME_PATTERN    = r'<span id="filename">(?P<N>.+?)</'
-    SIZE_PATTERN    = r'<span class="count">(?P<S>[\d.,]+) (?P<U>[\w^_]+)'
-    OFFLINE_PATTERN = r">(We can't find the file you are looking for)"
-
-    LINK_FREE_PATTERN = r'id="real\w*download"><a href="(https?://[\w\.]+\.openload\.co/dl/.*?)"'
+    # The API reference, that this implementation uses is available at https://openload.co/api
+    _API_BASE_URL = 'https://api.openload.co/1'
 
+    _DOWNLOAD_TICKET_URI_PATTERN = '/file/dlticket?file={0}'
+    _DOWNLOAD_FILE_URI_PATTERN = '/file/dl?file={0}&ticket={1}'
+    _FILE_INFO_URI_PATTERN = '/file/info?file={0}'
 
     def setup(self):
         self.multiDL     = True
         self.chunk_limit = 1
 
 
+    @classmethod
+    def get_info(cls, url="", html=""):
+        file_id = re.findall(cls._FILE_ID_PATTERN, url, re.I)
+        if not file_id:
+            return super(OpenloadIo, cls).get_info(url)
+
+        file_id = file_id[0]
+        info_json = cls._load_json(cls._FILE_INFO_URI_PATTERN.format(file_id))
+        file_info = info_json['result'][file_id]
+        return {'name': file_info['name'],
+                'size': file_info['size'],
+                'status': 3 if url.strip() else 8,
+                'url': url}
+
+
+    def handle_free(self, pyfile):
+        # If the link is being handled here, then it matches the file_id_pattern,
+        # therefore, we can call [0] safely.
+        file_id = re.findall(self._FILE_ID_PATTERN, pyfile.url, re.I)[0]
+
+        ticket_json = self._load_json(self._DOWNLOAD_TICKET_URI_PATTERN.format(file_id))
+
+        wait_time = ticket_json['result']['wait_time']
+        sleep(wait_time + 0.1)
+
+        ticket = ticket_json['result']['ticket']
+
+        download_json = self._load_json(self._DOWNLOAD_FILE_URI_PATTERN.format(file_id, ticket))
+        self.link = download_json['result']['url']
+
+
+    @classmethod
+    def _load_json(cls, uri):
+        return json.loads(
+            getURL(cls._API_BASE_URL + uri))
+
 getInfo = create_getInfo(OpenloadIo)
-- 
cgit v1.2.3