import logging
from aiogram import Bot, Dispatcher, types
from aiogram.contrib.fsm_storage.memory import MemoryStorage
from aiogram.dispatcher import FSMContext
from aiogram.dispatcher.filters import Command

from aiogram.types import InlineKeyboardMarkup, InlineKeyboardButton
from aiogram import executor

# Configure logging
logging.basicConfig(level=logging.INFO)

# Initialize bot and dispatcher
bot = Bot(token="5648389310:AAEofKzl4hqnZI6FxUeEH_Qab6br0uypApQ")
storage = MemoryStorage()
dp = Dispatcher(bot, storage=storage)


class ApplicationForm(StatesGroup):
    question1 = State()
    question2 = State()
    question3 = State()


@dp.message_handler(Command("start"))
async def start(message: types.Message):
    """
    Handler for the /start command. Initiates the application process.
    """
    await ApplicationForm.question1.set()
    await message.reply("Welcome to the application process! Please answer the following questions one by one.")

    # Ask the first question
    await message.reply("Question 1: What is your name?")


@dp.message_handler(state=ApplicationForm.question1)
async def process_question1(message: types.Message, state: FSMContext):
    """
    Handler for processing the answer to question 1.
    """
    # Save the answer to question 1 in the state
    await state.update_data(question1=message.text)

    # Ask the second question
    await ApplicationForm.next()
    await message.reply("Question 2: What is your age?")


@dp.message_handler(state=ApplicationForm.question2)
async def process_question2(message: types.Message, state: FSMContext):
    """
    Handler for processing the answer to question 2.
    """
    # Save the answer to question 2 in the state
    await state.update_data(question2=message.text)

    # Ask the third question
    await ApplicationForm.next()
    await message.reply("Question 3: Please provide the Telegram ID of the admin who invited you.")


@dp.message_handler(state=ApplicationForm.question3)
async def process_question3(message: types.Message, state: FSMContext):
    """
    Handler for processing the answer to question 3.
    """
    # Save the answer to question 3 in the state
    await state.update_data(question3=message.text)

    # Get the user's answers from the state
    user_data = await state.get_data()

    # Create the application message with the user's answers
    application_message = f"New application:\n\n" \
                          f"Question 1: {user_data['question1']}\n" \
                          f"Question 2: {user_data['question2']}\n" \
                          f"Question 3: {user_data['question3']}"

    # Create the inline keyboard with accept and reject buttons
    keyboard = InlineKeyboardMarkup(row_width=2)
    accept_button = InlineKeyboardButton("Accept", callback_data="accept")
    reject_button = InlineKeyboardButton("Reject", callback_data="reject")
    keyboard.add(accept_button, reject_button)

    # Send the application message to the admin chat with the inline keyboard
    await bot.send_message(chat_id="-1002103490242", text=application_message, reply_markup=keyboard)

    # Finish the application process
    await state.finish()
    await message.reply("Thank you for your application! It has been sent to the admins for review.")


@dp.callback_query_handler(lambda c: c.data == "accept" or c.data == "reject")
async def process_application_decision(callback_query: types.CallbackQuery):
    """
    Handler for processing the application decision (accept or reject).
    """
    # Get the decision from the callback data
    decision = callback_query.data

    # Send the decision message to the user
    if decision == "accept":
        await bot.send_message(chat_id=callback_query.from_user.id, text="Congratulations! Your application has been accepted.")
    else:
        await bot.send_message(chat_id=callback_query.from_user.id, text="Sorry, your application has been rejected.")

    # Answer the callback query
    await callback_query.answer()


if __name__ == "__main__":
    # Start the bot
    executor.start_polling(dp)