import logging
import os
from aiogram import Bot, Dispatcher, types
from aiogram.contrib.fsm_storage.memory import MemoryStorage
from aiogram.dispatcher import FSMContext
from aiogram.dispatcher.filters import Text
from aiogram.dispatcher.filters.state import State, StatesGroup
from aiogram.types import ParseMode
from aiogram.utils import executor
from dotenv import load_dotenv
from pymongo import MongoClient


# Загружаем переменные окружения
load_dotenv()

# Получаем токен бота из переменных окружения
BOT_TOKEN = os.getenv("6188948983:AAEnk6wVvLJlxrCl3iY26X-_q0RDhdIkthM")

# Создаем подключение к MongoDB
client = MongoClient("MongoDB//localhotst:27017/")
db = client["service_center"]
orders_collection = db["orders"]

# Настраиваем логирование
logging.basicConfig(level=logging.INFO)

# Инициализируем бота и диспетчер
bot = Bot(token=BOT_TOKEN)
dp = Dispatcher(bot, storage=MemoryStorage())


# Определяем состояния для FSM
class OrderStates(StatesGroup):
    waiting_for_full_name = State()
    waiting_for_phone_number = State()
    waiting_for_issue = State()


# Обрабатываем команду /start
@dp.message_handler(commands=["start"])
async def cmd_start(message: types.Message):
    await message.answer(
        "Привет! Я бот сервисного центра по ремонту телефонов. Чтобы записаться на ремонт, отправь мне свои данные.\n"
        "Напиши свою фамилию и имя:"
    )

    # Переводим пользователя в состояние "waiting_for_full_name"
    await OrderStates.waiting_for_full_name.set()


# Обрабатываем ответ с фамилией и именем
@dp.message_handler(state=OrderStates.waiting_for_full_name)
async def process_full_name(message: types.Message, state: FSMContext):
    # Сохраняем фамилию и имя в контексте
    async with state.proxy() as data:
        data["full_name"] = message.text

    await message.answer("Отлично! Теперь напиши свой номер телефона:")

    # Переводим пользователя в состояние "waiting_for_phone_number"
    await OrderStates.waiting_for_phone_number.set()


# Обрабатываем ответ с номером телефона
@dp.message_handler(state=OrderStates.waiting_for_phone_number)
async def process_phone_number(message: types.Message, state: FSMContext):
    # Сохраняем номер телефона в контексте
    async with state.proxy() as data:
        data["phone_number"] = message.text

    await message.answer("Опиши проблему со своим телефоном:")

    # Переводим пользователя в состояние "waiting_for_issue"
    await OrderStates.waiting_for_issue.set()


# Обрабатываем ответ с описанием проблемы
@dp.message_handler(state=OrderStates.waiting_for_issue)
async def process_issue(message: types.Message, state: FSMContext):
    # Сохраняем описание проблемы в контексте
    async with state.proxy() as data:
        data["issue"] = message.text

        # Записываем данные клиента в базу данных
        orders_collection.insert_one({
            "full_name": data["full_name"],
            "phone_number": data["phone_number"],
            "issue": data["issue"]
        })

    await message.answer(
        "Спасибо за заказ! Мы свяжемся с вами в ближайшее время."
    )

    # Завершаем FSM и сбрасываем состояние
    await state.finish()


# Обрабатываем ошибки FSM
@dp.errors_handler(exception=FSMContext.WaitingForActivationError)
async def waiting_for_activation_error_handler(update, exception):
    logging.error(f"FSM waiting for activation error: {exception}")
    await bot.send_message(chat_id=update.message.chat_id, text="Произошла ошибка. Попробуйте еще раз.")


# Запускаем бота
if name == "__main__":
    executor.start_polling(dp, skip_updates=True)