Fix free broker limitations

filipbekic01 · Sep 23, 2024 · 3ad1316 · 3ad1316
1 parent daef34e
commit 3ad1316
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 5 deletions.
diff --git a/backend/ResQueue/ResQueue/Endpoints/BrokerEndpoints.cs b/backend/ResQueue/ResQueue/Endpoints/BrokerEndpoints.cs
@@ -54,18 +54,39 @@ public static void MapBrokerEndpoints(this IEndpointRouteBuilder routes)
             });
 
         group.MapPost("",
-            async (IMongoCollection<Broker> collection, [FromBody] CreateBrokerDto dto, UserManager<User> userManager,
+            async (IMongoCollection<Broker> brokersCollection, [FromBody] CreateBrokerDto dto,
+                UserManager<User> userManager,
                 HttpContext httpContext) =>
             {
+                // Get user
                 var user = await userManager.GetUserAsync(httpContext.User);
                 if (user == null)
                 {
                     return Results.Unauthorized();
                 }
 
+                // Validate free plan broker count
+                if (user.Subscription is null)
+                {
+                    var filter = Builders<Broker>.Filter.And(
+                        Builders<Broker>.Filter.Eq(b => b.CreatedByUserId, user.Id),
+                        Builders<Broker>.Filter.Eq(b => b.DeletedAt, null)
+                    );
+
+                    if (await brokersCollection.Find(filter).AnyAsync())
+                    {
+                        return Results.Problem(new ProblemDetails
+                        {
+                            Title = "Free Plan Limit",
+                            Detail = "Upgrade to paid plan to unlock more broker slots.",
+                            Status = StatusCodes.Status403Forbidden
+                        });
+                    }
+                }
+
                 var broker = CreateBrokerDtoMapper.ToBroker(user.Id, dto);
 
-                await collection.InsertOneAsync(broker);
+                await brokersCollection.InsertOneAsync(broker);
 
                 return Results.Ok(BrokerMapper.ToDto(broker));
             });

diff --git a/frontend/src/dialogs/CreateBrokerDialog.vue b/frontend/src/dialogs/CreateBrokerDialog.vue
@@ -31,9 +31,11 @@ const newBroker = reactive<CreateBrokerDto>({
 })
 
 const createBroker = () => {
-  createBrokerAsync(newBroker).then((data) => {
-    dialogRef?.value.close(data)
-  })
+  createBrokerAsync(newBroker)
+    .then((data) => {
+      dialogRef?.value.close(data)
+    })
+    .catch((e) => toast.add(errorToToast(e)))
 }
 
 const testConnection = () => {