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

from database.db import async_session_maker
from services.user_service import UserService
from services.task_service import TaskService
from services.ai_service import ai_service
from utils.i18n import t

router = Router()


class TaskStates(StatesGroup):
    """Состояния для прохождения заданий"""
    doing_tasks = State()
    waiting_for_answer = State()


@router.message(Command("task"))
async def cmd_task(message: Message, state: FSMContext):
    """Обработчик команды /task"""
    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(t('ru', 'common.start_first'))
            return

        # Генерируем задания разных типов
        tasks = await TaskService.generate_mixed_tasks(
            session, user.id, count=5,
            learning_lang=user.learning_language,
            translation_lang=user.language_interface,
        )

        if not tasks:
            await message.answer(t(user.level.value and (user.language_interface or 'ru') or 'ru', 'tasks.no_words'))
            return

        # Сохраняем задания в состоянии
        await state.update_data(
            tasks=tasks,
            current_task_index=0,
            correct_count=0,
            user_id=user.id
        )
        await state.set_state(TaskStates.doing_tasks)

        # Показываем первое задание
        await show_current_task(message, state)


async def show_current_task(message: Message, state: FSMContext):
    """Показать текущее задание"""
    data = await state.get_data()
    tasks = data.get('tasks', [])
    current_index = data.get('current_task_index', 0)

    if current_index >= len(tasks):
        # Все задания выполнены
        await finish_tasks(message, state)
        return

    task = tasks[current_index]

    # Определяем язык пользователя
    async with async_session_maker() as session:
        user = await UserService.get_user_by_telegram_id(session, message.from_user.id)
    lang = (user.language_interface if user else 'ru') or 'ru'

    task_text = (
        t(lang, 'tasks.header', i=current_index + 1, n=len(tasks)) + "\n\n" +
        f"{task['question']}\n"
    )

    if task.get('transcription'):
        task_text += f"🔊 [{task['transcription']}]\n"

    task_text += t(lang, 'tasks.write_answer')

    await state.set_state(TaskStates.waiting_for_answer)
    await message.answer(task_text)


@router.message(TaskStates.waiting_for_answer)
async def process_answer(message: Message, state: FSMContext):
    """Обработка ответа пользователя"""
    user_answer = message.text.strip()
    data = await state.get_data()

    tasks = data.get('tasks', [])
    current_index = data.get('current_task_index', 0)
    correct_count = data.get('correct_count', 0)
    user_id = data.get('user_id')

    task = tasks[current_index]

    # Показываем индикатор проверки
    # Язык пользователя
    async with async_session_maker() as session:
        user = await UserService.get_user_by_telegram_id(session, message.from_user.id)
    lang = (user.language_interface if user else 'ru') or 'ru'

    checking_msg = await message.answer(t(lang, 'tasks.checking'))

    # Проверяем ответ через AI
    check_result = await ai_service.check_answer(
        question=task['question'],
        correct_answer=task['correct_answer'],
        user_answer=user_answer
    )

    await checking_msg.delete()

    is_correct = check_result.get('is_correct', False)
    feedback = check_result.get('feedback', '')

    # Формируем ответ
    if is_correct:
        result_text = t(lang, 'tasks.correct') + "\n\n"
        correct_count += 1
    else:
        result_text = t(lang, 'tasks.incorrect') + "\n\n"

    result_text += f"{t(lang, 'tasks.your_answer')}: <i>{user_answer}</i>\n"
    result_text += f"{t(lang, 'tasks.right_answer')}: <b>{task['correct_answer']}</b>\n\n"

    if feedback:
        result_text += f"💬 {feedback}\n\n"

    # Сохраняем результат в БД
    async with async_session_maker() as session:
        await TaskService.save_task_result(
            session=session,
            user_id=user_id,
            task_type=task['type'],
            content={
                'question': task['question'],
                'word': task['word']
            },
            user_answer=user_answer,
            correct_answer=task['correct_answer'],
            is_correct=is_correct,
            ai_feedback=feedback
        )

        # Обновляем статистику слова
        if 'word_id' in task:
            await TaskService.update_word_statistics(
                session=session,
                word_id=task['word_id'],
                is_correct=is_correct
            )

    # Обновляем счетчик
    await state.update_data(
        current_task_index=current_index + 1,
        correct_count=correct_count
    )

    # Показываем результат и кнопку "Далее"
    keyboard = InlineKeyboardMarkup(inline_keyboard=[
        [InlineKeyboardButton(text=t(lang, 'tasks.next_btn'), callback_data="next_task")],
        [InlineKeyboardButton(text=t(lang, 'tasks.stop_btn'), callback_data="stop_tasks")]
    ])

    await message.answer(result_text, reply_markup=keyboard)
    # После показа результата ждём нажатия кнопки – переключаемся в состояние doing_tasks
    await state.set_state(TaskStates.doing_tasks)


@router.callback_query(F.data == "next_task", TaskStates.doing_tasks)
async def next_task(callback: CallbackQuery, state: FSMContext):
    """Переход к следующему заданию"""
    await callback.message.delete()
    await show_current_task(callback.message, state)
    await callback.answer()


@router.callback_query(F.data == "stop_tasks", TaskStates.doing_tasks)
async def stop_tasks_callback(callback: CallbackQuery, state: FSMContext):
    """Остановить выполнение заданий через кнопку"""
    await state.clear()
    await callback.message.edit_reply_markup(reply_markup=None)
    async with async_session_maker() as session:
        user = await UserService.get_user_by_telegram_id(session, callback.from_user.id)
    lang = (user.language_interface if user else 'ru') or 'ru'
    await callback.message.answer(t(lang, 'tasks.finished'))
    await callback.answer()


@router.message(Command("stop"), TaskStates.doing_tasks)
@router.message(Command("stop"), TaskStates.waiting_for_answer)
async def stop_tasks(message: Message, state: FSMContext):
    """Остановить выполнение заданий командой /stop"""
    await state.clear()
    # Определяем язык пользователя
    async with async_session_maker() as session:
        user = await UserService.get_user_by_telegram_id(session, message.from_user.id)
    await message.answer(t((user.language_interface if user else 'ru') or 'ru', 'tasks.stopped'))


@router.message(Command("cancel"), TaskStates.doing_tasks)
@router.message(Command("cancel"), TaskStates.waiting_for_answer)
async def cancel_tasks(message: Message, state: FSMContext):
    """Отмена выполнения заданий командой /cancel"""
    await state.clear()
    async with async_session_maker() as session:
        user = await UserService.get_user_by_telegram_id(session, message.from_user.id)
    lang = (user.language_interface if user else 'ru') or 'ru'
    await message.answer(t(lang, 'tasks.cancelled'))


async def finish_tasks(message: Message, state: FSMContext):
    """Завершение всех заданий"""
    data = await state.get_data()
    tasks = data.get('tasks', [])
    correct_count = data.get('correct_count', 0)
    total_count = len(tasks)

    accuracy = int((correct_count / total_count) * 100) if total_count > 0 else 0

    # Определяем эмодзи на основе результата
    if accuracy >= 90:
        emoji = "🏆"
        comment_key = 'excellent'
    elif accuracy >= 70:
        emoji = "👍"
        comment_key = 'good'
    elif accuracy >= 50:
        emoji = "📚"
        comment_key = 'average'
    else:
        emoji = "💪"
        comment_key = 'poor'

    # Язык пользователя
    async with async_session_maker() as session:
        user = await UserService.get_user_by_telegram_id(session, message.from_user.id)
    lang = (user.language_interface if user else 'ru') or 'ru'

    result_text = (
        t(lang, 'tasks.finish_title', emoji=emoji) + "\n\n" +
        t(lang, 'tasks.correct_of', correct=correct_count, total=total_count) + "\n" +
        t(lang, 'tasks.accuracy', accuracy=accuracy) + "\n\n" +
        t(lang, f"tasks.comment.{comment_key}") + "\n\n" +
        t(lang, 'tasks.use_task') + "\n" +
        t(lang, 'tasks.use_stats')
    )

    await state.clear()
    await message.answer(result_text)


@router.message(Command("stats"))
async def cmd_stats(message: Message):
    """Обработчик команды /stats"""
    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(t('ru', 'common.start_first'))
            return

        # Получаем статистику
        stats = await TaskService.get_user_stats(session, user.id)
        lang = (user.language_interface if user else 'ru') or 'ru'

        stats_text = (
            t(lang, 'stats.header') + "\n\n" +
            t(lang, 'stats.total_words', n=stats['total_words']) + "\n" +
            t(lang, 'stats.studied_words', n=stats['reviewed_words']) + "\n" +
            t(lang, 'stats.total_tasks', n=stats['total_tasks']) + "\n" +
            t(lang, 'stats.correct_tasks', n=stats['correct_tasks']) + "\n" +
            t(lang, 'stats.accuracy', n=stats['accuracy']) + "\n\n"
        )

        if stats['total_words'] == 0:
            stats_text += t(lang, 'stats.hint_add_words')
        elif stats['total_tasks'] == 0:
            stats_text += t(lang, 'stats.hint_first_task')
        else:
            stats_text += t(lang, 'stats.hint_keep_practice')

        await message.answer(stats_text)