MAJ Mathilde

PyroFetes/Endpoints/Warehouse/UpdateWarehouseEndpoint.cs

@@ -2,21 +2,25 @@
 using API.DTO.Warehouse.Response;
 using FastEndpoints;
 using Microsoft.EntityFrameworkCore;
+using PyroFetes.Models;
 
 namespace PyroFetes.Endpoints.Warehouse;
 
-public class UpdateWarehouseEndpoint(PyroFetesDbContext pyrofetesdbcontext) : Endpoint<UpdateWarehouseDto, GetWarehouseDto>
+public class UpdateWarehouseEndpoint(PyroFetesDbContext db)
+    : Endpoint<UpdateWarehouseDto, GetWarehouseDto>
 {
     public override void Configure()
     {
-        Put("/api/warehouses/{@id}", x => new { x.Id });
+        // Utilise {id} plutôt que {@id}
+        Put("/api/warehouses/{id}", x => new { x.Id });
         AllowAnonymous();
     }
 
     public override async Task HandleAsync(UpdateWarehouseDto req, CancellationToken ct)
     {
-        Models.Warehouse? warehouseToEdit = await pyrofetesdbcontext
-            .Warehouses
+        // 🔹 On inclut les produits existants pour pouvoir les modifier
+        var warehouseToEdit = await db.Warehouses
+            .Include(w => w.WarehouseProducts)
             .SingleOrDefaultAsync(w => w.Id == req.Id, cancellationToken: ct);
 
         if (warehouseToEdit == null)
@@ -26,7 +30,7 @@ public class UpdateWarehouseEndpoint(PyroFetesDbContext pyrofetesdbcontext) : En
             return;
         }
 
-        // Mise à jour des champs
+        // 🔹 Mise à jour des champs de base
         warehouseToEdit.Name = req.Name;
         warehouseToEdit.MaxWeight = req.MaxWeight;
         warehouseToEdit.Current = req.Current;
@@ -35,9 +39,30 @@ public class UpdateWarehouseEndpoint(PyroFetesDbContext pyrofetesdbcontext) : En
         warehouseToEdit.ZipCode = req.ZipCode;
         warehouseToEdit.City = req.City;
 
-        await pyrofetesdbcontext.SaveChangesAsync(ct);
+        // 🔹 Gestion des produits associés
+        if (req.Products is not null)
+        {
+            // On supprime les anciens liens pour recréer proprement
+            var existingLinks = warehouseToEdit.WarehouseProducts.ToList();
+            if (existingLinks.Any())
+                db.WarehouseProducts.RemoveRange(existingLinks);
 
-        GetWarehouseDto responseDto = new()
+            foreach (var p in req.Products)
+            {
+                var newLink = new WarehouseProduct
+                {
+                    WarehouseId = warehouseToEdit.Id,
+                    ProductId = p.ProductId,
+                    Quantity = p.Quantity
+                };
+                db.WarehouseProducts.Add(newLink);
+            }
+        }
+
+        await db.SaveChangesAsync(ct);
+
+        // 🔹 On renvoie la version mise à jour
+        var response = new GetWarehouseDto
         {
             Id = warehouseToEdit.Id,
             Name = warehouseToEdit.Name,
@@ -46,9 +71,16 @@ public class UpdateWarehouseEndpoint(PyroFetesDbContext pyrofetesdbcontext) : En
             MinWeight = warehouseToEdit.MinWeight,
             Adress = warehouseToEdit.Address,
             ZipCode = warehouseToEdit.ZipCode,
-            City = warehouseToEdit.City
+            City = warehouseToEdit.City,
+            Products = warehouseToEdit.WarehouseProducts.Select(wp => new WarehouseProductDto
+            {
+                ProductId = wp.ProductId,
+                ProductName = wp.Product?.Name,
+                Quantity = wp.Quantity
+            }).ToList()
         };
 
-        await Send.OkAsync(responseDto, ct);
+        Console.WriteLine($"Entrepôt {warehouseToEdit.Name} mis à jour avec succès.");
+        await Send.OkAsync(response, ct);
     }
-}
+}