PagerMaid-Modify/pagermaid/modules/account.py

""" This module contains utils to configure your account. """

from os import remove
from telethon.errors import ImageProcessFailedError, PhotoCropSizeSmallError
from telethon.errors.rpcerrorlist import PhotoExtInvalidError, UsernameOccupiedError, AboutTooLongError, \
    FirstNameInvalidError, UsernameInvalidError
from telethon.tl.functions.account import UpdateProfileRequest, UpdateUsernameRequest
from telethon.tl.functions.photos import DeletePhotosRequest, GetUserPhotosRequest, UploadProfilePhotoRequest
from telethon.tl.functions.users import GetFullUserRequest
from telethon.tl.types import InputPhoto, MessageMediaPhoto, MessageEntityMentionName
from struct import error as StructError
from pagermaid import bot, log
from pagermaid.utils import lang
from pagermaid.listener import listener


@listener(is_plugin=False, outgoing=True, command="username",
          description=lang('username_des'),
          parameters="<username>")
async def username(context):
    """ Reconfigure your username. """
    if len(context.parameter) > 1:
        await context.edit(f"{lang('error_prefix')}{lang('arg_error')}｝")
    if len(context.parameter) == 1:
        result = context.parameter[0]
    else:
        result = ""
    try:
        await bot(UpdateUsernameRequest(result))
    except UsernameOccupiedError:
        await context.edit(f"{lang('error_prefix')}")
        return
    except UsernameInvalidError:
        await context.edit(f"{lang('error_prefix')}{lang('username_vaild')}")
        return
    await context.edit(lang('username_set'))
    if result == "":
        await log(lang('username_cancel'))
        return
    await log(f"{lang('username_whatset')}`{result}`")


@listener(is_plugin=False, outgoing=True, command="name",
          description=lang('name_des'),
          parameters="<first name> <last name>")
async def name(context):
    """ Updates your display name. """
    if len(context.parameter) == 2:
        first_name = context.parameter[0]
        last_name = context.parameter[1]
    elif len(context.parameter) == 1:
        first_name = context.parameter[0]
        last_name = " "
    else:
        await context.edit(f"{lang('error_prefix')}{lang('arg_error')}")
        return
    try:
        await bot(UpdateProfileRequest(
            first_name=first_name,
            last_name=last_name))
    except FirstNameInvalidError:
        await context.edit(f"{lang('error_prefix')}{lang('name_vaild')}")
        return
    await context.edit(lang('name_set'))
    if last_name != " ":
        await log(f"{lang('name_whatset')}`{first_name} {last_name}`.")
    else:
        await log(f"{lang('name_whatset')}`{first_name}`.")


@listener(is_plugin=False, outgoing=True, command="pfp",
          description=lang('pfp_des'))
async def pfp(context):
    """ Sets your profile picture. """
    reply = await context.get_reply_message()
    photo = None
    await context.edit(lang('pfp_process'))
    if reply.media:
        if isinstance(reply.media, MessageMediaPhoto):
            photo = await bot.download_media(message=reply.photo)
        elif "image" in reply.media.document.mime_type.split('/'):
            photo = await bot.download_file(reply.media.document)
        else:
            await context.edit(f"{lang('error_prefix')}{lang('pfp_e_notp')}")

    if photo:
        try:
            await bot(UploadProfilePhotoRequest(
                await bot.upload_file(photo)
            ))
            remove(photo)
            await context.edit("头像修改成功啦 ~")
        except PhotoCropSizeSmallError:
            await context.edit(f"{lang('error_prefix')}{lang('pfp_e_size')}")
        except ImageProcessFailedError:
            await context.edit(f"{lang('error_prefix')}{lang('pfp_e_img')}")
        except PhotoExtInvalidError:
            await context.edit(f"{lang('error_prefix')}{lang('pfp_e_notp')}")


@listener(is_plugin=False, outgoing=True, command="bio",
          description="设置咱的公开情报",
          parameters="<string>")
async def bio(context):
    """ Sets your bio. """
    try:
        await bot(UpdateProfileRequest(about=context.arguments))
    except AboutTooLongError:
        await context.edit(f"{lang('error_prefix')}{lang('bio_too_lang')}")
        return
    await context.edit(lang('bio_set'))
    if context.arguments == "":
        await log(lang('bio_cancel'))
        return
    await log(f"{lang('bio_whatset')}`{context.arguments}`.")


@listener(is_plugin=False, outgoing=True, command="rmpfp",
          description=lang('rmpfp_des'),
          parameters=f"<{lang('int')}>")
async def rmpfp(context):
    """ Removes your profile picture. """
    group = context.text[8:]
    if group == 'all':
        limit = 0
    elif group.isdigit():
        limit = int(group)
    else:
        limit = 1

    pfp_list = await bot(GetUserPhotosRequest(
        user_id=context.from_id,
        offset=0,
        max_id=0,
        limit=limit))
    input_photos = []
    for sep in pfp_list.photos:
        input_photos.append(
            InputPhoto(
                id=sep.id,
                access_hash=sep.access_hash,
                file_reference=sep.file_reference
            )
        )
    await bot(DeletePhotosRequest(id=input_photos))
    await context.edit(f"`{lang('rmpfp_p')}{len(input_photos)} {lang('rmpfp_l')}`")


@listener(is_plugin=False, outgoing=True, command="profile",
          description=lang('profile_des'),
          parameters="<username>")
async def profile(context):
    """ Queries profile of a user. """
    if len(context.parameter) > 1:
        await context.edit(f"{lang('error_prefix')}{lang('arg_error')}")
        return

    await context.edit(lang('profile_process'))
    if context.reply_to_msg_id:
        reply_message = await context.get_reply_message()
        user_id = reply_message.from_id
        target_user = await context.client(GetFullUserRequest(user_id))
    else:
        if len(context.parameter) == 1:
            user = context.parameter[0]
            if user.isnumeric():
                user = int(user)
        else:
            user_object = await context.client.get_me()
            user = user_object.id
        if context.message.entities is not None:
            if isinstance(context.message.entities[0], MessageEntityMentionName):
                return await context.client(GetFullUserRequest(context.message.entities[0].user_id))
        try:
            user_object = await context.client.get_entity(user)
            target_user = await context.client(GetFullUserRequest(user_object.id))
        except (TypeError, ValueError, OverflowError, StructError) as exception:
            if str(exception).startswith("Cannot find any entity corresponding to"):
                await context.edit(f"{lang('error_prefix')}{lang('profile_e_no')}")
                return
            if str(exception).startswith("No user has"):
                await context.edit(f"{lang('error_prefix')}{lang('profile_e_nou')}")
                return
            if str(exception).startswith("Could not find the input entity for") or isinstance(exception, StructError):
                await context.edit(f"{lang('error_prefix')}{lang('profile_e_nof')}")
                return
            if isinstance(exception, OverflowError):
                await context.edit(f"{lang('error_prefix')}{lang('profile_e_long')}")
                return
            raise exception
    user_type = "Bot" if target_user.user.bot else lang('profile_user')
    username_system = f"@{target_user.user.username}" if target_user.user.username is not None else (
        lang('profile_noset'))
    first_name = target_user.user.first_name.replace("\u2060", "")
    last_name = target_user.user.last_name.replace("\u2060", "") if target_user.user.last_name is not None else (
        lang('profile_noset')
    )
    biography = target_user.about if target_user.about is not None else lang('profile_nobio')
    verified = lang('profile_yes') if target_user.user.verified else lang('profile_no')
    restricted = lang('profile_yes') if target_user.user.restricted else lang('profile_no')
    caption = f"**{lang('profile_name')}:** \n" \
              f"{lang('profile_username')}: {username_system} \n" \
              f"ID: {target_user.user.id} \n" \
              f"{lang('profile_fname')}: {first_name} \n" \
              f"{lang('pfofile_lname')}: {last_name} \n" \
              f"{lang('profile_bio')}: {biography} \n" \
              f"{lang('profile_gic')}: {target_user.common_chats_count} \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={target_user.user.id})"
    photo = await context.client.download_profile_photo(
              target_user.user.id,
              "./" + str(target_user.user.id) + ".jpg",
              download_big=True
               )
    try:
        reply_to = context.message.reply_to_msg_id
        try:
            await context.client.send_file(
                context.chat_id,
                photo,
                caption=caption,
                link_preview=False,
                force_document=False,
                reply_to=reply_to
            )
            if not photo.startswith("http"):
                remove(photo)
            await context.delete()
            try:
                remove(photo)
            except:
                pass
            return
        except TypeError:
            await context.edit(caption)
    except:
        try:
            await context.client.send_file(
                context.chat_id,
                photo,
                caption=caption,
                link_preview=False,
                force_document=False
            )
            if not photo.startswith("http"):
                remove(photo)
            await context.delete()
            try:
                remove(photo)
            except:
                pass
            return
        except TypeError:
            await context.edit(caption)