X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fcore%2FCore.java;h=fbbb166012a29fbad8350d555e808a70a204dbcc;hb=793370e43152d61ca04699fdfccf39fd73707806;hp=7a0ec9c673c219b6f1ceb5166eecad7e9c8c1466;hpb=4d0705ef1773ee51717911bf5a1bc8f6ca71d339;p=Sone.git

diff --git a/src/main/java/net/pterodactylus/sone/core/Core.java b/src/main/java/net/pterodactylus/sone/core/Core.java
index 7a0ec9c..fbbb166 100644
--- 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,
 
@@ -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,18 @@ public class Core extends AbstractService {
 		return replies;
 	}
 
+	/**
+	 * 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
 	//