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

import logging

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

from core.service import URLService
from core.utils import bold
from yt_shared.emoji import SUCCESS_EMOJI


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: Client, message: Message) -> None:
        """Receive video URL and send to the download worker."""
        is_sent = await self._url_service.process_url(message.text, message.id)
        await message.reply(
            self._MSG_SEND_OK if is_sent else self._MSG_SEND_FAIL,
            parse_mode=ParseMode.HTML,
            reply_to_message_id=message.id,
        )