Rename “friend Sone” to just “friend.”

[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 2b0c0bf..72c36d8 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/core/Core.java
+++ b/src/main/java/net/pterodactylus/sone/core/Core.java
@@ -59,6 +59,9 @@ public class Core extends AbstractService {
        /** Interface to freenet. */
        private FreenetInterface freenetInterface;
 
+       /** The Sone downloader. */
+       private SoneDownloader soneDownloader;
+
        /** The local Sones. */
        private final Set<Sone> localSones = new HashSet<Sone>();
 
@@ -108,6 +111,7 @@ public class Core extends AbstractService {
         */
        public Core freenetInterface(FreenetInterface freenetInterface) {
                this.freenetInterface = freenetInterface;
+               soneDownloader = new SoneDownloader(freenetInterface);
                return this;
        }
 
@@ -150,11 +154,23 @@ public class Core extends AbstractService {
                        soneCache.put(sone.getId(), sone);
                        SoneInserter soneInserter = new SoneInserter(freenetInterface, sone);
                        soneInserter.start();
+                       soneDownloader.addSone(sone);
                        soneInserters.put(sone, soneInserter);
                }
        }
 
        /**
+        * Adds a remote Sone so that it is watched for updates.
+        *
+        * @param sone
+        *            The sone to watch
+        */
+       public void addRemoteSone(Sone sone) {
+               Sone updatedSone = soneCache.put(sone.getId(), sone);
+               soneDownloader.addSone(updatedSone);
+       }
+
+       /**
         * Creates a new Sone at a random location.
         *
         * @param name
@@ -201,7 +217,7 @@ public class Core extends AbstractService {
                Sone sone;
                try {
                        logger.log(Level.FINEST, "Creating new Sone “%s” at %s (%s)…", new Object[] { name, finalRequestUri, finalInsertUri });
-                       sone = new Sone(UUID.randomUUID(), name, new FreenetURI(finalRequestUri).setKeyType("USK"), new FreenetURI(finalInsertUri).setKeyType("USK"));
+                       sone = new Sone(UUID.randomUUID(), name, new FreenetURI(finalRequestUri).setKeyType("USK").setDocName("Sone-" + name), new FreenetURI(finalInsertUri).setKeyType("USK").setDocName("Sone-" + name));
                        sone.setProfile(new Profile());
                        /* set modification counter to 1 so it is inserted immediately. */
                        sone.setModificationCounter(1);
@@ -241,6 +257,7 @@ public class Core extends AbstractService {
         */
        @Override
        protected void serviceStop() {
+               soneDownloader.stop();
                /* stop all Sone inserters. */
                for (SoneInserter soneInserter : soneInserters.values()) {
                        soneInserter.stop();
@@ -314,6 +331,25 @@ public class Core extends AbstractService {
                                        Reply reply = new ReplyShell().setSone(replySone).setPost(replyPost).setTime(replyTime).setText(replyText).getShelled();
                                        replyCache.put(replyId, reply);
                                } while (true);
+
+                               /* load friends. */
+                               int friendCounter = 0;
+                               while (true) {
+                                       String friendPrefix = sonePrefix + "/Friend." + friendCounter++;
+                                       String friendId = configuration.getStringValue(friendPrefix + "/ID").getValue(null);
+                                       if (friendId == null) {
+                                               break;
+                                       }
+                                       Sone friendSone = soneCache.get(friendId);
+                                       if (friendSone instanceof SoneShell) {
+                                               String friendKey = configuration.getStringValue(friendPrefix + "/Key").getValue(null);
+                                               String friendName = configuration.getStringValue(friendPrefix + "/Name").getValue(null);
+                                               ((SoneShell) friendSone).setRequestUri(new FreenetURI(friendKey)).setName(friendName);
+                                       }
+                                       addRemoteSone(friendSone);
+                                       sone.addFriend(sone);
+                               }
+
                                sone.setModificationCounter(modificationCounter);
                                addSone(sone);
                        } catch (MalformedURLException mue1) {
@@ -368,6 +404,17 @@ public class Core extends AbstractService {
                                }
                                /* write null ID as terminator. */
                                configuration.getStringValue(sonePrefix + "/Reply." + replyId + "/ID").setValue(null);
+
+                               int friendId = 0;
+                               for (Sone friend : sone.getFriends()) {
+                                       String friendPrefix = sonePrefix + "/Friend." + friendId++;
+                                       configuration.getStringValue(friendPrefix + "/ID").setValue(friend.getId());
+                                       configuration.getStringValue(friendPrefix + "/Key").setValue(friend.getRequestUri().toString());
+                                       configuration.getStringValue(friendPrefix + "/Name").setValue(friend.getName());
+                               }
+                               /* write null ID as terminator. */
+                               configuration.getStringValue(sonePrefix + "/Friend." + friendId + "/ID").setValue(null);
+
                        }
                        /* write null ID as terminator. */
                        configuration.getStringValue("Sone/Sone." + soneId + "/ID").setValue(null);