Simplify album removal.

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

diff --git a/src/main/java/net/pterodactylus/sone/data/impl/DefaultAlbumBuilder.java b/src/main/java/net/pterodactylus/sone/data/impl/DefaultAlbumBuilder.java
index f3e771c..5148471 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/data/impl/DefaultAlbumBuilder.java
+++ b/src/main/java/net/pterodactylus/sone/data/impl/DefaultAlbumBuilder.java
@@ -18,6 +18,7 @@
 package net.pterodactylus.sone.data.impl;
 
 import net.pterodactylus.sone.data.Album;
+import net.pterodactylus.sone.data.Sone;
 import net.pterodactylus.sone.database.AlbumBuilder;
 
 /**
@@ -27,10 +28,18 @@ import net.pterodactylus.sone.database.AlbumBuilder;
  */
 public class DefaultAlbumBuilder extends AbstractAlbumBuilder {
 
+       private final Sone sone;
+       private final DefaultAlbum parent;
+
+       public DefaultAlbumBuilder(Sone sone, DefaultAlbum parent) {
+               this.sone = sone;
+               this.parent = parent;
+       }
+
        @Override
        public Album build() throws IllegalStateException {
                validate();
-               return randomId ? new DefaultAlbum() : new DefaultAlbum(id);
+               return new DefaultAlbum(getId(), sone, parent);
        }
 
 }