X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fdata%2FAlbum.java;h=76015c82bab446a0b72e9e2593517278bf407470;hb=f38f21ad0a7703d84f826b01c4e107e4c7ba5fe6;hp=792f5181ecd2ed9279435528b5da3c8dbe83ee98;hpb=cf9d466a99e30e2c23694da9037c12de6372aaae;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 792f518..76015c8 100644
--- a/src/main/java/net/pterodactylus/sone/data/Album.java
+++ b/src/main/java/net/pterodactylus/sone/data/Album.java
@@ -28,7 +28,7 @@ import net.pterodactylus.util.validation.Validation;
  *
  * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
  */
-public class Album {
+public class Album implements Fingerprintable {
 
 	/** The ID of this album. */
 	private final String id;
@@ -42,6 +42,9 @@ public class Album {
 	/** The images in this album. */
 	private final List<Image> images = new ArrayList<Image>();
 
+	/** The parent album. */
+	private Album parent;
+
 	/** The name of this album. */
 	private String name;
 
@@ -104,6 +107,30 @@ public class Album {
 	}
 
 	/**
+	 * Adds an album to this album.
+	 *
+	 * @param album
+	 *            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);
+		album.setParent(this);
+	}
+
+	/**
+	 * Removes an album from this album.
+	 *
+	 * @param album
+	 *            The album to remove
+	 */
+	public void removeAlbum(Album album) {
+		Validation.begin().isNotNull("Album", album).check().isEqual("Album Owner", album.sone, sone).isEqual("Album Parent", album.parent, this).check();
+		albums.remove(album);
+		album.removeParent();
+	}
+
+	/**
 	 * Returns the images in this album.
 	 *
 	 * @return The images in this album
@@ -113,6 +140,62 @@ public class Album {
 	}
 
 	/**
+	 * Adds the given image to this album.
+	 *
+	 * @param image
+	 *            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);
+	}
+
+	/**
+	 * Removes the given image from this album.
+	 *
+	 * @param image
+	 *            The image to remove
+	 */
+	public void removeImage(Image image) {
+		Validation.begin().isNotNull("Image", image).check().isEqual("Image Owner", image.getSone(), sone).check();
+		images.remove(image);
+	}
+
+	/**
+	 * Returns the parent album of this album.
+	 *
+	 * @return The parent album of this album, or {@code null} if this album
+	 *         does not have a parent
+	 */
+	public Album getParent() {
+		return parent;
+	}
+
+	/**
+	 * Sets the parent album of this album.
+	 *
+	 * @param parent
+	 *            The new parent album of this album
+	 * @return This album
+	 */
+	protected Album setParent(Album parent) {
+		Validation.begin().isNotNull("Album Parent", parent).check();
+		this.parent = parent;
+		return this;
+	}
+
+	/**
+	 * Removes the parent album of this album.
+	 *
+	 * @return This album
+	 */
+	protected Album removeParent() {
+		Validation.begin().isNotNull("Album Parent", parent).check();
+		this.parent = null;
+		return this;
+	}
+
+	/**
 	 * Returns the name of this album.
 	 *
 	 * @return The name of this album
@@ -157,6 +240,39 @@ public class Album {
 	}
 
 	//
+	// FINGERPRINTABLE METHODS
+	//
+
+	/**
+	 * {@inheritDoc}
+	 */
+	@Override
+	public String getFingerprint() {
+		StringBuilder fingerprint = new StringBuilder();
+		fingerprint.append("Album(");
+		fingerprint.append("ID(").append(id).append(')');
+		fingerprint.append("Name(").append(name).append(')');
+		fingerprint.append("Description(").append(description).append(')');
+
+		/* add nested albums. */
+		fingerprint.append("Albums(");
+		for (Album album : albums) {
+			fingerprint.append(album.getFingerprint());
+		}
+		fingerprint.append(')');
+
+		/* add images. */
+		fingerprint.append("Images(");
+		for (Image image : images) {
+			fingerprint.append(image.getFingerprint());
+		}
+		fingerprint.append(')');
+
+		fingerprint.append(')');
+		return fingerprint.toString();
+	}
+
+	//
 	// OBJECT METHODS
 	//