import asyncio
import logging
import sys
from os import getenv

from aiogram import Bot, Dispatcher, html, F
from aiogram.client.default import DefaultBotProperties
from aiogram.enums import ParseMode
from aiogram.filters import CommandStart, Command
from aiogram.types import Message, InlineKeyboardButton, CallbackQuery
from aiogram.fsm.context import FSMContext
from aiogram.fsm.state import State, StatesGroup
from aiogram.utils.keyboard import InlineKeyboardBuilder
from dotenv import load_dotenv

from database import Database

load_dotenv()
TOKEN = getenv("BOT_TOKEN")
dp = Dispatcher()
db = Database("users.db")

# Добавили состояние для ручного ввода сферы
class Registration(StatesGroup):
    waiting_for_sphere = State()
    waiting_for_custom_sphere = State() # <-- Новое состояние
    waiting_for_language = State()
    waiting_for_preferences = State()

@dp.message(CommandStart())
async def command_start_handler(message: Message) -> None:
    builder = InlineKeyboardBuilder()
    builder.row(InlineKeyboardButton(text="✅ Подписаться", callback_data="subscribe"))
    builder.row(InlineKeyboardButton(text="📄 Читать оферту", url="https://telegra.ph/Polzovatelskoe-soglashenie-i-Oferta-qwork-parse-bot-03-28"))

    text = (
        "<b>👋 Привет! Я твой персональный агент по Kwork.</b>\n\n"
        "💻 ⚠️ <b>ВАЖНО:</b> Этот бот предназначен <b>исключительно для IT-специалистов</b>.\n\n"
        "🔍 Я мониторю биржу 24/7 и мгновенно присылаю тебе свежие заказы.\n\n"
        "Нажимая кнопку «Подписаться», вы принимаете условия "
        "<a href='https://telegra.ph/Polzovatelskoe-soglashenie-i-Oferta-qwork-parse-bot-03-28'>публичной оферты</a>."
    )

    await message.answer(text, reply_markup=builder.as_markup(), disable_web_page_preview=True)

@dp.message(Command("profile"))
async def show_profile(message: Message):
    user_data = await db.get_user(message.from_user.id)
    
    if not user_data or user_data[0] is None:
        await message.answer("⚠️ Твой профиль еще не настроен. Нажми /start, чтобы начать.")
        return

    sphere, lang, prefs = user_data
    
    builder = InlineKeyboardBuilder()
    builder.row(InlineKeyboardButton(text="📝 Редактировать профиль", callback_data="subscribe")) # Используем тот же callback

    text = (
        "<b>👤 Твой профиль IT-специалиста:</b>\n\n"
        f"<b>🌐 Сфера:</b> {sphere}\n"
        f"<b>🛠 Стек:</b> {lang}\n"
        f"<b>⚙️ Предпочтения:</b> {prefs}\n\n"
        "Хочешь что-то изменить? Нажми кнопку ниже."
    )
    
    await message.answer(text, reply_markup=builder.as_markup())

@dp.callback_query(F.data == "subscribe")
async def subscribe_handler(callback: CallbackQuery, state: FSMContext):
    await db.add_user(callback.from_user.id)
    
    builder = InlineKeyboardBuilder()
    spheres = ["Backend", "Frontend", "Mobile", "DevOps", "Design", "QA"]
    for sphere in spheres:
        builder.add(InlineKeyboardButton(text=sphere, callback_data=f"sphere_{sphere}"))
    
    # Добавляем кнопку своего варианта
    builder.row(InlineKeyboardButton(text="⌨️ Свой вариант", callback_data="sphere_other"))
    builder.adjust(2)

    await callback.message.edit_text(
        "Отлично! Давай настроим профиль.\n<b>В какой сфере IT ты работаешь?</b>",
        reply_markup=builder.as_markup()
    )
    await state.set_state(Registration.waiting_for_sphere)
    await callback.answer()

@dp.callback_query(Registration.waiting_for_sphere)
async def sphere_chosen(callback: CallbackQuery, state: FSMContext):
    sphere = callback.data.split("_")[1]

    if sphere == "other":
        await callback.message.edit_text("Напиши свою сферу деятельности (например: <i>Data Science</i> или <i>GameDev</i>):")
        await state.set_state(Registration.waiting_for_custom_sphere)
    else:
        await state.update_data(sphere=sphere)
        await callback.message.edit_text(
            f"Выбрано: <b>{sphere}</b>\n\nКакой основной <b>язык программирования</b> или стек технологий используешь?"
        )
        await state.set_state(Registration.waiting_for_language)
    
    await callback.answer()

# Обработчик для текстового ввода своей сферы
@dp.message(Registration.waiting_for_custom_sphere)
async def custom_sphere_input(message: Message, state: FSMContext):
    sphere = message.text
    await state.update_data(sphere=sphere)
    await message.answer(
        f"Принято: <b>{sphere}</b>\n\nКакой основной <b>язык программирования</b> или стек технологий используешь?"
    )
    await state.set_state(Registration.waiting_for_language)

@dp.message(Registration.waiting_for_language)
async def language_chosen(message: Message, state: FSMContext):
    await state.update_data(language=message.text)
    
    # Создаем кнопку для пропуска
    builder = InlineKeyboardBuilder()
    builder.row(InlineKeyboardButton(text="⏩ Пропустить", callback_data="skip_preferences"))
    
    await message.answer(
        "Принято! И последнее: напиши свои <b>предпочтения по заказам</b> (фильтры).\n"
        "Например: 'чек от 5000р' или 'без правок'.\n\n"
        "<i>Если не хочешь заполнять сейчас, нажми кнопку ниже.</i>",
        reply_markup=builder.as_markup()
    )
    await state.set_state(Registration.waiting_for_preferences)

@dp.callback_query(Registration.waiting_for_preferences, F.data == "skip_preferences")
async def skip_preferences(callback: CallbackQuery, state: FSMContext):
    await state.update_data(preferences="Не указано") # Устанавливаем значение по умолчанию
    user_data = await state.get_data()
    
    await db.update_user_data(callback.from_user.id, user_data)
    
    await callback.message.edit_text(
        "✅ <b>Профиль успешно настроен!</b> (Фильтры пропущены)\n\n"
        f"Сфера: {user_data['sphere']}\n"
        f"Стек: {user_data['language']}\n"
        f"Фильтры: {user_data['preferences']}"
    )
    await state.clear()
    await callback.answer()

async def on_startup():
    await db.create_tables()
    print("Database ready")

async def main() -> None:
    bot = Bot(token=TOKEN, default=DefaultBotProperties(parse_mode=ParseMode.HTML))
    dp.startup.register(on_startup)
    await dp.start_polling(bot)

if __name__ == "__main__":
    logging.basicConfig(level=logging.INFO, stream=sys.stdout)
    asyncio.run(main())