feat: Add member and channel permissions calculation functions (#969)

* feat: added member and channel permissions functions

* ci: correct from checks.

* docs: update docs

* refactor: switch from ALL_PERMISSIONS to Permissions.ALL & DEFAULT

* feat: add default permissions if in DM

* fix: small fixes

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

Author: mAxYoLo01

interactions/api/models/channel.py

@@ -11,6 +11,7 @@
     define,
     field,
 )
+from .flags import Permissions
 from .misc import File, IDMixin, Overwrite, Snowflake
 from .user import User
 from .webhook import Webhook
@@ -1207,6 +1208,67 @@ async def join(self) -> None:
 
         await self._client.join_thread(int(self.id))
 
+    async def get_permissions_for(self, member: "Member") -> Permissions:
+        """
+        Returns the permissions of the member in this specific channel.
+
+        .. note::
+            The permissions returned by this function take into account role and
+            user overwrites that can be assigned to channels or categories. If you
+            don't need these overwrites, look into :meth:`.Member.get_guild_permissions`.
+
+        :param member: The member to get the permissions from
+        :type member: Member
+        :return: Permissions of the member in this channel
+        :rtype: Permissions
+        """
+        if not self.guild_id:
+            return Permissions.DEFAULT
+
+        from .guild import Guild
+
+        guild = Guild(**await self._client.get_guild(int(self.guild_id)), _client=self._client)
+
+        permissions = await member.get_guild_permissions(guild)
+
+        if permissions & Permissions.ADMINISTRATOR == Permissions.ADMINISTRATOR:
+            return Permissions.ALL
+
+        # @everyone role overwrites
+        from ...client.models.utils import search_iterable
+
+        overwrite_everyone = search_iterable(
+            self.permission_overwrites, lambda overwrite: int(overwrite.id) == int(self.guild_id)
+        )
+        if overwrite_everyone:
+            permissions &= ~int(overwrite_everyone[0].deny)
+            permissions |= int(overwrite_everyone[0].allow)
+
+        # Apply role specific overwrites
+        allow, deny = 0, 0
+        for role_id in member.roles:
+            overwrite_role = search_iterable(
+                self.permission_overwrites, lambda overwrite: int(overwrite.id) == int(role_id)
+            )
+            if overwrite_role:
+                allow |= int(overwrite_role[0].allow)
+                deny |= int(overwrite_role[0].deny)
+
+        if deny:
+            permissions &= ~deny
+        if allow:
+            permissions |= allow
+
+        # Apply member specific overwrites
+        overwrite_member = search_iterable(
+            self.permission_overwrites, lambda overwrite: int(overwrite.id) == int(member.id)
+        )
+        if overwrite_member:
+            permissions &= ~int(overwrite_member[0].deny)
+            permissions |= int(overwrite_member[0].allow)
+
+        return Permissions(permissions)
+
 
 @define()
 class Thread(Channel):

interactions/api/models/flags.py

@@ -93,6 +93,53 @@ class Permissions(IntFlag):
     START_EMBEDDED_ACTIVITIES = 1 << 39
     MODERATE_MEMBERS = 1 << 40
 
+    DEFAULT = (
+        ADD_REACTIONS
+        | VIEW_CHANNEL
+        | SEND_MESSAGES
+        | EMBED_LINKS
+        | ATTACH_FILES
+        | READ_MESSAGE_HISTORY
+        | MENTION_EVERYONE
+        | USE_EXTERNAL_EMOJIS
+    )
+    ALL = (
+        DEFAULT
+        | CREATE_INSTANT_INVITE
+        | KICK_MEMBERS
+        | BAN_MEMBERS
+        | ADMINISTRATOR
+        | MANAGE_CHANNELS
+        | MANAGE_GUILD
+        | VIEW_AUDIT_LOG
+        | PRIORITY_SPEAKER
+        | STREAM
+        | SEND_TTS_MESSAGES
+        | MANAGE_MESSAGES
+        | VIEW_GUILD_INSIGHTS
+        | CONNECT
+        | SPEAK
+        | MUTE_MEMBERS
+        | DEAFEN_MEMBERS
+        | MOVE_MEMBERS
+        | USE_VAD
+        | CHANGE_NICKNAME
+        | MANAGE_NICKNAMES
+        | MANAGE_ROLES
+        | MANAGE_WEBHOOKS
+        | MANAGE_EMOJIS_AND_STICKERS
+        | USE_APPLICATION_COMMANDS
+        | REQUEST_TO_SPEAK
+        | MANAGE_EVENTS
+        | MANAGE_THREADS
+        | CREATE_PUBLIC_THREADS
+        | CREATE_PRIVATE_THREADS
+        | USE_EXTERNAL_STICKERS
+        | SEND_MESSAGES_IN_THREADS
+        | START_EMBEDDED_ACTIVITIES
+        | MODERATE_MEMBERS
+    )
+
 
 class UserFlags(IntFlag):
     """An integer flag bitshift object representing the different user flags given by Discord."""

interactions/api/models/member.py

@@ -380,6 +380,7 @@ async def add_to_thread(
     def get_avatar_url(self, guild_id: Union[int, Snowflake, "Guild"]) -> Optional[str]:
         """
         Returns the URL of the member's avatar for the specified guild.
+
         :param guild_id: The id of the guild to get the member's avatar from
         :type guild_id: Union[int, Snowflake, "Guild"]
         :return: URL of the members's avatar (None will be returned if no avatar is set)
@@ -393,3 +394,32 @@ def get_avatar_url(self, guild_id: Union[int, Snowflake, "Guild"]) -> Optional[s
         url = f"https://cdn.discordapp.com/guilds/{_guild_id}/users/{int(self.user.id)}/avatars/{self.avatar}"
         url += ".gif" if self.avatar.startswith("a_") else ".png"
         return url
+
+    async def get_guild_permissions(self, guild: "Guild") -> Permissions:
+        """
+        Returns the permissions of the member for the specified guild.
+
+        .. note::
+            The permissions returned by this function will not take into account role and
+            user overwrites that can be assigned to channels or categories. If you need
+            these overwrites, look into :meth:`.Channel.get_permissions_for`.
+
+        :param guild: The guild of the member
+        :type guild: Guild
+        :return: Base permissions of the member in the specified guild
+        :rtype: Permissions
+        """
+        if int(guild.owner_id) == int(self.id):
+            return Permissions.ALL
+
+        role_everyone = await guild.get_role(int(guild.id))
+        permissions = int(role_everyone.permissions)
+
+        for role_id in self.roles:
+            role = await guild.get_role(role_id)
+            permissions |= int(role.permissions)
+
+        if permissions & Permissions.ADMINISTRATOR == Permissions.ADMINISTRATOR:
+            return Permissions.ALL
+
+        return Permissions(permissions)