refactor prenda Entity (la lista en marca)

Author: JulianKer

src/main/java/com/outfitlab/project/domain/model/MarcaModel.java

@@ -1,8 +1,10 @@
 package com.outfitlab.project.domain.model;
 
-import lombok.Getter;
+import com.outfitlab.project.infrastructure.model.PrendaEntity;
 
 import java.time.LocalDateTime;
+import java.util.ArrayList;
+import java.util.List;
 
 public class MarcaModel {
 
@@ -11,15 +13,17 @@ public class MarcaModel {
     private String logoUrl;
     private LocalDateTime createdAt;
     private LocalDateTime updatedAt;
+    private List<PrendaModel> prendas;
 
     public MarcaModel(){}
 
-    public MarcaModel(String codigoMarca, String nombre, String logoUrl, LocalDateTime createdAt, LocalDateTime updatedAt) {
+    public MarcaModel(String codigoMarca, String nombre, String logoUrl, LocalDateTime createdAt, LocalDateTime updatedAt, List<PrendaModel> prendas) {
         this.codigoMarca = codigoMarca;
         this.nombre = nombre;
         this.logoUrl = logoUrl;
         this.createdAt = createdAt;
         this.updatedAt = updatedAt;
+        this.prendas = prendas;
     }
 
     public String getCodigoMarca() {
@@ -61,4 +65,12 @@ public LocalDateTime getUpdatedAt() {
     public void setUpdatedAt(LocalDateTime updatedAt) {
         this.updatedAt = updatedAt;
     }
+
+    public List<PrendaModel> getPrendas() {
+        return prendas;
+    }
+
+    public void setPrendas(List<PrendaModel> prendas) {
+        this.prendas = prendas;
+    }
 }

src/main/java/com/outfitlab/project/infrastructure/model/MarcaEntity.java

@@ -2,6 +2,7 @@
 
 import com.fasterxml.jackson.annotation.JsonManagedReference;
 import com.outfitlab.project.domain.model.MarcaModel;
+import com.outfitlab.project.domain.model.PrendaModel;
 import jakarta.persistence.*;
 import lombok.AllArgsConstructor;
 import lombok.Getter;
@@ -63,24 +64,47 @@ public void addPrenda(PrendaEntity prenda) {
 
 
     // ------------- acá hacemos los dos convert ------------
-    public static MarcaModel convertToModel(MarcaEntity marcaEntity) {
+    public static MarcaModel convertToModel(MarcaEntity entity) {
+        if (entity == null) return null;
+
+        List<PrendaModel> prendasModel = new ArrayList<>();
+        if (entity.getPrendas() != null) {
+            for (PrendaEntity prendaEntity : entity.getPrendas()) {
+                prendasModel.add(PrendaEntity.convertToModel(prendaEntity));
+            }
+        }
+
         return new MarcaModel(
-                marcaEntity.getCodigoMarca(),
-                marcaEntity.getNombre(),
-                marcaEntity.getLogoUrl(),
-                marcaEntity.getCreatedAt(),
-                marcaEntity.getUpdatedAt()
+                entity.getCodigoMarca(),
+                entity.getNombre(),
+                entity.getLogoUrl(),
+                entity.getCreatedAt(),
+                entity.getUpdatedAt(),
+                prendasModel
         );
     }
 
-    public static MarcaEntity convertToEntity(MarcaModel marcaModel) {
-        return new MarcaEntity(
-                marcaModel.getCodigoMarca(),
-                marcaModel.getNombre(),
-                marcaModel.getLogoUrl(),
-                marcaModel.getCreatedAt(),
-                marcaModel.getUpdatedAt()
-        );
+    public static MarcaEntity convertToEntity(MarcaModel model) {
+        if (model == null) return null;
+
+        MarcaEntity entity = new MarcaEntity();
+        entity.setCodigoMarca(model.getCodigoMarca());
+        entity.setNombre(model.getNombre());
+        entity.setLogoUrl(model.getLogoUrl());
+        entity.setCreatedAt(model.getCreatedAt());
+        entity.setUpdatedAt(model.getUpdatedAt());
+
+        List<PrendaEntity> prendasEntity = new ArrayList<>();
+        if (model.getPrendas() != null) {
+            for (PrendaModel prendaModel : model.getPrendas()) {
+                PrendaEntity prendaEntity = PrendaEntity.convertToEntity(prendaModel);
+                prendaEntity.setMarca(entity);
+                prendasEntity.add(prendaEntity);
+            }
+        }
+
+        entity.setPrendas(prendasEntity);
+        return entity;
     }
     //--------------------------------------------------------
 }