X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fcore%2FCore.java;h=5fa330b175dec903f277559160d53b80fc4d23e2;hb=d50ceb969ffc7187d48127c14f8cd22ba135b137;hp=9e20c00fd23d5dca89669cc6262a357533481ab1;hpb=51341976b36a42fec8fb093ef4079e4b618844f9;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 9e20c00..5fa330b 100644
--- a/src/main/java/net/pterodactylus/sone/core/Core.java
+++ b/src/main/java/net/pterodactylus/sone/core/Core.java
@@ -49,6 +49,8 @@ import net.pterodactylus.sone.core.ConfigurationSoneParser.InvalidParentAlbumFou
 import net.pterodactylus.sone.core.ConfigurationSoneParser.InvalidPostFound;
 import net.pterodactylus.sone.core.ConfigurationSoneParser.InvalidPostReplyFound;
 import net.pterodactylus.sone.core.Options.DefaultOption;
+import net.pterodactylus.sone.core.SoneChangeDetector.PostProcessor;
+import net.pterodactylus.sone.core.SoneChangeDetector.PostReplyProcessor;
 import net.pterodactylus.sone.core.SoneInserter.SetInsertionDelay;
 import net.pterodactylus.sone.core.event.ImageInsertFinishedEvent;
 import net.pterodactylus.sone.core.event.MarkPostKnownEvent;
@@ -78,6 +80,7 @@ import net.pterodactylus.sone.data.TemporaryImage;
 import net.pterodactylus.sone.database.AlbumBuilder;
 import net.pterodactylus.sone.database.Database;
 import net.pterodactylus.sone.database.DatabaseException;
+import net.pterodactylus.sone.database.ImageBuilder;
 import net.pterodactylus.sone.database.PostBuilder;
 import net.pterodactylus.sone.database.PostProvider;
 import net.pterodactylus.sone.database.PostReplyBuilder;
@@ -610,6 +613,10 @@ public class Core extends AbstractService implements SoneProvider, PostProvider,
 		return database.getAlbum(albumId).orNull();
 	}
 
+	public ImageBuilder imageBuilder() {
+		return database.newImageBuilder();
+	}
+
 	/**
 	 * Returns the image with the given ID, creating it if necessary.
 	 *
@@ -941,7 +948,7 @@ public class Core extends AbstractService implements SoneProvider, PostProvider,
 	 *            {@code true} if the stored Sone should be updated regardless
 	 *            of the age of the given Sone
 	 */
-	public void updateSone(Sone sone, boolean soneRescueMode) {
+	public void updateSone(final Sone sone, boolean soneRescueMode) {
 		Optional<Sone> storedSone = getSone(sone.getId());
 		if (storedSone.isPresent()) {
 			if (!soneRescueMode && !(sone.getTime() > storedSone.get().getTime())) {
@@ -949,47 +956,42 @@ public class Core extends AbstractService implements SoneProvider, PostProvider,
 				return;
 			}
 			/* find removed posts. */
-			Collection<Post> removedPosts = new ArrayList<Post>();
-			Collection<Post> newPosts = new ArrayList<Post>();
-			Collection<Post> existingPosts = database.getPosts(sone.getId());
-			for (Post oldPost : existingPosts) {
-				if (!sone.getPosts().contains(oldPost)) {
-					removedPosts.add(oldPost);
-				}
-			}
-			/* find new posts. */
-			for (Post newPost : sone.getPosts()) {
-				if (existingPosts.contains(newPost)) {
-					continue;
+			SoneChangeDetector soneChangeDetector = new SoneChangeDetector(storedSone.get());
+			soneChangeDetector.onNewPosts(new PostProcessor() {
+				@Override
+				public void processPost(Post post) {
+					if (post.getTime() < getSoneFollowingTime(sone)) {
+						post.setKnown(true);
+					} else if (!post.isKnown()) {
+						eventBus.post(new NewPostFoundEvent(post));
+					}
 				}
-				if (newPost.getTime() < getSoneFollowingTime(sone)) {
-					newPost.setKnown(true);
-				} else if (!newPost.isKnown()) {
-					newPosts.add(newPost);
+			});
+			soneChangeDetector.onRemovedPosts(new PostProcessor() {
+				@Override
+				public void processPost(Post post) {
+					eventBus.post(new PostRemovedEvent(post));
 				}
-			}
-			/* store posts. */
-			database.storePosts(sone, sone.getPosts());
-			Collection<PostReply> newPostReplies = new ArrayList<PostReply>();
-			Collection<PostReply> removedPostReplies = new ArrayList<PostReply>();
-			if (!soneRescueMode) {
-				for (PostReply reply : storedSone.get().getReplies()) {
-					if (!sone.getReplies().contains(reply)) {
-						removedPostReplies.add(reply);
+			});
+			soneChangeDetector.onNewPostReplies(new PostReplyProcessor() {
+				@Override
+				public void processPostReply(PostReply postReply) {
+					if (postReply.getTime() < getSoneFollowingTime(sone)) {
+						postReply.setKnown(true);
+					} else if (!postReply.isKnown()) {
+						eventBus.post(new NewPostReplyFoundEvent(postReply));
 					}
 				}
-			}
-			Set<PostReply> storedReplies = storedSone.get().getReplies();
-			for (PostReply reply : sone.getReplies()) {
-				if (storedReplies.contains(reply)) {
-					continue;
-				}
-				if (reply.getTime() < getSoneFollowingTime(sone)) {
-					reply.setKnown(true);
-				} else if (!reply.isKnown()) {
-					newPostReplies.add(reply);
+			});
+			soneChangeDetector.onRemovedPostReplies(new PostReplyProcessor() {
+				@Override
+				public void processPostReply(PostReply postReply) {
+					eventBus.post(new PostReplyRemovedEvent(postReply));
 				}
-			}
+			});
+			soneChangeDetector.detectChanges(sone);
+			/* store posts. */
+			database.storePosts(sone, sone.getPosts());
 			database.storePostReplies(sone, sone.getReplies());
 			for (Album album : storedSone.get().getRootAlbum().getAlbums()) {
 				database.removeAlbum(album);
@@ -997,19 +999,7 @@ public class Core extends AbstractService implements SoneProvider, PostProvider,
 					database.removeImage(image);
 				}
 			}
-			for (Post removedPost : removedPosts) {
-				eventBus.post(new PostRemovedEvent(removedPost));
-			}
-			for (Post newPost : newPosts) {
-				eventBus.post(new NewPostFoundEvent(newPost));
-			}
-			for (PostReply removedPostReply : removedPostReplies) {
-				eventBus.post(new PostReplyRemovedEvent(removedPostReply));
-			}
-			for (PostReply newPostReply : newPostReplies) {
-				eventBus.post(new NewPostReplyFoundEvent(newPostReply));
-			}
-			for (Album album : sone.getRootAlbum().getAlbums()) {
+			for (Album album : toAllAlbums.apply(sone)) {
 				database.storeAlbum(album);
 				for (Image image : album.getImages()) {
 					database.storeImage(image);
@@ -1947,19 +1937,14 @@ public class Core extends AbstractService implements SoneProvider, PostProvider,
 		OwnIdentity ownIdentity = identityRemovedEvent.ownIdentity();
 		Identity identity = identityRemovedEvent.identity();
 		trustedIdentities.remove(ownIdentity, identity);
-		boolean foundIdentity = false;
 		for (Entry<OwnIdentity, Collection<Identity>> trustedIdentity : trustedIdentities.asMap().entrySet()) {
 			if (trustedIdentity.getKey().equals(ownIdentity)) {
 				continue;
 			}
 			if (trustedIdentity.getValue().contains(identity)) {
-				foundIdentity = true;
+				return;
 			}
 		}
-		if (foundIdentity) {
-			/* some local identity still trusts this identity, don’t remove. */
-			return;
-		}
 		Optional<Sone> sone = getSone(identity.getId());
 		if (!sone.isPresent()) {
 			/* TODO - we don’t have the Sone anymore. should this happen? */