import asyncio
import os

import aiohttp
from aiogram import Bot, Dispatcher, F, Router
from aiogram.filters import CommandStart
from aiogram.types import CallbackQuery, InlineKeyboardButton, InlineKeyboardMarkup, Message

BOT_TOKEN = os.getenv("BOT_TOKEN")
GRASSFLOW_API_KEY = os.getenv("GRASSFLOW_API_KEY")
GRASSFLOW_API_URL = "https://grassflow.ru/"

bot = Bot(token=BOT_TOKEN)
dp = Dispatcher()
router = Router()
pending_links: dict[int, list[str]] = {}

async def grassflow_request(session, method, path, payload=None):
    async with session.request(
        method,
        f"{GRASSFLOW_API_URL}{path}",
        json=payload,
        headers={
            "Authorization": f"Bearer {GRASSFLOW_API_KEY}",
            "Content-Type": "application/json",
        },
    ) as response:
        data = await response.json(content_type=None)
        if response.status >= 400:
            raise RuntimeError(data.get("message") or f"GrassFlow error {response.status}")
        return data

def sponsors_keyboard(sponsors):
    buttons = [[InlineKeyboardButton(text="Выполнить задание", url=item["link"])] for item in sponsors]
    buttons.append([InlineKeyboardButton(text="Проверить", callback_data="grassflow:check")])
    return InlineKeyboardMarkup(inline_keyboard=buttons)

@router.message(CommandStart())
async def start_handler(message: Message, session: aiohttp.ClientSession):
    tasks = await grassflow_request(session, "POST", "/sponsors", {
        "user_id": message.from_user.id,
        "chat_id": message.chat.id,
        "max_sponsors": 5,
    })
    sponsors = tasks.get("sponsors") or []
    if not sponsors:
        await message.answer("Сейчас нет доступных заданий.")
        return
    pending_links[message.from_user.id] = [item["link"] for item in sponsors]
    await message.answer("Подпишитесь на спонсоров, затем нажмите Проверить.", reply_markup=sponsors_keyboard(sponsors))

@router.callback_query(F.data == "grassflow:check")
async def check_handler(callback: CallbackQuery, session: aiohttp.ClientSession):
    links = pending_links.get(callback.from_user.id, [])
    result = await grassflow_request(session, "POST", "/sponsors/check", {"user_id": callback.from_user.id, "links": links})
    done = result.get("sponsors") or []
    if done and all(item.get("status") == "subscribed" for item in done):
        pending_links.pop(callback.from_user.id, None)
        await callback.message.answer("Все задания выполнены. Доступ открыт.")
    else:
        await callback.message.answer("Не все задания выполнены. Попробуйте ещё раз.")
    await callback.answer()

async def main():
    async with aiohttp.ClientSession(timeout=aiohttp.ClientTimeout(total=15)) as session:
        dp.workflow_data["session"] = session
        dp.include_router(router)
        await dp.start_polling(bot)

if __name__ == "__main__":
    asyncio.run(main())