1 files changed, 130 insertions, 4 deletions
diff --git a/api/viewsets.py b/api/viewsets.py
index de5ddaf6..e406e8c3 100644
--- a/api/viewsets.py
+++ b/api/viewsets.py
@@ -10,21 +10,56 @@ from rest_framework.viewsets import GenericViewSet, ModelViewSet, ViewSet
 from rest_framework_bulk import BulkCreateModelMixin
 
 from .models import (
-    DocumentationLink,
-    OffTopicChannelName,
+    DocumentationLink, MessageDeletionContext,
+    OffTopicChannelName, Role,
     SnakeFact, SnakeIdiom,
     SnakeName, SpecialSnake,
     Tag, User
 )
 from .serializers import (
-    DocumentationLinkSerializer,
-    OffTopicChannelNameSerializer,
+    DocumentationLinkSerializer, MessageDeletionContextSerializer,
+    OffTopicChannelNameSerializer, RoleSerializer,
     SnakeFactSerializer, SnakeIdiomSerializer,
     SnakeNameSerializer, SpecialSnakeSerializer,
     TagSerializer, UserSerializer
 )
 
 
+class DeletedMessageViewSet(GenericViewSet):
+    """
+    View providing support for posting bulk deletion logs generated by the bot.
+
+    ## Routes
+    ### POST /bot/deleted-messages
+    Post messages from bulk deletion logs.
+
+    #### Body schema
+    >>> {
+    ...     # The member ID of the original actor, if applicable.
+    ...     # If a member ID is given, it must be present on the site.
+    ...     'actor': Optional[int]
+    ...     'creation': datetime,
+    ...     'messages': [
+    ...         {
+    ...             'id': int,
+    ...             'author': int,
+    ...             'channel_id': int,
+    ...             'content': str,
+    ...             'embeds': [
+    ...                 # Discord embed objects
+    ...             ]
+    ...         }
+    ...     ]
+    ... }
+
+    #### Status codes
+    - 204: returned on success
+    """
+
+    queryset = MessageDeletionContext.objects.all()
+    serializer = MessageDeletionContextSerializer
+
+
 class DocumentationLinkViewSet(
     CreateModelMixin, DestroyModelMixin, ListModelMixin, RetrieveModelMixin, GenericViewSet
 ):
@@ -178,6 +213,97 @@ class OffTopicChannelNameViewSet(DestroyModelMixin, ViewSet):
         return Response(serialized.data)
 
 
+class RoleViewSet(ModelViewSet):
+    """
+    View providing CRUD access to the roles on our server, used
+    by the bot to keep a mirror of our server's roles on the site.
+
+    ## Routes
+    ### GET /bot/roles
+    Returns all roles in the database.
+
+    #### Response format
+    >>> [
+    ...     {
+    ...         'id': 267628507062992896,
+    ...         'name': "Admins",
+    ...         'colour': 1337,
+    ...         'permissions': 8
+    ...     }
+    ... ]
+
+    #### Status codes
+    - 200: returned on success
+
+    ### GET /bot/roles/<snowflake:int>
+    Gets a single role by ID.
+
+    #### Response format
+    >>> {
+    ...     'id': 267628507062992896,
+    ...     'name': "Admins",
+    ...     'colour': 1337,
+    ...     'permissions': 8
+    ... }
+
+    #### Status codes
+    - 200: returned on success
+    - 404: if a role with the given `snowflake` could not be found
+
+    ### POST /bot/roles
+    Adds a single new role.
+
+    #### Request body
+    >>> {
+    ...     'id': int,
+    ...     'name': str,
+    ...     'colour': int,
+    ...     'permissions': int,
+    ... }
+
+    #### Status codes
+    - 201: returned on success
+    - 400: if the body format is invalid
+
+    ### PUT /bot/roles/<snowflake:int>
+    Update the role with the given `snowflake`.
+    All fields in the request body are required.
+
+    #### Request body
+    >>> {
+    ...     'id': int,
+    ...     'name': str,
+    ...     'colour': int,
+    ...     'permissions': int
+    ... }
+
+    #### Status codes
+    - 200: returned on success
+    - 400: if the request body was invalid
+
+    ### PATCH /bot/roles/<snowflake:int>
+    Update the role with the given `snowflake`.
+    All fields in the request body are required.
+
+    >>> {
+    ...     'id': int,
+    ...     'name': str,
+    ...     'colour': int,
+    ...     'permissions': int
+    ... }
+
+    ### DELETE /bot/roles/<snowflake:int>
+    Deletes the role with the given `snowflake`.
+
+    #### Status codes
+    - 204: returned on success
+    - 404: if a role with the given `snowflake` does not exist
+    """
+
+    queryset = Role.objects.all()
+    serializer_class = RoleSerializer
+
+
 class SnakeFactViewSet(ListModelMixin, GenericViewSet):
     """
     View providing snake facts created by the Pydis community in the first code jam.