"""Обработчик команды /start и главного меню."""

from aiogram import Router
from aiogram.filters import CommandStart, Command
from aiogram.types import Message, InlineKeyboardMarkup, InlineKeyboardButton

from config import settings

router = Router()


def get_main_keyboard(user_id: int = None) -> InlineKeyboardMarkup:
    """Создать главное меню."""
    keyboard = InlineKeyboardMarkup(inline_keyboard=[
        [
            InlineKeyboardButton(text="🎨 Генерация (txt2img)", callback_data="gen_txt2img"),
        ],
        [
            InlineKeyboardButton(text="⚙️ Профили", callback_data="profiles_menu"),
            InlineKeyboardButton(text="🗑️ Настройки хранения", callback_data="storage_settings"),
        ],
        [
            InlineKeyboardButton(text="📊 Статус SD API", callback_data="check_sd_status"),
            InlineKeyboardButton(text="ℹ️ Помощь", callback_data="help"),
        ],
    ])

    # Добавляем кнопку админ-панели для админа
    if user_id and user_id == settings.ADMIN_ID:
        keyboard.inline_keyboard.insert(0, [
            InlineKeyboardButton(text="🛡️ Админ-панель", callback_data="admin_panel"),
        ])

    return keyboard


@router.message(CommandStart())
async def cmd_start(message: Message):
    """Обработка команды /start."""
    text = (
        f"👋 <b>Привет, {message.from_user.first_name}!</b>\n\n"
        "Я бот для генерации изображений через Stable Diffusion.\n"
        "Выберите действие из меню ниже:"
    )
    await message.answer(text, reply_markup=get_main_keyboard(message.from_user.id), parse_mode="HTML")


@router.message(Command("menu"))
async def cmd_menu(message: Message):
    """Обработка команды /menu."""
    await message.answer("📋 Главное меню:", reply_markup=get_main_keyboard(message.from_user.id))