from aiogram_dialog import Dialog, Window
from aiogram_dialog.widgets.kbd import Row, Start, Button
from aiogram_dialog.widgets.text import Format, Const
from aiogram_dialog.widgets.input import TextInput, ManagedTextInput
from aiogram_dialog import DialogManager
from aiogram.types import Message

from bot.services import check_pay_crypto_bot, check_pay_lolz
from bot.getters import getter_amount, getter_amount_cryptobot
from bot.states import UpBalanceCryptoSG, UpBalanceSG, UpBalanceLolzSG, ProfileSG
from bot.validations import error_number_handler
from typing import Any


class UpBalanceWindow(Window):
    def __init__(self, text: str, state: Any, on_error: Any, **kwargs):
        super().__init__(
            Const(text=text),
            TextInput(
                id="count",
                type_factory=str,
                on_success=self._on_success_callback,
                on_error=on_error,
            ),
            state=state,
            **kwargs
        )

    async def _on_success_callback(self, message: Message, widget: ManagedTextInput, dialog_manager: DialogManager, data: str):
        if not dialog_manager.current_context().dialog_data.get("error"):
            await dialog_manager.next()


class PaymentWindow(Window):
    def __init__(self, state: Any, check_pay_function: Any, getter: Any, **kwargs):
        super().__init__(
            Format(text="{paylink}", when=lambda data, widget, manager: not data.get("error")),
            Format(text="{error}", when=lambda data, widget, manager: data.get("error") is not None),
            Button(
                Const("🔍 Проверить оплату"),
                id="check_pay_button",
                on_click=check_pay_function,
                when=lambda data, widget, manager: not data.get("error") and data.get("paylink") is not None
            ),
            Button(
                Const("🔙 Отмена"),
                id="cancel_button",
                on_click=lambda callback, button, manager: manager.switch_to(ProfileSG.profile),
                when=lambda data, widget, manager: data.get("error") is not None
            ),
            state=state,
            getter=getter,
            **kwargs
        )


up_balance_dialog = Dialog(
    Window(
        Const("Выберите через что хотите пополнить баланс:"),
        Row(
            Start(
                Const("💸 Lolz"),
                id="paymethod_lolz",
                state=UpBalanceLolzSG.up_balance_lolz,
            ),
            Start(
                Const("💸 CryptoBot"),
                id="paymethod_crypto",
                state=UpBalanceCryptoSG.up_balance_crypto,
            ),
        ),
        Start(Const("🔙 Назад"), id="back", state=ProfileSG.profile),
        state=UpBalanceSG.up_balance,
    )
)


up_balance_lolz_dialog = Dialog(
    UpBalanceWindow(
        text="💸 Введите сумму пополнения",
        state=UpBalanceLolzSG.up_balance_lolz,
        on_error=error_number_handler,
    ),
    PaymentWindow(
        state=UpBalanceLolzSG.pay_link,
        check_pay_function=check_pay_lolz,
        getter=getter_amount,
    )

)


up_balance_cryptobot_dialog = Dialog(
    UpBalanceWindow(
        text="💸 Введите сумму пополнения",
        state=UpBalanceCryptoSG.up_balance_crypto,
        on_error=error_number_handler
    ),
    PaymentWindow(
        state=UpBalanceCryptoSG.pay_link,
        check_pay_function=check_pay_crypto_bot,
        getter=getter_amount_cryptobot,
    ),
)