From 99756673dca799a16bfa2ad7676f0bd24f25ba5e Mon Sep 17 00:00:00 2001
From: Oksamies <asko.soisalo@gmail.com>
Date: Thu, 17 Oct 2024 15:56:10 +0300
Subject: [PATCH] Add rated_packages_cyberstorm to current_user endpoint

Because new UI will not use package UUIDs in rating recognition
---
 .../api/experimental/views/current_user.py       | 16 +++++++++++++++-
 .../social/tests/test_current_user.py            |  2 ++
 2 files changed, 17 insertions(+), 1 deletion(-)

diff --git a/django/thunderstore/social/api/experimental/views/current_user.py b/django/thunderstore/social/api/experimental/views/current_user.py
index 83410bad4..161ab3caa 100644
--- a/django/thunderstore/social/api/experimental/views/current_user.py
+++ b/django/thunderstore/social/api/experimental/views/current_user.py
@@ -1,7 +1,8 @@
 import datetime
 from typing import List, Optional, Set, TypedDict
 
-from django.db.models import Q
+from django.db.models import Q, Value
+from django.db.models.functions import Concat
 from django.utils import timezone
 from drf_yasg.utils import swagger_auto_schema
 from pydantic import BaseModel
@@ -68,6 +69,7 @@ class UserProfile(TypedDict):
     connections: List[SocialAuthConnection]
     subscription: SubscriptionStatus
     rated_packages: List[str]
+    rated_packages_cyberstorm: List[str]
     teams: List[str]
     teams_full: List[UserTeam]
 
@@ -78,6 +80,7 @@ class UserProfileSerializer(serializers.Serializer):
     connections = SocialAuthConnectionSerializer(many=True)
     subscription = SubscriptionStatusSerializer()
     rated_packages = serializers.ListField()
+    rated_packages_cyberstorm = serializers.ListField()
     teams = (
         serializers.ListField()
     )  # This is in active use by the Django frontend react components at least
@@ -91,6 +94,7 @@ def get_empty_profile() -> UserProfile:
         "connections": [],
         "subscription": get_subscription_status(user=None),
         "rated_packages": [],
+        "rated_packages_cyberstorm": [],
         "teams": [],
         "teams_full": [],
     }
@@ -107,6 +111,15 @@ def get_user_profile(user: UserType) -> UserProfile:
         ),
     )
 
+    rated_packages_cyberstorm = list(
+        user.package_ratings.select_related("package")
+        .annotate(P=Concat("package__namespace__name", Value("-"), "package__name"))
+        .values_list(
+            "P",
+            flat=True,
+        )
+    )
+
     teams = get_teams(user)
 
     return UserProfileSerializer(
@@ -116,6 +129,7 @@ def get_user_profile(user: UserType) -> UserProfile:
             "connections": get_social_auth_connections(user),
             "subscription": get_subscription_status(user),
             "rated_packages": rated_packages,
+            "rated_packages_cyberstorm": rated_packages_cyberstorm,
             "teams": [x.name for x in teams],
             "teams_full": teams,
         }
diff --git a/django/thunderstore/social/tests/test_current_user.py b/django/thunderstore/social/tests/test_current_user.py
index b9f853527..c888e8982 100644
--- a/django/thunderstore/social/tests/test_current_user.py
+++ b/django/thunderstore/social/tests/test_current_user.py
@@ -35,6 +35,7 @@ def test_current_user_info__for_unauthenticated_user__is_empty_structure(
     assert len(user_info["capabilities"]) == 0
     assert len(user_info["connections"]) == 0
     assert len(user_info["rated_packages"]) == 0
+    assert len(user_info["rated_packages_cyberstorm"]) == 0
     assert len(user_info["teams"]) == 0
 
 
@@ -53,6 +54,7 @@ def test_current_user_info__for_authenticated_user__has_basic_values(
     assert user_info["username"] == "Test"
     assert type(user_info["capabilities"]) == list
     assert type(user_info["rated_packages"]) == list
+    assert type(user_info["rated_packages_cyberstorm"]) == list
 
 
 @pytest.mark.django_db