X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fdatabase%2Fmemory%2FConfigurationLoader.java;h=1691ddb7ffa09bba3dcfaee188ef4b5f74e80781;hb=f229fe41f708d2b275c20ceb9aba5993761218a3;hp=b9ac859fead3bde807e5c3c311491425a7dc63b8;hpb=d9ecda2e1e4411a0b5da8fb760091b6f659e961f;p=Sone.git

diff --git a/src/main/java/net/pterodactylus/sone/database/memory/ConfigurationLoader.java b/src/main/java/net/pterodactylus/sone/database/memory/ConfigurationLoader.java
index b9ac859..1691ddb 100644
--- a/src/main/java/net/pterodactylus/sone/database/memory/ConfigurationLoader.java
+++ b/src/main/java/net/pterodactylus/sone/database/memory/ConfigurationLoader.java
@@ -1,9 +1,14 @@
 package net.pterodactylus.sone.database.memory;
 
+import static java.util.logging.Level.WARNING;
+
+import java.util.Collection;
 import java.util.HashSet;
 import java.util.Set;
+import java.util.logging.Logger;
 
 import net.pterodactylus.util.config.Configuration;
+import net.pterodactylus.util.config.ConfigurationException;
 
 /**
  * Helper class for interacting with a {@link Configuration}.
@@ -12,12 +17,22 @@ import net.pterodactylus.util.config.Configuration;
  */
 public class ConfigurationLoader {
 
+	private static final Logger logger =
+			Logger.getLogger("Sone.Database.Memory.Configuration");
 	private final Configuration configuration;
 
 	public ConfigurationLoader(Configuration configuration) {
 		this.configuration = configuration;
 	}
 
+	public synchronized Set<String> loadFriends(String localSoneId) {
+		return loadIds("Sone/" + localSoneId + "/Friends");
+	}
+
+	public void saveFriends(String soneId, Collection<String> friends) {
+		saveIds("Sone/" + soneId + "/Friends", friends);
+	}
+
 	public synchronized Set<String> loadKnownPosts() {
 		return loadIds("KnownPosts");
 	}
@@ -45,4 +60,25 @@ public class ConfigurationLoader {
 		return ids;
 	}
 
+	public synchronized void saveBookmarkedPosts(
+			Set<String> bookmarkedPosts) {
+		saveIds("Bookmarks/Post", bookmarkedPosts);
+	}
+
+	private void saveIds(String prefix, Collection<String> ids) {
+		try {
+			int idCounter = 0;
+			for (String id : ids) {
+				configuration
+						.getStringValue(prefix + "/" + idCounter++ + "/ID")
+						.setValue(id);
+			}
+			configuration
+					.getStringValue(prefix + "/" + idCounter + "/ID")
+					.setValue(null);
+		} catch (ConfigurationException ce1) {
+			logger.log(WARNING, "Could not save bookmarked posts!", ce1);
+		}
+	}
+
 }