feat: create Context.has_permissions() & Member.has_permissions() (#1056)

* feat: create `Context.has_permissions()` & `Member.has_permissions()`

* ci: correct from checks.

* Update interactions/api/models/member.py

Co-authored-by: EdVraz <88881326+EdVraz@users.noreply.github.com>

* fix: guild could be unbound

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

Author: 3 people

interactions/api/models/channel.py

@@ -1231,13 +1231,9 @@ async def get_permissions_for(self, member: "Member") -> Permissions:
         if not self.guild_id:
             return Permissions.DEFAULT
 
-        from .guild import Guild
+        permissions = await member.get_guild_permissions(self.guild_id)
 
-        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:
+        if Permissions.ADMINISTRATOR in permissions:
             return Permissions.ALL
 
         # @everyone role overwrites

interactions/api/models/member.py

@@ -507,7 +507,9 @@ def get_avatar_url(
         url += ".gif" if self.avatar.startswith("a_") else ".png"
         return url
 
-    async def get_guild_permissions(self, guild: "Guild") -> Permissions:
+    async def get_guild_permissions(
+        self, guild_id: Optional[Union[int, Snowflake, "Guild"]] = MISSING
+    ) -> Permissions:
         """
         Returns the permissions of the member for the specified guild.
 
@@ -521,6 +523,18 @@ async def get_guild_permissions(self, guild: "Guild") -> Permissions:
         :return: Base permissions of the member in the specified guild
         :rtype: Permissions
         """
+        from .guild import Guild
+
+        if guild_id is MISSING:
+            _guild_id = self.guild_id
+            if isinstance(_guild_id, LibraryException):
+                raise _guild_id
+
+        else:
+            _guild_id = int(guild_id) if not isinstance(guild_id, Guild) else int(guild_id.id)
+
+        guild = Guild(**await self._client.get_guild(int(_guild_id)), _client=self._client)
+
         if int(guild.owner_id) == int(self.id):
             return Permissions.ALL
 
@@ -531,7 +545,49 @@ async def get_guild_permissions(self, guild: "Guild") -> Permissions:
             role = await guild.get_role(role_id)
             permissions |= int(role.permissions)
 
-        if permissions & Permissions.ADMINISTRATOR == Permissions.ADMINISTRATOR:
+        if Permissions.ADMINISTRATOR in Permissions(permissions):
             return Permissions.ALL
 
         return Permissions(permissions)
+
+    async def has_permissions(
+        self,
+        *permissions: Union[int, Permissions],
+        channel: Optional[Channel] = MISSING,
+        guild_id: Optional[Union[int, Snowflake, "Guild"]] = MISSING,
+        operator: str = "and",
+    ) -> bool:
+        """
+        Returns whether the member has the permissions passed.
+
+        .. note::
+            If the channel argument is present, the function will look if the member has the permissions in the specified channel.
+            If the argument is missing, then it will only consider the member's guild permissions.
+
+        :param *permissions: The list of permissions
+        :type *permissions: Union[int, Permissions]
+        :param channel: The channel where to check for permissions
+        :type channel: Channel
+        :param guild_id: The id of the guild
+        :type guild_id: Optional[Union[int, Snowflake, Guild]]
+        :param operator: The operator to use to calculate permissions. Possible values: `and`, `or`. Defaults to `and`.
+        :type operator: str
+        :return: Whether the member has the permissions
+        :rtype: bool
+        """
+        perms = (
+            await self.get_guild_permissions(guild_id)
+            if channel is MISSING
+            else await channel.get_permissions_for(self)
+        )
+
+        if operator == "and":
+            for perm in permissions:
+                if perm not in perms:
+                    return False
+            return True
+        else:
+            for perm in permissions:
+                if perm in perms:
+                    return True
+            return False

interactions/client/context.py

@@ -310,6 +310,30 @@ async def popup(self, modal: Modal) -> dict:
 
         return payload
 
+    async def has_permissions(
+        self, *permissions: Union[int, Permissions], operator: str = "and"
+    ) -> bool:
+        """
+        Returns whether the author of the interaction has the permissions in the given context.
+
+        :param *permissions: The list of permissions
+        :type *permissions: Union[int, Permissions]
+        :param operator: The operator to use to calculate permissions. Possible values: `and`, `or`. Defaults to `and`.
+        :type operator: str
+        :return: Whether the author has the permissions
+        :rtype: bool
+        """
+        if operator == "and":
+            for perm in permissions:
+                if perm not in self.author.permissions:
+                    return False
+            return True
+        else:
+            for perm in permissions:
+                if perm in self.author.permissions:
+                    return True
+            return False
+
 
 @define()
 class CommandContext(_Context):