from datetime import datetime
from typing import Any

import pytz

from aiogram import F, Router
from aiogram.exceptions import TelegramBadRequest
from aiogram.types import CallbackQuery

from filters.admin import IsAdminFilter
from logger import logger
from utils.csv_export import export_hot_leads_csv, export_keys_csv, export_payments_csv, export_users_csv

from ..panel.keyboard import AdminPanelCallback, build_admin_back_kb
from .keyboard import build_stats_kb


router = Router()


@router.callback_query(
    AdminPanelCallback.filter(F.action == "stats"),
    IsAdminFilter(),
)
async def handle_stats(callback_query: CallbackQuery, session: Any):
    try:
        total_users = await session.fetchval("SELECT COUNT(*) FROM users")
        total_keys = await session.fetchval("SELECT COUNT(*) FROM keys")
        total_referrals = await session.fetchval("SELECT COUNT(*) FROM referrals")

        total_payments_today = int(
            await session.fetchval("SELECT COALESCE(SUM(amount), 0) FROM payments WHERE created_at >= CURRENT_DATE")
        )
        total_payments_week = int(
            await session.fetchval(
                "SELECT COALESCE(SUM(amount), 0) FROM payments WHERE created_at >= date_trunc('week', CURRENT_DATE)"
            )
        )
        total_payments_month = int(
            await session.fetchval(
                "SELECT COALESCE(SUM(amount), 0) FROM payments WHERE created_at >= date_trunc('month', CURRENT_DATE)"
            )
        )
        total_payments_last_month = int(
            await session.fetchval(
                """
                SELECT COALESCE(SUM(amount), 0)
                FROM payments
                WHERE created_at >= date_trunc('month', CURRENT_DATE - interval '1 month')
                AND created_at < date_trunc('month', CURRENT_DATE)
                """
            )
        )
        total_payments_all_time = int(await session.fetchval("SELECT COALESCE(SUM(amount), 0) FROM payments"))

        all_keys = await session.fetch("SELECT created_at, expiry_time FROM keys")

        def count_subscriptions_by_duration(keys):
            periods = {"trial": 0, "1": 0, "3": 0, "6": 0, "12": 0}
            for key in keys:
                try:
                    duration_days = (key["expiry_time"] - key["created_at"]) / (1000 * 60 * 60 * 24)

                    if duration_days <= 29:
                        periods["trial"] += 1
                    elif duration_days <= 89:
                        periods["1"] += 1
                    elif duration_days <= 179:
                        periods["3"] += 1
                    elif duration_days <= 359:
                        periods["6"] += 1
                    else:
                        periods["12"] += 1
                except Exception as e:
                    logger.error(f"Error processing key duration: {e}")
                    continue
            return periods

        subs_all_time = count_subscriptions_by_duration(all_keys)

        registrations_today = await session.fetchval("SELECT COUNT(*) FROM users WHERE created_at >= CURRENT_DATE")
        registrations_week = await session.fetchval(
            "SELECT COUNT(*) FROM users WHERE created_at >= date_trunc('week', CURRENT_DATE)"
        )
        registrations_month = await session.fetchval(
            "SELECT COUNT(*) FROM users WHERE created_at >= date_trunc('month', CURRENT_DATE)"
        )

        users_updated_today = await session.fetchval("SELECT COUNT(*) FROM users WHERE updated_at >= CURRENT_DATE")

        active_keys = await session.fetchval(
            "SELECT COUNT(*) FROM keys WHERE expiry_time > $1",
            int(datetime.utcnow().timestamp() * 1000),
        )
        expired_keys = total_keys - active_keys
        moscow_tz = pytz.timezone("Europe/Moscow")
        update_time = datetime.now(moscow_tz).strftime("%d.%m.%y %H:%M:%S")

        hot_leads_count = await session.fetchval("""
            SELECT COUNT(DISTINCT u.tg_id)
            FROM users u
            JOIN payments p ON u.tg_id = p.tg_id
            LEFT JOIN keys k ON u.tg_id = k.tg_id
            WHERE p.status = 'success'
            AND k.tg_id IS NULL
        """)

        stats_message = (
            "📊 <b>Статистика проекта</b>\n\n"
            "👤 <b>Пользователи:</b>\n"
            f"├ 🗓️ За день: <b>{registrations_today}</b>\n"
            f"├ 📆 За неделю: <b>{registrations_week}</b>\n"
            f"├ 🗓️ За месяц: <b>{registrations_month}</b>\n"
            f"└ 🌐 Всего: <b>{total_users}</b>\n\n"
            "💡 <b>Активность:</b>\n"
            f"└ 👥 Сегодня были активны: <b>{users_updated_today}</b>\n\n"
            "🤝 <b>Реферальная система:</b>\n"
            f"└ 👥 Всего привлечено: <b>{total_referrals}</b>\n\n"
            "🔐 <b>Подписки:</b>\n"
            f"├ 📦 Всего сгенерировано: <b>{total_keys}</b>\n"
            f"├ ✅ Активных: <b>{active_keys}</b>\n"
            f"├ ❌ Просроченных: <b>{expired_keys}</b>\n"
            f"└ 📋 По срокам:\n"
            f"     • 🎁 Триал: <b>{subs_all_time['trial']}</b>\n"
            f"     • 🗓️ 1 мес: <b>{subs_all_time['1']}</b>\n"
            f"     • 🗓️ 3 мес: <b>{subs_all_time['3']}</b>\n"
            f"     • 🗓️ 6 мес: <b>{subs_all_time['6']}</b>\n"
            f"     • 🗓️ 12 мес: <b>{subs_all_time['12']}</b>\n\n"
            "💰 <b>Финансы:</b>\n"
            f"├ 📅 За день: <b>{total_payments_today} ₽</b>\n"
            f"├ 📆 За неделю: <b>{total_payments_week} ₽</b>\n"
            f"├ 📆 За месяц: <b>{total_payments_month} ₽</b>\n"
            f"├ 📆 За прошлый месяц: <b>{total_payments_last_month} ₽</b>\n"
            f"└ 🏦 Всего: <b>{total_payments_all_time} ₽</b>\n\n"
            f"🔥 <b>Горящие лиды</b>: <b>{hot_leads_count}</b> (платили, но не продлили)\n\n"
            f"⏱️ <i>Последнее обновление:</i> <code>{update_time}</code>"
        )

        await callback_query.message.edit_text(text=stats_message, reply_markup=build_stats_kb())
    except TelegramBadRequest as e:
        if "message is not modified" not in str(e):
            logger.error(f"Error in user_stats_menu: {e}")
    except Exception as e:
        logger.error(f"Error in user_stats_menu: {e}")
        await callback_query.answer("Произошла ошибка при получении статистики", show_alert=True)


@router.callback_query(
    AdminPanelCallback.filter(F.action == "stats_export_users_csv"),
    IsAdminFilter(),
)
async def handle_export_users_csv(callback_query: CallbackQuery, session: Any):
    kb = build_admin_back_kb("stats")
    try:
        export = await export_users_csv(session)
        await callback_query.message.answer_document(document=export, caption="📥 Экспорт пользователей в CSV")
    except Exception as e:
        logger.error(f"Ошибка при экспорте пользователей в CSV: {e}")
        await callback_query.message.edit_text(text=f"❗ Произошла ошибка при экспорте: {e}", reply_markup=kb)


@router.callback_query(
    AdminPanelCallback.filter(F.action == "stats_export_payments_csv"),
    IsAdminFilter(),
)
async def handle_export_payments_csv(callback_query: CallbackQuery, session: Any):
    kb = build_admin_back_kb("stats")
    try:
        export = await export_payments_csv(session)
        await callback_query.message.answer_document(document=export, caption="📥 Экспорт платежей в CSV")
    except Exception as e:
        logger.error(f"Ошибка при экспорте платежей в CSV: {e}")
        await callback_query.message.edit_text(text=f"❗ Произошла ошибка при экспорте: {e}", reply_markup=kb)


@router.callback_query(
    AdminPanelCallback.filter(F.action == "stats_export_hot_leads_csv"),
    IsAdminFilter(),
)
async def handle_export_hot_leads_csv(callback_query: CallbackQuery, session: Any):
    kb = build_admin_back_kb("stats")
    try:
        export = await export_hot_leads_csv(session)
        await callback_query.message.answer_document(document=export, caption="📥 Экспорт горящих лидов")
    except Exception as e:
        logger.error(f"Ошибка при экспорте 'горящих лидов': {e}")
        await callback_query.message.edit_text(text=f"❗ Произошла ошибка при экспорте: {e}", reply_markup=kb)


@router.callback_query(
    AdminPanelCallback.filter(F.action == "stats_export_keys_csv"),
    IsAdminFilter(),
)
async def handle_export_keys_csv(callback_query: CallbackQuery, session: Any):
    kb = build_admin_back_kb("stats")
    try:
        export = await export_keys_csv(session)
        await callback_query.message.answer_document(document=export, caption="📥 Экспорт подписок в CSV")
    except Exception as e:
        logger.error(f"Ошибка при экспорте подписок в CSV: {e}")
        await callback_query.message.edit_text(text=f"❗ Произошла ошибка при экспорте: {e}", reply_markup=kb)