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 d098456..e944f1f 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/core/Core.java
+++ b/src/main/java/net/pterodactylus/sone/core/Core.java
@@ -330,6 +330,21 @@ public class Core extends AbstractService {
         *            The request URI to load the Sone from
         */
        public void loadSone(final String requestUri) {
+               loadSone(requestUri, null);
+       }
+
+       /**
+        * Loads the Sone from the given request URI. The fetching of the data is
+        * performed in a new thread so this method returns immediately. If
+        * {@code insertUri} is not {@code null} the loaded Sone is converted into a
+        * local Sone and available using as any other local Sone.
+        *
+        * @param requestUri
+        *            The request URI to load the Sone from
+        * @param insertUri
+        *            The insert URI of the Sone
+        */
+       public void loadSone(final String requestUri, final String insertUri) {
                new Thread(new Runnable() {
 
                        @Override
@@ -340,7 +355,12 @@ public class Core extends AbstractService {
                                        FetchResult fetchResult = freenetInterface.fetchUri(realRequestUri);
                                        Sone parsedSone = soneDownloader.parseSone(null, fetchResult, realRequestUri);
                                        if (parsedSone != null) {
-                                               addSone(parsedSone);
+                                               if (insertUri != null) {
+                                                       parsedSone.setInsertUri(new FreenetURI(insertUri));
+                                                       addLocalSone(parsedSone);
+                                               } else {
+                                                       addSone(parsedSone);
+                                               }
                                        }
                                } catch (MalformedURLException mue1) {
                                        logger.log(Level.INFO, "Could not create URI from “" + requestUri + "”.", mue1);
@@ -536,10 +556,20 @@ public class Core extends AbstractService {
                                        String friendKey = configuration.getStringValue(friendPrefix + "/Key").getValue(null);
                                        String friendName = configuration.getStringValue(friendPrefix + "/Name").getValue(null);
                                        friendSone.setRequestUri(new FreenetURI(friendKey)).setName(friendName);
-                                       loadSone(friendKey);
                                        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) {
@@ -548,6 +578,28 @@ 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);
+               }
+
                logger.exiting(Core.class.getName(), "loadConfiguration()");
        }
 
@@ -602,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);
                }