from aiogram import F, Router
from aiogram.enums import ParseMode
from aiogram.fsm.context import FSMContext
from aiogram.fsm.state import State, StatesGroup
from aiogram.types import CallbackQuery, Message

from config import USERNAME_BOT
from database import (
    create_tracking_source,
    get_all_tracking_sources,
    get_tracking_source_stats,
)
from filters.admin import IsAdminFilter
from logger import logger

from ..panel.keyboard import AdminPanelCallback
from .keyboard import (
    AdminAdsCallback,
    build_ads_delete_confirm_kb,
    build_ads_kb,
    build_ads_list_kb,
    build_ads_stats_kb,
    build_cancel_input_kb,
)


router = Router()


class AdminAdsState(StatesGroup):
    waiting_for_new_name = State()
    waiting_for_new_code = State()


@router.callback_query(AdminPanelCallback.filter(F.action == "ads"), IsAdminFilter())
async def handle_ads_menu(callback_query: CallbackQuery):
    await callback_query.message.edit_text(text="📊 <b>Аналитика рекламы:</b>", reply_markup=build_ads_kb())


@router.callback_query(AdminAdsCallback.filter(F.action == "create"), IsAdminFilter())
async def handle_ads_create(callback_query: CallbackQuery, state: FSMContext):
    await state.set_state(AdminAdsState.waiting_for_new_name)
    await callback_query.message.edit_text(
        "📝 Введите <b>название</b> новой ссылки:", reply_markup=build_cancel_input_kb()
    )


@router.message(AdminAdsState.waiting_for_new_name, IsAdminFilter())
async def handle_ads_name_input(message: Message, state: FSMContext):
    name = message.text.strip()
    await state.update_data(name=name)
    await state.set_state(AdminAdsState.waiting_for_new_code)
    await message.answer(
        f"🔗 Введите <b>код ссылки</b> для: <code>{name}</code>.", reply_markup=build_cancel_input_kb()
    )


@router.message(AdminAdsState.waiting_for_new_code, IsAdminFilter())
async def handle_ads_code_input(message: Message, state: FSMContext, session):
    code = message.text.strip()
    data = await state.get_data()
    name = data["name"]
    code_with_prefix = f"utm_{code}"

    try:
        await create_tracking_source(
            name=name, code=code_with_prefix, type_="utm", created_by=message.from_user.id, session=session
        )
        stats = await get_tracking_source_stats(code_with_prefix, session)
        if not stats:
            await message.answer("❌ Источник не найден или не содержит данных.")
            return
        msg = format_ads_stats(stats, USERNAME_BOT)
        await message.answer(
            text=msg,
            reply_markup=build_ads_stats_kb(code_with_prefix),
        )

    except Exception as e:
        logger.error(f"Ошибка при создании ссылки: {e}")
        await message.answer("❌ Произошла ошибка при создании ссылки.")
    finally:
        await state.clear()


@router.callback_query(AdminAdsCallback.filter(F.action == "list"), IsAdminFilter())
async def handle_ads_list(callback_query: CallbackQuery, session):
    try:
        ads = await get_all_tracking_sources(session)
        reply_markup = build_ads_list_kb(ads, current_page=1, total_pages=1)
        await callback_query.message.edit_text(
            "📋 Выберите ссылку для просмотра статистики:", reply_markup=reply_markup
        )
    except Exception as e:
        logger.error(f"Ошибка при получении списка UTM: {e}")
        await callback_query.message.edit_text("Произошла ошибка при получении списка.")


@router.callback_query(AdminAdsCallback.filter(F.action == "view"), IsAdminFilter())
async def handle_ads_view(callback_query: CallbackQuery, callback_data: AdminAdsCallback, session):
    code = callback_data.code
    try:
        stats = await get_tracking_source_stats(code, session)
        if not stats:
            await callback_query.message.edit_text("❌ Источник не найден или не содержит данных.")
            return
        msg = format_ads_stats(stats, USERNAME_BOT)
        await callback_query.message.edit_text(
            text=msg, reply_markup=build_ads_stats_kb(code), parse_mode=ParseMode.HTML
        )
    except Exception as e:
        logger.error(f"Ошибка при просмотре статистики: {e}")
        await callback_query.message.edit_text("❌ Ошибка при получении статистики.")


@router.callback_query(AdminAdsCallback.filter(F.action == "delete_confirm"), IsAdminFilter())
async def handle_ads_delete_confirm(callback_query: CallbackQuery, callback_data: AdminAdsCallback):
    code = callback_data.code
    await callback_query.message.edit_text(
        text=f"Вы уверены, что хотите удалить ссылку <code>{code}</code>?",
        reply_markup=build_ads_delete_confirm_kb(code),
    )


@router.callback_query(AdminAdsCallback.filter(F.action == "delete"), IsAdminFilter())
async def handle_ads_delete(callback_query: CallbackQuery, callback_data: AdminAdsCallback, session):
    code = callback_data.code
    try:
        await session.execute("UPDATE users SET source_code = NULL WHERE source_code = $1", code)
        await session.execute("DELETE FROM tracking_sources WHERE code = $1", code)
        await callback_query.message.edit_text(f"🗑️ Ссылка <code>{code}</code> удалена.", reply_markup=build_ads_kb())
    except Exception as e:
        logger.error(f"Ошибка при удалении метки {code}: {e}", exc_info=True)
        await callback_query.message.edit_text("❌ Не удалось удалить ссылку.")


def format_ads_stats(stats: dict, username_bot: str) -> str:
    return (
        f"<b>📊 <u>Статистика по рекламной ссылке</u></b>\n\n"
        f"📌 <b>Название:</b> {stats['name']}\n"
        f"🔗 <b>Ссылка:</b> <code>https://t.me/{username_bot}?start={stats['code']}</code>\n"
        f"🕓 <b>Создана:</b> {stats['created_at'].strftime('%d.%m.%Y %H:%M')}\n\n"
        f"💡 <b>Активность:</b>\n"
        f"└ 🆕 <b>Регистраций:</b> <b>{stats.get('registrations', 0)}</b>\n"
        f"└ 🧪 <b>Триалов:</b> <b>{stats.get('trials', 0)}</b>\n"
        f"\n💰 <b>Финансовая информация:</b>\n"
        f"└ 💳 <b>Покупок:</b> <b>{stats.get('payments', 0)}</b>\n\n"
        f"<i>Просмотр статистики и управление рекламными ссылками</i>."
    )


@router.callback_query(AdminAdsCallback.filter(F.action == "cancel_input"), IsAdminFilter())
async def handle_ads_cancel_input(callback_query: CallbackQuery, state: FSMContext):
    await state.clear()
    await callback_query.message.edit_text(text="📊 <b>Аналитика рекламы:</b>", reply_markup=build_ads_kb())