Store the Sone ID instead of the Sone itself in the session.

[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 77b80cf..13f1589 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/core/Core.java
+++ b/src/main/java/net/pterodactylus/sone/core/Core.java
@@ -22,11 +22,13 @@ import java.util.Collections;
 import java.util.HashSet;
 import java.util.List;
 import java.util.Set;
+import java.util.UUID;
 import java.util.logging.Level;
 import java.util.logging.Logger;
 
 import net.pterodactylus.sone.data.Sone;
 import net.pterodactylus.util.config.Configuration;
+import net.pterodactylus.util.config.ConfigurationException;
 import net.pterodactylus.util.logging.Logging;
 import net.pterodactylus.util.service.AbstractService;
 import net.pterodactylus.util.text.StringEscaper;
@@ -112,6 +114,14 @@ public class Core extends AbstractService {
                loadConfiguration();
        }
 
+       /**
+        * {@inheritDoc}
+        */
+       @Override
+       protected void serviceStop() {
+               saveConfiguration();
+       }
+
        //
        // PRIVATE METHODS
        //
@@ -136,11 +146,13 @@ public class Core extends AbstractService {
                }
 
                /* parse local Sones. */
+               logger.log(Level.INFO, "Loading %d Sones…", allSoneNames.size());
                for (String soneName : allSoneNames) {
+                       String id = configuration.getStringValue("Sone/Name." + soneName + "/ID").getValue(null);
                        String insertUri = configuration.getStringValue("Sone/Name." + soneName + "/InsertURI").getValue(null);
                        String requestUri = configuration.getStringValue("Sone/Name." + soneName + "/RequestURI").getValue(null);
                        try {
-                               localSones.add(new Sone(soneName, new FreenetURI(requestUri), new FreenetURI(insertUri)));
+                               localSones.add(new Sone(UUID.fromString(id), soneName, new FreenetURI(requestUri), new FreenetURI(insertUri)));
                        } catch (MalformedURLException mue1) {
                                logger.log(Level.WARNING, "Could not create Sone from requestUri (“" + requestUri + "”) and insertUri (“" + insertUri + "”)!", mue1);
                        }
@@ -149,4 +161,32 @@ public class Core extends AbstractService {
                logger.exiting(Core.class.getName(), "loadConfiguration()");
        }
 
+       /**
+        * Saves the configuraiton.
+        */
+       private void saveConfiguration() {
+
+               /* get the names of all Sones. */
+               Set<String> soneNames = new HashSet<String>();
+               for (Sone sone : localSones) {
+                       soneNames.add(sone.getName());
+               }
+               String soneNamesString = StringEscaper.escapeWords(soneNames);
+
+               logger.log(Level.INFO, "Storing %d Sones…", soneNames.size());
+               try {
+                       /* store names of all Sones. */
+                       configuration.getStringValue("Sone/Names").setValue(soneNamesString);
+
+                       /* store all Sones. */
+                       for (Sone sone : localSones) {
+                               configuration.getStringValue("Sone/Name." + sone.getName() + "/ID").setValue(sone.getId());
+                               configuration.getStringValue("Sone/Name." + sone.getName() + "/RequestURI").setValue(sone.getRequestUri().toString());
+                               configuration.getStringValue("Sone/Name." + sone.getName() + "/InsertURI").setValue(sone.getInsertUri().toString());
+                       }
+               } catch (ConfigurationException ce1) {
+                       logger.log(Level.WARNING, "Could not store configuration!", ce1);
+               }
+       }
+
 }