from aiogram import Router, F
from aiogram.fsm.context import FSMContext
from aiogram.types import Message, CallbackQuery
from aiogram.fsm.state import State, StatesGroup
from sqlalchemy import select, insert, update
from models import User
from database import async_session
from keyboards import get_subscription_keyboard

router = Router()

# Машина состояний
class Register(StatesGroup):
    waiting_for_fullname = State()
    waiting_for_study_group = State()

@router.message(F.text == "/start")
async def start_handler(message: Message, state: FSMContext):
    user_id = message.from_user.id

    async with async_session() as session:
        result = await session.execute(select(User).where(User.user_id == user_id))
        user = result.scalar_one_or_none()

        if user:
            # Приветствие при повторном запуске
            text = f"👋 Привет, {user.full_name or 'друг'}!\n" \
                   f"Группа: {user.study_group or 'не указана'}\n" \
                   "Используй кнопки ниже, чтобы управлять подпиской:"
            await message.answer(text, reply_markup=get_subscription_keyboard(user.is_subscribed))
        else:
            # Первый запуск — регистрация
            await state.set_state(Register.waiting_for_fullname)
            await message.answer("👋 Привет! Пожалуйста, напиши свои ФИО одной строкой:")

@router.message(Register.waiting_for_fullname)
async def get_fullname(message: Message, state: FSMContext):
    await state.update_data(full_name=message.text)
    await state.set_state(Register.waiting_for_study_group)
    await message.answer("Теперь укажи номер своей группы (например, Б11-111):")

@router.message(Register.waiting_for_study_group)
async def get_study_group(message: Message, state: FSMContext):
    data = await state.get_data()
    full_name = data["full_name"]
    study_group = message.text

    async with async_session() as session:
        await session.execute(insert(User).values(
            user_id=message.from_user.id,
            full_name=full_name,
            study_group=study_group,
            is_subscribed=True
        ))
        await session.commit()

    await message.answer(f"✅ Регистрация завершена!\nФИО: {full_name}\nГруппа: {study_group}")
    await message.answer("Теперь ты можешь управлять подпиской ниже 👇", reply_markup=get_subscription_keyboard(True))
    await state.clear()