feat: Add support for voice message meta-fields

Signed-off-by: Mathieu Corsham <[email protected]>

Author: mccoderpy

discord/file.py

@@ -23,6 +23,7 @@
 FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
 DEALINGS IN THE SOFTWARE.
 """
+from __future__ import annotations
 
 import os.path
 
@@ -71,14 +72,18 @@ class File:
         The file description can be set to attached images to show a alternative text.
     """
 
-    __slots__ = ('fp', 'filename', 'description', 'spoiler', '_original_pos', '_owner', '_closer')
-
-    def __init__(self,
-                 fp: Union[io.IOBase, str, bytes, 'PathLike[str]', 'PathLike[bytes]', int],
-                 filename: Optional[str] = None,
-                 description: Optional[str] = None,
-                 *,
-                 spoiler: bool = False):
+    __slots__ = ('fp', 'filename', 'description', 'spoiler', 'duration', 'waveform', '_original_pos', '_owner', '_closer')
+
+    def __init__(
+            self,
+            fp: Union[io.IOBase, str, bytes, PathLike[str], PathLike[bytes], int],
+            filename: Optional[str] = None,
+            description: Optional[str] = None,
+            *,
+            spoiler: bool = False,
+            duration: Optional[float] = None,
+            waveform: Optional[str] = None
+    ):
         self.fp = fp
 
         if isinstance(fp, io.IOBase):
@@ -113,6 +118,8 @@ def __init__(self,
             self.filename = 'SPOILER_' + self.filename
 
         self.spoiler = spoiler or (self.filename is not None and self.filename.startswith('SPOILER_'))
+        self.duration: Optional[float] = duration
+        self.waveform: Optional[str] = waveform
 
     def reset(self, *, seek=True):
         # The `seek` parameter is needed because
@@ -132,11 +139,17 @@ def close(self):
             self._closer()
 
     def to_dict(self, file_index: int) -> dict:
-        return {
+        base = {
             'id': file_index,
             'description': self.description,
             'filename': self.filename
         }
+        if self.duration is not None:
+            base['duration_secs'] = self.duration
+        if self.waveform is not None:
+            base['waveform'] = self.waveform
+        return base
+            
 
 
 class UploadFile(File):

discord/message.py

@@ -27,6 +27,7 @@
 
 import asyncio
 from datetime import datetime
+from base64 import b64decode
 import re
 import io
 
@@ -134,7 +135,7 @@ class Attachment(Hashable):
     .. versionchanged:: 1.7
         Attachment can now be cast to :class:`str` and is hashable.
     .. versionchanged:: 2.0
-        The :attr:`ephemeral` and :attr:`description` attributes were added.
+        The :attr:`ephemeral`, :attr:`description`, :attr:`duration` and :attr:`waveform` attributes were added.
 
     Attributes
     ------------
@@ -161,10 +162,12 @@ class Attachment(Hashable):
         Whether the attachment is ephemeral (part of an ephemeral message or was provided in a slash-command option).
     description: Optional[:class:`str`]
         The description for the file.
+    duration: Optional[:class:`float`]
+        The duration of the audio file (currently only for voice messages).
     """
 
     __slots__ = ('id', 'size', 'height', 'width', 'ephemeral', 'description',
-                 'filename', 'url', 'proxy_url', '_http', 'content_type')
+                 'filename', 'url', 'proxy_url', '_http', 'content_type', 'duration', '_waveform')
 
     def __init__(self, *, data: AttachmentPayload, state: ConnectionState):
         self.id: int = int(data['id'])
@@ -178,10 +181,17 @@ def __init__(self, *, data: AttachmentPayload, state: ConnectionState):
         self.proxy_url = data.get('proxy_url')
         self._http: HTTPClient = state.http
         self.content_type: str = data.get('content_type')
+        self.duration: Optional[float] = data.get('duration_secs')
+        self._waveform: Optional[List[int]] = data.get('waveform')
 
     def is_spoiler(self) -> bool:
         """:class:`bool`: Whether this attachment contains a spoiler."""
         return self.filename.startswith('SPOILER_')
+    
+    @property
+    def waveform(self) -> Optional[bytearray]:
+        """:class:`Optional[List[:class:`int`]]`: The waveform of the audio file (currently only for voice messages)."""
+        return b64decode(self._waveform) if self._waveform else None
 
     def __repr__(self) -> str:
         return '<Attachment id={0.id} filename={0.filename!r} url={0.url!r}>'.format(self)

discord/types/guild.py

@@ -42,6 +42,7 @@
 from .sticker import GuildSticker
 
 __all__ = (
+    'PermissionFlags',
     'UnavailableGuild',
     'PartialGuild',
     'Guild',
@@ -104,6 +105,56 @@
     'ROLE_SUBSCRIPTIONS_ENABLED',
     'ROLE_SUBSCRIPTIONS_ENABLED_FOR_PURCHASE'
 ]
+PermissionFlags = Literal[
+    'create_instant_invite',
+    'kick_members',
+    'ban_members',
+    'administrator',
+    'manage_channels',
+    'manage_guild',
+    'add_reactions',
+    'view_audit_log',
+    'priority_speaker',
+    'stream',
+    'read_messages',
+    'send_messages',
+    'send_tts_messages',
+    'manage_messages',
+    'embed_links',
+    'attach_files',
+    'read_message_history',
+    'mention_everyone',
+    'external_emojis',
+    'view_guild_insights',
+    'connect',
+    'speak',
+    'mute_members',
+    'deafen_members',
+    'move_members',
+    'use_voice_activation',
+    'change_nickname',
+    'manage_nicknames',
+    'manage_roles',
+    'manage_webhooks',
+    'manage_expressions',
+    'create_expressions',
+    'use_slash_commands',
+    'request_to_speak',
+    'manage_events',
+    'create_events',
+    'manage_threads',
+    'create_public_threads',
+    'create_private_threads',
+    'use_external_stickers',
+    'send_messages_in_threads',
+    'start_embedded_activities',
+    'moderate_members',
+    'view_creator_monetization_analytics',
+    'use_soundboard',
+    'use_external_sounds',
+    'send_voice_messages'
+]
+
 
 
 class UnavailableGuild(TypedDict):

discord/types/message.py

@@ -165,6 +165,8 @@ class Attachment(TypedDict):
     height: NotRequired[Optional[int]]
     width: NotRequired[Optional[int]]
     ephemeral: NotRequired[Literal[True]]
+    duration_secs: NotRequired[float]
+    waveform: NotRequired[str]
 
 
 class PartialMessage(TypedDict):