Merge pull request #1858 from whitedemon938/master

Added : mediafile.cc ddl generator

Author: anasty17

bot/helper/mirror_leech_utils/download_utils/direct_link_generator.py

@@ -40,6 +40,8 @@ def direct_link_generator(link):
         return uploadhaven(link)
     elif "fuckingfast.co" in domain:
         return fuckingfast_dl(link)
+    elif "mediafile.cc" in domain:
+        return mediafile(link)
     elif "mediafire.com" in domain:
         return mediafire(link)
     elif "osdn.net" in domain:
@@ -399,6 +401,30 @@ def uploadhaven(url):
     except Exception as e:
         raise DirectDownloadLinkException(f"ERROR: {str(e)}") from e
 
+def mediafile(url):
+    """
+    Generate a direct download link for mediafile.cc URLs.
+    @param url: URL from mediafile.cc
+    @return: Direct download link
+    """
+    try:
+        res = get(url, allow_redirects=True)
+        match = re.search(r"href='([^']+)'", res.text)
+        if not match:
+            raise DirectDownloadLinkException("ERROR: Unable to find link data")        
+        download_url = match.group(1)
+        sleep(60)
+        res = get(download_url, headers={'Referer': url}, cookies=res.cookies)
+        postvalue = re.search(r'showFileInformation(.*);', res.text)
+        if not postvalue:
+            raise DirectDownloadLinkException("ERROR: Unable to find post value")       
+        postid = postvalue.group(1).replace('(','').replace(')','')
+        response = post('https://mediafile.cc/account/ajax/file_details',data={"u": postid}, headers={"X-Requested-With": "XMLHttpRequest"})
+        html = response.json()['html']
+        return [i for i in re.findall(r'https://[^\s"\']+', html) if 'download_token' in i][1]
+    except Exception as e:
+        raise DirectDownloadLinkException(f"ERROR: {str(e)}") from e
+
 def mediafire(url, session=None):
     if "/folder/" in url:
         return mediafireFolder(url)