Section, TextDisplay

Author: NeloBlivion

discord/components.py

@@ -27,7 +27,7 @@
 
 from typing import TYPE_CHECKING, Any, ClassVar, TypeVar
 
-from .enums import ButtonStyle, ChannelType, ComponentType, InputTextStyle, try_enum
+from .enums import ButtonStyle, ChannelType, ComponentType, InputTextStyle, SeparatorSpacingSize, try_enum
 from .partial_emoji import PartialEmoji, _EmojiTag
 from .utils import MISSING, get_slots
 
@@ -36,9 +36,19 @@
     from .types.components import ActionRow as ActionRowPayload
     from .types.components import ButtonComponent as ButtonComponentPayload
     from .types.components import Component as ComponentPayload
+    from .types.components import BaseComponent as BaseComponentPayload
     from .types.components import InputText as InputTextComponentPayload
     from .types.components import SelectMenu as SelectMenuPayload
     from .types.components import SelectOption as SelectOptionPayload
+    from .types.components import TextDisplayComponent as TextDisplayComponentPayload
+    from .types.components import SectionComponent as SectionComponentPayload
+    from .types.components import UnfurledMediaItem as UnfurledMediaItemPayload
+    from .types.components import ThumbnailComponent as ThumbnailComponentPayload
+    from .types.components import MediaGalleryItem as MediaGalleryItemPayload
+    from .types.components import MediaGalleryComponent as MediaGalleryComponentPayload
+    from .types.components import FileComponent as FileComponentPayload
+    from .types.components import SeparatorComponent as SeparatorComponentPayload
+    from .types.components import ContainerComponent as ContainerComponentPayload
 
 __all__ = (
     "Component",
@@ -47,11 +57,12 @@
     "SelectMenu",
     "SelectOption",
     "InputText",
+    "Section",
+    "TextDisplay",
 )
 
 C = TypeVar("C", bound="Component")
 
-
 class Component:
     """Represents a Discord Bot UI Kit Component.
 
@@ -69,12 +80,15 @@ class Component:
     ----------
     type: :class:`ComponentType`
         The type of component.
+    id: :class:`str`
+        The component's ID.
     """
 
-    __slots__: tuple[str, ...] = ("type",)
+    __slots__: tuple[str, ...] = ("type", "id")
 
     __repr_info__: ClassVar[tuple[str, ...]]
     type: ComponentType
+    id: str
 
     def __repr__(self) -> str:
         attrs = " ".join(f"{key}={getattr(self, key)!r}" for key in self.__repr_info__)
@@ -494,16 +508,100 @@ def to_dict(self) -> SelectOptionPayload:
         return payload
 
 
+class Section(Component):
+    """Represents a Section from Components V2.
+
+    This is a component that contains other components such as :class:`TextDisplay` and :class:`Thumbnail`.
+
+    This inherits from :class:`Component`.
+
+    .. versionadded:: 2.7
+
+    Attributes
+    ----------
+    components: List[:class:`Component`]
+        The components contained in this section.
+    accessory: Optional[:class:`Component`]
+        The accessory attached to this Section.
+    """
+
+    __slots__: tuple[str, ...] = ("components", "accessory")
+
+    __repr_info__: ClassVar[tuple[str, ...]] = __slots__
+
+    def __init__(self, data: SectionComponentPayload):
+        self.type: ComponentType = try_enum(ComponentType, data["type"])
+        self.id: str = data.get("id")
+        self.components: list[Component] = [_component_factory(d) for d in data.get("components", [])]
+        self.accessory: Component | None = None
+        if _accessory := data.get("accessory"):
+            self.accessory = _component_factory(_accessory)
+
+    def to_dict(self) -> SectionComponentPayload:
+        payload = {
+            "type": int(self.type),
+            "id": self.id,
+            "components": [c.to_dict() for c in self.components]
+        }
+        if self.accessory:
+            payload["accessory"] = self.accessory.to_dict()
+        return payload
+
+
+class TextDisplay(Component):
+    """Represents a Text Display from Components V2.
+
+    This is a component that displays text.
+
+    This inherits from :class:`Component`.
+
+    .. versionadded:: 2.7
+
+    Attributes
+    ----------
+    content: :class:`str`
+        The component's text content.
+    """
+
+    __slots__: tuple[str, ...] = ("content",)
+
+    __repr_info__: ClassVar[tuple[str, ...]] = __slots__
+
+    def __init__(self, data: TextDisplayComponentPayload):
+        self.type: ComponentType = try_enum(ComponentType, data["type"])
+        self.id: str = data.get("id")
+        self.content: str = data.get("content")
+
+    def to_dict(self) -> TextDisplayComponentPayload:
+        return {
+            "type": int(self.type),
+            "id": self.id,
+            "content": self.content
+        }
+
+
+COMPONENT_MAPPINGS = {
+    1: ActionRow,
+    2: Button,
+    3: SelectMenu,
+    4: InputText,
+    5: SelectMenu,
+    6: SelectMenu,
+    7: SelectMenu,
+    8: SelectMenu,
+    9: Section,
+    10: TextDisplay,
+    11: None,
+    12: None,
+    13: None,
+    14: None,
+    17: None,
+}
+
 def _component_factory(data: ComponentPayload) -> Component:
     component_type = data["type"]
-    if component_type == 1:
-        return ActionRow(data)
-    elif component_type == 2:
-        return Button(data)  # type: ignore
-    elif component_type == 4:
-        return InputText(data)  # type: ignore
-    elif component_type in (3, 5, 6, 7, 8):
-        return SelectMenu(data)  # type: ignore
+    if cls := COMPONENT_MAPPINGS.get(component_type):
+        return cls(data)
     else:
         as_enum = try_enum(ComponentType, component_type)
         return Component._raw_construct(type=as_enum)