Add blacklisting functionality.

[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 e0f2649..941710c 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/core/Core.java
+++ b/src/main/java/net/pterodactylus/sone/core/Core.java
@@ -91,6 +91,9 @@ public class Core extends AbstractService {
        /** The Sone downloader. */
        private SoneDownloader soneDownloader;
 
+       /** The Sone blacklist. */
+       private final Set<Sone> blacklistedSones = new HashSet<Sone>();
+
        /** The local Sones. */
        private final Set<Sone> localSones = new HashSet<Sone>();
 
@@ -163,7 +166,17 @@ public class Core extends AbstractService {
         * @return The local Sones
         */
        public Set<Sone> getSones() {
-               return Collections.unmodifiableSet(localSones);
+               return Filters.filteredSet(localSones, new Filter<Sone>() {
+
+                       /**
+                        * {@inheritDoc}
+                        */
+                       @Override
+                       @SuppressWarnings("synthetic-access")
+                       public boolean filterObject(Sone sone) {
+                               return !blacklistedSones.contains(sone);
+                       }
+               });
        }
 
        /**
@@ -189,7 +202,17 @@ public class Core extends AbstractService {
         * @return All known sones
         */
        public Collection<Sone> getKnownSones() {
-               return soneCache.values();
+               return Filters.filteredCollection(soneCache.values(), new Filter<Sone>() {
+
+                       /**
+                        * {@inheritDoc}
+                        */
+                       @Override
+                       @SuppressWarnings("synthetic-access")
+                       public boolean filterObject(Sone sone) {
+                               return !blacklistedSones.contains(sone);
+                       }
+               });
        }
 
        /**
@@ -202,13 +225,33 @@ public class Core extends AbstractService {
 
                        @Override
                        @SuppressWarnings("synthetic-access")
-                       public boolean filterObject(Sone object) {
-                               return !localSones.contains(object);
+                       public boolean filterObject(Sone sone) {
+                               return !blacklistedSones.contains(sone) && !localSones.contains(sone);
                        }
                });
        }
 
        /**
+        * Returns all blacklisted Sones.
+        *
+        * @return All blacklisted Sones
+        */
+       public Collection<Sone> getBlacklistedSones() {
+               return Collections.unmodifiableCollection(blacklistedSones);
+       }
+
+       /**
+        * Checks whether the given Sone is blacklisted.
+        *
+        * @param sone
+        *            The Sone to check
+        * @return {@code true} if this Sone is blacklisted, {@code false} otherwise
+        */
+       public boolean isBlacklistedSone(Sone sone) {
+               return blacklistedSones.contains(sone);
+       }
+
+       /**
         * Returns the status of the given Sone.
         *
         * @param sone
@@ -329,6 +372,38 @@ public class Core extends AbstractService {
        }
 
        /**
+        * Blackslists the given Sone.
+        *
+        * @param sone
+        *            The Sone to blacklist
+        */
+       public void blacklistSone(Sone sone) {
+               if (blacklistedSones.add(sone)) {
+                       soneDownloader.removeSone(sone);
+                       if (localSones.remove(sone)) {
+                               SoneInserter soneInserter = soneInserters.remove(sone);
+                               soneInserter.stop();
+                       }
+               }
+       }
+
+       /**
+        * Unblacklists the given Sone.
+        *
+        * @param sone
+        *            The Sone to unblacklist
+        */
+       public void unblacklistSone(Sone sone) {
+               if (blacklistedSones.remove(sone)) {
+                       if (sone.getInsertUri() != null) {
+                               addLocalSone(sone);
+                       } else {
+                               addSone(sone);
+                       }
+               }
+       }
+
+       /**
         * Creates a new Sone at a random location.
         *
         * @param name
@@ -789,6 +864,23 @@ public class Core extends AbstractService {
                        }
                }
 
+               /* load all blacklisted Sones. */
+               int blacklistedSonesCounter = 0;
+               while (true) {
+                       String blacklistedSonePrefix = "BlacklistedSone." + blacklistedSonesCounter++;
+                       String blacklistedSoneId = configuration.getStringValue(blacklistedSonePrefix + "/ID").getValue(null);
+                       if (blacklistedSoneId == null) {
+                               break;
+                       }
+                       String blacklistedSoneName = configuration.getStringValue(blacklistedSonePrefix + "/Name").getValue(null);
+                       String blacklistedSoneKey = configuration.getStringValue(blacklistedSonePrefix + "/Key").getValue(null);
+                       try {
+                               blacklistSone(getSone(blacklistedSoneId).setName(blacklistedSoneName).setRequestUri(new FreenetURI(blacklistedSoneKey)));
+                       } catch (MalformedURLException mue1) {
+                               logger.log(Level.WARNING, "Could not create blacklisted Sone from requestUri (“" + blacklistedSoneKey + "”)!", mue1);
+                       }
+               }
+
                /* load all remote Sones. */
                for (Sone remoteSone : getRemoteSones()) {
                        loadSone(remoteSone);
@@ -897,6 +989,17 @@ public class Core extends AbstractService {
                        }
                        configuration.getStringValue("KnownSone." + knownSonesCounter + "/ID").setValue(null);
 
+                       /* write all blacklisted Sones. */
+                       int blacklistedSonesCounter = 0;
+                       for (Sone blacklistedSone : getBlacklistedSones()) {
+                               String blacklistedSonePrefix = "BlacklistedSone." + blacklistedSonesCounter++;
+                               configuration.getStringValue(blacklistedSonePrefix + "/ID").setValue(blacklistedSone.getId());
+                               configuration.getStringValue(blacklistedSonePrefix + "/Name").setValue(blacklistedSone.getName());
+                               configuration.getStringValue(blacklistedSonePrefix + "/Key").setValue(blacklistedSone.getRequestUri().toString());
+                               /* TODO - store all known stuff? */
+                       }
+                       configuration.getStringValue("BlacklistedSone." + blacklistedSonesCounter + "/ID").setValue(null);
+
                } catch (ConfigurationException ce1) {
                        logger.log(Level.WARNING, "Could not store configuration!", ce1);
                }