from aiogram import types, Router, F
from aiogram.filters import Command
from tonsdk.utils import Address

from app.core._blockchain.ton.connect import TonConnect
from sqlalchemy import select, and_
from app.core._keyboards import get_inline_keyboard
from app.core._utils.tg_process_template import tg_process_template
from app.core.models.wallet_connection import WalletConnection
from app.core._config import PROJECT_HOST

main_router = Router()


async def send_home_menu(chat_wrap, user, wallet_connection, **kwargs):
    return await tg_process_template(
        chat_wrap, user.translated('home_menu').format(
            wallet_address=(
                Address(wallet_connection.wallet_address).to_string(1, 1, 1)
                    if wallet_connection else user.translated('p_walletNotConnected')
            ),
            name=user.front_format(plain_text=True),
        ), keyboard=get_inline_keyboard([
            [{
                'text': user.translated('ownedContent_button'),
                'callback_data': 'ownedContent'
            }],
            [{
                'text': user.translated('disconnectWallet_button'),
                'callback_data': 'disconnectWallet'
            }] if wallet_connection else []
        ]), **kwargs
    )


async def send_connect_wallets_list(db_session, chat_wrap, user, **kwargs):
    # Try to restore existing connection via DB
    result = await db_session.execute(select(WalletConnection).where(
        and_(WalletConnection.user_id == user.id, WalletConnection.invalidated == False, WalletConnection.network == 'ton')
    ).order_by(WalletConnection.created.desc()))
    ton_connection = result.scalars().first()
    ton_connect = TonConnect.by_key(ton_connection.keys["connection_key"]) if ton_connection else TonConnect()
    if ton_connection:
        await ton_connect.restore_connection()
    wallets = ton_connect._sdk_client.get_wallets()
    message_text = user.translated("connectWalletsList_menu")
    return await tg_process_template(
        chat_wrap, message_text,
        keyboard=get_inline_keyboard([
            [
                {
                    'text': f"{wallets[i]['name']}",
                    'callback_data': f"initTonconnect_{wallets[i]['app_name']}"
                } if i < len(wallets) else None,
                {
                    'text': f"{wallets[i + 1]['name']}",
                    'callback_data': f"initTonconnect_{wallets[i + 1]['app_name']}"
                } if i + 1 < len(wallets) else None,
            ]
            for i in range(0, len(wallets), 2)
        ]), **kwargs
    )


async def t_home_menu(__msg, **extra):
    memory, user, db_session, chat_wrap = extra['memory'], extra['user'], extra['db_session'], extra['chat_wrap']
    if extra.get('state'):
        await extra['state'].clear()

    if isinstance(__msg, types.CallbackQuery):
        message_id = __msg.message.message_id
    # elif isinstance(__msg, types.Message):
    #     message_id = __msg.message_id
    else:
        message_id = None

    wallet_connection = (await db_session.execute(select(WalletConnection).where(
        and_(WalletConnection.user_id == user.id, WalletConnection.invalidated == False)
    ))).scalars().first()

    # if not wallet_connection:
    #     return await send_connect_wallets_list(db_session, chat_wrap, user, message_id=message_id)

    return await send_home_menu(chat_wrap, user, wallet_connection, message_id=message_id)


async def t_admin_panel(message: types.Message, **extra):
    user = extra.get('user')
    chat_wrap = extra.get('chat_wrap')
    admin_host = (PROJECT_HOST or '').rstrip('/')
    if not user or not getattr(user, 'is_admin', False):
        await chat_wrap.send_message("Доступ к админ-панели ограничен.")
        return
    if not admin_host:
        await chat_wrap.send_message("Адрес админ-панели не настроен на этой ноде.")
        return
    admin_url = f"{admin_host}/admin"
    buttons = []
    if admin_url.startswith('https://'):
        buttons.append({
            'text': 'Открыть в Telegram',
            'web_app': types.WebAppInfo(url=admin_url),
        })
    buttons.append({
        'text': 'Открыть в браузере',
        'url': admin_url,
    })
    keyboard = get_inline_keyboard([buttons]) if buttons else None
    await chat_wrap.send_message(
        "Админ-панель доступна по кнопке ниже.",
        keyboard=keyboard,
    )


main_router.message.register(t_home_menu, Command('start'))
main_router.message.register(t_admin_panel, Command('admin'))
main_router.callback_query.register(t_home_menu, F.data == 'home')
router = main_router