Set Sone in album builder, remote getOrCreate method from core.

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

diff --git a/src/main/java/net/pterodactylus/sone/data/impl/AbstractAlbumBuilder.java b/src/main/java/net/pterodactylus/sone/data/impl/AbstractAlbumBuilder.java
index 312c795..8e15b6f 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/data/impl/AbstractAlbumBuilder.java
+++ b/src/main/java/net/pterodactylus/sone/data/impl/AbstractAlbumBuilder.java
@@ -19,6 +19,7 @@ package net.pterodactylus.sone.data.impl;
 
 import static com.google.common.base.Preconditions.checkState;
 
+import net.pterodactylus.sone.data.Sone;
 import net.pterodactylus.sone.database.AlbumBuilder;
 
 /**
@@ -34,6 +35,7 @@ public abstract class AbstractAlbumBuilder implements AlbumBuilder {
 
        /** The ID of the album to create. */
        protected String id;
+       protected Sone sone;
 
        @Override
        public AlbumBuilder randomId() {
@@ -47,6 +49,11 @@ public abstract class AbstractAlbumBuilder implements AlbumBuilder {
                return this;
        }
 
+       public AlbumBuilder by(Sone sone) {
+               this.sone = sone;
+               return this;
+       }
+
        //
        // PROTECTED METHODS
        //
@@ -59,6 +66,7 @@ public abstract class AbstractAlbumBuilder implements AlbumBuilder {
         */
        protected void validate() throws IllegalStateException {
                checkState((randomId && (id == null)) || (!randomId && (id != null)), "exactly one of random ID or custom ID must be set");
+               checkState(sone != null, "Sone must not be null");
        }
 
 }