from telethon import TelegramClient, events
from textual.app import App, ComposeResult
from textual.containers import Horizontal, VerticalScroll, Vertical
from textual.widgets import Static, Footer, Label, Input, Button
from widgets.chat import Chat
from widgets.dialog import Dialog
from textual.screen import Screen
#from telegram.client import TelegramClientWrapper
from tokens import api_id, api_hash

class ChatScreen(Screen):
    """Класс экрана чатов, он же основной экран приложения"""

    def __init__(self, name = None, id = None, classes = None, telegram_client = None):
        super().__init__(name, id, classes)
        self.telegram_client = telegram_client
        self.telegram_client.on(events.NewMessage())(self.update_chat_list)

    def on_mount(self):
        self.chat_container = self.query_one("#main_container").query_one("#chats").query_one("#chat_container")

        self.limit = 100
        for i in range(self.limit):
            chat = Chat(id=f"chat-{i + 1}", notify_func=self.notify)
            self.chat_container.mount(chat)
        #self.mount_chats(limit=25)

    def mount_chats(self, limit: int):
        self.limit = limit
        chats_amount = len(self.chat_container.query(Chat))
        if limit > chats_amount:
            for i in range(limit - chats_amount):
                chat = Chat(id=f"chat-{i + 1 + (limit - chats_amount)}")
                self.chat_container.mount(chat)
        elif not (limit == chats_amount):
            for i in range(chats_amount - limit):
                self.chat_container.query(Chat).last().remove()

    async def update_chat_list(self):
        dialogs = await self.telegram_client.get_dialogs(limit=self.limit)

        for i in range(len(dialogs)):
            chat = self.chat_container.query_one(f"#chat-{i + 1}")
            chat.username = str(dialogs[i].name)
            chat.msg = str(dialogs[i].message.message)
            chat.peer_id = dialogs[i].id
            #self.notify("Новое сообщение")    #колхоз дебаг

    def compose(self) -> ComposeResult:
        yield Footer()
        with Horizontal(id="main_container"):
            with Horizontal(id="chats"):
                yield VerticalScroll(Static(id="chat_container"))
                #TODO: сделать кнопку чтобы прогрузить больше чатов, это оптимизация

            yield Dialog()

class AuthScreen(Screen):
    """Это будет классом экрана логина"""

    def __init__(self, name = None, id = None, classes = None, telegram_client: TelegramClient | None = None):
        super().__init__(name, id, classes)
        self.client = telegram_client
        self.ac = self.query_one("#auth_container")

    def compose(self):
        with Vertical(id="auth_container"):
            yield Label("Добро пожаловать в Telegram TUI")
            yield Input(placeholder="Номер телефона", id="phone")
            yield Input(placeholder="Код", id="code", disabled=True)
            yield Input(placeholder="Пароль", id="password", password=True, disabled=True)

    async def on_submit(self, event: Input.Submitted) -> None:
        if event.button.id == "phone":
            await self.client.send_code_request(event.value)
            for i in ("#phone", "#password"):
                self.ac.query_one(i).disabled = True
            self.ac.query_one("#code").disabled = False
        elif event.button.id == "code":
            await self.client.sign_in(event.value)
            for i in ("#phone", "#code"):
                self.ac.query_one(i).disabled = True
            self.ac.query_one("#password").disabled = False
        elif event.button.id == "password":
            await self.client.sign_in(event.value)
            for i in ("#phone", "#password"):
                self.ac.query_one(i).disabled = True
            self.ac.query_one("#code").disabled = True

class TelegramTUI(App):
    """Класс приложения"""

    CSS_PATH = "../tcss/style.tcss"
    #SCREENS = {"chats": ChatScreen}

    def __init__(self):
        super().__init__()

    async def on_mount(self) -> None:
        self.telegram_client = TelegramClient("user", api_id, api_hash)
        await self.telegram_client.start()

        if not await self.telegram_client.is_user_authorized():
            auth_screen = AuthScreen(telegram_client=self.telegram_client)
            self.install_screen(auth_screen, name="auth")
            self.push_screen("auth")

        """chat_screen = ChatScreen(telegram_client=self.telegram_client)
        self.install_screen(chat_screen, name="chats")
        self.push_screen("chats")
        await self.telegram_client.start()
        await chat_screen.update_chat_list()"""

    async def on_exit_app(self):
        await self.telegram_client.disconnect()
        return super()._on_exit_app()