new_lottery_bot/src/controllers/bot_controller.py

from aiogram.types import Message, CallbackQuery
from aiogram import F
import logging

from src.interfaces.base import IBotController, ILotteryService, IUserService, IKeyboardBuilder, IMessageFormatter
from src.interfaces.base import ILotteryRepository, IParticipationRepository
from src.core.config import ADMIN_IDS
from src.core.telegram_config import get_parse_mode

logger = logging.getLogger(__name__)


class BotController(IBotController):
    """Основной контроллер бота"""
    
    def __init__(
        self,
        lottery_service: ILotteryService,
        user_service: IUserService,
        keyboard_builder: IKeyboardBuilder,
        message_formatter: IMessageFormatter,
        lottery_repo: ILotteryRepository,
        participation_repo: IParticipationRepository
    ):
        self.lottery_service = lottery_service
        self.user_service = user_service
        self.keyboard_builder = keyboard_builder
        self.message_formatter = message_formatter
        self.lottery_repo = lottery_repo
        self.participation_repo = participation_repo
    
    def is_admin(self, user_id: int) -> bool:
        """Проверить, является ли пользователь администратором"""
        return user_id in ADMIN_IDS
    
    async def handle_start(self, message: Message):
        """Обработать команду /start"""
        from src.utils.keyboards import get_main_reply_keyboard
        import logging
        
        logger = logging.getLogger(__name__)
        
        user = await self.user_service.get_or_create_user(
            telegram_id=message.from_user.id,
            username=message.from_user.username,
            first_name=message.from_user.first_name,
            last_name=message.from_user.last_name
        )
        
        # Логирование статуса регистрации
        logger.info(f"User {message.from_user.id}: is_registered={user.is_registered}, is_admin={self.is_admin(message.from_user.id)}")
        
        welcome_text = f"👋 Добро пожаловать, {user.first_name or 'дорогой пользователь'}!\n\n"
        welcome_text += "🎲 Это бот для участия в розыгрышах.\n\n"
        
        if user.is_registered:
            welcome_text += "✅ Вы уже зарегистрированы в системе!"
        else:
            welcome_text += "📝 Для участия в розыгрышах необходимо зарегистрироваться."
        
        # Inline клавиатура
        inline_keyboard = self.keyboard_builder.get_main_keyboard(
            is_admin=self.is_admin(message.from_user.id),
            is_registered=user.is_registered
        )
        
        # Обычная клавиатура
        reply_keyboard = get_main_reply_keyboard(
            is_admin=self.is_admin(message.from_user.id),
            is_registered=user.is_registered
        )
        
        await message.answer(
            welcome_text,
            reply_markup=reply_keyboard  # Обычная клавиатура
        )
        
        # Отправляем inline клавиатуру отдельным сообщением
        await message.answer(
            "Выберите действие:",
            reply_markup=inline_keyboard
        )
    
    async def handle_active_lotteries(self, callback: CallbackQuery):
        """Показать активные розыгрыши"""
        lotteries = await self.lottery_repo.get_active()
        
        if not lotteries:
            await callback.answer("❌ Нет активных розыгрышей", show_alert=True)
            return
        
        text = "🎲 <b>Активные розыгрыши:</b>\n\n"
        
        for lottery in lotteries:
            participants_count = await self.participation_repo.get_count_by_lottery(lottery.id)
            lottery_info = self.message_formatter.format_lottery_info(lottery, participants_count)
            text += lottery_info + "\n" + "="*30 + "\n\n"
        
        # Получаем информацию о регистрации пользователя
        user = await self.user_service.get_or_create_user(
            telegram_id=callback.from_user.id,
            username=callback.from_user.username,
            first_name=callback.from_user.first_name,
            last_name=callback.from_user.last_name
        )
        
        keyboard = self.keyboard_builder.get_main_keyboard(
            is_admin=self.is_admin(callback.from_user.id),
            is_registered=user.is_registered
        )
        
        try:
            await callback.message.edit_text(
                text,
                reply_markup=keyboard,
                parse_mode=get_parse_mode("inline_keyboard")
            )
        except Exception as e:
            # Если сообщение не изменилось - просто отвечаем на callback
            if "message is not modified" in str(e):
                await callback.answer("✅ Уже показаны активные розыгрыши")
            else:
                # Другие ошибки - пробуем отправить новое сообщение
                await callback.answer()
                await callback.message.answer(
                    text,
                    reply_markup=keyboard,
                    parse_mode=get_parse_mode("inline_keyboard")
                )