X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fdata%2FImage.java;fp=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fdata%2FImage.java;h=d6e7114c338cc57d4d2f27f10fd240528ad0b6f4;hp=f38a4facedfde21660f1668374bf615439106f8e;hb=5e142aaabb86f62c3962a4bf60b8a79f4e822f5b;hpb=70ffa07939b79f80257cf5140768e9b90e7e7800

diff --git a/src/main/java/net/pterodactylus/sone/data/Image.java b/src/main/java/net/pterodactylus/sone/data/Image.java
index f38a4fa..d6e7114 100644
--- a/src/main/java/net/pterodactylus/sone/data/Image.java
+++ b/src/main/java/net/pterodactylus/sone/data/Image.java
@@ -34,6 +34,9 @@ public class Image implements Fingerprintable {
 	/** The Sone the image belongs to. */
 	private Sone sone;
 
+	/** The album this image belongs to. */
+	private Album album;
+
 	/** The request key of the image. */
 	private String key;
 
@@ -108,6 +111,29 @@ public class Image implements Fingerprintable {
 	}
 
 	/**
+	 * Returns the album this image belongs to.
+	 *
+	 * @return The album this image belongs to
+	 */
+	public Album getAlbum() {
+		return album;
+	}
+
+	/**
+	 * Sets the album this image belongs to. The album of an image can only be
+	 * set once, and it is usually called by {@link Album#addImage(Image)}.
+	 *
+	 * @param album
+	 *            The album this image belongs to
+	 * @return This image
+	 */
+	public Image setAlbum(Album album) {
+		Validation.begin().isNull("Current Album", this.album).isNotNull("New Album", album).check().isEqual("Album Owner and Image Owner", album.getSone(), getSone()).check();
+		this.album = album;
+		return this;
+	}
+
+	/**
 	 * Returns the request key of this image.
 	 *
 	 * @return The request key of this image