X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fcore%2FFreenetInterface.java;h=4b33139a4862d1d99e52f88f3302a2f7dab033ba;hp=a06b5b006964a8359f34a1cb621f5d292c4cb2cb;hb=f814ad7a876bea03db77b30532eab3e7e03a05d8;hpb=c4ae226ec5052116cefc542ae2017036a7bc6332

diff --git a/src/main/java/net/pterodactylus/sone/core/FreenetInterface.java b/src/main/java/net/pterodactylus/sone/core/FreenetInterface.java
index a06b5b0..4b33139 100644
--- a/src/main/java/net/pterodactylus/sone/core/FreenetInterface.java
+++ b/src/main/java/net/pterodactylus/sone/core/FreenetInterface.java
@@ -23,6 +23,7 @@ import static java.util.logging.Level.WARNING;
 import static java.util.logging.Logger.getLogger;
 import static net.pterodactylus.sone.freenet.Key.routingKey;
 
+import java.io.IOException;
 import java.net.MalformedURLException;
 import java.util.Collections;
 import java.util.HashMap;
@@ -30,6 +31,8 @@ import java.util.Map;
 import java.util.logging.Level;
 import java.util.logging.Logger;
 
+import javax.annotation.Nonnull;
+
 import net.pterodactylus.sone.core.event.ImageInsertAbortedEvent;
 import net.pterodactylus.sone.core.event.ImageInsertFailedEvent;
 import net.pterodactylus.sone.core.event.ImageInsertFinishedEvent;
@@ -44,6 +47,7 @@ import com.google.inject.Inject;
 import com.google.inject.Singleton;
 
 import freenet.client.ClientMetadata;
+import freenet.client.FetchContext;
 import freenet.client.FetchException;
 import freenet.client.FetchException.FetchExceptionMode;
 import freenet.client.FetchResult;
@@ -51,16 +55,21 @@ import freenet.client.HighLevelSimpleClient;
 import freenet.client.InsertBlock;
 import freenet.client.InsertContext;
 import freenet.client.InsertException;
+import freenet.client.Metadata;
 import freenet.client.async.BaseClientPutter;
 import freenet.client.async.ClientContext;
+import freenet.client.async.ClientGetCallback;
+import freenet.client.async.ClientGetter;
 import freenet.client.async.ClientPutCallback;
 import freenet.client.async.ClientPutter;
+import freenet.client.async.SnoopMetadata;
 import freenet.client.async.USKCallback;
 import freenet.keys.FreenetURI;
 import freenet.keys.InsertableClientSSK;
 import freenet.keys.USK;
 import freenet.node.Node;
 import freenet.node.RequestClient;
+import freenet.node.RequestClientBuilder;
 import freenet.node.RequestStarter;
 import freenet.support.api.Bucket;
 import freenet.support.api.RandomAccessBucket;
@@ -93,17 +102,8 @@ public class FreenetInterface {
 	/** The not-Sone-related USK callbacks. */
 	private final Map<FreenetURI, USKCallback> uriUskCallbacks = Collections.synchronizedMap(new HashMap<FreenetURI, USKCallback>());
 
-	private final RequestClient imageInserts = new RequestClient() {
-		@Override
-		public boolean persistent() {
-			return false;
-		}
-
-		@Override
-		public boolean realTimeFlag() {
-			return true;
-		}
-	};
+	private final RequestClient imageInserts = new RequestClientBuilder().realTime().build();
+	private final RequestClient imageLoader = new RequestClientBuilder().realTime().build();
 
 	/**
 	 * Creates a new Freenet interface.
@@ -148,6 +148,59 @@ public class FreenetInterface {
 		}
 	}
 
+	public void startFetch(final FreenetURI uri, final BackgroundFetchCallback backgroundFetchCallback) {
+		ClientGetCallback callback = new ClientGetCallback() {
+			@Override
+			public void onSuccess(FetchResult result, ClientGetter state) {
+				try {
+					backgroundFetchCallback.loaded(uri, result.getMimeType(), result.asByteArray());
+				} catch (IOException e) {
+					backgroundFetchCallback.failed(uri);
+				}
+			}
+
+			@Override
+			public void onFailure(FetchException e, ClientGetter state) {
+				backgroundFetchCallback.failed(uri);
+			}
+
+			@Override
+			public void onResume(ClientContext context) throws ResumeFailedException {
+				/* do nothing. */
+			}
+
+			@Override
+			public RequestClient getRequestClient() {
+				return imageLoader;
+			}
+		};
+		SnoopMetadata snoop = new SnoopMetadata() {
+			@Override
+			public boolean snoopMetadata(Metadata meta, ClientContext context) {
+				String mimeType = meta.getMIMEType();
+				boolean cancel = (mimeType == null) || backgroundFetchCallback.cancelForMimeType(uri, mimeType);
+				if (cancel) {
+					backgroundFetchCallback.failed(uri);
+				}
+				return cancel;
+			}
+		};
+		FetchContext fetchContext = client.getFetchContext();
+		try {
+			ClientGetter clientGetter = client.fetch(uri, 2097152, callback, fetchContext, RequestStarter.INTERACTIVE_PRIORITY_CLASS);
+			clientGetter.setMetaSnoop(snoop);
+			clientGetter.restart(uri, fetchContext.filterData, node.clientCore.clientContext);
+		} catch (FetchException fe) {
+			/* stupid exception that can not actually be thrown! */
+		}
+	}
+
+	public interface BackgroundFetchCallback {
+		boolean cancelForMimeType(@Nonnull FreenetURI uri, @Nonnull String mimeType);
+		void loaded(@Nonnull FreenetURI uri, @Nonnull String mimeType, @Nonnull byte[] data);
+		void failed(@Nonnull FreenetURI uri);
+	}
+
 	/**
 	 * Inserts the image data of the given {@link TemporaryImage} and returns
 	 * the given insert token that can be used to add listeners or cancel the