X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fdata%2Fimpl%2FAbstractImageBuilder.java;h=209ef0f3003052e9db38cbae5e7f2c4dd4afd8ce;hp=1b9e3a86037aa6cdd37b014043060ff1d533ca68;hb=fc2165a3e44887c18d6faca054cc4efa384ca797;hpb=58497297d2b9a18cd2877a226870acfe9e8837af

diff --git a/src/main/java/net/pterodactylus/sone/data/impl/AbstractImageBuilder.java b/src/main/java/net/pterodactylus/sone/data/impl/AbstractImageBuilder.java
index 1b9e3a8..209ef0f 100644
--- a/src/main/java/net/pterodactylus/sone/data/impl/AbstractImageBuilder.java
+++ b/src/main/java/net/pterodactylus/sone/data/impl/AbstractImageBuilder.java
@@ -16,8 +16,10 @@
  */
 package net.pterodactylus.sone.data.impl;
 
+import static com.google.common.base.Preconditions.checkNotNull;
 import static com.google.common.base.Preconditions.checkState;
 
+import net.pterodactylus.sone.data.Album;
 import net.pterodactylus.sone.data.Sone;
 import net.pterodactylus.sone.database.ImageBuilder;
 
@@ -29,6 +31,8 @@ import net.pterodactylus.sone.database.ImageBuilder;
  */
 public abstract class AbstractImageBuilder implements ImageBuilder {
 
+	protected final Album album;
+
 	/** Whether to create an album with a random ID. */
 	protected boolean randomId;
 
@@ -41,6 +45,10 @@ public abstract class AbstractImageBuilder implements ImageBuilder {
 	protected int width;
 	protected int height;
 
+	public AbstractImageBuilder(Album album) {
+		this.album = checkNotNull(album, "album must not be null");
+	}
+
 	@Override
 	public ImageBuilder randomId() {
 		randomId = true;