X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fdata%2FAlbum.java;h=62357d1913045c44418ec0ed1456df8c0daf1ba1;hb=63c32a955bb06e5b3c92f87f62b9914bd9606281;hp=50b72546c2f198fb779c1a51f007bf0a9c3fe735;hpb=ca43577fb67eaa9199a4f42e62d4c802814ec9fe;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 50b7254..62357d1 100644
--- a/src/main/java/net/pterodactylus/sone/data/Album.java
+++ b/src/main/java/net/pterodactylus/sone/data/Album.java
@@ -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;
 
@@ -110,8 +113,9 @@ public class Album {
 	 *            The album to add
 	 */
 	public void addAlbum(Album album) {
-		Validation.begin().isNotNull("Album", album).check().isEqual("Album Owner", album.sone, sone).check();
+		Validation.begin().isNotNull("Album", album).check().isEqual("Album Owner", album.sone, sone).isNull("Album Parent", album.parent).check();
 		albums.add(album);
+		album.setParent(this);
 	}
 
 	/**
@@ -121,8 +125,9 @@ public class Album {
 	 *            The album to remove
 	 */
 	public void removeAlbum(Album album) {
-		Validation.begin().isNotNull("Album", album).check().isEqual("Album Owner", album.sone, sone).check();
+		Validation.begin().isNotNull("Album", album).check().isEqual("Album Owner", album.sone, sone).isEqual("Album Parent", album.parent, this).check();
 		albums.remove(album);
+		album.removeParent();
 	}
 
 	/**
@@ -135,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