Simplify image builders.

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

diff --git a/src/main/java/net/pterodactylus/sone/data/impl/ImageBuilderImpl.java b/src/main/java/net/pterodactylus/sone/data/impl/ImageBuilderImpl.java
index 83e1dbc..d3f060d 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/data/impl/ImageBuilderImpl.java
+++ b/src/main/java/net/pterodactylus/sone/data/impl/ImageBuilderImpl.java
@@ -17,8 +17,7 @@
 
 package net.pterodactylus.sone.data.impl;
 
-import static java.util.UUID.randomUUID;
-
+import net.pterodactylus.sone.data.Album;
 import net.pterodactylus.sone.data.Image;
 import net.pterodactylus.sone.data.ImageImpl;
 import net.pterodactylus.sone.database.ImageBuilder;
@@ -30,12 +29,14 @@ import net.pterodactylus.sone.database.ImageBuilder;
  */
 public class ImageBuilderImpl extends AbstractImageBuilder {
 
+       public ImageBuilderImpl(Album album) {
+               super(album);
+       }
+
        @Override
        public Image build() throws IllegalStateException {
                validate();
-               String id = randomId ? randomUUID().toString() : this.id;
-               long creationTime = createdNow ? System.currentTimeMillis() : this.creationTime;
-               return new ImageImpl(id, sone, creationTime, key, width, height);
+               return new ImageImpl(getId(), sone, album, key, getCreationTime(), width, height);
        }
 
 }