X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fcore%2FCore.java;h=806ac21a10dfc7238cda765d29c7b75516a58220;hb=e58ffeff69ff27c1174114e882cc9eab56b3bf8e;hp=5b1b9f629f221cd424ce45952a7fea3246fa0916;hpb=2d7f57c45bd5ecd2649c8e42f9096571524cacff;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 5b1b9f6..806ac21 100644
--- a/src/main/java/net/pterodactylus/sone/core/Core.java
+++ b/src/main/java/net/pterodactylus/sone/core/Core.java
@@ -102,6 +102,10 @@ public class Core implements IdentityListener {
 	/* synchronize access on itself. */
 	private final Map<Sone, SoneStatus> soneStatuses = new HashMap<Sone, SoneStatus>();
 
+	/** Locked local Sones. */
+	/* synchronize on itself. */
+	private final Set<Sone> lockedSones = new HashSet<Sone>();
+
 	/** Sone inserters. */
 	/* synchronize access on this on localSones. */
 	private final Map<Sone, SoneInserter> soneInserters = new HashMap<Sone, SoneInserter>();
@@ -241,6 +245,19 @@ public class Core implements IdentityListener {
 	}
 
 	/**
+	 * Returns whether the given Sone is currently locked.
+	 *
+	 * @param sone
+	 *            The sone to check
+	 * @return {@code true} if the Sone is locked, {@code false} if it is not
+	 */
+	public boolean isLocked(Sone sone) {
+		synchronized (lockedSones) {
+			return lockedSones.contains(sone);
+		}
+	}
+
+	/**
 	 * Returns all Sones, remote and local.
 	 *
 	 * @return All Sones
@@ -605,6 +622,33 @@ public class Core implements IdentityListener {
 	//
 
 	/**
+	 * Locks the given Sone. A locked Sone will not be inserted by
+	 * {@link SoneInserter} until it is {@link #unlockSone(Sone) unlocked}
+	 * again.
+	 *
+	 * @param sone
+	 *            The sone to lock
+	 */
+	public void lockSone(Sone sone) {
+		synchronized (lockedSones) {
+			lockedSones.add(sone);
+		}
+	}
+
+	/**
+	 * Unlocks the given Sone.
+	 *
+	 * @see #lockSone(Sone)
+	 * @param sone
+	 *            The sone to unlock
+	 */
+	public void unlockSone(Sone sone) {
+		synchronized (lockedSones) {
+			lockedSones.remove(sone);
+		}
+	}
+
+	/**
 	 * Adds a local Sone from the given ID which has to be the ID of an own
 	 * identity.
 	 *
@@ -668,6 +712,8 @@ public class Core implements IdentityListener {
 						return;
 					}
 					logger.log(Level.INFO, "Trying to restore Sone from Freenet…");
+					coreListenerManager.fireRescuingSone(sone);
+					lockSone(sone);
 					long edition = sone.getLatestEdition();
 					while (!stopped && (edition >= 0) && isSoneRescueMode()) {
 						logger.log(Level.FINE, "Downloading edition " + edition + "…");
@@ -675,6 +721,7 @@ public class Core implements IdentityListener {
 						--edition;
 					}
 					logger.log(Level.INFO, "Finished restoring Sone from Freenet, starting Inserter…");
+					coreListenerManager.fireRescuedSone(sone);
 					soneInserter.start();
 				}
 
@@ -763,6 +810,7 @@ public class Core implements IdentityListener {
 				synchronized (newPosts) {
 					for (Post post : sone.getPosts()) {
 						if (!storedSone.getPosts().contains(post) && !knownPosts.contains(post.getId())) {
+							post.setSone(getSone(post.getSone().getId()));
 							newPosts.add(post.getId());
 							coreListenerManager.fireNewPostFound(post);
 						}