Implement “like” button.

[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 7a0ec9c..a4b3569 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 {
         */
        public enum SoneStatus {
 
+               /** The Sone is unknown, i.e. not yet downloaded. */
+               unknown,
+
                /** The Sone is idle, i.e. not being downloaded or inserted. */
                idle,
 
@@ -88,7 +91,7 @@ public class Core extends AbstractService {
        private final Map<Sone, SoneInserter> soneInserters = new HashMap<Sone, SoneInserter>();
 
        /** The Sones’ statuses. */
-       private final Map<Sone, SoneStatus> soneStatuses = new HashMap<Sone, SoneStatus>();
+       private final Map<Sone, SoneStatus> soneStatuses = Collections.synchronizedMap(new HashMap<Sone, SoneStatus>());
 
        /* various caches follow here. */
 
@@ -105,7 +108,7 @@ public class Core extends AbstractService {
         * Creates a new core.
         */
        public Core() {
-               super("Sone Core");
+               super("Sone Core", false);
        }
 
        //
@@ -159,7 +162,7 @@ public class Core extends AbstractService {
                if (!soneCache.containsKey(soneId)) {
                        Sone sone = new Sone(soneId);
                        soneCache.put(soneId, sone);
-                       setSoneStatus(sone, SoneStatus.idle);
+                       setSoneStatus(sone, SoneStatus.unknown);
                }
                return soneCache.get(soneId);
        }
@@ -302,6 +305,7 @@ public class Core extends AbstractService {
         */
        public void addLocalSone(Sone sone) {
                if (localSones.add(sone)) {
+                       setSoneStatus(sone, SoneStatus.idle);
                        SoneInserter soneInserter = new SoneInserter(this, freenetInterface, sone);
                        soneInserter.start();
                        soneInserters.put(sone, soneInserter);
@@ -399,6 +403,9 @@ public class Core extends AbstractService {
                                try {
                                        FreenetURI realRequestUri = new FreenetURI(requestUri).setMetaString(new String[] { "sone.xml" });
                                        FetchResult fetchResult = freenetInterface.fetchUri(realRequestUri);
+                                       if (fetchResult == null) {
+                                               return;
+                                       }
                                        Sone parsedSone = soneDownloader.parseSone(null, fetchResult, realRequestUri);
                                        if (parsedSone != null) {
                                                if (insertUri != null) {
@@ -428,10 +435,19 @@ public class Core extends AbstractService {
                        @SuppressWarnings("synthetic-access")
                        public void run() {
                                FreenetURI realRequestUri = sone.getRequestUri().setMetaString(new String[] { "sone.xml" });
-                               FetchResult fetchResult = freenetInterface.fetchUri(realRequestUri);
-                               Sone parsedSone = soneDownloader.parseSone(sone, fetchResult, realRequestUri);
-                               if (parsedSone != null) {
-                                       addSone(parsedSone);
+                               setSoneStatus(sone, SoneStatus.downloading);
+                               try {
+                                       FetchResult fetchResult = freenetInterface.fetchUri(realRequestUri);
+                                       if (fetchResult == null) {
+                                               /* TODO - mark Sone as bad. */
+                                               return;
+                                       }
+                                       Sone parsedSone = soneDownloader.parseSone(sone, fetchResult, realRequestUri);
+                                       if (parsedSone != null) {
+                                               addSone(parsedSone);
+                                       }
+                               } finally {
+                                       setSoneStatus(sone, (sone.getTime() == 0) ? SoneStatus.unknown : SoneStatus.idle);
                                }
                        }
                }, "Sone Downloader").start();
@@ -506,6 +522,35 @@ public class Core extends AbstractService {
                return replies;
        }
 
+       /**
+        * Gets all Sones that like the given post.
+        *
+        * @param post
+        *            The post to check for
+        * @return All Sones that like the post
+        */
+       public Set<Sone> getLikes(final Post post) {
+               return Filters.filteredSet(getSones(), new Filter<Sone>() {
+
+                       @Override
+                       public boolean filterObject(Sone sone) {
+                               return sone.isLikedPostId(post.getId());
+                       }
+               });
+       }
+
+       /**
+        * Deletes the given reply. It is removed from its Sone and from the reply
+        * cache.
+        *
+        * @param reply
+        *            The reply to remove
+        */
+       public void deleteReply(Reply reply) {
+               reply.getSone().removeReply(reply);
+               replyCache.remove(reply.getId());
+       }
+
        //
        // SERVICE METHODS
        //
@@ -617,6 +662,17 @@ public class Core extends AbstractService {
                                        sone.addBlockedSoneId(blockedSoneId);
                                }
 
+                               /* load liked post IDs. */
+                               int likedPostIdCounter = 0;
+                               while (true) {
+                                       String likedPostIdPrefix = sonePrefix + "/LikedPostId." + likedPostIdCounter++;
+                                       String likedPostId = configuration.getStringValue(likedPostIdPrefix + "/ID").getValue(null);
+                                       if (likedPostId == null) {
+                                               break;
+                                       }
+                                       sone.addLikedPostId(likedPostId);
+                               }
+
                                sone.setModificationCounter(modificationCounter);
                                addLocalSone(sone);
                        } catch (MalformedURLException mue1) {
@@ -710,6 +766,14 @@ public class Core extends AbstractService {
                                }
                                configuration.getStringValue(sonePrefix + "/BlockedSone." + blockedSoneCounter + "/ID").setValue(null);
 
+                               /* write all liked posts. */
+                               int likedPostIdCounter = 0;
+                               for (String soneLikedPostId : sone.getLikedPostIds()) {
+                                       String likedPostIdPrefix = sonePrefix + "/LikedPostId." + likedPostIdCounter++;
+                                       configuration.getStringValue(likedPostIdPrefix + "/ID").setValue(soneLikedPostId);
+                               }
+                               configuration.getStringValue(sonePrefix + "/LikedPostId." + likedPostIdCounter + "/ID").setValue(null);
+
                        }
                        /* write null ID as terminator. */
                        configuration.getStringValue("Sone/Sone." + soneId + "/ID").setValue(null);