Save and load known Sones.

[Sone.git] / src / main / java / net / pterodactylus / sone / core / Core.java

diff --git a/src/main/java/net/pterodactylus/sone/core/Core.java b/src/main/java/net/pterodactylus/sone/core/Core.java
index c861fd1..e944f1f 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/core/Core.java
+++ b/src/main/java/net/pterodactylus/sone/core/Core.java
@@ -559,6 +559,17 @@ public class Core extends AbstractService {
                                        sone.addFriend(friendSone);
                                }
 
+                               /* load blocked Sone IDs. */
+                               int blockedSoneCounter = 0;
+                               while (true) {
+                                       String blockedSonePrefix = sonePrefix + "/BlockedSone." + blockedSoneCounter++;
+                                       String blockedSoneId = configuration.getStringValue(blockedSonePrefix + "/ID").getValue(null);
+                                       if (blockedSoneId == null) {
+                                               break;
+                                       }
+                                       sone.addBlockedSoneId(blockedSoneId);
+                               }
+
                                sone.setModificationCounter(modificationCounter);
                                addLocalSone(sone);
                        } catch (MalformedURLException mue1) {
@@ -567,6 +578,23 @@ public class Core extends AbstractService {
                } while (true);
                logger.log(Level.INFO, "Loaded %d Sones.", getSones().size());
 
+               /* load all known Sones. */
+               int knownSonesCounter = 0;
+               while (true) {
+                       String knownSonePrefix = "KnownSone." + knownSonesCounter++;
+                       String knownSoneId = configuration.getStringValue(knownSonePrefix + "/ID").getValue(null);
+                       if (knownSoneId == null) {
+                               break;
+                       }
+                       String knownSoneName = configuration.getStringValue(knownSonePrefix + "/Name").getValue(null);
+                       String knownSoneKey = configuration.getStringValue(knownSonePrefix + "/Key").getValue(null);
+                       try {
+                               getSone(knownSoneId).setName(knownSoneName).setRequestUri(new FreenetURI(knownSoneKey));
+                       } catch (MalformedURLException mue1) {
+                               logger.log(Level.WARNING, "Could not create Sone from requestUri (“" + knownSoneKey + "”)!", mue1);
+                       }
+               }
+
                /* load all remote Sones. */
                for (Sone remoteSone : getRemoteSones()) {
                        loadSone(remoteSone);
@@ -626,10 +654,29 @@ public class Core extends AbstractService {
                                /* write null ID as terminator. */
                                configuration.getStringValue(sonePrefix + "/Friend." + friendId + "/ID").setValue(null);
 
+                               /* write all blocked Sones. */
+                               int blockedSoneCounter = 0;
+                               for (String blockedSoneId : sone.getBlockedSoneIds()) {
+                                       String blockedSonePrefix = sonePrefix + "/BlockedSone." + blockedSoneCounter++;
+                                       configuration.getStringValue(blockedSonePrefix + "/ID").setValue(blockedSoneId);
+                               }
+                               configuration.getStringValue(sonePrefix + "/BlockedSone." + blockedSoneCounter + "/ID").setValue(null);
+
                        }
                        /* write null ID as terminator. */
                        configuration.getStringValue("Sone/Sone." + soneId + "/ID").setValue(null);
 
+                       /* write all known Sones. */
+                       int knownSonesCounter = 0;
+                       for (Sone knownSone : getRemoteSones()) {
+                               String knownSonePrefix = "KnownSone." + knownSonesCounter++;
+                               configuration.getStringValue(knownSonePrefix + "/ID").setValue(knownSone.getId());
+                               configuration.getStringValue(knownSonePrefix + "/Name").setValue(knownSone.getName());
+                               configuration.getStringValue(knownSonePrefix + "/Key").setValue(knownSone.getRequestUri().toString());
+                               /* TODO - store all known stuff? */
+                       }
+                       configuration.getStringValue("KnownSone." + knownSonesCounter + "/ID").setValue(null);
+
                } catch (ConfigurationException ce1) {
                        logger.log(Level.WARNING, "Could not store configuration!", ce1);
                }