PagerMaid-Pyro/pagermaid/modules/account.py

from os import remove

from pyrogram.errors import UsernameNotOccupied, PeerIdInvalid
from pyrogram.types import User

from pagermaid import Config
from pagermaid.listener import listener
from pagermaid.utils import lang, Message

import contextlib


@listener(is_plugin=False, command="profile",
          description=lang('profile_des'),
          parameters="<username>")
async def profile(message: Message):
    """ Queries profile of a user. """
    if len(message.parameter) > 1:
        await message.edit(f"{lang('error_prefix')}{lang('arg_error')}")
        return
    if not Config.SILENT:
        message = await message.edit(lang('profile_process'))
    if message.reply_to_message:
        user = message.reply_to_message.from_user
        if not user:
            return await message.edit(f"{lang('error_prefix')}{lang('profile_e_no')}")
    else:
        if len(message.parameter) == 1:
            user = message.parameter[0]
            if user.isdigit():
                user = int(user)
        else:
            user = await message.bot.get_me()
        if message.entities is not None:
            if message.entities[0].type == "text_mention":
                user = message.entities[0].user
            elif message.entities[0].type == "phone_number":
                user = int(message.parameter[0])
            else:
                return await message.edit(f"{lang('error_prefix')}{lang('arg_error')}")
        if not isinstance(user, User):
            try:
                user = await message.bot.get_users(user)
            except PeerIdInvalid:
                return await message.edit(f"{lang('error_prefix')}{lang('profile_e_nof')}")
            except UsernameNotOccupied:
                return await message.edit(f"{lang('error_prefix')}{lang('profile_e_nou')}")
            except OverflowError:
                return await message.edit(f"{lang('error_prefix')}{lang('profile_e_long')}")
            except Exception as exception:
                raise exception
    user_type = "Bot" if user.is_bot else lang('profile_user')
    username_system = f"@{user.username}" if user.username is not None else lang('profile_noset')
    if not user.first_name:
        await message.edit(f"{lang('error_prefix')}{lang('profile_e_no')}")
        return
    first_name = user.first_name.replace("\u2060", "")
    last_name = user.last_name.replace("\u2060", "") if user.last_name is not None else lang('profile_noset')
    verified = lang('profile_yes') if user.is_verified else lang('profile_no')
    restricted = lang('profile_yes') if user.is_restricted else lang('profile_no')
    caption = f"**{lang('profile_name')}:** \n" \
              f"{lang('profile_username')}: {username_system} \n" \
              f"ID: {user.id} \n" \
              f"{lang('profile_fname')}: {first_name} \n" \
              f"{lang('profile_lname')}: {last_name} \n" \
              f"{lang('profile_verified')}: {verified} \n" \
              f"{lang('profile_restricted')}: {restricted} \n" \
              f"{lang('profile_type')}: {user_type} \n" \
              f"[{first_name}](tg://user?id={user.id})"
    photo = await message.bot.download_media(user.photo.big_file_id)
    reply_to = message.reply_to_message
    try:
        await message.bot.send_photo(
            message.chat.id,
            photo,
            caption=caption,
            reply_to_message_id=reply_to.id if reply_to else None
        )
        await message.delete()
        return remove(photo)
    except TypeError:
        await message.edit(caption)


@listener(is_plugin=False, outgoing=True, command="block",
          need_admin=True,
          description=lang('block_des'),
          parameters="(username/uid/reply)")
async def block_user(message: Message):
    """ Block a user. """
    if len(message.parameter) > 1:
        await message.edit(f"{lang('error_prefix')}{lang('arg_error')}")
        return
    if not Config.SILENT:
        message = await message.edit(lang('block_process'))
    user = message.obtain_user()
    if not user:
        return await message.edit(f"{lang('error_prefix')}{lang('arg_error')}")
    with contextlib.suppress(Exception):
        if await message.bot.block_user(user):
            return await message.edit(f"{lang('block_success')} `{user}`")
    await message.edit(f"`{user}` {lang('block_exist')}")


@listener(is_plugin=False, outgoing=True, command="unblock",
          need_admin=True,
          description=lang('unblock_des'),
          parameters="<username/uid/reply>")
async def unblock_user(message: Message):
    """ Unblock a user. """
    if len(message.parameter) > 1:
        await message.edit(f"{lang('error_prefix')}{lang('arg_error')}")
        return
    if not Config.SILENT:
        message = await message.edit(lang('unblock_process'))
    user = message.obtain_user()
    if not user:
        return await message.edit(f"{lang('error_prefix')}{lang('arg_error')}")
    with contextlib.suppress(Exception):
        if await message.bot.unblock_user(user):
            return await message.edit(f"{lang('unblock_success')} `{user}`")
    await message.edit(f"`{user}` {lang('unblock_exist')}")