From 2d7f57c45bd5ecd2649c8e42f9096571524cacff Mon Sep 17 00:00:00 2001
From: =?utf8?q?David=20=E2=80=98Bombe=E2=80=99=20Roden?=
 <bombe@pterodactylus.net>
Date: Wed, 17 Nov 2010 13:47:39 +0100
Subject: [PATCH] =?utf8?q?Implement=20a=20=E2=80=9CSone=20rescue=20mode.?=
 =?utf8?q?=E2=80=9D?=
MIME-Version: 1.0
Content-Type: text/plain; charset=utf8
Content-Transfer-Encoding: 8bit

This mode will when a local Sone is added try to download all previously
known editions of the Sone and merge all posts, replies, and likes it can
find. The SoneInserter is not started until all editions have been tried
(or the rescue mode is turned off).
---
 .../java/net/pterodactylus/sone/core/Core.java     | 58 +++++++++++++++++-----
 1 file changed, 46 insertions(+), 12 deletions(-)

diff --git a/src/main/java/net/pterodactylus/sone/core/Core.java b/src/main/java/net/pterodactylus/sone/core/Core.java
index ccf2cd3..5b1b9f6 100644
--- a/src/main/java/net/pterodactylus/sone/core/Core.java
+++ b/src/main/java/net/pterodactylus/sone/core/Core.java
@@ -651,17 +651,31 @@ public class Core implements IdentityListener {
 			sone.setClient(new Client("Sone", SonePlugin.VERSION.toString()));
 			/* TODO - load posts ’n stuff */
 			localSones.put(ownIdentity.getId(), sone);
-			SoneInserter soneInserter = new SoneInserter(this, freenetInterface, sone);
+			final SoneInserter soneInserter = new SoneInserter(this, freenetInterface, sone);
 			soneInserters.put(sone, soneInserter);
 			setSoneStatus(sone, SoneStatus.idle);
 			loadSone(sone);
-			soneInserter.start();
+			if (!isSoneRescueMode()) {
+				soneInserter.start();
+			}
 			new Thread(new Runnable() {
 
 				@Override
 				@SuppressWarnings("synthetic-access")
 				public void run() {
-					soneDownloader.fetchSone(sone);
+					if (!isSoneRescueMode()) {
+						soneDownloader.fetchSone(sone);
+						return;
+					}
+					logger.log(Level.INFO, "Trying to restore Sone from Freenet…");
+					long edition = sone.getLatestEdition();
+					while (!stopped && (edition >= 0) && isSoneRescueMode()) {
+						logger.log(Level.FINE, "Downloading edition " + edition + "…");
+						soneDownloader.fetchSone(sone, sone.getRequestUri().setKeyType("SSK").setDocName("Sone-" + edition));
+						--edition;
+					}
+					logger.log(Level.INFO, "Finished restoring Sone from Freenet, starting Inserter…");
+					soneInserter.start();
 				}
 
 			}, "Sone Downloader").start();
@@ -734,14 +748,17 @@ public class Core implements IdentityListener {
 	 */
 	public void updateSone(Sone sone) {
 		if (hasSone(sone.getId())) {
+			boolean soneRescueMode = isLocalSone(sone) && isSoneRescueMode();
 			Sone storedSone = getSone(sone.getId());
-			if (!(sone.getTime() > storedSone.getTime())) {
+			if (!soneRescueMode && !(sone.getTime() > storedSone.getTime())) {
 				logger.log(Level.FINE, "Downloaded Sone %s is not newer than stored Sone %s.", new Object[] { sone, storedSone });
 				return;
 			}
 			synchronized (posts) {
-				for (Post post : storedSone.getPosts()) {
-					posts.remove(post.getId());
+				if (!soneRescueMode) {
+					for (Post post : storedSone.getPosts()) {
+						posts.remove(post.getId());
+					}
 				}
 				synchronized (newPosts) {
 					for (Post post : sone.getPosts()) {
@@ -754,8 +771,10 @@ public class Core implements IdentityListener {
 				}
 			}
 			synchronized (replies) {
-				for (Reply reply : storedSone.getReplies()) {
-					replies.remove(reply.getId());
+				if (!soneRescueMode) {
+					for (Reply reply : storedSone.getReplies()) {
+						replies.remove(reply.getId());
+					}
 				}
 				synchronized (newReplies) {
 					for (Reply reply : sone.getReplies()) {
@@ -771,10 +790,25 @@ public class Core implements IdentityListener {
 				storedSone.setTime(sone.getTime());
 				storedSone.setClient(sone.getClient());
 				storedSone.setProfile(sone.getProfile());
-				storedSone.setPosts(sone.getPosts());
-				storedSone.setReplies(sone.getReplies());
-				storedSone.setLikePostIds(sone.getLikedPostIds());
-				storedSone.setLikeReplyIds(sone.getLikedReplyIds());
+				if (soneRescueMode) {
+					for (Post post : sone.getPosts()) {
+						storedSone.addPost(post);
+					}
+					for (Reply reply : sone.getReplies()) {
+						storedSone.addReply(reply);
+					}
+					for (String likedPostId : sone.getLikedPostIds()) {
+						storedSone.addLikedPostId(likedPostId);
+					}
+					for (String likedReplyId : sone.getLikedReplyIds()) {
+						storedSone.addLikedReplyId(likedReplyId);
+					}
+				} else {
+					storedSone.setPosts(sone.getPosts());
+					storedSone.setReplies(sone.getReplies());
+					storedSone.setLikePostIds(sone.getLikedPostIds());
+					storedSone.setLikeReplyIds(sone.getLikedReplyIds());
+				}
 				storedSone.setLatestEdition(sone.getRequestUri().getEdition());
 			}
 		}
-- 
2.7.4