X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fdata%2FAlbum.java;h=d53ba6e20d27ced5f0b15eb2ffcf7a67cc0b6004;hp=67bfefdc4f0e78c4bb32bd7292260ae6bee1124c;hb=55ee8d58d39bf1e6f0edc3baebb1b1119f92a6c1;hpb=fef93a83c728ca853ea93872e870130512af3fcb

diff --git a/src/main/java/net/pterodactylus/sone/data/Album.java b/src/main/java/net/pterodactylus/sone/data/Album.java
index 67bfefd..d53ba6e 100644
--- a/src/main/java/net/pterodactylus/sone/data/Album.java
+++ b/src/main/java/net/pterodactylus/sone/data/Album.java
@@ -21,6 +21,8 @@ import java.util.ArrayList;
 import java.util.List;
 import java.util.UUID;
 
+import net.pterodactylus.util.validation.Validation;
+
 /**
  * Container for images that can also contain nested {@link Album}s.
  *
@@ -65,6 +67,7 @@ public class Album {
 	 *            The Sone this album belongs to
 	 */
 	public Album(String id, Sone sone) {
+		Validation.begin().isNotNull("Album ID", id).isNotNull("Album Owner", sone).check();
 		this.id = id;
 		this.sone = sone;
 	}
@@ -101,6 +104,28 @@ 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).check();
+		albums.add(album);
+	}
+
+	/**
+	 * 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).check();
+		albums.remove(album);
+	}
+
+	/**
 	 * Returns the images in this album.
 	 *
 	 * @return The images in this album
@@ -110,6 +135,28 @@ 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 name of this album.
 	 *
 	 * @return The name of this album
@@ -126,6 +173,7 @@ public class Album {
 	 * @return This album
 	 */
 	public Album setName(String name) {
+		Validation.begin().isNotNull("Album Name", name).check();
 		this.name = name;
 		return this;
 	}
@@ -147,6 +195,7 @@ public class Album {
 	 * @return This album
 	 */
 	public Album setDescription(String description) {
+		Validation.begin().isNotNull("Album Description", description).check();
 		this.description = description;
 		return this;
 	}