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_empty", "message", locale=inline_query.from_user).format(locale(spoil["category"], "message", "spoiler_categories")) if spoil["description"] == "" else 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")}/?avatar_id={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")}/?avatar_id={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
        )