yt-dlp-bot/bot/core/callbacks.py

import logging

from pyrogram.enums import ParseMode
from pyrogram.types import Message

from core.bot import VideoBot
from core.service import URLService
from core.utils import bold
from yt_shared.emoji import SUCCESS_EMOJI
from yt_shared.schemas.url import URL


class TelegramCallback:
    _MSG_SEND_OK = f'{SUCCESS_EMOJI} {bold("URL sent for download")}'
    _MSG_SEND_FAIL = f'🛑 {bold("Failed to send URL for download")}'

    def __init__(self) -> None:
        self._log = logging.getLogger(self.__class__.__name__)
        self._url_service = URLService()

    async def on_message(self, client: VideoBot, message: Message) -> None:
        """Receive video URL and send to the download worker."""
        url = URL(
            url=message.text, from_user_id=message.from_user.id, message_id=message.id
        )
        is_sent = await self._url_service.process_url(url)
        await message.reply(
            self._MSG_SEND_OK.format(url=url.url) if is_sent else self._MSG_SEND_FAIL,
            parse_mode=ParseMode.HTML,
            reply_to_message_id=message.id,
        )