feat: context menu context is now generated

Author: LordOfPolls

interactions/client/client.py

@@ -101,6 +101,7 @@
     ModalContext,
     ComponentContext,
     AutocompleteContext,
+    ContextMenuContext,
 )
 from interactions.models.internal.listener import Listener
 from interactions.models.internal.tasks import Task
@@ -257,7 +258,9 @@ def __init__(
         activity: Union[Activity, str] = None,
         auto_defer: Absent[Union[AutoDefer, bool]] = MISSING,
         autocomplete_context: Type[BaseContext] = AutocompleteContext,
+        basic_logging: bool = False,
         component_context: Type[BaseContext] = ComponentContext,
+        context_menu_context: Type[BaseContext] = ContextMenuContext,
         debug_scope: Absent["Snowflake_Type"] = MISSING,
         delete_unused_application_cmds: bool = False,
         disable_dm_commands: bool = False,
@@ -268,19 +271,18 @@ def __init__(
         intents: Union[int, Intents] = Intents.DEFAULT,
         interaction_context: Type[InteractionContext] = InteractionContext,
         logger: logging.Logger = MISSING,
-        owner_ids: Iterable["Snowflake_Type"] = (),
+        logging_level: int = logging.INFO,
         modal_context: Type[BaseContext] = ModalContext,
+        owner_ids: Iterable["Snowflake_Type"] = (),
         send_command_tracebacks: bool = True,
         shard_id: int = 0,
         show_ratelimit_tracebacks: bool = False,
         slash_context: Type[BaseContext] = SlashContext,
         status: Status = Status.ONLINE,
+        sync_ext: bool = True,
         sync_interactions: bool = True,
         token: str | None = None,
-        sync_ext: bool = True,
         total_shards: int = 1,
-        basic_logging: bool = False,
-        logging_level: int = logging.INFO,
         **kwargs,
     ) -> None:
         if logger is MISSING:
@@ -332,6 +334,8 @@ def __init__(
         """The object to instantiate for Modal Context"""
         self.slash_context: Type[BaseContext] = slash_context
         """The object to instantiate for Slash Context"""
+        self.context_menu_context: Type[BaseContext] = context_menu_context
+        """The object to instantiate for Context Menu Context"""
 
         self.token: str | None = token
 
@@ -1553,7 +1557,10 @@ async def get_context(self, data: dict) -> InteractionContext:
             case InteractionType.MODAL_RESPONSE:
                 cls = self.modal_context.from_dict(self, data)
             case InteractionType.APPLICATION_COMMAND:
-                cls = self.slash_context.from_dict(self, data)
+                if data["data"].get("target_id"):
+                    cls = self.context_menu_context.from_dict(self, data)
+                else:
+                    cls = self.slash_context.from_dict(self, data)
             case _:
                 self.logger.warning(f"Unknown interaction type [{data['type']}] - please update or report this.")
                 cls = self.interaction_context.from_dict(self, data)

interactions/models/internal/context.py

@@ -10,6 +10,7 @@
 from interactions.models.discord.components import BaseComponent
 from interactions.models.discord.file import UPLOADABLE_TYPE
 from interactions.models.discord.sticker import Sticker
+from interactions.models.discord.user import Member, User
 
 from interactions.models.internal.command import BaseCommand
 from interactions.client.mixins.modal import ModalMixin
@@ -21,6 +22,7 @@
     MessageFlags,
     InteractionType,
     ComponentType,
+    CommandType,
 )
 from interactions.models.discord.message import (
     AllowedMentions,
@@ -584,6 +586,8 @@ class ContextMenuContext(InteractionContext, ModalMixin):
     """The id of the target of the context menu."""
     editing_origin: bool
     """Whether you have deferred the interaction and are editing the original response."""
+    target_type: None | CommandType
+    """The type of the target of the context menu."""
 
     def __init__(self, client: "interactions.Client") -> None:
         super().__init__(client)
@@ -592,7 +596,8 @@ def __init__(self, client: "interactions.Client") -> None:
     @classmethod
     def from_dict(cls, client: "interactions.Client", payload: dict) -> Self:
         instance = super().from_dict(client, payload)
-        instance.target_id = payload["target_id"]
+        instance.target_id = Snowflake(payload["data"]["target_id"])
+        instance.target_type = CommandType(payload["data"]["type"])
         return instance
 
     async def defer(self, *, ephemeral: bool = False, edit_origin: bool = False) -> None:
@@ -619,6 +624,16 @@ async def defer(self, *, ephemeral: bool = False, edit_origin: bool = False) ->
         self.ephemeral = ephemeral
         self.editing_origin = edit_origin
 
+    @property
+    def target(self) -> None | Message | User | Member:
+        """
+        The target of the context menu.
+
+        Returns:
+            The target of the context menu.
+        """
+        return self.resolved.get(self.target_id)
+
 
 class ComponentContext(InteractionContext):
     values: list[str]