Store DefaultAlbum in DefaultImage.

[Sone.git] / src / main / java / net / pterodactylus / sone / data / impl / DefaultImageBuilder.java

diff --git a/src/main/java/net/pterodactylus/sone/data/impl/DefaultImageBuilder.java b/src/main/java/net/pterodactylus/sone/data/impl/DefaultImageBuilder.java
index 7f7f53f..12f20c2 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/data/impl/DefaultImageBuilder.java
+++ b/src/main/java/net/pterodactylus/sone/data/impl/DefaultImageBuilder.java
@@ -17,8 +17,10 @@
 
 package net.pterodactylus.sone.data.impl;
 
-import net.pterodactylus.sone.data.Album;
+import static com.google.common.base.Preconditions.checkNotNull;
+
 import net.pterodactylus.sone.data.Image;
+import net.pterodactylus.sone.data.Sone;
 import net.pterodactylus.sone.database.ImageBuilder;
 
 /**
@@ -28,8 +30,12 @@ import net.pterodactylus.sone.database.ImageBuilder;
  */
 public class DefaultImageBuilder extends AbstractImageBuilder {
 
-       public DefaultImageBuilder(Album album) {
-               super(album);
+       protected final Sone sone;
+       protected final DefaultAlbum album;
+
+       public DefaultImageBuilder(Sone sone, DefaultAlbum album) {
+               this.sone = checkNotNull(sone, "sone must not be null");
+               this.album = checkNotNull(album, "album must not be null");
        }
 
        @Override