hyperboria/nexus/views/telegram/document_list_widget.py

from typing import (
    List,
    Optional,
)

from idm.api.proto import chat_manager_service_pb2
from library.telegram.common import close_button
from nexus.translations import t
from nexus.views.telegram.base_holder import BaseHolder
from telethon import Button


class DocumentListWidget:
    def __init__(
        self,
        chat: chat_manager_service_pb2.Chat,
        document_holders: List[BaseHolder],
        bot_name,
        header: Optional[str] = None,
        has_next: bool = False,
        session_id: Optional[str] = None,
        message_id: Optional[int] = None,
        request_id: Optional[str] = None,
        cmd: str = None,
        page: int = 0,
        page_size: int = 5,
    ):
        self.chat = chat
        self.document_holders = document_holders
        self.bot_name = bot_name
        self.header = header
        self.cmd = cmd
        self.has_next = has_next
        self.session_id = session_id
        self.message_id = message_id
        self.request_id = request_id
        self.page = page
        self.page_size = page_size

    async def render(self) -> tuple[str, Optional[list]]:
        if not len(self.document_holders):
            return t('COULD_NOT_FIND_ANYTHING', self.chat.language), [close_button(self.session_id)]

        serp_elements = []
        for position, document_holder in enumerate(self.document_holders):
            serp_elements.append(
                document_holder
                    .view_builder(self.chat.language)
                    .add_short_description()
                    .add_new_line()
                    .add_links(bot_name=self.bot_name)
                    .build()
            )

        serp = '\n\n'.join(serp_elements)

        if self.header:
            serp = f'**{self.header}**\n\n{serp}'

        buttons = []
        if self.cmd and self.message_id and self.session_id and (self.has_next or self.page > 0):
            buttons = [
                Button.inline(
                    text='<<1' if self.page > 1 else ' ',
                    data=f'/{self.cmd}_{self.session_id}_{self.message_id}_0'
                    if self.page > 1 else '/noop',
                ),
                Button.inline(
                    text=f'<{self.page}' if self.page > 0 else ' ',
                    data=f'/{self.cmd}_{self.session_id}_{self.message_id}_{self.page - 1}'
                    if self.page > 0 else '/noop',
                ),
                Button.inline(
                    text=f'{self.page + 2}>' if self.has_next else ' ',
                    data=f'/{self.cmd}_{self.session_id}_{self.message_id}_{self.page + 1}'
                    if self.has_next else '/noop',
                )
            ]
        buttons.append(close_button(self.session_id))
        return serp, buttons