package net.pterodactylus.sone.core;
+import java.io.InputStream;
import java.net.MalformedURLException;
import java.util.ArrayList;
import java.util.Collection;
import java.util.logging.Level;
import java.util.logging.Logger;
+import net.pterodactylus.sone.core.Options.DefaultOption;
+import net.pterodactylus.sone.core.Options.Option;
+import net.pterodactylus.sone.core.Options.OptionWatcher;
import net.pterodactylus.sone.core.SoneException.Type;
import net.pterodactylus.sone.data.Post;
import net.pterodactylus.sone.data.Profile;
*/
public class Core extends AbstractService {
+ /** The default Sones. */
+ private static final Set<String> defaultSones = new HashSet<String>();
+
+ static {
+ /* Sone of Sone. */
+ defaultSones.add("USK@eRHt0ceFsHjRZ11j6dd68RSdIvfd8f9YjJLZ9lnhEyo,iJWjIWh6TkMZm1NY8qBranKTIuwsCPkVPG6T6c6ft-I,AQACAAE/Sone/3");
+ /* Sone of Bombe. */
+ defaultSones.add("USK@RuW~uAO35Ipne896-1OmaVJNPuYE4ZIB5oZ5ziaU57A,7rV3uiyztXBDt03DCoRiNwiGjgFCJuznM9Okc1opURU,AQACAAE/Sone/24");
+ }
+
/**
* Enumeration for the possible states of a {@link Sone}.
*
/** The logger. */
private static final Logger logger = Logging.getLogger(Core.class);
+ /** The options. */
+ private final Options options = new Options();
+
/** The configuration. */
private Configuration configuration;
/** The Sone downloader. */
private SoneDownloader soneDownloader;
+ /** The Sone blacklist. */
+ private final Set<Sone> blacklistedSones = Collections.synchronizedSet(new HashSet<Sone>());
+
/** The local Sones. */
- private final Set<Sone> localSones = new HashSet<Sone>();
+ private final Set<Sone> localSones = Collections.synchronizedSet(new HashSet<Sone>());
/** Sone inserters. */
- private final Map<Sone, SoneInserter> soneInserters = new HashMap<Sone, SoneInserter>();
+ private final Map<Sone, SoneInserter> soneInserters = Collections.synchronizedMap(new HashMap<Sone, SoneInserter>());
/** The Sones’ statuses. */
private final Map<Sone, SoneStatus> soneStatuses = Collections.synchronizedMap(new HashMap<Sone, SoneStatus>());
/* various caches follow here. */
/** Cache for all known Sones. */
- private final Map<String, Sone> soneCache = new HashMap<String, Sone>();
+ private final Map<String, Sone> soneCache = Collections.synchronizedMap(new HashMap<String, Sone>());
/** Cache for all known posts. */
- private final Map<String, Post> postCache = new HashMap<String, Post>();
+ private final Map<String, Post> postCache = Collections.synchronizedMap(new HashMap<String, Post>());
/** Cache for all known replies. */
- private final Map<String, Reply> replyCache = new HashMap<String, Reply>();
+ private final Map<String, Reply> replyCache = Collections.synchronizedMap(new HashMap<String, Reply>());
/**
* Creates a new core.
//
/**
+ * Returns the options of the Sone plugin.
+ *
+ * @return The options of the Sone plugin
+ */
+ public Options getOptions() {
+ return options;
+ }
+
+ /**
* Sets the configuration of the core.
*
* @param configuration
* @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);
+ }
+ });
}
/**
* @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);
+ }
+ });
}
/**
@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
}
/**
+ * 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
* if a Sone error occurs
*/
public Sone createSone(String name) throws SoneException {
- return createSone(name, "Sone-" + name, null, null);
+ return createSone(name, "Sone", null, null);
}
/**
} else {
addSone(parsedSone);
}
+ setSoneStatus(parsedSone, SoneStatus.idle);
}
} catch (MalformedURLException mue1) {
logger.log(Level.INFO, "Could not create URI from “" + requestUri + "”.", mue1);
}
/**
+ * Loads a Sone from an input stream.
+ *
+ * @param soneInputStream
+ * The input stream to load the Sone from
+ * @return The parsed Sone, or {@code null} if the Sone could not be parsed
+ */
+ public Sone loadSone(InputStream soneInputStream) {
+ Sone parsedSone = soneDownloader.parseSone(soneInputStream);
+ if (parsedSone == null) {
+ return null;
+ }
+ if (parsedSone.getInsertUri() != null) {
+ addLocalSone(parsedSone);
+ } else {
+ addSone(parsedSone);
+ }
+ return parsedSone;
+ }
+
+ /**
* Loads and updates the given Sone.
*
* @param sone
SoneInserter soneInserter = soneInserters.remove(sone);
soneInserter.stop();
localSones.remove(sone);
+ soneStatuses.remove(sone);
+ soneCache.remove(sone.getId());
}
/**
* 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>() {
+ public Collection<Sone> getLikes(final Post post) {
+ return Filters.filteredCollection(getKnownSones(), new Filter<Sone>() {
@Override
public boolean filterObject(Sone sone) {
}
/**
+ * Gets all Sones that like the given reply.
+ *
+ * @param reply
+ * The reply to check for
+ * @return All Sones that like the reply
+ */
+ public Collection<Sone> getLikes(final Reply reply) {
+ return Filters.filteredCollection(getKnownSones(), new Filter<Sone>() {
+
+ @Override
+ public boolean filterObject(Sone sone) {
+ return sone.isLikedReplyId(reply.getId());
+ }
+ });
+ }
+
+ /**
* Deletes the given reply. It is removed from its Sone and from the reply
* cache.
*
//
/**
+ * Adds some default Sones.
+ */
+ private void addDefaultSones() {
+ for (String soneUri : defaultSones) {
+ loadSone(soneUri);
+ }
+ }
+
+ /**
* Loads the configuration.
*/
+ @SuppressWarnings("unchecked")
private void loadConfiguration() {
logger.entering(Core.class.getName(), "loadConfiguration()");
+ boolean firstStart = configuration.getBooleanValue("FirstStart").getValue(true);
+ if (firstStart) {
+ logger.log(Level.INFO, "First start of Sone, adding a couple of default Sones…");
+ addDefaultSones();
+ try {
+ configuration.getBooleanValue("FirstStart").setValue(false);
+ } catch (ConfigurationException ce1) {
+ logger.log(Level.WARNING, "Could not clear “first start” flag!");
+ }
+ }
+
+ options.addIntegerOption("InsertionDelay", new DefaultOption<Integer>(60, new OptionWatcher<Integer>() {
+
+ @Override
+ public void optionChanged(Option<Integer> option, Integer oldValue, Integer newValue) {
+ SoneInserter.setInsertionDelay(newValue);
+ }
+
+ }));
+
+ options.addBooleanOption("ClearOnNextRestart", new DefaultOption<Boolean>(false));
+ options.addBooleanOption("ReallyClearOnNextRestart", new DefaultOption<Boolean>(false));
+
+ if (firstStart) {
+ return;
+ }
+
+ options.getBooleanOption("ClearOnNextRestart").set(configuration.getBooleanValue("Option/ClearOnNextRestart").getValue(null));
+ options.getBooleanOption("ReallyClearOnNextRestart").set(configuration.getBooleanValue("Option/ReallyClearOnNextRestart").getValue(null));
+ boolean clearConfiguration = options.getBooleanOption("ClearOnNextRestart").get() && options.getBooleanOption("ReallyClearOnNextRestart").get();
+ options.getBooleanOption("ClearOnNextRestart").set(null);
+ options.getBooleanOption("ReallyClearOnNextRestart").set(null);
+ if (clearConfiguration) {
+ /* stop loading the configuration. */
+ addDefaultSones();
+ return;
+ }
+
+ options.getIntegerOption("InsertionDelay").set(configuration.getIntValue("Option/InsertionDelay").getValue(null));
+
/* parse local Sones. */
logger.log(Level.INFO, "Loading Sones…");
int soneId = 0;
String firstName = configuration.getStringValue(sonePrefix + "/Profile/FirstName").getValue(null);
String middleName = configuration.getStringValue(sonePrefix + "/Profile/MiddleName").getValue(null);
String lastName = configuration.getStringValue(sonePrefix + "/Profile/LastName").getValue(null);
+ Integer birthDay = configuration.getIntValue(sonePrefix + "/Profile/BirthDay").getValue(null);
+ Integer birthMonth = configuration.getIntValue(sonePrefix + "/Profile/BirthMonth").getValue(null);
+ Integer birthYear = configuration.getIntValue(sonePrefix + "/Profile/BirthYear").getValue(null);
try {
Profile profile = new Profile();
profile.setFirstName(firstName);
profile.setMiddleName(middleName);
profile.setLastName(lastName);
+ profile.setBirthDay(birthDay).setBirthMonth(birthMonth).setBirthYear(birthYear);
Sone sone = getSone(id).setName(name).setTime(time).setRequestUri(new FreenetURI(requestUri)).setInsertUri(new FreenetURI(insertUri));
sone.setProfile(profile);
int postId = 0;
sone.addLikedPostId(likedPostId);
}
+ /* load liked reply IDs. */
+ int likedReplyIdCounter = 0;
+ while (true) {
+ String likedReplyIdPrefix = sonePrefix + "/LikedReplyId." + likedReplyIdCounter++;
+ String likedReplyId = configuration.getStringValue(likedReplyIdPrefix + "/ID").getValue(null);
+ if (likedReplyId == null) {
+ break;
+ }
+ sone.addLikedReplyId(likedReplyId);
+ }
+
sone.setModificationCounter(modificationCounter);
addLocalSone(sone);
} catch (MalformedURLException mue1) {
}
}
+ /* 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);
+ String blacklistedSoneInsertKey = configuration.getStringValue(blacklistedSonePrefix + "/InsertKey").getValue(null);
+ try {
+ blacklistSone(getSone(blacklistedSoneId).setName(blacklistedSoneName).setRequestUri(new FreenetURI(blacklistedSoneKey)).setInsertUri((blacklistedSoneInsertKey != null) ? new FreenetURI(blacklistedSoneInsertKey) : null));
+ } 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);
private void saveConfiguration() {
Set<Sone> sones = getSones();
logger.log(Level.INFO, "Storing %d Sones…", sones.size());
+
try {
+ /* store the options first. */
+ configuration.getIntValue("Option/InsertionDelay").setValue(options.getIntegerOption("InsertionDelay").getReal());
+ configuration.getBooleanValue("Option/ClearOnNextRestart").setValue(options.getBooleanOption("ClearOnNextRestart").getReal());
+ configuration.getBooleanValue("Option/ReallyClearOnNextRestart").setValue(options.getBooleanOption("ReallyClearOnNextRestart").getReal());
+
/* store all Sones. */
int soneId = 0;
for (Sone sone : localSones) {
configuration.getStringValue(sonePrefix + "/Profile/FirstName").setValue(profile.getFirstName());
configuration.getStringValue(sonePrefix + "/Profile/MiddleName").setValue(profile.getMiddleName());
configuration.getStringValue(sonePrefix + "/Profile/LastName").setValue(profile.getLastName());
+ configuration.getIntValue(sonePrefix + "/Profile/BirthDay").setValue(profile.getBirthDay());
+ configuration.getIntValue(sonePrefix + "/Profile/BirthMonth").setValue(profile.getBirthMonth());
+ configuration.getIntValue(sonePrefix + "/Profile/BirthYear").setValue(profile.getBirthYear());
int postId = 0;
for (Post post : sone.getPosts()) {
String postPrefix = sonePrefix + "/Post." + postId++;
}
configuration.getStringValue(sonePrefix + "/LikedPostId." + likedPostIdCounter + "/ID").setValue(null);
+ /* write all liked replies. */
+ int likedReplyIdCounter = 0;
+ for (String soneLikedReplyId : sone.getLikedReplyIds()) {
+ String likedReplyIdPrefix = sonePrefix + "/LikedReplyId." + likedReplyIdCounter++;
+ configuration.getStringValue(likedReplyIdPrefix + "/ID").setValue(soneLikedReplyId);
+ }
+ configuration.getStringValue(sonePrefix + "/LikedReplyId." + likedReplyIdCounter + "/ID").setValue(null);
+
}
/* write null ID as terminator. */
configuration.getStringValue("Sone/Sone." + soneId + "/ID").setValue(null);
}
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());
+ configuration.getStringValue(blacklistedSonePrefix + "/InsertKey").setValue((blacklistedSone.getInsertUri() != null) ? blacklistedSone.getInsertUri().toString() : null);
+ /* TODO - store all known stuff? */
+ }
+ configuration.getStringValue("BlacklistedSone." + blacklistedSonesCounter + "/ID").setValue(null);
+
} catch (ConfigurationException ce1) {
logger.log(Level.WARNING, "Could not store configuration!", ce1);
}