Move Sone parsing into its own class.

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

diff --git a/src/main/java/net/pterodactylus/sone/data/impl/DefaultAlbum.java b/src/main/java/net/pterodactylus/sone/data/impl/DefaultAlbum.java
index c1640f9..0006dc4 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/data/impl/DefaultAlbum.java
+++ b/src/main/java/net/pterodactylus/sone/data/impl/DefaultAlbum.java
@@ -90,11 +90,14 @@ public class DefaultAlbum extends AbstractAlbum {
        public ImageBuilder newImageBuilder() throws IllegalStateException {
                return new AbstractImageBuilder() {
                        @Override
-                       public Image build() throws IllegalStateException {
+                       public Image build(Optional<ImageCreated> imageCreated) throws IllegalStateException {
                                validate();
-                               DefaultImage memoryImage = new DefaultImage(database, getId(), sone, DefaultAlbum.this.id, key, getCreationTime(), width, height);
-                               database.storeImage(memoryImage);
-                               return memoryImage;
+                               DefaultImage image = new DefaultImage(database, getId(), sone, DefaultAlbum.this.id, key, getCreationTime(), width, height);
+                               database.storeImage(image);
+                               if (imageCreated.isPresent()) {
+                                       imageCreated.get().imageCreated(image);
+                               }
+                               return image;
                        }
                };
        }