X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fcore%2FWebOfTrustUpdater.java;h=ded6da945dffaec39247919683091ebf919d4840;hp=0ee5a2cc573079a8e618fb9d82b957b5e522c36d;hb=95404485c0987d8192337e7f44f37a09ed7818ea;hpb=cfbde5f4b68cb6e185aa1ea1d0b2dcfc4d224cb2

diff --git a/src/main/java/net/pterodactylus/sone/core/WebOfTrustUpdater.java b/src/main/java/net/pterodactylus/sone/core/WebOfTrustUpdater.java
index 0ee5a2c..ded6da9 100644
--- a/src/main/java/net/pterodactylus/sone/core/WebOfTrustUpdater.java
+++ b/src/main/java/net/pterodactylus/sone/core/WebOfTrustUpdater.java
@@ -31,6 +31,7 @@ import net.pterodactylus.sone.freenet.wot.WebOfTrustConnector;
 import net.pterodactylus.sone.freenet.wot.WebOfTrustException;
 import net.pterodactylus.util.logging.Logging;
 import net.pterodactylus.util.service.AbstractService;
+import net.pterodactylus.util.validation.Validation;
 
 /**
  * Updates WebOfTrust identity data in a background thread because communicating
@@ -118,6 +119,90 @@ public class WebOfTrustUpdater extends AbstractService {
 		}
 	}
 
+	/**
+	 * Adds the given context to the given own identity.
+	 *
+	 * @param ownIdentity
+	 *            The own identity to add the context to
+	 * @param context
+	 *            The context to add
+	 */
+	public void addContext(OwnIdentity ownIdentity, String context) {
+		addContextWait(ownIdentity, context, false);
+	}
+
+	/**
+	 * Adds the given context to the given own identity, waiting for completion
+	 * of the operation.
+	 *
+	 * @param ownIdentity
+	 *            The own identity to add the context to
+	 * @param context
+	 *            The context to add
+	 * @return {@code true} if the context was added successfully, {@code false}
+	 *         otherwise
+	 */
+	public boolean addContextWait(OwnIdentity ownIdentity, String context) {
+		return addContextWait(ownIdentity, context, true);
+	}
+
+	/**
+	 * Adds the given context to the given own identity, waiting for completion
+	 * of the operation.
+	 *
+	 * @param ownIdentity
+	 *            The own identity to add the context to
+	 * @param context
+	 *            The context to add
+	 * @param wait
+	 *            {@code true} to wait for the end of the operation,
+	 *            {@code false} to return immediately
+	 * @return {@code true} if the context was added successfully, {@code false}
+	 *         if the context was not added successfully, or if the job should
+	 *         not wait for completion
+	 */
+	private boolean addContextWait(OwnIdentity ownIdentity, String context, boolean wait) {
+		AddContextJob addContextJob = new AddContextJob(ownIdentity, context);
+		if (!updateJobs.contains(addContextJob)) {
+			logger.log(Level.FINER, "Adding Context Job: " + addContextJob);
+			try {
+				updateJobs.put(addContextJob);
+			} catch (InterruptedException ie1) {
+				/* the queue is unbounded so it should never block. */
+			}
+			if (wait) {
+				return addContextJob.waitForCompletion();
+			}
+		} else if (wait) {
+			for (WebOfTrustUpdateJob updateJob : updateJobs) {
+				if (updateJob.equals(addContextJob)) {
+					return updateJob.waitForCompletion();
+				}
+			}
+		}
+		return false;
+	}
+
+	/**
+	 * Removes the given context from the given own identity.
+	 *
+	 * @param ownIdentity
+	 *            The own identity to remove the context from
+	 * @param context
+	 *            The context to remove
+	 */
+	public void removeContext(OwnIdentity ownIdentity, String context) {
+		RemoveContextJob removeContextJob = new RemoveContextJob(ownIdentity, context);
+		if (!updateJobs.contains(removeContextJob)) {
+			logger.log(Level.FINER, "Adding Context Job: " + removeContextJob);
+			try {
+				updateJobs.put(removeContextJob);
+			} catch (InterruptedException ie1) {
+				/* the queue is unbounded so it should never block. */
+			}
+		}
+	}
+
 	//
 	// SERVICE METHODS
 	//
@@ -386,6 +471,143 @@ public class WebOfTrustUpdater extends AbstractService {
 				finish(false);
 			}
 		}
+
+	}
+
+	/**
+	 * Base class for context updates of an {@link OwnIdentity}.
+	 *
+	 * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
+	 */
+	private class WebOfTrustContextUpdateJob extends WebOfTrustUpdateJob {
+
+		/** The own identity whose contexts to manage. */
+		protected final OwnIdentity ownIdentity;
+
+		/** The context to update. */
+		protected final String context;
+
+		/**
+		 * Creates a new context update job.
+		 *
+		 * @param ownIdentity
+		 *            The own identity to update
+		 * @param context
+		 *            The context to update
+		 */
+		@SuppressWarnings("synthetic-access")
+		public WebOfTrustContextUpdateJob(OwnIdentity ownIdentity, String context) {
+			Validation.begin().isNotNull("OwnIdentity", ownIdentity).isNotNull("Context", context).check();
+			this.ownIdentity = ownIdentity;
+			this.context = context;
+		}
+
+		//
+		// OBJECT METHODS
+		//
+
+		/**
+		 * {@inheritDoc}
+		 */
+		@Override
+		public boolean equals(Object object) {
+			if ((object == null) || !object.getClass().equals(getClass())) {
+				return false;
+			}
+			WebOfTrustContextUpdateJob updateJob = (WebOfTrustContextUpdateJob) object;
+			return updateJob.ownIdentity.equals(ownIdentity) && updateJob.context.equals(context);
+		}
+
+		/**
+		 * {@inheritDoc}
+		 */
+		@Override
+		public int hashCode() {
+			return getClass().hashCode() ^ ownIdentity.hashCode() ^ context.hashCode();
+		}
+
+		/**
+		 * {@inheritDoc}
+		 */
+		@Override
+		public String toString() {
+			return String.format("%s[ownIdentity=%s,context=%s]", getClass().getSimpleName(), ownIdentity, context);
+		}
+
+	}
+
+	/**
+	 * Job that adds a context to an {@link OwnIdentity}.
+	 *
+	 * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
+	 */
+	private class AddContextJob extends WebOfTrustContextUpdateJob {
+
+		/**
+		 * Creates a new add-context job.
+		 *
+		 * @param ownIdentity
+		 *            The own identity whose contexts to manage
+		 * @param context
+		 *            The context to add
+		 */
+		public AddContextJob(OwnIdentity ownIdentity, String context) {
+			super(ownIdentity, context);
+		}
+
+		/**
+		 * {@inheritDoc}
+		 */
+		@Override
+		@SuppressWarnings("synthetic-access")
+		public void run() {
+			try {
+				webOfTrustConnector.addContext(ownIdentity, context);
+				ownIdentity.addContext(context);
+				finish(true);
+			} catch (PluginException pe1) {
+				logger.log(Level.WARNING, String.format("Could not add Context “%2$s” to Own Identity %1$s!", ownIdentity, context), pe1);
+				finish(false);
+			}
+		}
+
+	}
+
+	/**
+	 * Job that removes a context from an {@link OwnIdentity}.
+	 *
+	 * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
+	 */
+	private class RemoveContextJob extends WebOfTrustContextUpdateJob {
+
+		/**
+		 * Creates a new remove-context job.
+		 *
+		 * @param ownIdentity
+		 *            The own identity whose contexts to manage
+		 * @param context
+		 *            The context to remove
+		 */
+		public RemoveContextJob(OwnIdentity ownIdentity, String context) {
+			super(ownIdentity, context);
+		}
+
+		/**
+		 * {@inheritDoc}
+		 */
+		@Override
+		@SuppressWarnings("synthetic-access")
+		public void run() {
+			try {
+				webOfTrustConnector.removeContext(ownIdentity, context);
+				ownIdentity.removeContext(context);
+				finish(true);
+			} catch (PluginException pe1) {
+				logger.log(Level.WARNING, String.format("Could not remove Context “%2$s” to Own Identity %1$s!", ownIdentity, context), pe1);
+				finish(false);
+			}
+		}
+
 	}
 
 }