tuxbot-bot/tuxbot/core/checks.py

from typing import Dict

import discord
from discord.ext import commands
from discord.ext.commands import (
    bot_has_permissions,
    has_permissions,
    is_owner,
)

from tuxbot.core.utils.functions.extra import ContextPlus

__all__ = [
    "bot_has_permissions",
    "has_permissions",
    "is_owner",
    "is_mod",
    "is_admin",
    "check_permissions",
    "guild_owner_or_permissions",
]


def is_mod():
    """Is the user a moderator ?"""

    async def pred(ctx):
        if await ctx.bot.is_owner(ctx.author):
            return True
        permissions: discord.Permissions = ctx.channel.permissions_for(
            ctx.author
        )
        return permissions.manage_messages

    return commands.check(pred)


def is_admin():
    """Is the user admin ?"""

    async def pred(ctx):
        if await ctx.bot.is_owner(ctx.author):
            return True
        permissions: discord.Permissions = ctx.channel.permissions_for(
            ctx.author
        )
        return permissions.administrator

    return commands.check(pred)


async def check_permissions(ctx: ContextPlus, **perms: Dict[str, bool]):
    """Does a user have any perms ?

    Parameters
    ----------
    ctx:ContextPlus
        Command context.
    **perms:dict
        Perms to verify.
    """
    if await ctx.bot.is_owner(ctx.author):
        return True

    if not perms:
        return False

    resolved = ctx.channel.permissions_for(ctx.author)

    return all(
        getattr(resolved, name, None) == value for name, value in perms.items()
    )


def guild_owner_or_permissions(**perms: Dict[str, bool]):
    """Is a user the guild's owner or does this user have any perms ?

    Parameters
    ----------
    **perms:dict
        Perms to verify.
    """

    async def pred(ctx):
        if ctx.author is ctx.guild.owner:
            return True
        return await check_permissions(ctx, **perms)

    return commands.check(pred)