from functools import wraps

from telegram_downloader_bot.logger import log
from telegram_downloader_bot.settings import settings

allowed_ids = settings.allowed_ids.split(",")
allowed_ids = [int(x) for x in allowed_ids]


def protected(func):
    @wraps(func)
    async def wrapper(client, message):
        if int(message.from_user.id) not in allowed_ids:
            log.warning(
                f"User with ID {message.from_user.id} attempted"
                "to text this bot!")
            log.info(
                f"Only users allowed are: {' '.join(allowed_ids)}")
            return await message.reply_text("You are not on the list!")
        return await func(client, message)
    return wrapper