225 lines
10 KiB
Python
225 lines
10 KiB
Python
# Moon-Userbot - telegram userbot
|
||
# Copyright (C) 2020-present Moon Userbot Organization
|
||
#
|
||
# This program is free software: you can redistribute it and/or modify
|
||
# it under the terms of the GNU General Public License as published by
|
||
# the Free Software Foundation, either version 3 of the License, or
|
||
# (at your option) any later version.
|
||
|
||
# This program is distributed in the hope that it will be useful,
|
||
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
# GNU General Public License for more details.
|
||
|
||
# You should have received a copy of the GNU General Public License
|
||
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||
|
||
import hashlib
|
||
import os
|
||
|
||
import requests
|
||
from pyrogram import Client, filters
|
||
from pyrogram.types import Message
|
||
|
||
from utils.scripts import restart
|
||
from utils.misc import modules_help, prefix
|
||
|
||
|
||
BASE_PATH = os.path.abspath(os.getcwd())
|
||
|
||
|
||
@Client.on_message(filters.command(["modhash", "mh"], prefix) & filters.me)
|
||
async def get_mod_hash(_, message: Message):
|
||
if len(message.command) == 1:
|
||
return
|
||
url = message.command[1].lower()
|
||
resp = requests.get(url)
|
||
if not resp.ok:
|
||
await message.edit(
|
||
f"<b>Troubleshooting with downloading module <code>{url}</code></b>"
|
||
)
|
||
return
|
||
|
||
await message.edit(
|
||
f"<b>Module hash: <code>{hashlib.sha256(resp.content).hexdigest()}</code>\n"
|
||
f"Link: <code>{url}</code>\nFile: <code>{url.split('/')[-1]}</code></b>"
|
||
)
|
||
|
||
|
||
@Client.on_message(filters.command(["loadmod", "lm"], prefix) & filters.me)
|
||
async def loadmod(_, message: Message):
|
||
if (
|
||
not (
|
||
message.reply_to_message
|
||
and message.reply_to_message.document
|
||
and message.reply_to_message.document.file_name.endswith(".py")
|
||
)
|
||
and len(message.command) == 1
|
||
):
|
||
await message.edit("<b>Specify module to download</b>")
|
||
return
|
||
|
||
if len(message.command) > 1:
|
||
url = message.command[1].lower()
|
||
|
||
if url.startswith(
|
||
"https://raw.githubusercontent.com/Moon-Userbot/custom_modules/main/"
|
||
):
|
||
module_name = url.split("/")[-1].split(".")[0]
|
||
elif "/" not in url and "." not in url:
|
||
module_name = url.lower()
|
||
url = f"https://raw.githubusercontent.com/Moon-Userbot/custom_modules/main/{module_name}.py"
|
||
else:
|
||
modules_hashes = requests.get(
|
||
"https://raw.githubusercontent.com/Moon-Userbot/custom_modules/main/modules_hashes.txt"
|
||
).text
|
||
resp = requests.get(url)
|
||
|
||
if not resp.ok:
|
||
await message.edit(
|
||
f"<b>Troubleshooting with downloading module <code>{url}</code></b>"
|
||
)
|
||
return
|
||
|
||
if hashlib.sha256(resp.content).hexdigest() not in modules_hashes:
|
||
return await message.edit(
|
||
"<b>Only <a href=https://github.com/Moon-Userbot/custom_modules/tree/main/modules_hashes.txt>"
|
||
"verified</a> modules or from the official "
|
||
"<a href=https://github.com/Moon-Userbot/custom_modules>"
|
||
"custom_modules</a> repository are supported!</b>",
|
||
disable_web_page_preview=True,
|
||
)
|
||
|
||
module_name = url.split("/")[-1].split(".")[0]
|
||
|
||
resp = requests.get(url)
|
||
if not resp.ok:
|
||
await message.edit(f"<b>Module <code>{module_name}</code> is not found</b>")
|
||
return
|
||
|
||
if not os.path.exists(f"{BASE_PATH}/modules/custom_modules"):
|
||
os.mkdir(f"{BASE_PATH}/modules/custom_modules")
|
||
|
||
with open(f"./modules/custom_modules/{module_name}.py", "wb") as f:
|
||
f.write(resp.content)
|
||
else:
|
||
file_name = await message.reply_to_message.download()
|
||
module_name = message.reply_to_message.document.file_name[:-3]
|
||
|
||
with open(file_name, "rb") as f:
|
||
content = f.read()
|
||
|
||
modules_hashes = requests.get(
|
||
"https://raw.githubusercontent.com/Moon-Userbot/custom_modules/main/modules_hashes.txt"
|
||
).text
|
||
|
||
if hashlib.sha256(content).hexdigest() not in modules_hashes:
|
||
os.remove(file_name)
|
||
return await message.edit(
|
||
"<b>Only <a href=https://github.com/Moon-Userbot/custom_modules/tree/main/modules_hashes.txt>"
|
||
"verified</a> modules or from the official "
|
||
"<a href=https://github.com/Moon-Userbot/custom_modules>"
|
||
"custom_modules</a> repository are supported!</b>",
|
||
disable_web_page_preview=True,
|
||
)
|
||
else:
|
||
os.rename(file_name, f"./modules/custom_modules/{module_name}.py")
|
||
|
||
await message.edit(f"<b>The module <code>{module_name}</code> is loaded!</b>")
|
||
restart()
|
||
|
||
|
||
@Client.on_message(filters.command(["unloadmod", "ulm"], prefix) & filters.me)
|
||
async def unload_mods(_, message: Message):
|
||
if len(message.command) <= 1:
|
||
return
|
||
|
||
module_name = message.command[1].lower()
|
||
|
||
if module_name.startswith(
|
||
"https://raw.githubusercontent.com/Moon-Userbot/custom_modules/main/"
|
||
):
|
||
module_name = module_name.split("/")[-1].split(".")[0]
|
||
|
||
if os.path.exists(f"{BASE_PATH}/modules/custom_modules/{module_name}.py"):
|
||
os.remove(f"{BASE_PATH}/modules/custom_modules/{module_name}.py")
|
||
await message.edit(f"<b>The module <code>{module_name}</code> removed!</b>")
|
||
restart()
|
||
elif os.path.exists(f"{BASE_PATH}/modules/{module_name}.py"):
|
||
await message.edit(
|
||
"<b>It is forbidden to remove built-in modules, it will disrupt the updater</b>"
|
||
)
|
||
else:
|
||
await message.edit(f"<b>Module <code>{module_name}</code> is not found</b>")
|
||
|
||
|
||
@Client.on_message(filters.command(["loadallmods"], prefix) & filters.me)
|
||
async def load_all_mods(_, message: Message):
|
||
await message.edit("<b>Fetching info...</b>")
|
||
|
||
if not os.path.exists(f"{BASE_PATH}/modules/custom_modules"):
|
||
os.mkdir(f"{BASE_PATH}/modules/custom_modules")
|
||
|
||
modules_list = requests.get(
|
||
"https://api.github.com/repos/Moon-Userbot/custom_modules/contents/"
|
||
).json()
|
||
|
||
new_modules = {}
|
||
for module_info in modules_list:
|
||
if not module_info["name"].endswith(".py"):
|
||
continue
|
||
if os.path.exists(f'{BASE_PATH}/modules/custom_modules/{module_info["name"]}'):
|
||
continue
|
||
new_modules[module_info["name"][:-3]] = module_info["download_url"]
|
||
if not new_modules:
|
||
return await message.edit("<b>All modules already loaded</b>")
|
||
|
||
await message.edit(f'<b>Loading new modules: {" ".join(new_modules.keys())}</b>')
|
||
for name, url in new_modules.items():
|
||
with open(f"./modules/custom_modules/{name}.py", "wb") as f:
|
||
f.write(requests.get(url).content)
|
||
|
||
await message.edit(
|
||
f'<b>Successfully loaded new modules: {" ".join(new_modules.keys())}</b>'
|
||
)
|
||
restart()
|
||
|
||
|
||
@Client.on_message(filters.command(["updateallmods"], prefix) & filters.me)
|
||
async def updateallmods(_, message: Message):
|
||
await message.edit("<b>Updating modules...</b>")
|
||
|
||
if not os.path.exists(f"{BASE_PATH}/modules/custom_modules"):
|
||
os.mkdir(f"{BASE_PATH}/modules/custom_modules")
|
||
|
||
modules_installed = list(os.walk("modules/custom_modules"))[0][2]
|
||
|
||
if not modules_installed:
|
||
return await message.edit("<b>You don't have any modules installed</b>")
|
||
|
||
for module_name in modules_installed:
|
||
if not module_name.endswith(".py"):
|
||
continue
|
||
|
||
resp = requests.get(
|
||
f"https://raw.githubusercontent.com/Moon-Userbot/custom_modules/main/{module_name}"
|
||
)
|
||
if not resp.ok:
|
||
modules_installed.remove(module_name)
|
||
continue
|
||
|
||
with open(f"./modules/custom_modules/{module_name}", "wb") as f:
|
||
f.write(resp.content)
|
||
|
||
await message.edit(f"<b>Successfully updated {len(modules_installed)} modules</b>")
|
||
|
||
|
||
modules_help["loader"] = {
|
||
"loadmod [module_name]*": "Download module.\n"
|
||
"Only modules from the official custom_modules repository and proven "
|
||
"modules whose hashes are in modules_hashes.txt are supported",
|
||
"unloadmod [module_name]*": "Delete module",
|
||
"modhash [link]*": "Get module hash by link",
|
||
"loadallmods": "Load all custom modules (use it at your own risk)",
|
||
"updateallmods": "Update all custom modules",
|
||
} |