Telegram/modules/inline.py

"""Module responsible for providing answers to
all inline queries that bot receives"""

from datetime import datetime
from os import path, sep
from app import app, isAnAdmin
from pyrogram.types import (
    InlineQueryResultArticle,
    InputTextMessageContent,
    InlineQuery,
    InlineKeyboardMarkup,
    InlineKeyboardButton,
)
from pyrogram.client import Client
from pyrogram.enums.chat_type import ChatType
from pyrogram.enums.chat_members_filter import ChatMembersFilter
from dateutil.relativedelta import relativedelta
from classes.errors.holo_user import UserNotFoundError, UserInvalidError
from classes.holo_user import HoloUser
from modules.logging import logWrite
from modules.utils import configGet, jsonLoad, locale
from modules.database import col_applications, col_spoilers
from bson.objectid import ObjectId
from bson.errors import InvalidId


@app.on_inline_query()
async def inline_answer(client: Client, inline_query: InlineQuery):
    results = []

    if configGet("allow_external", "features", "spoilers") is True:
        if inline_query.query.startswith("spoiler:"):
            try:
                spoil = col_spoilers.find_one(
                    {"_id": ObjectId(inline_query.query.removeprefix("spoiler:"))}
                )

                if spoil is not None:
                    desc = locale(
                        "spoiler_described",
                        "message",
                        locale=inline_query.from_user,
                    ).format(
                        locale(spoil["category"], "message", "spoiler_categories"),
                        spoil["description"],
                    )

                    results = [
                        InlineQueryResultArticle(
                            title=locale(
                                "title",
                                "inline",
                                "spoiler",
                                locale=inline_query.from_user,
                            ),
                            description=locale(
                                "description",
                                "inline",
                                "spoiler",
                                locale=inline_query.from_user,
                            ),
                            input_message_content=InputTextMessageContent(
                                desc, disable_web_page_preview=True
                            ),
                            reply_markup=InlineKeyboardMarkup(
                                [
                                    [
                                        InlineKeyboardButton(
                                            locale(
                                                "spoiler_view",
                                                "button",
                                                locale=inline_query.from_user,
                                            ),
                                            callback_data=f'sid_{inline_query.query.removeprefix("spoiler:")}',
                                        )
                                    ]
                                ]
                            ),
                        )
                    ]

            except InvalidId:
                results = []

            await inline_query.answer(results=results)

            return

    if inline_query.chat_type in [ChatType.CHANNEL]:
        await inline_query.answer(
            results=[
                InlineQueryResultArticle(
                    title=locale(
                        "title", "inline", "not_pm", locale=inline_query.from_user
                    ),
                    input_message_content=InputTextMessageContent(
                        locale(
                            "message_content",
                            "inline",
                            "not_pm",
                            locale=inline_query.from_user,
                        )
                    ),
                    description=locale(
                        "description", "inline", "not_pm", locale=inline_query.from_user
                    ),
                )
            ]
        )
        return

    results_forbidden = [
        InlineQueryResultArticle(
            title=locale("title", "inline", "forbidden", locale=inline_query.from_user),
            input_message_content=InputTextMessageContent(
                locale(
                    "message_content",
                    "inline",
                    "forbidden",
                    locale=inline_query.from_user,
                )
            ),
            description=locale(
                "description", "inline", "forbidden", locale=inline_query.from_user
            ),
        )
    ]

    try:
        holo_user = HoloUser(inline_query.from_user)
    except (UserNotFoundError, UserInvalidError):
        logWrite(
            f"Could not find application of {inline_query.from_user.id}, ignoring inline query",
            debug=True,
        )
        await inline_query.answer(results=results_forbidden)
        return

    if path.exists(path.join(configGet("cache", "locations"), "group_members")) and (
        inline_query.from_user.id
        not in jsonLoad(path.join(configGet("cache", "locations"), "group_members"))
    ):
        if path.exists(path.join(configGet("cache", "locations"), "admins")) and (
            inline_query.from_user.id
            not in jsonLoad(path.join(configGet("cache", "locations"), "admins"))
        ):
            logWrite(
                f"{inline_query.from_user.id} is not an admin and not in members group, ignoring inline query",
                debug=True,
            )
            await inline_query.answer(results=results_forbidden)
            return

    if holo_user.application_approved() or (await isAnAdmin(holo_user.id) is True):
        max_results = (
            configGet("inline_preview_count") if inline_query.query != "" else 200
        )

        list_of_users = []
        async for m in app.get_chat_members(
            configGet("users", "groups"),
            limit=max_results,
            filter=ChatMembersFilter.SEARCH,
            query=inline_query.query,
        ):
            list_of_users.append(m)

        for match in list_of_users:
            application = col_applications.find_one({"user": match.user.id})

            if application is None:
                continue

            application_content = []
            i = 1

            for question in application["application"]:
                if i == 2:
                    age = relativedelta(datetime.now(), application["application"]["2"])
                    application_content.append(
                        f"{locale(f'question{i}', 'message', 'question_titles', locale=inline_query.from_user)} {application['application']['2'].strftime('%d.%m.%Y')} ({age.years} р.)"
                    )
                elif i == 3:
                    if application["application"]["3"]["countryCode"] == "UA":
                        application_content.append(
                            f"{locale(f'question{i}', 'message', 'question_titles', locale=inline_query.from_user)} {application['application']['3']['name']}"
                        )
                    else:
                        application_content.append(
                            f"{locale(f'question{i}', 'message', 'question_titles', locale=inline_query.from_user)} {application['application']['3']['name']} ({application['application']['3']['adminName1']}, {application['application']['3']['countryName']})"
                        )
                else:
                    application_content.append(
                        f"{locale(f'question{i}', 'message', 'question_titles', locale=inline_query.from_user)} {application['application'][question]}"
                    )

                i += 1

            if match.user.photo != None:
                try:
                    if not path.exists(
                        f'{configGet("cache", "locations")}{sep}avatars{sep}{match.user.photo.big_file_id}'
                    ):
                        print(
                            f'Downloaded avatar {match.user.photo.big_file_id} of {match.user.id} and uploaded to {configGet("api")}/avatars/{match.user.photo.big_file_id}',
                            flush=True,
                        )
                        await app.download_media(
                            match.user.photo.big_file_id,
                            file_name=f'{configGet("cache", "locations")}{sep}avatars{sep}{match.user.photo.big_file_id}',
                        )
                    results.append(
                        InlineQueryResultArticle(
                            title=str(match.user.first_name),
                            input_message_content=InputTextMessageContent(
                                locale(
                                    "message_content",
                                    "inline",
                                    "user",
                                    locale=inline_query.from_user,
                                ).format(
                                    match.user.first_name,
                                    match.user.username,
                                    "\n".join(application_content),
                                )
                            ),
                            description=locale(
                                "description",
                                "inline",
                                "user",
                                locale=inline_query.from_user,
                            ).format(match.user.first_name, match.user.username),
                            thumb_url=f'{configGet("api")}/avatars/{match.user.photo.big_file_id}',
                        )
                    )
                except ValueError:
                    results.append(
                        InlineQueryResultArticle(
                            title=str(match.user.first_name),
                            input_message_content=InputTextMessageContent(
                                locale(
                                    "message_content",
                                    "inline",
                                    "user",
                                    locale=inline_query.from_user,
                                ).format(
                                    match.user.first_name,
                                    match.user.username,
                                    "\n".join(application_content),
                                )
                            ),
                            description=locale(
                                "description",
                                "inline",
                                "user",
                                locale=inline_query.from_user,
                            ).format(match.user.first_name, match.user.username),
                        )
                    )
                except FileNotFoundError:
                    results.append(
                        InlineQueryResultArticle(
                            title=str(match.user.first_name),
                            input_message_content=InputTextMessageContent(
                                locale(
                                    "message_content",
                                    "inline",
                                    "user",
                                    locale=inline_query.from_user,
                                ).format(
                                    match.user.first_name,
                                    match.user.username,
                                    "\n".join(application_content),
                                )
                            ),
                            description=locale(
                                "description",
                                "inline",
                                "user",
                                locale=inline_query.from_user,
                            ).format(match.user.first_name, match.user.username),
                        )
                    )
            else:
                results.append(
                    InlineQueryResultArticle(
                        title=str(match.user.first_name),
                        input_message_content=InputTextMessageContent(
                            locale(
                                "message_content",
                                "inline",
                                "user",
                                locale=inline_query.from_user,
                            ).format(
                                match.user.first_name,
                                match.user.username,
                                "\n".join(application_content),
                            )
                        ),
                        description=locale(
                            "description",
                            "inline",
                            "user",
                            locale=inline_query.from_user,
                        ).format(match.user.first_name, match.user.username),
                    )
                )

        await inline_query.answer(results=results, cache_time=10, is_personal=True)