Update: Fix issue error mega download

Special thanks to @adekmaulana

Signed-off-by: MoveAngel <[email protected]>

Author: MoveAngel

Dockerfile

@@ -10,7 +10,8 @@ RUN sed -e 's;^#http\(.*\)/edge/community;http\1/edge/community;g' -i /etc/apk/r
 # Installing Packages
 #
 RUN apk add --no-cache=true --update \
-        bash \
+    coreutils \
+    bash \
     build-base \
     bzip2-dev \
     curl \
@@ -54,8 +55,6 @@ RUN apk add --no-cache=true --update \
     zlib-dev \
     jpeg 
 
-  
-
 
 RUN python3 -m ensurepip \
     && pip3 install --upgrade pip setuptools \

requirements.txt

@@ -43,3 +43,4 @@ telegraph
 urbandict>=0.5
 wikipedia>=1.4.0
 coffeehouse
+wget

userbot/__init__.py

@@ -155,6 +155,8 @@
     "bin/megadown",
     "https://raw.githubusercontent.com/yshalsager/cmrudl.py/master/cmrudl.py":
     "bin/cmrudl"
+    "https://raw.githubusercontent.com/adekmaulana/python-scripts/master/shell/megadirect":
+    "bin/megadirect"
 }
 
 for binary, path in binaries.items():

userbot/modules/mega_downloader.py

@@ -0,0 +1,112 @@
+# Copyright (C) 2020 Adek Maulana.
+# All rights reserved.
+#
+# Redistribution and use of this script, with or without modification, is
+# permitted provided that the following conditions are met:
+#
+# 1. Redistributions of this script must retain the above copyright
+#    notice, this list of conditions and the following disclaimer.
+#
+#  THIS SOFTWARE IS PROVIDED BY THE AUTHOR "AS IS" AND ANY EXPRESS OR IMPLIED
+#  WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
+#  MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO
+#  EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+#  SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
+#  PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
+#  OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
+#  WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
+#  OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
+#  ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+
+from humanize import naturalsize
+from subprocess import PIPE, Popen
+
+import re
+import json
+import wget
+import os
+
+from userbot import CMD_HELP
+from userbot.events import register
+
+
+def subprocess_run(cmd):
+    subproc = Popen(cmd, stdout=PIPE, stderr=PIPE,
+                    shell=True, universal_newlines=True)
+    talk = subproc.communicate()
+    exitCode = subproc.returncode
+    if exitCode != 0:
+        print('An error was detected while running the subprocess:\n'
+              'exit code: %d\n'
+              'stdout: %s\n'
+              'stderr: %s' % (exitCode, talk[0], talk[1]))
+    return talk
+
+
+@register(outgoing=True, pattern=r"^.mega(?: |$)(.*)")
+async def mega_downloader(event):
+    await event.edit("`Processing...`")
+    textx = await event.get_reply_message()
+    link = event.pattern_match.group(1)
+    if link:
+        pass
+    elif textx:
+        link = textx.text
+    else:
+        await event.edit("`Usage: .mega <mega url>`")
+        return
+    reply = ''
+    if not link:
+        reply = "`No MEGA.nz link found!`"
+        await event.edit(reply)
+    await event.edit("`Downloading...`")
+    reply += mega_download(link)
+    await event.edit(reply)
+
+
+def mega_download(url: str) -> str:
+    reply = ''
+    try:
+        link = re.findall(r'\bhttps?://.*mega.*\.nz\S+', url)[0]
+    except IndexError:
+        reply = "`No MEGA.nz link found`\n"
+        return reply
+    cmd = f'bin/megadirect {link}'
+    result = subprocess_run(cmd)
+    try:
+        data = json.loads(result[0])
+        print(data)
+    except json.JSONDecodeError:
+        reply += "`Error: Can't extract the link`\n"
+        return reply
+    file_name = data['file_name']
+    file_size = naturalsize(int(data['file_size']))
+    file_url = data['url']
+    file_hex = data['hex']
+    file_raw_hex = data['raw_hex']
+    if wget.download(file_url, out=file_name):
+        encrypt_file(file_name, file_hex, file_raw_hex)
+        reply += (f"`{file_name}`\n"
+                  f"Size: {file_size}\n"
+                  "\n"
+                  "Successfully downloaded...")
+    else:
+        reply += "Failed to download..."
+    return reply
+
+
+def encrypt_file(file_name, file_hex, file_raw_hex):
+    os.rename(file_name, f"old_{file_name}")
+    cmd = f"cat old_{file_name} | openssl enc -d -aes-128-ctr -K {file_hex} -iv {file_raw_hex} > {file_name}"
+    subprocess_run(cmd)
+    os.remove(f"old_{file_name}")
+    return
+
+
+CMD_HELP.update({
+    "mega":
+    ".mega <mega url>\"
+    "\nUsage: Reply to a mega link or paste your mega link to"
+    "\ndownload the file into your userbot server"
+    "\n\nOnly support for *FILE* only."
+})