X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fcore%2FFreenetInterface.java;h=30ba69a23e00f36e24de7b35a9e2d163cddb5a27;hb=b29cf0908e6dfd2b55220a3a7e44200f2fe5b19e;hp=8f9ec8fe12a0bd3c503a5439fc23792a29b3088d;hpb=087cfe551f0e87ebb128820045bac5b2c04b6ac2;p=Sone.git

diff --git a/src/main/java/net/pterodactylus/sone/core/FreenetInterface.java b/src/main/java/net/pterodactylus/sone/core/FreenetInterface.java
index 8f9ec8f..30ba69a 100644
--- a/src/main/java/net/pterodactylus/sone/core/FreenetInterface.java
+++ b/src/main/java/net/pterodactylus/sone/core/FreenetInterface.java
@@ -17,23 +17,35 @@
 
 package net.pterodactylus.sone.core;
 
+import java.net.MalformedURLException;
+import java.util.HashMap;
+import java.util.Map;
 import java.util.logging.Level;
 import java.util.logging.Logger;
 
+import net.pterodactylus.sone.data.Sone;
 import net.pterodactylus.util.logging.Logging;
-import net.pterodactylus.util.service.AbstractService;
+
+import com.db4o.ObjectContainer;
+
 import freenet.client.FetchException;
 import freenet.client.FetchResult;
 import freenet.client.HighLevelSimpleClient;
+import freenet.client.HighLevelSimpleClientImpl;
+import freenet.client.InsertException;
+import freenet.client.async.ClientContext;
+import freenet.client.async.USKCallback;
 import freenet.keys.FreenetURI;
+import freenet.keys.USK;
 import freenet.node.Node;
+import freenet.node.RequestStarter;
 
 /**
  * Contains all necessary functionality for interacting with the Freenet node.
  *
  * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
  */
-public class FreenetInterface extends AbstractService {
+public class FreenetInterface {
 
 	/** The logger. */
 	private static final Logger logger = Logging.getLogger(FreenetInterface.class);
@@ -44,6 +56,9 @@ public class FreenetInterface extends AbstractService {
 	/** The high-level client to use for requests. */
 	private final HighLevelSimpleClient client;
 
+	/** The USK callbacks. */
+	private final Map<String, USKCallback> soneUskCallbacks = new HashMap<String, USKCallback>();
+
 	/**
 	 * Creates a new Freenet interface.
 	 *
@@ -69,11 +84,111 @@ public class FreenetInterface extends AbstractService {
 	 * @return The result of the fetch, or {@code null} if an error occured
 	 */
 	public FetchResult fetchUri(FreenetURI uri) {
+		FetchResult fetchResult = null;
+		FreenetURI currentUri = new FreenetURI(uri);
+		while (true) {
+			try {
+				fetchResult = client.fetch(currentUri);
+				return fetchResult;
+			} catch (FetchException fe1) {
+				if (fe1.getMode() == FetchException.PERMANENT_REDIRECT) {
+					currentUri = fe1.newURI;
+					continue;
+				}
+				logger.log(Level.WARNING, "Could not fetch “" + uri + "”!", fe1);
+				return null;
+			}
+		}
+	}
+
+	/**
+	 * Creates a key pair.
+	 *
+	 * @return The request key at index 0, the insert key at index 1
+	 */
+	public String[] generateKeyPair() {
+		FreenetURI[] keyPair = client.generateKeyPair("");
+		return new String[] { keyPair[1].toString(), keyPair[0].toString() };
+	}
+
+	/**
+	 * Inserts a directory into Freenet.
+	 *
+	 * @param insertUri
+	 *            The insert URI
+	 * @param manifestEntries
+	 *            The directory entries
+	 * @param defaultFile
+	 *            The name of the default file
+	 * @return The generated URI
+	 * @throws SoneException
+	 *             if an insert error occurs
+	 */
+	public FreenetURI insertDirectory(FreenetURI insertUri, HashMap<String, Object> manifestEntries, String defaultFile) throws SoneException {
+		try {
+			return client.insertManifest(insertUri, manifestEntries, defaultFile);
+		} catch (InsertException ie1) {
+			throw new SoneException(null, ie1);
+		}
+	}
+
+	/**
+	 * Registers the USK for the given Sone and notifies the given
+	 * {@link SoneDownloader} if an update was found.
+	 *
+	 * @param sone
+	 *            The Sone to watch
+	 * @param soneDownloader
+	 *            The Sone download to notify on updates
+	 */
+	public void registerUsk(final Sone sone, final SoneDownloader soneDownloader) {
+		try {
+			logger.log(Level.FINE, "Registering Sone “%s” for USK updates at %s…", new Object[] { sone, sone.getRequestUri().setMetaString(new String[] { "sone.xml" }) });
+			USKCallback uskCallback = new USKCallback() {
+
+				@Override
+				@SuppressWarnings("synthetic-access")
+				public void onFoundEdition(long edition, USK key, ObjectContainer objectContainer, ClientContext clientContext, boolean metadata, short codec, byte[] data, boolean newKnownGood, boolean newSlotToo) {
+					logger.log(Level.FINE, "Found USK update for Sone “%s” at %s, new known good: %s, new slot too: %s.", new Object[] { sone, key, newKnownGood, newSlotToo });
+					if (newKnownGood) {
+						sone.updateUris(key.suggestedEdition);
+						soneDownloader.fetchSone(sone);
+					}
+				}
+
+				@Override
+				public short getPollingPriorityProgress() {
+					return RequestStarter.INTERACTIVE_PRIORITY_CLASS;
+				}
+
+				@Override
+				public short getPollingPriorityNormal() {
+					return RequestStarter.INTERACTIVE_PRIORITY_CLASS;
+				}
+			};
+			soneUskCallbacks.put(sone.getId(), uskCallback);
+			node.clientCore.uskManager.subscribe(USK.create(sone.getRequestUri()), uskCallback, true, (HighLevelSimpleClientImpl) client);
+		} catch (MalformedURLException mue1) {
+			logger.log(Level.WARNING, "Could not subscribe USK “" + sone.getRequestUri() + "”!", mue1);
+		}
+	}
+
+	/**
+	 * Unsubscribes the request URI of the given Sone.
+	 *
+	 * @param sone
+	 *            The Sone to unregister
+	 */
+	public void unregisterUsk(Sone sone) {
+		USKCallback uskCallback = soneUskCallbacks.remove(sone.getId());
+		if (uskCallback == null) {
+			return;
+		}
 		try {
-			return client.fetch(uri);
-		} catch (FetchException fe1) {
-			logger.log(Level.WARNING, "Could not fetch “" + uri + "”!", fe1);
-			return null;
+			logger.log(Level.FINEST, "Unsubscribing from USK for %s…", new Object[] { sone });
+			node.clientCore.uskManager.unsubscribe(USK.create(sone.getRequestUri()), uskCallback);
+		} catch (MalformedURLException mue1) {
+			logger.log(Level.FINE, "Could not unsubscribe USK “" + sone.getRequestUri() + "”!", mue1);
 		}
 	}