feat: Add support for age-restricted commands (#1208)

* feat: Add support for age-restricted commands

* refactor: pre-commited

* Update command.py

* Update command.py

Author: Damego

interactions/client/bot.py

@@ -1068,6 +1068,7 @@ def command(
         description_localizations: Optional[Dict[Union[str, Locale], str]] = MISSING,
         default_member_permissions: Optional[Union[int, Permissions]] = MISSING,
         dm_permission: Optional[bool] = MISSING,
+        nsfw: Optional[bool] = MISSING,
         default_scope: bool = True,
     ) -> Callable[[Callable[..., Coroutine]], Command]:
         """
@@ -1121,6 +1122,10 @@ async def sudo(ctx):
             The dictionary of localization for the ``description`` field. This enforces the same restrictions as the ``description`` field.
         :param Optional[Union[int, Permissions]] default_member_permissions: The permissions bit value of :class:`.Permissions`. If not given, defaults to :attr:`.Permissions.USE_APPLICATION_COMMANDS`
         :param Optional[bool] dm_permission: The application permissions if executed in a Direct Message. Defaults to ``True``.
+        :param Optional[bool] nsfw:
+            .. versionadded:: 4.4.0
+
+            Indicates whether the command is age-restricted. Defaults to ``False``
         :param Optional[bool] default_scope:
             .. versionadded:: 4.3.0
 
@@ -1139,6 +1144,7 @@ def decorator(coro: Callable[..., Coroutine]) -> Command:
                 scope=scope,
                 default_member_permissions=default_member_permissions,
                 dm_permission=dm_permission,
+                nsfw=nsfw,
                 name_localizations=name_localizations,
                 description_localizations=description_localizations,
                 default_scope=default_scope,
@@ -1157,6 +1163,7 @@ def message_command(
         name_localizations: Optional[Dict[Union[str, Locale], Any]] = MISSING,
         default_member_permissions: Optional[Union[int, Permissions]] = MISSING,
         dm_permission: Optional[bool] = MISSING,
+        nsfw: Optional[bool] = MISSING,
         default_scope: bool = True,
     ) -> Callable[[Callable[..., Coroutine]], Command]:
         """
@@ -1183,6 +1190,10 @@ async def context_menu_name(ctx):
             The dictionary of localization for the ``name`` field. This enforces the same restrictions as the ``name`` field.
         :param Optional[Union[int, Permissions]] default_member_permissions: The permissions bit value of :class:`.Permissions`. If not given, defaults to :attr:`.Permissions.USE_APPLICATION_COMMANDS`
         :param Optional[bool] dm_permission: The application permissions if executed in a Direct Message. Defaults to ``True``.
+        :param Optional[bool] nsfw:
+            .. versionadded:: 4.4.0
+
+            Indicates whether the command is age-restricted. Defaults to ``False``
         :param Optional[bool] default_scope:
             .. versionadded:: 4.3.0
 
@@ -1198,6 +1209,7 @@ def decorator(coro: Callable[..., Coroutine]) -> Command:
                 scope=scope,
                 default_member_permissions=default_member_permissions,
                 dm_permission=dm_permission,
+                nsfw=nsfw,
                 name_localizations=name_localizations,
                 default_scope=default_scope,
             )(coro)
@@ -1212,6 +1224,7 @@ def user_command(
         name_localizations: Optional[Dict[Union[str, Locale], Any]] = MISSING,
         default_member_permissions: Optional[Union[int, Permissions]] = MISSING,
         dm_permission: Optional[bool] = MISSING,
+        nsfw: Optional[bool] = MISSING,
         default_scope: bool = True,
     ) -> Callable[[Callable[..., Coroutine]], Command]:
         """
@@ -1239,6 +1252,10 @@ async def context_menu_name(ctx):
         :param Optional[Union[int, Permissions]] default_member_permissions:
         The permissions bit value of :class:`.Permissions`. If not given, defaults to :attr:`.Permissions.USE_APPLICATION_COMMANDS`
         :param Optional[bool] dm_permission: The application permissions if executed in a Direct Message. Defaults to ``True``.
+        :param Optional[bool] nsfw:
+            .. versionadded:: 4.4.0
+
+            Indicates whether the command is age-restricted. Defaults to ``False``
         :param Optional[bool] default_scope:
             .. versionadded:: 4.3.0
 
@@ -1254,6 +1271,7 @@ def decorator(coro: Callable[..., Coroutine]) -> Command:
                 scope=scope,
                 default_member_permissions=default_member_permissions,
                 dm_permission=dm_permission,
+                nsfw=nsfw,
                 name_localizations=name_localizations,
                 default_scope=default_scope,
             )(coro)

interactions/client/decor.py

@@ -20,7 +20,8 @@ def command(
     name_localizations: Optional[Dict[Union[str, Locale], str]] = MISSING,
     description_localizations: Optional[Dict[Union[str, Locale], str]] = MISSING,
     default_member_permissions: Optional[Union[int, Permissions]] = MISSING,
-    dm_permission: Optional[bool] = MISSING
+    dm_permission: Optional[bool] = MISSING,
+    nsfw: Optional[bool] = MISSING,
 ) -> Union[List[dict], dict]:  # sourcery skip: low-code-quality
     """
     A wrapper designed to interpret the client-facing API for
@@ -78,6 +79,7 @@ def command(
         )
     )
     _dm_permission: bool = True if dm_permission is MISSING else dm_permission
+    _nsfw: bool = False if nsfw is MISSING else nsfw
 
     payloads: list = []
 
@@ -102,6 +104,7 @@ def command(
                 description_localizations=_description_localizations,
                 default_member_permissions=_default_member_permissions,
                 dm_permission=_dm_permission,
+                nsfw=_nsfw,
             )
             payloads.append(payload._json)
     else:
@@ -114,6 +117,7 @@ def command(
             description_localizations=_description_localizations,
             default_member_permissions=_default_member_permissions,
             dm_permission=_dm_permission,
+            nsfw=_nsfw,
         )
         return payload._json
 

interactions/client/models/command.py

@@ -173,9 +173,10 @@ class ApplicationCommand(DictSerializerMixin):
     :ivar str description: The description of the application command.
     :ivar Optional[List[Option]] options: The "options"/arguments of the application command.
     :ivar Optional[bool] default_permission: The default permission accessibility state of the application command.
+    :ivar Optional[bool] nsfw: Indicates whether the command is age-restricted.
     :ivar int version: The Application Command version autoincrement identifier.
     :ivar str default_member_permissions: The default member permission state of the application command.
-    :ivar boolean dm_permission: The application permissions if executed in a Direct Message.
+    :ivar bool dm_permission: The application permissions if executed in a Direct Message.
     :ivar Optional[Dict[Union[str, Locale], str]] name_localizations: The localisation dictionary for the application command name, if any.
     :ivar Optional[Dict[Union[str, Locale], str]] description_localizations: The localisation dictionary for the application command description, if any.
     """
@@ -188,6 +189,7 @@ class ApplicationCommand(DictSerializerMixin):
     description: str = field()
     options: Optional[List[Option]] = field(converter=convert_list(Option), default=None)
     default_permission: Optional[bool] = field(default=None)
+    nsfw: Optional[bool] = field(default=None)
     version: int = field(default=None)
     default_member_permissions: str = field()
     dm_permission: bool = field(default=None)
@@ -386,6 +388,7 @@ class Command(DictSerializerMixin):
     :ivar Optional[Union[int, Guild, List[int], List[Guild]]] scope: The scope of the command.
     :ivar Optional[str] default_member_permissions: The default member permissions of the command.
     :ivar Optional[bool] dm_permission: The DM permission of the command.
+    :ivar Optional[bool] nsfw: Indicates whether the command is age-restricted. Defaults to ``False``.
     :ivar Optional[Dict[Union[str, Locale], str]] name_localizations: The dictionary of localization for the ``name`` field. This enforces the same restrictions as the ``name`` field.
     :ivar Optional[Dict[Union[str, Locale], str]] description_localizations: The dictionary of localization for the ``description`` field. This enforces the same restrictions as the ``description`` field.
     :ivar bool default_scope: Whether the command should use the default scope. Defaults to ``True``.
@@ -407,6 +410,7 @@ class Command(DictSerializerMixin):
     scope: Optional[Union[int, Guild, List[int], List[Guild]]] = field(default=MISSING)
     default_member_permissions: Optional[str] = field(default=MISSING)
     dm_permission: Optional[bool] = field(default=MISSING)
+    nsfw: Optional[bool] = field(default=None)
     name_localizations: Optional[Dict[Union[str, Locale], str]] = field(default=MISSING)
     description_localizations: Optional[Dict[Union[str, Locale], str]] = field(default=MISSING)
     default_scope: bool = field(default=True)
@@ -479,6 +483,7 @@ def full_data(self) -> Union[dict, List[dict]]:
             description_localizations=self.description_localizations,
             default_member_permissions=self.default_member_permissions,
             dm_permission=self.dm_permission,
+            nsfw=self.nsfw,
         )
 
     @property