X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fdata%2FAlbum.java;h=d808e7c40337fb2072a9cb7f5dbd343f0157d648;hb=77b74fe98ac3880ad8491048f3a482feb3518ba6;hp=76015c82bab446a0b72e9e2593517278bf407470;hpb=f38f21ad0a7703d84f826b01c4e107e4c7ba5fe6;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 76015c8..d808e7c 100644
--- a/src/main/java/net/pterodactylus/sone/data/Album.java
+++ b/src/main/java/net/pterodactylus/sone/data/Album.java
@@ -34,7 +34,7 @@ public class Album implements Fingerprintable {
 	private final String id;
 
 	/** The Sone this album belongs to. */
-	private final Sone sone;
+	private Sone sone;
 
 	/** Nested albums. */
 	private final List<Album> albums = new ArrayList<Album>();
@@ -51,14 +51,14 @@ public class Album implements Fingerprintable {
 	/** The description of this album. */
 	private String description;
 
+	/** The index of the album picture. */
+	private int albumImage = -1;
+
 	/**
 	 * Creates a new album with a random ID.
-	 *
-	 * @param sone
-	 *            The Sone this album belongs to
 	 */
-	public Album(Sone sone) {
-		this(UUID.randomUUID().toString(), sone);
+	public Album() {
+		this(UUID.randomUUID().toString());
 	}
 
 	/**
@@ -66,13 +66,10 @@ public class Album implements Fingerprintable {
 	 *
 	 * @param id
 	 *            The ID of the album
-	 * @param sone
-	 *            The Sone this album belongs to
 	 */
-	public Album(String id, Sone sone) {
-		Validation.begin().isNotNull("Album ID", id).isNotNull("Album Owner", sone).check();
+	public Album(String id) {
+		Validation.begin().isNotNull("Album ID", id).check();
 		this.id = id;
-		this.sone = sone;
 	}
 
 	//
@@ -98,11 +95,25 @@ public class Album implements Fingerprintable {
 	}
 
 	/**
+	 * Sets the owner of the album. The owner can only be set as long as the
+	 * current owner is {@code null}.
+	 *
+	 * @param sone
+	 *            The album owner
+	 * @return This album
+	 */
+	public Album setSone(Sone sone) {
+		Validation.begin().isNull("Current Album Owner", this.sone).isNotNull("New Album Owner", sone).check();
+		this.sone = sone;
+		return this;
+	}
+
+	/**
 	 * Returns the nested albums.
 	 *
 	 * @return The nested albums
 	 */
-	public List<Album> getNestedAlbums() {
+	public List<Album> getAlbums() {
 		return new ArrayList<Album>(albums);
 	}
 
@@ -146,7 +157,7 @@ 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();
+		Validation.begin().isNotNull("Image", image).check().isNotNull("Image Owner", image.getSone()).check().isEqual("Image Owner", image.getSone(), sone).check();
 		images.add(image);
 	}
 
@@ -162,6 +173,19 @@ public class Album implements Fingerprintable {
 	}
 
 	/**
+	 * 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 == -1) {
+			return null;
+		}
+		return images.get(albumImage);
+	}
+
+	/**
 	 * Returns the parent album of this album.
 	 *
 	 * @return The parent album of this album, or {@code null} if this album
@@ -285,7 +309,7 @@ public class Album implements Fingerprintable {
 			return false;
 		}
 		Album album = (Album) object;
-		return sone.equals(album.sone) && id.equals(album.id);
+		return id.equals(album.id);
 	}
 
 }