feat: display warning and prevent creation on limit

api/audit/serializers.py

@@ -2,14 +2,14 @@
 
 from audit.models import AuditLog
 from environments.serializers import EnvironmentSerializerLight
-from projects.serializers import ProjectSerializer
+from projects.serializers import ProjectListSerializer
 from users.serializers import UserListSerializer
 
 
 class AuditLogSerializer(serializers.ModelSerializer):
     author = UserListSerializer()
     environment = EnvironmentSerializerLight()
-    project = ProjectSerializer()
+    project = ProjectListSerializer()
 
     class Meta:
         model = AuditLog

api/environments/models.py

@@ -121,6 +121,10 @@ class Environment(
         help_text="If true, will hide sensitive data(e.g: traits, description etc) from the SDK endpoints",
     )
 
+    @property
+    def total_segment_overrides(self) -> int:
+        return self.feature_segments.count()
+
     objects = EnvironmentManager()
 
     class Meta:

api/environments/serializers.py

@@ -10,15 +10,15 @@
     ReadOnlyIfNotValidPlanMixin,
 )
 from projects.models import Project
-from projects.serializers import ProjectSerializer
+from projects.serializers import ProjectListSerializer
 from util.drf_writable_nested.serializers import (
     DeleteBeforeUpdateWritableNestedModelSerializer,
 )
 
 
 class EnvironmentSerializerFull(serializers.ModelSerializer):
     feature_states = FeatureStateSerializerFull(many=True)
-    project = ProjectSerializer()
+    project = ProjectListSerializer()
 
     class Meta:
         model = Environment
@@ -65,15 +65,20 @@ def get_use_mv_v2_evaluation(self, instance: Environment) -> bool:
         return instance.use_identity_composite_key_for_hashing
 
 
+class EnvironmentRetrieveSerializerLight(EnvironmentSerializerLight):
+    class Meta(EnvironmentSerializerLight.Meta):
+        fields = EnvironmentSerializerLight.Meta.fields + ("total_segment_overrides",)
+
+
 class EnvironmentSerializerWithMetadata(
     SerializerWithMetadata,
     DeleteBeforeUpdateWritableNestedModelSerializer,
-    EnvironmentSerializerLight,
+    EnvironmentRetrieveSerializerLight,
 ):
     metadata = MetadataSerializer(required=False, many=True)
 
-    class Meta(EnvironmentSerializerLight.Meta):
-        fields = EnvironmentSerializerLight.Meta.fields + ("metadata",)
+    class Meta(EnvironmentRetrieveSerializerLight.Meta):
+        fields = EnvironmentRetrieveSerializerLight.Meta.fields + ("metadata",)
 
     def get_organisation_from_validated_data(self, validated_data) -> Organisation:
         return validated_data.get("project").organisation

api/organisations/views.py

@@ -49,7 +49,7 @@
     PermissionModelSerializer,
     UserObjectPermissionsSerializer,
 )
-from projects.serializers import ProjectSerializer
+from projects.serializers import ProjectListSerializer
 from users.serializers import UserIdSerializer
 from webhooks.mixins import TriggerSampleWebhookMixin
 from webhooks.webhooks import WebhookType
@@ -118,7 +118,7 @@ def create(self, request, **kwargs):
     def projects(self, request, pk):
         organisation = self.get_object()
         projects = organisation.projects.all()
-        return Response(ProjectSerializer(projects, many=True).data)
+        return Response(ProjectListSerializer(projects, many=True).data)
 
     @action(detail=True, methods=["POST"])
     def invite(self, request, pk):

api/projects/models.py

@@ -136,6 +136,14 @@ def is_edge_project_by_default(self) -> bool:
             and self.created_date >= settings.EDGE_RELEASE_DATETIME
         )
 
+    @property
+    def total_features(self) -> int:
+        return self.features.count()
+
+    @property
+    def total_segments(self) -> int:
+        return self.segments.count()
+
     def is_feature_name_valid(self, feature_name: str) -> bool:
         """
         Validate the feature name based on the feature_name_regex attribute.

api/projects/serializers.py

@@ -12,7 +12,7 @@
 from users.serializers import UserListSerializer, UserPermissionGroupSerializer
 
 
-class ProjectSerializer(serializers.ModelSerializer):
+class ProjectListSerializer(serializers.ModelSerializer):
     migration_status = serializers.SerializerMethodField(
         help_text="Edge migration status of the project; can be one of: "
         + ", ".join([k.value for k in ProjectIdentityMigrationStatus])
@@ -39,10 +39,8 @@ class Meta:
     def get_migration_status(self, obj: Project) -> str:
         if not settings.PROJECT_METADATA_TABLE_NAME_DYNAMO:
             migration_status = ProjectIdentityMigrationStatus.NOT_APPLICABLE.value
-
         elif obj.is_edge_project_by_default:
             migration_status = ProjectIdentityMigrationStatus.MIGRATION_COMPLETED.value
-
         else:
             migration_status = IdentityMigrator(obj.id).migration_status.value
 
@@ -58,6 +56,17 @@ def get_use_edge_identities(self, obj: Project) -> bool:
         )
 
 
+class ProjectRetrieveSerializer(ProjectListSerializer):
+    class Meta(ProjectListSerializer.Meta):
+        fields = ProjectListSerializer.Meta.fields + (
+            "max_segments_allowed",
+            "max_features_allowed",
+            "max_segment_overrides_allowed",
+            "total_features",
+            "total_segments",
+        )
+
+
 class CreateUpdateUserProjectPermissionSerializer(
     CreateUpdateUserPermissionSerializerABC
 ):

api/projects/tests/test_serializers.py

@@ -4,10 +4,13 @@
 from django.utils import timezone
 
 from environments.dynamodb.types import ProjectIdentityMigrationStatus
-from projects.serializers import ProjectSerializer
+from projects.serializers import (
+    ProjectListSerializer,
+    ProjectRetrieveSerializer,
+)
 
 
-def test_ProjectSerializer_get_migration_status_returns_migration_not_applicable_if_not_configured(
+def test_ProjectListSerializer_get_migration_status_returns_migration_not_applicable_if_not_configured(
     mocker, project, settings
 ):
     # Given
@@ -16,7 +19,7 @@ def test_ProjectSerializer_get_migration_status_returns_migration_not_applicable
         "projects.serializers.IdentityMigrator", autospec=True
     )
 
-    serializer = ProjectSerializer()
+    serializer = ProjectListSerializer()
 
     # When
     migration_status = serializer.get_migration_status(project)
@@ -26,7 +29,7 @@ def test_ProjectSerializer_get_migration_status_returns_migration_not_applicable
     mocked_identity_migrator.assert_not_called()
 
 
-def test_ProjectSerializer_get_migration_status_returns_migration_completed_for_new_projects(
+def test_ProjectListSerializer_get_migration_status_returns_migration_completed_for_new_projects(
     mocker, project, settings
 ):
     # Given
@@ -36,7 +39,7 @@ def test_ProjectSerializer_get_migration_status_returns_migration_completed_for_
         "projects.serializers.IdentityMigrator", autospec=True
     )
 
-    serializer = ProjectSerializer()
+    serializer = ProjectListSerializer()
 
     # When
     migration_status = serializer.get_migration_status(project)
@@ -46,7 +49,7 @@ def test_ProjectSerializer_get_migration_status_returns_migration_completed_for_
     mocked_identity_migrator.assert_not_called()
 
 
-def test_ProjectSerializer_get_migration_status_calls_migrator_with_correct_arguments_for_old_projects(
+def test_ProjectListSerializer_get_migration_status_calls_migrator_with_correct_arguments_for_old_projects(
     mocker, project, settings
 ):
     # Given
@@ -57,7 +60,7 @@ def test_ProjectSerializer_get_migration_status_calls_migrator_with_correct_argu
 
     settings.EDGE_RELEASE_DATETIME = timezone.now()
 
-    serializer = ProjectSerializer()
+    serializer = ProjectListSerializer()
 
     # When
     migration_status = serializer.get_migration_status(project)
@@ -78,9 +81,32 @@ def test_ProjectSerializer_get_migration_status_calls_migrator_with_correct_argu
         (ProjectIdentityMigrationStatus.NOT_APPLICABLE.value, False),
     ],
 )
-def test_ProjectSerializer_get_use_edge_identities(project, migration_status, expected):
+def test_ProjectListSerializer_get_use_edge_identities(
+    project, migration_status, expected
+):
     # Given
-    serializer = ProjectSerializer(context={"migration_status": migration_status})
+    serializer = ProjectListSerializer(context={"migration_status": migration_status})
+
+    # When/Then
+    assert expected is serializer.get_use_edge_identities(project)
+
+
+@pytest.mark.parametrize(
+    "migration_status, expected",
+    [
+        (ProjectIdentityMigrationStatus.MIGRATION_COMPLETED.value, True),
+        (ProjectIdentityMigrationStatus.MIGRATION_IN_PROGRESS.value, False),
+        (ProjectIdentityMigrationStatus.MIGRATION_NOT_STARTED.value, False),
+        (ProjectIdentityMigrationStatus.NOT_APPLICABLE.value, False),
+    ],
+)
+def test_ProjectRetrieveSerializer_get_use_edge_identities(
+    project, migration_status, expected
+):
+    # Given
+    serializer = ProjectRetrieveSerializer(
+        context={"migration_status": migration_status}
+    )
 
     # When/Then
     assert expected is serializer.get_use_edge_identities(project)

api/projects/tests/test_views.py

@@ -157,6 +157,67 @@ def setUp(self):
     def _get_detail_url(self, project_id):
         return reverse("api-v1:projects:project-detail", args=[project_id])
 
+    def test_get_project_data(self):
+        # Given
+        project_name = "Test project"
+        hide_disabled_flags = False
+        enable_dynamo_db = False
+        prevent_flag_defaults = True
+        enable_realtime_updates = False
+        only_allow_lower_case_feature_names = True
+
+        Project.objects.create(
+            name=project_name,
+            organisation=self.organisation,
+            hide_disabled_flags=hide_disabled_flags,
+            enable_dynamo_db=enable_dynamo_db,
+            prevent_flag_defaults=prevent_flag_defaults,
+            enable_realtime_updates=enable_realtime_updates,
+            only_allow_lower_case_feature_names=only_allow_lower_case_feature_names,
+        )
+        url = reverse("api-v1:projects:project-list")
+
+        # When
+        response = self.client.get(url)
+
+        # Then
+        assert response.status_code == status.HTTP_200_OK
+        assert response.json()[0]["name"] == project_name
+        assert response.json()[0]["hide_disabled_flags"] is hide_disabled_flags
+        assert response.json()[0]["enable_dynamo_db"] is enable_dynamo_db
+        assert response.json()[0]["prevent_flag_defaults"] is prevent_flag_defaults
+        assert response.json()[0]["enable_realtime_updates"] is enable_realtime_updates
+        assert (
+            response.json()[0]["only_allow_lower_case_feature_names"]
+            is only_allow_lower_case_feature_names
+        )
+        assert "max_segments_allowed" not in response.json()[0].keys()
+        assert "max_features_allowed" not in response.json()[0].keys()
+        assert "max_segment_overrides_allowed" not in response.json()[0].keys()
+        assert "total_features" not in response.json()[0].keys()
+        assert "total_segments" not in response.json()[0].keys()
+
+    def test_get_project_data_by_id(self):
+        # Given
+        project_name = "Test project"
+        project = Project.objects.create(
+            name=project_name,
+            organisation=self.organisation,
+        )
+        url = reverse("api-v1:projects:project-detail", args=[project.id])
+
+        # When
+        response = self.client.get(url)
+
+        # Then
+        assert response.status_code == status.HTTP_200_OK
+        assert response.json()["name"] == project_name
+        assert response.json()["max_segments_allowed"] == 100
+        assert response.json()["max_features_allowed"] == 400
+        assert response.json()["max_segment_overrides_allowed"] == 100
+        assert response.json()["total_features"] == 0
+        assert response.json()["total_segments"] == 0
+
     def test_create_project_returns_403_if_user_is_not_organisation_admin(self):
         # Given
         not_permitted_user = FFAdminUser.objects.create(email="[email protected]")

api/projects/views.py

@@ -44,7 +44,8 @@
     CreateUpdateUserProjectPermissionSerializer,
     ListUserPermissionGroupProjectPermissionSerializer,
     ListUserProjectPermissionSerializer,
-    ProjectSerializer,
+    ProjectListSerializer,
+    ProjectRetrieveSerializer,
 )
 
 
@@ -70,7 +71,14 @@
     ),
 )
 class ProjectViewSet(viewsets.ModelViewSet):
-    serializer_class = ProjectSerializer
+    def get_serializer_class(self):
+        if self.action == "list":
+            return ProjectListSerializer
+        elif self.action == "retrieve":
+            return ProjectRetrieveSerializer
+
+        return ProjectListSerializer
+
     permission_classes = [ProjectPermissions | MasterAPIKeyProjectPermissions]
     pagination_class = None
 

frontend/common/providers/FeatureListProvider.js

@@ -1,5 +1,6 @@
 import React from 'react'
 import FeatureListStore from 'common/stores/feature-list-store'
+import ProjectStore from 'common/stores/project-store'
 
 const FeatureListProvider = class extends React.Component {
   static displayName = 'FeatureListProvider'
@@ -11,8 +12,9 @@ const FeatureListProvider = class extends React.Component {
       isLoading: FeatureListStore.isLoading,
       isSaving: FeatureListStore.isSaving,
       lastSaved: FeatureListStore.getLastSaved(),
+      maxFeaturesAllowed: ProjectStore.getMaxFeaturesAllowed(),
       projectFlags: FeatureListStore.getProjectFlags(),
-      usageData: FeatureListStore.getFeatureUsage(),
+      totalFeatures: ProjectStore.getTotalFeatures(),
     }
     ES6Component(this)
     this.listenTo(FeatureListStore, 'change', () => {
@@ -22,7 +24,9 @@ const FeatureListProvider = class extends React.Component {
         isLoading: FeatureListStore.isLoading,
         isSaving: FeatureListStore.isSaving,
         lastSaved: FeatureListStore.getLastSaved(),
+        maxFeaturesAllowed: ProjectStore.getMaxFeaturesAllowed(),
         projectFlags: FeatureListStore.getProjectFlags(),
+        totalFeatures: ProjectStore.getTotalFeatures(),
         usageData: FeatureListStore.getFeatureUsage(),
       })
     })

frontend/common/stores/project-store.js

@@ -161,6 +161,21 @@ const store = Object.assign({}, BaseStore, {
       return Promise.resolve(store.getEnvironmentIdFromKey(apiKey))
     })
   },
+  getMaxFeaturesAllowed: () => {
+    return store.model && store.model.max_features_allowed
+  },
+  getMaxSegmentOverridesAllowed: () => {
+    return store.model && store.model.max_segment_overrides_allowed
+  },
+  getMaxSegmentsAllowed: () => {
+    return store.model && store.model.max_segments_allowed
+  },
+  getTotalFeatures: () => {
+    return store.model && store.model.total_features
+  },
+  getTotalSegments: () => {
+    return store.model && store.model.total_segments
+  },
   getEnvs: () => store.model && store.model.environments,
   id: 'project',
   model: null,