diff --git a/01-lecture/quemepongo/src/main/java/quemepongo/model/usuario/guardarropa/Guardarropa.java b/01-lecture/quemepongo/src/main/java/quemepongo/model/usuario/guardarropa/Guardarropa.java new file mode 100644 index 0000000..c44ff24 --- /dev/null +++ b/01-lecture/quemepongo/src/main/java/quemepongo/model/usuario/guardarropa/Guardarropa.java @@ -0,0 +1,48 @@ +package quemepongo.model.usuario.guardarropa; + +import java.util.HashMap; +import java.util.LinkedHashSet; +import java.util.Map; +import java.util.Set; +import quemepongo.model.prenda.Categoria; +import quemepongo.model.prenda.Prenda; + +/** + * Guardarropa, repositorio de Prendas. + * + * @since 05.27.2021 + * @version 1.0 + */ +public class Guardarropa { + + /** + * Prendas categorizadas. + * + * @since Iteración V + */ + Map> prendas = new HashMap>(); + + public Guardarropa() { + + Categoria[] categorias = Categoria.values(); + + for (Categoria categoria : categorias) { + prendas.put(categoria, new LinkedHashSet()); + } + } + + /** + * Agrega una prenda al guardarropas. + * + * @param prenda la prenda a añadir + * @return el guardarropa + * @since Iteración V + */ + public Guardarropa agregarPrenda(Prenda prenda) { + + Set prendasDeCategoria = prendas.get(prenda.categoria()); + prendasDeCategoria.add(prenda); + + return this; + } +}