add project listing

Author: nconrad

app/tests.py

@@ -842,6 +842,61 @@ def testAccessFilesOnly(self):
         })
 
 
+class TestProjectsView(TestCase):
+    def testListAllProjectsWithExpectedShape(self):
+        from manifests.models import NodeData, NodeBuildProject
+
+        alpha = Project.objects.create(name="Alpha")
+        beta = Project.objects.create(name="Beta")
+
+        alpha_user_1 = create_random_user()
+        alpha_user_2 = create_random_user()
+        beta_user = create_random_user()
+
+        UserMembership.objects.create(project=alpha, user=alpha_user_1)
+        UserMembership.objects.create(project=alpha, user=alpha_user_2)
+        UserMembership.objects.create(project=beta, user=beta_user)
+
+        alpha_node_1 = Node.objects.create(vsn="W001")
+        alpha_node_2 = Node.objects.create(vsn="W002")
+        beta_node = Node.objects.create(vsn="W010")
+
+        NodeMembership.objects.create(project=alpha, node=alpha_node_1)
+        NodeMembership.objects.create(project=alpha, node=alpha_node_2)
+        NodeMembership.objects.create(project=beta, node=beta_node)
+
+        # Create NodeData with projects from manifests
+        sage_project = NodeBuildProject.objects.create(name="Sage")
+        sgt_project = NodeBuildProject.objects.create(name="SGT")
+
+        NodeData.objects.create(vsn="W001", project=sage_project)
+        NodeData.objects.create(vsn="W002", project=sgt_project)
+        NodeData.objects.create(vsn="W010", project=sage_project)
+
+        # Login a user to access the authenticated endpoint
+        self.client.force_login(alpha_user_1)
+
+        r = self.client.get("/projects/")
+        self.assertEqual(r.status_code, status.HTTP_200_OK)
+
+        data = r.json()
+        self.assertEqual([item["name"] for item in data], ["Alpha", "Beta"])
+
+        alpha_data = data[0]
+        self.assertEqual(alpha_data["member_count"], 2)
+        self.assertEqual(
+            alpha_data["nodes"],
+            [
+                {"vsn": "W001", "project": "Sage"},
+                {"vsn": "W002", "project": "SGT"},
+            ],
+        )
+
+        beta_data = data[1]
+        self.assertEqual(beta_data["member_count"], 1)
+        self.assertEqual(beta_data["nodes"], [{"vsn": "W010", "project": "Sage"}])
+
+
 class TestAuth(TestCase):
     """
     TestAuth tests that our post Globus login, create user and logout flows work as expected.

app/urls.py

@@ -51,12 +51,14 @@
             views.UserProjectsView.as_view(),
             name="user-projects",
         ),
+        path("projects/", views.ProjectsView.as_view(), name="projects"),
         # keeping old profiles/ path for now. replaced by users/.
         path(
             "profiles/<str:username>/access",
             views.UserAccessView.as_view(permission_classes=[AllowAny]),
             name="user-access-old",
         ),
+
         # keeping compatibility with existing portal user profiles
         path("user_profile/<str:username>", views.UserProfileView.as_view()),
         # feedback

app/views.py

@@ -18,6 +18,7 @@
 from rest_framework.permissions import IsAdminUser, IsAuthenticated, AllowAny
 from rest_framework.authtoken.models import Token
 from rest_framework import status
+from django.db.models import Count
 from django.contrib.auth import views as auth_views
 from django.contrib.auth.mixins import LoginRequiredMixin
 from django_slack import slack_message
@@ -192,6 +193,47 @@ def get(self, request: Request, username: str, format=None) -> Response:
         })
 
 
+class ProjectsView(APIView):
+    permission_classes = [IsAuthenticated]
+
+    def get(self, request: Request, format=None) -> Response:
+        from manifests.models import NodeData
+
+        projects = (
+            Project.objects.annotate(member_count=Count("users", distinct=True))
+            .prefetch_related("nodemembership_set__node")
+            .order_by("name")
+        )
+
+        data = []
+        for project in projects:
+            nodes = []
+            for membership in sorted(
+                project.nodemembership_set.all(),
+                key=lambda m: (m.node.vsn or ""),
+            ):
+                if membership.node.vsn:
+                    # Get the NodeData from manifests to access its project
+                    try:
+                        node_data = NodeData.objects.get(vsn=membership.node.vsn)
+                        node_project = node_data.project.name if node_data.project else None
+                    except NodeData.DoesNotExist:
+                        node_project = None
+
+                    nodes.append({
+                        "vsn": membership.node.vsn,
+                        "project": node_project,
+                    })
+
+            data.append({
+                "name": project.name,
+                "nodes": nodes,
+                "member_count": project.member_count,
+            })
+
+        return Response(data)
+
+
 class NodeAuthorizedKeysView(APIView):
     permission_classes = [AllowAny]