tg_bot_language/bot/handlers/settings.py

from aiogram import Router, F
from aiogram.filters import Command
from aiogram.types import Message, CallbackQuery, InlineKeyboardMarkup, InlineKeyboardButton
from aiogram.fsm.context import FSMContext

from database.db import async_session_maker
from database.models import LanguageLevel
from bot.handlers.start import main_menu_keyboard
from services.user_service import UserService

router = Router()


def _is_en(user) -> bool:
    try:
        return (getattr(user, 'language_interface', 'ru') or 'ru') == 'en'
    except Exception:
        return False


def _is_ja(user) -> bool:
    try:
        return (getattr(user, 'language_interface', 'ru') or 'ru') == 'ja'
    except Exception:
        return False


def get_settings_keyboard(user) -> InlineKeyboardMarkup:
    """Создать клавиатуру настроек"""
    is_en = _is_en(user)
    is_ja = _is_ja(user)
    keyboard = InlineKeyboardMarkup(inline_keyboard=[
        [InlineKeyboardButton(
            text=(
                "📊 Level: " if is_en else ("📊 レベル: " if is_ja else "📊 Уровень: ")
            ) + f"{user.level.value}",
            callback_data="settings_level"
        )],
        [InlineKeyboardButton(
            text=(
                "🎯 Learning language: " if is_en else ("🎯 学習言語: " if is_ja else "🎯 Язык изучения: ")
            ) + (user.learning_language.upper()),
            callback_data="settings_learning"
        )],
        [InlineKeyboardButton(
            text=(
                "🌐 Interface language: " if is_en else ("🌐 インターフェース言語: " if is_ja else "🌐 Язык интерфейса: ")
            ) + ("🇬🇧 English" if getattr(user, 'language_interface', 'ru') == 'en' else ("🇯🇵 日本語" if getattr(user, 'language_interface', 'ru') == 'ja' else "🇷🇺 Русский")),
            callback_data="settings_language"
        )],
        [InlineKeyboardButton(
            text=("❌ Close" if is_en else ("❌ 閉じる" if is_ja else "❌ Закрыть")),
            callback_data="settings_close"
        )]
    ])
    return keyboard


def get_level_keyboard(user=None) -> InlineKeyboardMarkup:
    """Клавиатура выбора уровня"""
    lang = getattr(user, 'language_interface', 'ru') if user is not None else 'ru'
    if lang == 'en':
        levels = [
            ("A1 - Beginner", "set_level_A1"),
            ("A2 - Elementary", "set_level_A2"),
            ("B1 - Intermediate", "set_level_B1"),
            ("B2 - Upper-intermediate", "set_level_B2"),
            ("C1 - Advanced", "set_level_C1"),
            ("C2 - Proficient", "set_level_C2"),
        ]
    elif lang == 'ja':
        levels = [
            ("A1 - 初級", "set_level_A1"),
            ("A2 - 初級(上)", "set_level_A2"),
            ("B1 - 中級", "set_level_B1"),
            ("B2 - 中級(上)", "set_level_B2"),
            ("C1 - 上級", "set_level_C1"),
            ("C2 - ネイティブ", "set_level_C2"),
        ]
    else:
        levels = [
            ("A1 - Начальный", "set_level_A1"),
            ("A2 - Элементарный", "set_level_A2"),
            ("B1 - Средний", "set_level_B1"),
            ("B2 - Выше среднего", "set_level_B2"),
            ("C1 - Продвинутый", "set_level_C1"),
            ("C2 - Профессиональный", "set_level_C2"),
        ]

    keyboard = []
    for level_name, callback_data in levels:
        keyboard.append([InlineKeyboardButton(text=level_name, callback_data=callback_data)])

    back_label = "⬅️ Back" if lang == 'en' else ("⬅️ 戻る" if lang == 'ja' else "⬅️ Назад")
    keyboard.append([InlineKeyboardButton(text=back_label, callback_data="settings_back")])

    return InlineKeyboardMarkup(inline_keyboard=keyboard)


def get_language_keyboard(user=None) -> InlineKeyboardMarkup:
    """Клавиатура выбора языка интерфейса"""
    lang = getattr(user, 'language_interface', 'ru') if user is not None else 'ru'
    back = "⬅️ Back" if lang == 'en' else ("⬅️ 戻る" if lang == 'ja' else "⬅️ Назад")
    keyboard = InlineKeyboardMarkup(inline_keyboard=[
        [InlineKeyboardButton(text="🇷🇺 Русский", callback_data="set_lang_ru")],
        [InlineKeyboardButton(text="🇬🇧 English", callback_data="set_lang_en")],
        [InlineKeyboardButton(text="🇯🇵 日本語", callback_data="set_lang_ja")],
        [InlineKeyboardButton(text=back, callback_data="settings_back")]
    ])
    return keyboard


def get_learning_language_keyboard(user=None) -> InlineKeyboardMarkup:
    """Клавиатура выбора языка изучения"""
    lang = getattr(user, 'language_interface', 'ru') if user is not None else 'ru'
    back = "⬅️ Back" if lang == 'en' else ("⬅️ 戻る" if lang == 'ja' else "⬅️ Назад")

    # Пары (код -> подпись)
    options = [
        ("en", "🇬🇧 English" if lang != 'ja' else "🇬🇧 英語"),
        ("es", "🇪🇸 Spanish" if lang == 'en' else ("🇪🇸 スペイン語" if lang == 'ja' else "🇪🇸 Испанский")),
        ("de", "🇩🇪 German" if lang == 'en' else ("🇩🇪 ドイツ語" if lang == 'ja' else "🇩🇪 Немецкий")),
        ("fr", "🇫🇷 French" if lang == 'en' else ("🇫🇷 フランス語" if lang == 'ja' else "🇫🇷 Французский")),
        ("ja", "🇯🇵 Japanese" if lang == 'en' else ("🇯🇵 日本語" if lang == 'ja' else "🇯🇵 Японский")),
    ]

    keyboard = [[InlineKeyboardButton(text=label, callback_data=f"set_learning_{code}")] for code, label in options]
    keyboard.append([InlineKeyboardButton(text=back, callback_data="settings_back")])
    return InlineKeyboardMarkup(inline_keyboard=keyboard)


@router.message(Command("settings"))
async def cmd_settings(message: Message):
    """Обработчик команды /settings"""
    async with async_session_maker() as session:
        user = await UserService.get_user_by_telegram_id(session, message.from_user.id)

        if not user:
            await message.answer("Сначала запусти бота командой /start")
            return

        lang = getattr(user, 'language_interface', 'ru')
        title = "⚙️ <b>Settings</b>\n\n" if lang == 'en' else ("⚙️ <b>設定</b>\n\n" if lang == 'ja' else "⚙️ <b>Настройки</b>\n\n")
        level_label = "📊 English level: " if lang == 'en' else ("📊 英語レベル: " if lang == 'ja' else "📊 Уровень английского: ")
        lang_label = "🌐 Interface language: " if lang == 'en' else ("🌐 インターフェース言語: " if lang == 'ja' else "🌐 Язык интерфейса: ")
        lang_value = 'English' if lang == 'en' else ('日本語' if lang == 'ja' else ('Русский' if user.language_interface == 'ru' else 'English'))
        footer = "Choose what to change:" if lang == 'en' else ("変更したい項目を選択:" if lang == 'ja' else "Выбери, что хочешь изменить:")
        settings_text = f"{title}{level_label}<b>{user.level.value}</b>\n{lang_label}<b>{lang_value}</b>\n\n{footer}"

        await message.answer(settings_text, reply_markup=get_settings_keyboard(user))


@router.callback_query(F.data == "settings_level")
async def settings_level(callback: CallbackQuery):
    """Показать выбор уровня"""
    async with async_session_maker() as session:
        user = await UserService.get_user_by_telegram_id(session, callback.from_user.id)
    lang = getattr(user, 'language_interface', 'ru') if user else 'ru'
    title = "📊 <b>Choose your English level:</b>\n\n" if lang == 'en' else ("📊 <b>英語レベルを選択:</b>\n\n" if lang == 'ja' else "📊 <b>Выбери свой уровень английского:</b>\n\n")
    body = (
        "<b>A1-A2</b> - Beginner\n<b>B1-B2</b> - Intermediate\n<b>C1-C2</b> - Advanced\n\n" if lang == 'en' else (
        "<b>A1-A2</b> - 初級\n<b>B1-B2</b> - 中級\n<b>C1-C2</b> - 上級\n\n" if lang == 'ja' else 
        "<b>A1-A2</b> - Начинающий\n<b>B1-B2</b> - Средний\n<b>C1-C2</b> - Продвинутый\n\n"
    ))
    tail = "This affects difficulty of suggested words and tasks." if lang == 'en' else ("これは提案される単語や課題の難易度に影響します。" if lang == 'ja' else "Это влияет на сложность предлагаемых слов и заданий.")
    await callback.message.edit_text(title + body + tail, reply_markup=get_level_keyboard(user))
    await callback.answer()


@router.callback_query(F.data == "settings_learning")
async def settings_learning(callback: CallbackQuery):
    """Показать выбор языка изучения"""
    async with async_session_maker() as session:
        user = await UserService.get_user_by_telegram_id(session, callback.from_user.id)
    lang = getattr(user, 'language_interface', 'ru') if user else 'ru'
    title = "🎯 <b>Select learning language:</b>\n\n" if lang == 'en' else ("🎯 <b>学習言語を選択:</b>\n\n" if lang == 'ja' else "🎯 <b>Выбери язык изучения:</b>\n\n")
    await callback.message.edit_text(title, reply_markup=get_learning_language_keyboard(user))
    await callback.answer()


@router.callback_query(F.data.startswith("set_learning_"))
async def set_learning_language(callback: CallbackQuery):
    """Установить язык изучения"""
    code = callback.data.split("_")[-1]
    async with async_session_maker() as session:
        user = await UserService.get_user_by_telegram_id(session, callback.from_user.id)
        if user:
            await UserService.update_user_learning_language(session, user.id, code)
            lang = getattr(user, 'language_interface', 'ru')
            if lang == 'en':
                text = f"✅ Learning language: <b>{code.upper()}</b>"
                back = "⬅️ Back to settings"
            elif lang == 'ja':
                text = f"✅ 学習言語: <b>{code.upper()}</b>"
                back = "⬅️ 設定に戻る"
            else:
                text = f"✅ Язык изучения: <b>{code.upper()}</b>"
                back = "⬅️ К настройкам"
            await callback.message.edit_text(
                text,
                reply_markup=InlineKeyboardMarkup(inline_keyboard=[[InlineKeyboardButton(text=back, callback_data="settings_back")]])
            )
    await callback.answer()


@router.callback_query(F.data.startswith("set_level_"))
async def set_level(callback: CallbackQuery):
    """Установить уровень"""
    level_str = callback.data.split("_")[-1]  # A1, A2, B1, B2, C1, C2

    async with async_session_maker() as session:
        user = await UserService.get_user_by_telegram_id(session, callback.from_user.id)

        if user:
            # Обновляем уровень
            await UserService.update_user_level(session, user.id, LanguageLevel[level_str])

            lang = getattr(user, 'language_interface', 'ru')
            changed = "✅ Level changed to " if lang == 'en' else ("✅ レベルが変更されました: " if lang == 'ja' else "✅ Уровень изменен на ")
            msg = changed + f"<b>{level_str}</b>\n\n" + ("You will now receive words and tasks matching your level!" if lang == 'en' else ("これからレベルに合った単語と課題が出題されます！" if lang == 'ja' else "Теперь ты будешь получать слова и задания, соответствующие твоему уровню!"))
            back = "⬅️ Back to settings" if lang == 'en' else ("⬅️ 設定に戻る" if lang == 'ja' else "⬅️ К настройкам")
            await callback.message.edit_text(
                msg,
                reply_markup=InlineKeyboardMarkup(inline_keyboard=[[InlineKeyboardButton(text=back, callback_data="settings_back")]])
            )

    await callback.answer()


@router.callback_query(F.data == "settings_language")
async def settings_language(callback: CallbackQuery):
    """Показать выбор языка"""
    async with async_session_maker() as session:
        user = await UserService.get_user_by_telegram_id(session, callback.from_user.id)
    lang = getattr(user, 'language_interface', 'ru') if user else 'ru'
    title = "🌐 <b>Select interface language:</b>\n\n" if lang == 'en' else ("🌐 <b>インターフェース言語を選択:</b>\n\n" if lang == 'ja' else "🌐 <b>Выбери язык интерфейса:</b>\n\n")
    desc = "This will change the language of bot messages." if lang == 'en' else ("ボットの表示言語が変更されます。" if lang == 'ja' else "Это изменит язык всех сообщений бота.")
    await callback.message.edit_text(
        title + desc,
        reply_markup=get_language_keyboard(user)
    )
    await callback.answer()


@router.callback_query(F.data.startswith("set_lang_"))
async def set_language(callback: CallbackQuery):
    """Установить язык"""
    lang = callback.data.split("_")[-1]  # ru | en | ja

    async with async_session_maker() as session:
        user = await UserService.get_user_by_telegram_id(session, callback.from_user.id)

        if user:
            await UserService.update_user_language(session, user.id, lang)
            if lang == 'en':
                text = "✅ Interface language: <b>English</b>"
                back = "⬅️ Back"
            elif lang == 'ja':
                text = "✅ インターフェース言語: <b>日本語</b>"
                back = "⬅️ 戻る"
            else:
                text = "✅ Язык интерфейса: <b>Русский</b>"
                back = "⬅️ К настройкам"
            await callback.message.edit_text(
                text,
                reply_markup=InlineKeyboardMarkup(inline_keyboard=[[InlineKeyboardButton(text=back, callback_data="settings_back")]])
            )
            # Обновляем клавиатуру чата на выбранный язык
            menu_updated = "Main menu updated ⤵️" if lang == 'en' else ("メインメニューを更新しました ⤵️" if lang == 'ja' else "Клавиатура обновлена ⤵️")
            await callback.message.answer(menu_updated, reply_markup=main_menu_keyboard(lang))

    await callback.answer()


@router.callback_query(F.data == "settings_back")
async def settings_back(callback: CallbackQuery):
    """Вернуться к настройкам"""
    async with async_session_maker() as session:
        user = await UserService.get_user_by_telegram_id(session, callback.from_user.id)

        if user:
            lang = getattr(user, 'language_interface', 'ru')
            title = "⚙️ <b>Settings</b>\n\n" if lang == 'en' else ("⚙️ <b>設定</b>\n\n" if lang == 'ja' else "⚙️ <b>Настройки</b>\n\n")
            level_label = "📊 English level: " if lang == 'en' else ("📊 英語レベル: " if lang == 'ja' else "📊 Уровень английского: ")
            lang_label = "🌐 Interface language: " if lang == 'en' else ("🌐 インターフェース言語: " if lang == 'ja' else "🌐 Язык интерфейса: ")
            lang_value = 'English' if user.language_interface == 'en' else ('日本語' if user.language_interface == 'ja' else 'Русский')
            footer = "Choose what to change:" if lang == 'en' else ("変更したい項目を選択:" if lang == 'ja' else "Выбери, что хочешь изменить:")
            settings_text = f"{title}{level_label}<b>{user.level.value}</b>\n{lang_label}<b>{lang_value}</b>\n\n{footer}"

            await callback.message.edit_text(settings_text, reply_markup=get_settings_keyboard(user))

    await callback.answer()


@router.callback_query(F.data == "settings_close")
async def settings_close(callback: CallbackQuery):
    """Закрыть настройки"""
    await callback.message.delete()
    await callback.answer()