import asyncio
import os
from asyncio import Queue, sleep
from dataclasses import dataclass
from functools import cached_property
from http import HTTPStatus

from fastapi import Request, Response
from telegram import Update
from telegram.ext import Application, CommandHandler, ContextTypes

from app.constants import API_PREFIX
from settings.config import Settings


class Bot:
    def __init__(self, settings: Settings) -> None:
        self.application: Application = (  # type: ignore
            Application.builder().token(token=settings.TELEGRAM_API_TOKEN).build()
        )
        self.add_handlers()
        self.settings = settings

    async def set_webhook(self) -> None:
        await self.application.initialize()
        # await self.application.start()
        await self.application.bot.set_webhook(url=self.webhook_url)

    async def delete_webhook(self) -> None:
        await self.application.bot.delete_webhook()

    @staticmethod
    async def help_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
        """Send a message when the command /help is issued."""
        if update.message:
            await update.message.reply_text('Help!')
        return None

    def add_handlers(self) -> None:
        self.application.add_handler(CommandHandler('help', self.help_command))

    async def polling(self) -> None:
        await self.application.initialize()
        await self.application.start()
        await self.application.updater.start_polling()  # type: ignore

    async def shutdown(self) -> None:
        await self.application.updater.shutdown()  # type: ignore

    @cached_property
    def webhook_url(self) -> str:
        return os.path.join(
            self.settings.WEBHOOK_HOST.strip('/'),
            API_PREFIX.strip('/'),
            self.settings.WEBHOOK_PATH.strip('/'),
            self.settings.TELEGRAM_API_TOKEN.strip('/'),
        )


@dataclass
class BotQueue:
    bot: Bot
    queue: Queue = asyncio.Queue()  # type: ignore

    async def put_updates_on_queue(self, request: Request) -> Response:
        """
        Listen {WEBHOOK_PATH}/{TELEGRAM_WEB_TOKEN} path and proxy post request to bot
        """
        data = await request.json()
        tg_update = Update.de_json(data=data, bot=self.bot.application.bot)
        self.queue.put_nowait(tg_update)

        return Response(status_code=HTTPStatus.ACCEPTED)

    async def get_updates_from_queue(self) -> None:
        while True:
            update = await self.queue.get()
            await self.bot.application.process_update(update)
            await sleep(0.1)