115 lines
3.7 KiB
Python
115 lines
3.7 KiB
Python
from pyrogram.types import User
|
|
|
|
from app import BOT, DB, Config, Message, bot
|
|
from app.plugins.admin.fbans import _User
|
|
from app.utils.helpers import extract_user_data, get_name
|
|
|
|
|
|
async def init_task():
|
|
sudo = await DB.SUDO.find_one({"_id": "sudo_switch"})
|
|
if sudo:
|
|
Config.SUDO = sudo["value"]
|
|
Config.SUDO_USERS = [sudo_user["_id"] async for sudo_user in DB.SUDO_USERS.find()]
|
|
|
|
|
|
@bot.add_cmd(cmd="sudo")
|
|
async def sudo(bot: BOT, message: Message):
|
|
"""
|
|
CMD: SUDO
|
|
INFO: Enable/Disable sudo..
|
|
FLAGS: -c to check sudo status.
|
|
USAGE:
|
|
.sudo | .sudo -c
|
|
"""
|
|
if "-c" in message.flags:
|
|
await message.reply(text=f"Sudo is enabled: <b>{Config.SUDO}</b> .", del_in=8)
|
|
return
|
|
value = not Config.SUDO
|
|
Config.SUDO = value
|
|
await DB.add_data(collection=DB.SUDO, id="sudo_switch", data={"value": value})
|
|
await message.reply(text=f"Sudo is enabled: <b>{value}</b>!", del_in=8)
|
|
|
|
|
|
@bot.add_cmd(cmd="addsudo")
|
|
async def add_sudo(bot: BOT, message: Message) -> Message | None:
|
|
"""
|
|
CMD: ADDSUDO
|
|
INFO: Add Sudo User.
|
|
FLAGS: -temp to temporarily add until bot restarts.
|
|
USAGE:
|
|
.addsudo [-temp] [ UID | @ | Reply to Message ]
|
|
"""
|
|
response = await message.reply("Extracting User info...")
|
|
user, _ = await message.extract_user_n_reason()
|
|
if isinstance(user, str):
|
|
await response.edit(user)
|
|
return
|
|
if not isinstance(user, User):
|
|
user: _User = _User(id=message.text_list[1])
|
|
if user.id in Config.SUDO_USERS:
|
|
await response.edit(text=f"{get_name(user)} already in Sudo!", del_in=5)
|
|
return
|
|
response_str = f"{user.mention} added to Sudo List."
|
|
Config.SUDO_USERS.append(user.id)
|
|
if "-temp" not in message.flags:
|
|
await DB.add_data(
|
|
collection=DB.SUDO_USERS, id=user.id, data=extract_user_data(user)
|
|
)
|
|
else:
|
|
response_str += "\n<b>Temporary</b>: True"
|
|
await response.edit(text=response_str, del_in=5)
|
|
await bot.log(text=response_str)
|
|
|
|
|
|
@bot.add_cmd(cmd="delsudo")
|
|
async def remove_sudo(bot: BOT, message: Message) -> Message | None:
|
|
"""
|
|
CMD: DELSUDO
|
|
INFO: Add Remove User.
|
|
FLAGS: -temp to temporarily remove until bot restarts.
|
|
USAGE:
|
|
.delsudo [-temp] [ UID | @ | Reply to Message ]
|
|
"""
|
|
response = await message.reply("Extracting User info...")
|
|
user, _ = await message.extract_user_n_reason()
|
|
if isinstance(user, str):
|
|
await response.edit(user)
|
|
return
|
|
if not isinstance(user, User):
|
|
user: _User = _User(id=message.text_list[1])
|
|
|
|
if user.id not in Config.SUDO_USERS:
|
|
await response.edit(text=f"{get_name(user)} not in Sudo!", del_in=5)
|
|
return
|
|
Config.SUDO_USERS.remove(user.id)
|
|
response_str = f"{user.mention} removed from Sudo List."
|
|
if "-temp" not in message.flags:
|
|
await DB.delete_data(collection=DB.SUDO_USERS, id=user.id)
|
|
else:
|
|
response_str += "\n<b>Temporary</b>: True"
|
|
await response.edit(text=response_str, del_in=5)
|
|
await bot.log(text=response_str)
|
|
|
|
|
|
@bot.add_cmd(cmd="vsudo")
|
|
async def sudo_list(bot: BOT, message: Message):
|
|
"""
|
|
CMD: VSUDO
|
|
INFO: View Sudo Users.
|
|
FLAGS: -id to get UIDs
|
|
USAGE:
|
|
.vsudo | .vsudo -id
|
|
"""
|
|
output: str = ""
|
|
total = 0
|
|
async for user in DB.SUDO_USERS.find():
|
|
output += f'<b>• {user["name"]}</b>\n'
|
|
if "-id" in message.flags:
|
|
output += f' <code>{user["_id"]}</code>\n'
|
|
total += 1
|
|
if not total:
|
|
await message.reply("You don't have any SUDO USERS.")
|
|
return
|
|
output: str = f"List of <b>{total}</b> SUDO USERS:\n\n{output}"
|
|
await message.reply(output, del_in=30, block=False)
|