Add function that transforms a Sone into its nice name.

[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 cd6475a..a2f198f 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/data/impl/DefaultAlbumBuilder.java
+++ b/src/main/java/net/pterodactylus/sone/data/impl/DefaultAlbumBuilder.java
@@ -20,24 +20,32 @@ package net.pterodactylus.sone.data.impl;
 import net.pterodactylus.sone.data.Album;
 import net.pterodactylus.sone.data.Sone;
 import net.pterodactylus.sone.database.AlbumBuilder;
+import net.pterodactylus.sone.database.Database;
 
 /**
- * {@link AlbumBuilder} implementation that creates {@link DefaultAlbum} objects.
+ * {@link AlbumBuilder} implementation that creates {@link DefaultAlbum}
+ * objects.
  *
  * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
  */
 public class DefaultAlbumBuilder extends AbstractAlbumBuilder {
 
+       private final Database database;
        private final Sone sone;
+       private final String parentId;
 
-       public DefaultAlbumBuilder(Sone sone) {
+       public DefaultAlbumBuilder(Database database, Sone sone, String parentId) {
+               this.database = database;
                this.sone = sone;
+               this.parentId = parentId;
        }
 
        @Override
        public Album build() throws IllegalStateException {
                validate();
-               return new DefaultAlbum(getId(), sone);
+               DefaultAlbum album = new DefaultAlbum(database, getId(), sone, parentId);
+               database.storeAlbum(album);
+               return album;
        }
 
 }