from config import Config
import aiohttp
from aiogram.types import BufferedInputFile
from utils.antispam import admin_required
from aiogram import Dispatcher, Bot
from aiogram.types import Message
from models.state import BotState
from aiogram.filters import Command
from logging import getLogger
from aiogram.types import PollAnswer
from storage.message_storage import save_message

logger = getLogger(__name__)
API_URL = "http://127.0.0.1:7700/speak"

from config import Config
import aiohttp
from aiogram.types import BufferedInputFile
from utils.antispam import admin_required
from aiogram import Dispatcher, Bot
from aiogram.types import Message
from models.state import BotState
from aiogram.filters import Command
from logging import getLogger
from aiogram.types import PollAnswer
from storage.message_storage import save_message

logger = getLogger(__name__)
API_URL = "http://127.0.0.1:7700/speak"

def register_handlers(dp: Dispatcher, state: BotState, bot: Bot):
    @dp.message(Command("poll"))
    @admin_required(5)
    async def send_poll(message: Message):
        for chat_id in Config.CHAT_IDS:
            try:
                poll_msg = await bot.send_poll(
                    chat_id=chat_id,
                    question="Кто опоздает?",
                    options=["Я", "Не знаю", "Наверное"],
                    is_anonymous=False,
                    allows_multiple_answers=False
                )
                # сохраняем сам опрос
                save_message(poll_msg.chat.id, poll_msg.message_id)
                logger.info(f"Опрос отправлен в чат {chat_id}")

            except Exception as e:
                logger.error(f"Ошибка при отправке в чат {chat_id}: {e}")

        confirm_msg = await message.answer("✅ Сообщение отправлено.")
        save_message(confirm_msg.chat.id, confirm_msg.message_id)

    @dp.poll_answer()
    async def handle_poll_answer(poll_answer: PollAnswer):
        user = poll_answer.user
        option_ids = poll_answer.option_ids

        # username или fallback на имя
        username = f"@{user.username}" if user.username else user.first_name

        # всегда пишем в первый чат из Config.CHAT_IDS

        if option_ids and option_ids[0] == 0:
            msg = await bot.send_message(
                chat_id=6394047531,
                text=f"{username} опоздает"
            )
            save_message(msg.chat.id, msg.message_id)
        else:
            msg = await bot.send_message(
                chat_id=6394047531,
                text=f"{username} выбрал вариант {option_ids}"
            )
            save_message(msg.chat.id, msg.message_id)