X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fdata%2Fimpl%2FAbstractSoneBuilder.java;h=5ec8ec4244933d5aea7025820909bbc280c87442;hb=f8672b1385173a103d7f085d8e9cd43bc5762d71;hp=1ce38e4ed0c358384091f915b701217e04614186;hpb=3752ebf20fcb0c71287f2b7fed358b14c7649d7f;p=Sone.git

diff --git a/src/main/java/net/pterodactylus/sone/data/impl/AbstractSoneBuilder.java b/src/main/java/net/pterodactylus/sone/data/impl/AbstractSoneBuilder.java
index 1ce38e4..5ec8ec4 100644
--- a/src/main/java/net/pterodactylus/sone/data/impl/AbstractSoneBuilder.java
+++ b/src/main/java/net/pterodactylus/sone/data/impl/AbstractSoneBuilder.java
@@ -2,7 +2,12 @@ package net.pterodactylus.sone.data.impl;
 
 import static com.google.common.base.Preconditions.checkState;
 
+import java.util.Collection;
+import java.util.HashSet;
+
 import net.pterodactylus.sone.data.Client;
+import net.pterodactylus.sone.data.Post;
+import net.pterodactylus.sone.data.PostReply;
 import net.pterodactylus.sone.database.SoneBuilder;
 import net.pterodactylus.sone.freenet.wot.Identity;
 import net.pterodactylus.sone.freenet.wot.OwnIdentity;
@@ -15,9 +20,10 @@ import net.pterodactylus.sone.freenet.wot.OwnIdentity;
 public abstract class AbstractSoneBuilder implements SoneBuilder {
 
 	protected Identity identity;
-	protected boolean local;
 	protected long lastUpdated;
 	protected Client client;
+	protected final Collection<Post> posts = new HashSet<Post>();
+	protected final Collection<PostReply> postReplies = new HashSet<PostReply>();
 
 	@Override
 	public SoneBuilder from(Identity identity) {
@@ -26,12 +32,6 @@ public abstract class AbstractSoneBuilder implements SoneBuilder {
 	}
 
 	@Override
-	public SoneBuilder local() {
-		this.local = true;
-		return this;
-	}
-
-	@Override
 	public SoneBuilder lastUpdated(long lastUpdated) {
 		this.lastUpdated = lastUpdated;
 		return this;
@@ -43,12 +43,30 @@ public abstract class AbstractSoneBuilder implements SoneBuilder {
 		return this;
 	}
 
+	@Override
+	public SoneBuilder withPosts(Collection<Post> posts) {
+		this.posts.clear();
+		this.posts.addAll(posts);
+		return this;
+	}
+
+	@Override
+	public SoneBuilder withPostReplies(Collection<PostReply> postReplies) {
+		this.postReplies.clear();
+		this.postReplies.addAll(postReplies);
+		return this;
+	}
+
 	protected void validate() throws IllegalStateException {
 		checkState(identity != null, "identity must not be null");
-		checkState(!local || (identity instanceof OwnIdentity),
-				"can not create local Sone from remote identity");
 		checkState(lastUpdated > 0, "last update time must be set");
 		checkState(client != null, "client must not be null");
 	}
 
+	protected void validateLocal() throws IllegalStateException {
+		validate();
+		checkState(identity instanceof OwnIdentity,
+				"identity must be an own identity for a local Sone");
+	}
+
 }