Collections events

Core/Services/CollectionsService.cs

@@ -1,4 +1,5 @@
 using Infrastructure.Repositories;
+using Serilog;
 using Shared.Dtos;
 using Shared.Dtos.FromClient.Collections;
 using Shared.Exceptions;

Infrastructure/Repositories/CollectionsRepository.cs

@@ -39,6 +39,17 @@ public async Task<Collection> UpdateCollection(Collection collection)
     public async Task DeleteCollection(Collection collection)
     {
         await using var applicationDbContext = await dbContextFactory.CreateDbContextAsync();
+
+        // Remove the collection reference from plants
+        var plantsToUpdate = await applicationDbContext.Plants
+            .Where(p => p.CollectionId == collection.CollectionId)
+            .ToListAsync();
+
+        foreach (var plant in plantsToUpdate)
+        {
+            plant.CollectionId = null;
+        }
+        applicationDbContext.Plants.UpdateRange(plantsToUpdate);
         applicationDbContext.Collections.Remove(collection);
         await applicationDbContext.SaveChangesAsync();
     }

api/Events/Collections/Client/ClientWantsToDeleteCollection.cs

@@ -1,8 +1,10 @@
 using api.Events.Collections.Server;
+using api.Events.User.ServerResponses;
 using api.Extensions;
 using Core.Services;
 using Fleck;
 using lib;
+using Shared.Exceptions;
 using Shared.Models;
 
 namespace api.Events.Collections.Client;
@@ -16,8 +18,18 @@ public class ClientWantsToDeleteCollection(CollectionsService collectionsService
 {
     public override async Task Handle(ClientWantsToDeleteCollectionDto dto, IWebSocketConnection socket)
     {
-        var email = jwtService.GetEmailFromJwt(dto.Jwt!);
-        await collectionsService.DeleteCollection(dto.CollectionId, email);
-        socket.SendDto(new ServerDeletesCollection());
+        try
+        {
+            var email = jwtService.GetEmailFromJwt(dto.Jwt!);
+            await collectionsService.DeleteCollection(dto.CollectionId, email);
+            socket.SendDto(new ServerDeletesCollection());
+        }
+        catch (Exception e) when (e is not NotFoundException)
+        {
+            socket.SendDto(new ServerRejectsUpdate
+            {
+                Error = "Could not delete collection. Please try again later."
+            });
+        }
     }
 }