X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fcore%2FSoneDownloader.java;h=4328f02078ec45a6d29370cf68b52288c8568421;hb=7815943b9ecc1199e11dd8be9736d6f3e892118e;hp=42f67029b78cd41e760eb1e7ae08bc44e91aca4c;hpb=5b54d9007b20d2cd8f52decdeb2baaf1445f33cb;p=Sone.git

diff --git a/src/main/java/net/pterodactylus/sone/core/SoneDownloader.java b/src/main/java/net/pterodactylus/sone/core/SoneDownloader.java
index 42f6702..4328f02 100644
--- a/src/main/java/net/pterodactylus/sone/core/SoneDownloader.java
+++ b/src/main/java/net/pterodactylus/sone/core/SoneDownloader.java
@@ -55,6 +55,9 @@ public class SoneDownloader extends AbstractService {
 	/** The logger. */
 	private static final Logger logger = Logging.getLogger(SoneDownloader.class);
 
+	/** The maximum protocol version. */
+	private static final int MAX_PROTOCOL_VERSION = 0;
+
 	/** The core. */
 	private final Core core;
 
@@ -114,11 +117,25 @@ public class SoneDownloader extends AbstractService {
 	 *            The Sone to fetch
 	 */
 	public void fetchSone(Sone sone) {
+		fetchSone(sone, sone.getRequestUri());
+	}
+
+	/**
+	 * Fetches the updated Sone. This method can be used to fetch a Sone from a
+	 * specific URI (which happens when {@link Core#isSoneRescueMode() „Sone
+	 * rescue mode“} is active).
+	 *
+	 * @param sone
+	 *            The Sone to fetch
+	 * @param soneUri
+	 *            The URI to fetch the Sone from
+	 */
+	public void fetchSone(Sone sone, FreenetURI soneUri) {
 		if (core.getSoneStatus(sone) == SoneStatus.downloading) {
 			return;
 		}
-		logger.log(Level.FINE, "Starting fetch for Sone “%s” from %s…", new Object[] { sone, sone.getRequestUri().setMetaString(new String[] { "sone.xml" }) });
-		FreenetURI requestUri = sone.getRequestUri().setMetaString(new String[] { "sone.xml" });
+		logger.log(Level.FINE, "Starting fetch for Sone “%s” from %s…", new Object[] { sone, soneUri });
+		FreenetURI requestUri = soneUri.setMetaString(new String[] { "sone.xml" });
 		core.setSoneStatus(sone, SoneStatus.downloading);
 		try {
 			Pair<FreenetURI, FetchResult> fetchResults = freenetInterface.fetchUri(requestUri);
@@ -155,7 +172,12 @@ public class SoneDownloader extends AbstractService {
 			soneInputStream = soneBucket.getInputStream();
 			Sone parsedSone = parseSone(originalSone, soneInputStream);
 			if (parsedSone != null) {
-				parsedSone.setRequestUri(requestUri.setMetaString(new String[0]));
+				parsedSone.setLatestEdition(requestUri.getEdition());
+				if (requestUri.getKeyType().equals("USK")) {
+					parsedSone.setRequestUri(requestUri.setMetaString(new String[0]));
+				} else {
+					parsedSone.setRequestUri(requestUri.setKeyType("USK").setDocName("Sone").setMetaString(new String[0]));
+				}
 			}
 			return parsedSone;
 		} catch (IOException ioe1) {
@@ -202,6 +224,27 @@ public class SoneDownloader extends AbstractService {
 			return null;
 		}
 
+		Integer protocolVersion = null;
+		String soneProtocolVersion = soneXml.getValue("protocol-version", null);
+		if (soneProtocolVersion != null) {
+			protocolVersion = Numbers.safeParseInteger(soneProtocolVersion);
+		}
+		if (protocolVersion == null) {
+			logger.log(Level.INFO, "No protocol version found, assuming 0.");
+			protocolVersion = 0;
+		}
+
+		if (protocolVersion < 0) {
+			logger.log(Level.WARNING, "Invalid protocol version: " + protocolVersion + "! Not parsing Sone.");
+			return null;
+		}
+
+		/* check for valid versions. */
+		if (protocolVersion > MAX_PROTOCOL_VERSION) {
+			logger.log(Level.WARNING, "Unknown protocol version: " + protocolVersion + "! Not parsing Sone.");
+			return null;
+		}
+
 		String soneTime = soneXml.getValue("time", null);
 		if (soneTime == null) {
 			/* TODO - mark Sone as bad. */
@@ -276,6 +319,7 @@ public class SoneDownloader extends AbstractService {
 		} else {
 			for (SimpleXML postXml : postsXml.getNodes("post")) {
 				String postId = postXml.getValue("id", null);
+				String postRecipientId = postXml.getValue("recipient", null);
 				String postTime = postXml.getValue("time", null);
 				String postText = postXml.getValue("text", null);
 				if ((postId == null) || (postTime == null) || (postText == null)) {
@@ -284,7 +328,11 @@ public class SoneDownloader extends AbstractService {
 					return null;
 				}
 				try {
-					posts.add(core.getPost(postId).setSone(sone).setTime(Long.parseLong(postTime)).setText(postText));
+					Post post = core.getPost(postId).setSone(sone).setTime(Long.parseLong(postTime)).setText(postText);
+					if ((postRecipientId != null) && (postRecipientId.length() == 43)) {
+						post.setRecipient(core.getSone(postRecipientId));
+					}
+					posts.add(post);
 				} catch (NumberFormatException nfe1) {
 					/* TODO - mark Sone as bad. */
 					logger.log(Level.WARNING, "Downloaded post for Sone %s with invalid time: %s", new Object[] { sone, postTime });
@@ -353,6 +401,7 @@ public class SoneDownloader extends AbstractService {
 			sone.setPosts(posts);
 			sone.setReplies(replies);
 			sone.setLikePostIds(likedPostIds);
+			sone.setLikeReplyIds(likedReplyIds);
 		}
 
 		return sone;