from aiogram.filters import BaseFilter
from aiogram.types import CallbackQuery, Message

from bot.config import ADMIN_ID


class AdminFilter(BaseFilter):
    async def __call__(self, message: Message) -> bool:
        return message.from_user.id == ADMIN_ID  # type: ignore[union-attr]


class AdminCallbackFilter(BaseFilter):
    async def __call__(self, callback: CallbackQuery) -> bool:
        return callback.from_user.id == ADMIN_ID