feat: port serverlist into Central Command (#102)

* fix: run pre-commit checks from inside uv venv

* feat: ported current serverlist logic to cc

* feat: include rate limiting to avoid being taken down by funny users

* chore: baby serverlist migrations

* feat: add API tests for BabyServer and update permissions for ListBabyServersView

* feat: improved test coverage

Author: corp-0

.github/workflows/main.yml

@@ -68,7 +68,7 @@ jobs:
                   cd src
                   uv run manage.py makemigrations --check
                   uv run manage.py migrate
-                  uv run manage.py test tests/
+                  uv run manage.py test
 
     docker:
         needs: [ lint, unit_test ]

.pre-commit-config.yaml

@@ -13,22 +13,20 @@ repos:
         args: ["--allow-multiple-documents"]
       - id: debug-statements
       - id: trailing-whitespace
-        exclude: >-
-          ^.*.md$
+        exclude: ^.*.md$
 
   - repo: https://github.com/astral-sh/ruff-pre-commit
     rev: v0.7.0
     hooks:
       - id: ruff
-        args: [ --fix ]
+        args: [--fix]
       - id: ruff-format
 
-  # local mypy because of stub dependencies
   - repo: local
     hooks:
       - id: typecheck
-        name: Typecheck
-        entry: mypy .
-        types: [python]
+        name: Typecheck (uv)
+        entry: uv run mypy
         language: system
         pass_filenames: false
+        args: ["."]

dev-compose.yml

@@ -10,9 +10,15 @@ services:
     ports:
       - "5432:5432"
 
+  cache:
+    image: memcached:latest
+    ports:
+      - "11211:11211"
+
   web:
     depends_on:
-    - db
+      - db
+      - cache
     build: .
     env_file: ./.env
     ports:

docker-compose.yml

@@ -8,7 +8,13 @@ services:
     volumes:
       - db-data:/var/lib/postgresql/data
 
+  cache:
+    image: memcached:latest
+
   web:
+    depends_on:
+      - db
+      - cache
     image: unitystation/central-command:latest
     environment:
       - DEBUG=0

pyproject.toml

@@ -13,6 +13,7 @@ dependencies = [
     "psycopg2-binary~=2.9.9",
     "python-dotenv~=0.19.2",
     "whitenoise~=6.6.0",
+    "pymemcache>=4.0,<5.0",
 ]
 
 [build-system]

src/accounts/api/serializers.py

@@ -93,7 +93,7 @@ def validate(self, data):
 
         if not account_confirmation.is_token_valid():
             raise serializers.ValidationError({"token": "Token is invalid or expired."})
-        return {"token": data["token"]}
+        return {"token": data["token"], "account_confirmation": account_confirmation}
 
 
 class EmailSerializer(serializers.Serializer):

src/accounts/api/views.py

@@ -324,7 +324,7 @@ def post(self, request):
         if not serializer.is_valid():
             return ErrorResponse(serializer.errors, status.HTTP_400_BAD_REQUEST)
 
-        account_confirmation = AccountConfirmation.objects.get(token=serializer.validated_data["token"])
+        account_confirmation: AccountConfirmation = serializer.validated_data["account_confirmation"]
         account = account_confirmation.account
 
         account.is_confirmed = True

src/baby_serverlist/__init__.py

@@ -0,0 +1,10 @@
+from django.contrib import admin
+
+from .models import BabyServer
+
+
+@admin.register(BabyServer)
+class BabyServerAdmin(admin.ModelAdmin):
+    list_display = ("id", "owner", "whitelisted", "serverlist_token")
+    search_fields = ("id", "owner__email", "owner__unique_identifier")
+    list_filter = ("whitelisted",)

src/baby_serverlist/admin.py

@@ -0,0 +1,31 @@
+from rest_framework import serializers
+
+
+class ServerStatusSerializer(serializers.Serializer):
+    ServerToken = serializers.CharField()
+    Passworded = serializers.BooleanField()
+    ServerName = serializers.CharField()
+    ForkName = serializers.CharField()
+    BuildVersion = serializers.IntegerField()
+    CurrentMap = serializers.CharField()
+    GameMode = serializers.CharField()
+    IngameTime = serializers.CharField()
+    RoundTime = serializers.CharField()
+    PlayerCount = serializers.IntegerField()
+    PlayerCountMax = serializers.IntegerField()
+    ServerIP = serializers.CharField()
+    ServerPort = serializers.IntegerField()
+    WinDownload = serializers.CharField()
+    OSXDownload = serializers.CharField()
+    LinuxDownload = serializers.CharField()
+    fps = serializers.IntegerField()
+    GoodFileVersion = serializers.CharField()
+
+
+class ActiveServersSerializer(serializers.Serializer):
+    CashDateTime = serializers.DateTimeField()
+    servers = ServerStatusSerializer(many=True)
+
+
+class RegenerateServerlistTokenSerializer(serializers.Serializer):
+    server_id = serializers.UUIDField()