Add Docker Compose project grouping functionality and UI integration

- Updated `list_running_containers` to retrieve all containers and include labels.
- Introduced `get_grouped_docker_compose_projects` to group containers by their Docker Compose project labels.
- Added new API endpoint `ListDockerComposeProjectsView` for fetching grouped projects.
- Enhanced frontend with a new UI for selecting Docker Compose projects and their containers, including error handling and loading states.
- Refactored project creation modal to reset Docker selection state when opened.

Author: fr4iser90

backend/core/docker_service.py

@@ -9,14 +9,15 @@ def list_running_containers():
     """
     try:
         client = docker.from_env()
-        containers = client.containers.list()
+        containers = client.containers.list(all=True) # Get all containers, not just running, to see labels of stopped ones too
         container_list = []
         for container in containers:
             container_list.append({
                 "id": container.short_id,
                 "name": container.name,
                 "image": container.attrs['Config']['Image'],
-                "status": container.status
+                "status": container.status,
+                "labels": container.labels # Add labels
             })
         if not container_list:
             return "No running containers found."
@@ -26,6 +27,37 @@ def list_running_containers():
     except Exception as e:
         return f"An unexpected error occurred: {str(e)}"
 
+def get_grouped_docker_compose_projects():
+    """
+    Lists containers and groups them by the 'com.docker.compose.project' label.
+    Returns a dictionary where keys are compose project names and values are lists of container dicts.
+    Containers without the project label are grouped under '__ungrouped__'.
+    """
+    containers_or_error = list_running_containers() # This now includes labels
+    if isinstance(containers_or_error, str):
+        return containers_or_error # Return error message if fetching containers failed
+
+    grouped_projects = {}
+    for container in containers_or_error:
+        project_name = container.get("labels", {}).get("com.docker.compose.project")
+        # Also consider com.docker.compose.project.name for some compose versions/setups
+        if not project_name:
+            project_name = container.get("labels", {}).get("com.docker.compose.project.name")
+
+        if project_name:
+            if project_name not in grouped_projects:
+                grouped_projects[project_name] = []
+            grouped_projects[project_name].append(container)
+        else:
+            if '__ungrouped__' not in grouped_projects:
+                grouped_projects['__ungrouped__'] = []
+            grouped_projects['__ungrouped__'].append(container)
+    
+    # Convert to a list of dicts for easier frontend consumption if preferred, or keep as dict
+    # Example: [{ "projectName": "my_project", "containers": [...] }]
+    # For now, returning the dictionary direkte
+    return grouped_projects
+
 def get_container_code_paths(container_id: str):
     """
     Retrieves the host paths for volume mounts of a given container.
@@ -65,23 +97,37 @@ def get_container_code_paths(container_id: str):
 
 if __name__ == '__main__':
     print("Attempting to list running containers...")
-    running_containers = list_running_containers()
-    if isinstance(running_containers, str): # Error message
-        print(running_containers)
-    elif running_containers:
-        print("Running containers:")
-        for cont in running_containers:
-            print(f"  ID: {cont['id']}, Name: {cont['name']}, Image: {cont['image']}, Status: {cont['status']}")
-            print(f"    Attempting to get code paths for {cont['id']} ({cont['name']})...")
-            paths = get_container_code_paths(cont['id'])
-            if isinstance(paths, str): # Error message
-                print(f"    Error/Info: {paths}")
-            elif paths:
-                print("    Found potential host paths:")
-                for path in paths:
-                    print(f"      - {path}")
-            else: # Empty list but no error string
-                print(f"    No specific host paths identified for {cont['id']}.")
+    # Test the new grouping function
+    grouped_containers = get_grouped_docker_compose_projects()
+    if isinstance(grouped_containers, str):
+        print(f"Error getting grouped projects: {grouped_containers}")
+    else:
+        print("\nGrouped Docker Compose Projects:")
+        for project_name, containers_in_project in grouped_containers.items():
+            print(f"  Project: {project_name}")
+            for cont in containers_in_project:
+                print(f"    - ID: {cont['id']}, Name: {cont['name']}, Status: {cont['status']}, Image: {cont['image']}")
+                # Optional: print labels for verification
+                # print(f"      Labels: {cont['labels']}") 
 
-    else: # Empty list from list_running_containers, but not an error string.
-        print("No running containers were found to inspect.") 
+    # Original test for list_running_containers and get_container_code_paths (can be kept for direct testing)
+    # running_containers = list_running_containers()
+    # if isinstance(running_containers, str): # Error message
+    #     print(running_containers)
+    # elif running_containers:
+    #     print("\nIndividual Running containers (for path testing):")
+    #     for cont in running_containers:
+    #         if cont['status'] == 'running': # Only try to get paths for actually running containers for this test part
+    #             print(f"  ID: {cont['id']}, Name: {cont['name']}")
+    #             print(f"    Attempting to get code paths for {cont['id']} ({cont['name']})...")
+    #             paths = get_container_code_paths(cont['id'])
+    #             if isinstance(paths, str): # Error message
+    #                 print(f"    Error/Info: {paths}")
+    #             elif paths:
+    #                 print("    Found potential host paths:")
+    #                 for path in paths:
+    #                     print(f"      - {path}")
+    #             else: # Empty list but no error string
+    #                 print(f"    No specific host paths identified for {cont['id']}.")
+    # else: # Empty list from list_running_containers, but not an error string.
+    #     print("No running containers were found to inspect for paths.") 

backend/core/urls.py

@@ -5,7 +5,7 @@
     SecurityToolViewSet, ScanConfigurationViewSet, UserProfileViewSet,
     ApiKeyViewSet, ScanTriggerViewSet, ScanJobViewSet, 
     ProjectMembershipViewSet, UserViewSet, CIScanTriggerViewSet,
-    ListDockerContainersView, GetDockerContainerPathsView
+    ListDockerContainersView, GetDockerContainerPathsView, ListDockerComposeProjectsView
 )
 
 router = DefaultRouter()
@@ -26,4 +26,5 @@
     path('', include(router.urls)),
     path('docker/containers/', ListDockerContainersView.as_view(), name='docker-list-containers'),
     path('docker/containers/<str:container_id>/paths/', GetDockerContainerPathsView.as_view(), name='docker-container-paths'),
+    path('docker/compose-projects/', ListDockerComposeProjectsView.as_view(), name='docker-list-compose-projects'),
 ] 

backend/core/views.py

@@ -28,7 +28,7 @@
 from .authentication import ApiKeyAuthentication # Import the custom authentication class
 # Docker API Integration Imports
 from rest_framework.views import APIView
-from .docker_service import list_running_containers, get_container_code_paths # Assuming docker_service.py is in the same app directory
+from .docker_service import list_running_containers, get_container_code_paths, get_grouped_docker_compose_projects # Added get_grouped_docker_compose_projects
 # End Docker API Integration Imports
 User = get_user_model()
 
@@ -622,3 +622,26 @@ def get(self, request, container_id, *args, **kwargs):
         if not paths_or_error: # Empty list, but valid response
             return Response([], status=status.HTTP_200_OK)
         return Response(paths_or_error, status=status.HTTP_200_OK)
+
+class ListDockerComposeProjectsView(APIView):
+    """
+    Lists Docker containers grouped by their 'com.docker.compose.project' label.
+    Requires admin privileges.
+    """
+    permission_classes = [IsAdminUser]
+
+    def get(self, request, *args, **kwargs):
+        grouped_projects_or_error = get_grouped_docker_compose_projects()
+        if isinstance(grouped_projects_or_error, str): # Error message returned from service
+            return Response({"error": grouped_projects_or_error}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
+        
+        # Transform the dictionary into a list of objects for a more common API response structure
+        # e.g., [{"project_name": "my_project", "containers": [...]}, ...]
+        response_data = []
+        for project_name, containers in grouped_projects_or_error.items():
+            response_data.append({
+                "compose_project_name": project_name,
+                "containers": containers
+            })
+
+        return Response(response_data, status=status.HTTP_200_OK)