Added userbot

wbb/__init__.py

@@ -44,5 +44,11 @@ MOD_NOLOAD = []
 bot_start_time = time.time()
 mongo_client = MongoClient(MONGO_DB_URI)
 db = mongo_client.wbb
+
+print("\nStarting Helper Userbot")
+app2 = Client("userbot", api_id=API_ID, api_hash=API_HASH)
+
+print("\nStarting Main Bot\n")
 app = Client("wbb", bot_token=BOT_TOKEN, api_id=API_ID, api_hash=API_HASH)
+
 arq = ARQ(ARQ_API)

wbb/__main__.py

@@ -4,7 +4,7 @@ import importlib
 import uvloop
 from pyrogram import filters, idle
 from pyrogram.types import InlineKeyboardButton, InlineKeyboardMarkup
-from wbb import app
+from wbb import app, app2
 from wbb.utils import get_info, paginate_modules
 from wbb.utils import botinfo
 from wbb.modules.sudoers import bot_sys_stats
@@ -18,7 +18,9 @@ HELPABLE = {}
 
 async def start_bot():
     global COMMANDS_COUNT
+    await app2.start()
     await app.start()
+    
     await get_info(app)
 
     for module in ALL_MODULES:

wbb/modules/sudoers.py

@@ -1,5 +1,7 @@
 import re
 import time
+import os
+import subprocess
 from pyrogram import filters, types
 import speedtest
 import psutil
@@ -21,7 +23,8 @@ __HELP__ = '''/log - To Get Logs From Last Run.
 /stats - To Check System Status.
 /gstats - To Check Bot's Global Stats.
 /gban - To Ban A User Globally.
-/broadcast - To Broadcast A Message To All Groups.'''
+/broadcast - To Broadcast A Message To All Groups.
+/update - To Update And Restart The Bot'''
 
 
 # Logs Module
@@ -263,3 +266,12 @@ async def broadcast_message(_, message):
         except Exception:
             pass
     await message.reply_text(f"**Broadcasted Message In {sent} Chats.**")
+
+
+# Update
+
+
+@app.on_message(filters.command("update") & filters.user(SUDOERS))
+async def update_restart(_, message):
+    await message.reply_text(subprocess.check_output(["git", "pull"]).decode("UTF-8"))
+    os.execvp("python3", ["python3", "-m", "wbb"])

wbb/modules/userbot.py

@@ -0,0 +1,150 @@
+import sys
+import traceback
+from io import StringIO
+import os
+import re
+import subprocess
+from inspect import getfullargspec
+from wbb import app, OWNER_ID
+from pyrogram import filters
+from pyrogram.types import Message
+
+__MODULE__ = "Userbot"
+__HELP__ = ".l - Execute Python Code\n.sh - Execute Shell Code"
+
+
+# Eval and Sh module from nana-remix
+
+async def aexec(code, client, message):
+    exec(
+        "async def __aexec(client, message): "
+        + "".join(f"\n {a}" for a in code.split("\n"))
+    )
+    return await locals()["__aexec"](client, message)
+
+
+async def edit_or_reply(msg: Message, **kwargs):
+    func = msg.edit_text if msg.from_user.is_self else msg.reply
+    spec = getfullargspec(func.__wrapped__).args
+    await func(**{k: v for k, v in kwargs.items() if k in spec})
+
+
+@app2.on_message(filters.user(OWNER_ID) & ~filters.forwarded &
+                ~filters.via_bot & filters.command("eval"))
+async def executor(client, message):
+    try:
+        cmd = message.text.split(" ", maxsplit=1)[1]
+    except IndexError:
+        await message.delete()
+        return
+    reply_to_id = message.message_id
+    if message.reply_to_message:
+        reply_to_id = message.reply_to_message.message_id
+    old_stderr = sys.stderr
+    old_stdout = sys.stdout
+    redirected_output = sys.stdout = StringIO()
+    redirected_error = sys.stderr = StringIO()
+    stdout, stderr, exc = None, None, None
+    try:
+        await aexec(cmd, client, message)
+    except Exception:
+        exc = traceback.format_exc()
+    stdout = redirected_output.getvalue()
+    stderr = redirected_error.getvalue()
+    sys.stdout = old_stdout
+    sys.stderr = old_stderr
+    evaluation = ""
+    if exc:
+        evaluation = exc
+    elif stderr:
+        evaluation = stderr
+    elif stdout:
+        evaluation = stdout
+    else:
+        evaluation = "Success"
+    final_output = (f"**OUTPUT**:\n```{evaluation.strip()}```")
+    if len(final_output) > 4096:
+        filename = "output.txt"
+        with open(filename, "w+", encoding="utf8") as out_file:
+            out_file.write(str(evaluation.strip()))
+        await message.reply_document(
+            document=filename,
+            caption=cmd,
+            disable_notification=True,
+            reply_to_message_id=reply_to_id,
+        )
+        os.remove(filename)
+        await message.delete()
+    else:
+        await edit_or_reply(message, text=final_output)
+
+
+@app2.on_message(
+    filters.user(OWNER_ID)
+    & ~filters.forwarded
+    & ~filters.via_bot
+    & filters.command('sh'),
+)
+async def shellrunner(client, message):
+    if len(message.command) < 2:
+        await edit_or_reply(message, text='**Usage:**\n/sh git pull')
+        return
+    text = message.text.split(None, 1)[1]
+    if '\n' in text:
+        code = text.split('\n')
+        output = ''
+        for x in code:
+            shell = re.split(""" (?=(?:[^'"]|'[^']*'|"[^"]*")*$)""", x)
+            try:
+                process = subprocess.Popen(
+                    shell, stdout=subprocess.PIPE, stderr=subprocess.PIPE,
+                )
+            except Exception as err:
+                print(err)
+                await edit_or_reply(
+                    message,
+                    text=f"**Error:**\n```{err}```")
+            output += f'**{code}**\n'
+            output += process.stdout.read()[:-1].decode('utf-8')
+            output += '\n'
+    else:
+        shell = re.split(""" (?=(?:[^'"]|'[^']*'|"[^"]*")*$)""", text)
+        for a in range(len(shell)):
+            shell[a] = shell[a].replace('"', '')
+        try:
+            process = subprocess.Popen(
+                shell, stdout=subprocess.PIPE, stderr=subprocess.PIPE,
+            )
+        except Exception as err:
+            print(err)
+            exc_type, exc_obj, exc_tb = sys.exc_info()
+            errors = traceback.format_exception(
+                etype=exc_type, value=exc_obj, tb=exc_tb,
+            )
+            await edit_or_reply(
+                message,
+                text=f"**Error:**\n```{''.join(errors)}```")
+            return
+        output = process.stdout.read()[:-1].decode('utf-8')
+    if str(output) == '\n':
+        output = None
+    if output:
+        if len(output) > 4096:
+            with open('output.txt', 'w+') as file:
+                file.write(output)
+            await app.send_document(
+                message.chat.id,
+                'output.txt',
+                reply_to_message_id=message.message_id,
+                caption='`Output`',
+            )
+            os.remove('output.txt')
+            return
+        await edit_or_reply(
+            message,
+            text=f"**Output:**\n```{output}```"
+        )
+    else:
+        await edit_or_reply(
+            message,
+            text='**Output: **\n`No output`')