X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fdata%2FAlbum.java;h=792f5181ecd2ed9279435528b5da3c8dbe83ee98;hb=cf9d466a99e30e2c23694da9037c12de6372aaae;hp=a85a6abd5c97fe4297499e07333b6dae9c054f90;hpb=750189cec7fe3b815fa9b98625d66c61715eb443;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 a85a6ab..792f518 100644
--- a/src/main/java/net/pterodactylus/sone/data/Album.java
+++ b/src/main/java/net/pterodactylus/sone/data/Album.java
@@ -19,6 +19,9 @@ package net.pterodactylus.sone.data;
 
 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.
@@ -27,6 +30,12 @@ import java.util.List;
  */
 public class Album {
 
+	/** The ID of this album. */
+	private final String id;
+
+	/** The Sone this album belongs to. */
+	private final Sone sone;
+
 	/** Nested albums. */
 	private final List<Album> albums = new ArrayList<Album>();
 
@@ -36,11 +45,56 @@ public class Album {
 	/** The name of this album. */
 	private String name;
 
+	/** The description of this album. */
+	private String description;
+
+	/**
+	 * 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);
+	}
+
+	/**
+	 * Creates a new album with the given ID.
+	 *
+	 * @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();
+		this.id = id;
+		this.sone = sone;
+	}
+
 	//
 	// ACCESSORS
 	//
 
 	/**
+	 * Returns the ID of this album.
+	 *
+	 * @return The ID of this album
+	 */
+	public String getId() {
+		return id;
+	}
+
+	/**
+	 * Returns the Sone this album belongs to.
+	 *
+	 * @return The Sone this album belongs to
+	 */
+	public Sone getSone() {
+		return sone;
+	}
+
+	/**
 	 * Returns the nested albums.
 	 *
 	 * @return The nested albums
@@ -75,8 +129,47 @@ public class Album {
 	 * @return This album
 	 */
 	public Album setName(String name) {
+		Validation.begin().isNotNull("Album Name", name).check();
 		this.name = name;
 		return this;
 	}
 
+	/**
+	 * Returns the description of this album.
+	 *
+	 * @return The description of this album
+	 */
+	public String getDescription() {
+		return description;
+	}
+
+	/**
+	 * Sets the description of this album.
+	 *
+	 * @param description
+	 *            The description of this album
+	 * @return This album
+	 */
+	public Album setDescription(String description) {
+		Validation.begin().isNotNull("Album Description", description).check();
+		this.description = description;
+		return this;
+	}
+
+	//
+	// OBJECT METHODS
+	//
+
+	/**
+	 * {@inheritDoc}
+	 */
+	@Override
+	public boolean equals(Object object) {
+		if (!(object instanceof Album)) {
+			return false;
+		}
+		Album album = (Album) object;
+		return sone.equals(album.sone) && id.equals(album.id);
+	}
+
 }