Solo_bot/handlers/admin/ads/ads_handler.py

from aiogram import F, Router
from aiogram.fsm.context import FSMContext
from aiogram.fsm.state import State, StatesGroup
from aiogram.types import CallbackQuery, Message
from sqlalchemy import delete, select, update
from sqlalchemy.ext.asyncio import AsyncSession
import re

from config import USERNAME_BOT
from database import create_tracking_source, get_tracking_source_stats
from database.models import TrackingSource, User
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>.\n\n"
        f"💡 <b>Правила:</b> только латинские буквы и цифры",
        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: AsyncSession
):
    code = message.text.strip()
    data = await state.get_data()
    name = data["name"]

    if not re.match(r'^[a-zA-Z0-9]+$', code):
        await message.answer(
            "❌ Код может содержать только латинские буквы и цифры\n"
            "Введите код заново:",
            reply_markup=build_cancel_input_kb(),
        )
        return
    
    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(session, code_with_prefix)
        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}", exc_info=True)
        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: AsyncSession, callback_data: AdminAdsCallback):
    try:
        result = await session.execute(
            select(TrackingSource).order_by(TrackingSource.created_at.desc())
        )
        ads = result.scalars().all()
        items_per_page = 6
        if callback_data.code and callback_data.code.isdigit():
            current_page = int(callback_data.code)
        else:
            current_page = 1
        total_pages = (len(ads) + items_per_page - 1) // items_per_page
        reply_markup = build_ads_list_kb(ads, current_page, total_pages)
        await callback_query.message.edit_text(
            "📋 Выберите ссылку для просмотра статистики:", reply_markup=reply_markup
        )
    except Exception as e:
        logger.error(f"Ошибка при получении списка UTM: {e}", exc_info=True)
        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: AsyncSession,
):
    code = callback_data.code
    try:
        stats = await get_tracking_source_stats(session, code)
        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)
        )
    except Exception as e:
        logger.error(f"Ошибка при просмотре статистики: {e}", exc_info=True)
        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: AsyncSession,
):
    code = callback_data.code
    try:
        await session.execute(
            update(User).where(User.source_code == code).values(source_code=None)
        )
        await session.execute(delete(TrackingSource).where(TrackingSource.code == code))
        await session.commit()
        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("❌ Не удалось удалить ссылку.")


@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()
    )


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\n"
        f"💰 <b>Финансовая информация:</b>\n"
        f"├ 💳 <b>Покупок:</b> <b>{stats.get('payments', 0)}</b>\n"
        f"└ 💸 <b>Сумма:</b> <b>{round(stats.get('total_amount', 0), 2)} ₽</b>\n\n"
        f"<i>Просмотр статистики и управление рекламными ссылками</i>."
    )