X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fdata%2FAlbum.java;h=89823a81786f4158e8bc0eff21a411f3fa06217a;hb=e71cce762443757a193067b13a82fb7ae92b28b2;hp=60fb296373a037dc7d8237415ecf00b772199169;hpb=e31ead2b972074c1e5ecc5213b2bdc34c453d8b6;p=Sone.git diff --git a/src/main/java/net/pterodactylus/sone/data/Album.java b/src/main/java/net/pterodactylus/sone/data/Album.java index 60fb296..89823a8 100644 --- a/src/main/java/net/pterodactylus/sone/data/Album.java +++ b/src/main/java/net/pterodactylus/sone/data/Album.java @@ -18,7 +18,9 @@ package net.pterodactylus.sone.data; import java.util.ArrayList; +import java.util.LinkedHashMap; import java.util.List; +import java.util.Map; import java.util.UUID; import net.pterodactylus.util.validation.Validation; @@ -40,17 +42,20 @@ public class Album implements Fingerprintable { private final List albums = new ArrayList(); /** The images in this album. */ - private final List images = new ArrayList(); + private final Map images = new LinkedHashMap(); /** The parent album. */ private Album parent; - /** The name of this album. */ - private String name; + /** The title of this album. */ + private String title; /** The description of this album. */ private String description; + /** The ID of the album picture. */ + private String albumImage; + /** * Creates a new album with a random ID. */ @@ -100,7 +105,7 @@ public class Album implements Fingerprintable { * @return This album */ public Album setSone(Sone sone) { - Validation.begin().isNull("Current Album Owner", this.sone).isNotNull("New Album Owner", sone).check().isEqual("New Album Owner", sone, this.sone).check(); + Validation.begin().isNotNull("New Album Owner", sone).isEither("Old Album Owner", this.sone, null, sone).check(); this.sone = sone; return this; } @@ -110,7 +115,7 @@ public class Album implements Fingerprintable { * * @return The nested albums */ - public List getNestedAlbums() { + public List getAlbums() { return new ArrayList(albums); } @@ -121,9 +126,11 @@ public class Album implements Fingerprintable { * The album to add */ public void addAlbum(Album album) { - Validation.begin().isNotNull("Album", album).check().isEqual("Album Owner", album.sone, sone).isNull("Album Parent", album.parent).check(); - albums.add(album); + Validation.begin().isNotNull("Album", album).check().isEqual("Album Owner", album.sone, sone).isEither("Old Album Parent", this.parent, null, album.parent).check(); album.setParent(this); + if (!albums.contains(album)) { + albums.add(album); + } } /** @@ -144,7 +151,7 @@ public class Album implements Fingerprintable { * @return The images in this album */ public List getImages() { - return new ArrayList(images); + return new ArrayList(images.values()); } /** @@ -154,8 +161,17 @@ public class Album implements Fingerprintable { * The image to add */ public void addImage(Image image) { - Validation.begin().isNotNull("Image", image).check().isEqual("Image Owner", image.getSone(), sone).check(); - images.add(image); + Validation.begin().isNotNull("Image", image).check().isNotNull("Image Owner", image.getSone()).check().isEqual("Image Owner", image.getSone(), sone).check(); + if (image.getAlbum() != null) { + image.getAlbum().removeImage(image); + } + image.setAlbum(this); + if (images.isEmpty()) { + albumImage = image.getId(); + } + if (!images.containsKey(image.getId())) { + images.put(image.getId(), image); + } } /** @@ -167,6 +183,47 @@ public class Album implements Fingerprintable { public void removeImage(Image image) { Validation.begin().isNotNull("Image", image).check().isEqual("Image Owner", image.getSone(), sone).check(); images.remove(image); + if (image.getId().equals(albumImage)) { + if (images.isEmpty()) { + albumImage = null; + } else { + albumImage = images.values().iterator().next().getId(); + } + } + } + + /** + * Returns the album image of this album, or {@code null} if no album image + * has been set. + * + * @return The image to show when this album is listed + */ + public Image getAlbumImage() { + if (albumImage == null) { + return null; + } + return images.get(albumImage); + } + + /** + * Sets the ID of the album image. + * + * @param id + * The ID of the album image + * @return This album + */ + public Album setAlbumImage(String id) { + this.albumImage = id; + return this; + } + + /** + * Returns whether this album contains any other albums or images. + * + * @return {@code true} if this album is empty, {@code false} otherwise + */ + public boolean isEmpty() { + return albums.isEmpty() && images.isEmpty(); } /** @@ -198,30 +255,29 @@ public class Album implements Fingerprintable { * @return This album */ protected Album removeParent() { - Validation.begin().isNotNull("Album Parent", parent).check(); this.parent = null; return this; } /** - * Returns the name of this album. + * Returns the title of this album. * - * @return The name of this album + * @return The title of this album */ - public String getName() { - return name; + public String getTitle() { + return title; } /** - * Sets the name of this album. + * Sets the title of this album. * - * @param name - * The name of this album + * @param title + * The title of this album * @return This album */ - public Album setName(String name) { - Validation.begin().isNotNull("Album Name", name).check(); - this.name = name; + public Album setTitle(String title) { + Validation.begin().isNotNull("Album Title", title).check(); + this.title = title; return this; } @@ -259,7 +315,7 @@ public class Album implements Fingerprintable { StringBuilder fingerprint = new StringBuilder(); fingerprint.append("Album("); fingerprint.append("ID(").append(id).append(')'); - fingerprint.append("Name(").append(name).append(')'); + fingerprint.append("Title(").append(title).append(')'); fingerprint.append("Description(").append(description).append(')'); /* add nested albums. */ @@ -271,8 +327,10 @@ public class Album implements Fingerprintable { /* add images. */ fingerprint.append("Images("); - for (Image image : images) { - fingerprint.append(image.getFingerprint()); + for (Image image : images.values()) { + if (image.isInserted()) { + fingerprint.append(image.getFingerprint()); + } } fingerprint.append(')'); @@ -288,6 +346,14 @@ public class Album implements Fingerprintable { * {@inheritDoc} */ @Override + public int hashCode() { + return id.hashCode(); + } + + /** + * {@inheritDoc} + */ + @Override public boolean equals(Object object) { if (!(object instanceof Album)) { return false;