X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Ffcp%2Fhighlevel%2FFcpClient.java;h=bc1092eacc912b2a1000c5fdc6637526ac85c9ec;hb=8ba2bf0baac5fa3b872c4a1f5020a15d825e525c;hp=d090d645ae9fb3ef94f27835ce5e03243d9fdf59;hpb=30ce501106648904c6079bba147103750d801ce4;p=jFCPlib.git
diff --git a/src/main/java/net/pterodactylus/fcp/highlevel/FcpClient.java b/src/main/java/net/pterodactylus/fcp/highlevel/FcpClient.java
index d090d64..bc1092e 100644
--- a/src/main/java/net/pterodactylus/fcp/highlevel/FcpClient.java
+++ b/src/main/java/net/pterodactylus/fcp/highlevel/FcpClient.java
@@ -18,6 +18,8 @@
package net.pterodactylus.fcp.highlevel;
+import static com.google.common.collect.FluentIterable.from;
+
import java.io.Closeable;
import java.io.IOException;
import java.io.InputStream;
@@ -32,6 +34,7 @@ import java.util.Map;
import java.util.Map.Entry;
import java.util.Set;
import java.util.concurrent.CountDownLatch;
+import java.util.concurrent.atomic.AtomicReference;
import net.pterodactylus.fcp.AddPeer;
import net.pterodactylus.fcp.AllData;
@@ -68,10 +71,8 @@ import net.pterodactylus.fcp.RemovePeer;
import net.pterodactylus.fcp.SSKKeypair;
import net.pterodactylus.fcp.SimpleProgress;
import net.pterodactylus.fcp.WatchGlobal;
-import net.pterodactylus.util.filter.Filter;
-import net.pterodactylus.util.filter.Filters;
-import net.pterodactylus.util.io.TemporaryInputStream;
-import net.pterodactylus.util.thread.ObjectWrapper;
+
+import com.google.common.base.Predicate;
/**
* High-level FCP client that hides the details of the underlying FCP
@@ -307,7 +308,8 @@ public class FcpClient implements Closeable {
}
/**
- * Returns the file with the given URI.
+ * Returns the file with the given URI. The retrieved data will be run
+ * through Freenetâs content filter.
*
* @param uri
* The URI to get
@@ -318,6 +320,24 @@ public class FcpClient implements Closeable {
* if an FCP error occurs
*/
public GetResult getURI(final String uri) throws IOException, FcpException {
+ return getURI(uri, true);
+ }
+
+ /**
+ * Returns the file with the given URI.
+ *
+ * @param uri
+ * The URI to get
+ * @param filterData
+ * {@code true} to filter the retrieved data, {@code false}
+ * otherwise
+ * @return The result of the get request
+ * @throws IOException
+ * if an I/O error occurs
+ * @throws FcpException
+ * if an FCP error occurs
+ */
+ public GetResult getURI(final String uri, final boolean filterData) throws IOException, FcpException {
checkConnected(true);
final GetResult getResult = new GetResult();
new ExtendedFcpAdapter() {
@@ -329,6 +349,7 @@ public class FcpClient implements Closeable {
@SuppressWarnings("synthetic-access")
public void run() throws IOException {
ClientGet clientGet = new ClientGet(uri, identifier);
+ clientGet.setFilterData(filterData);
fcpConnection.sendMessage(clientGet);
}
@@ -342,7 +363,9 @@ public class FcpClient implements Closeable {
String newUri = getFailed.getRedirectURI();
getResult.realUri(newUri);
try {
- fcpConnection.sendMessage(new ClientGet(newUri, identifier));
+ ClientGet clientGet = new ClientGet(newUri, identifier);
+ clientGet.setFilterData(filterData);
+ fcpConnection.sendMessage(clientGet);
} catch (IOException ioe1) {
getResult.success(false).exception(ioe1);
completionLatch.countDown();
@@ -358,13 +381,7 @@ public class FcpClient implements Closeable {
if (!allData.getIdentifier().equals(identifier)) {
return;
}
- InputStream temporaryInputStream;
- try {
- temporaryInputStream = new TemporaryInputStream(allData.getPayloadInputStream());
- getResult.success(true).contentType(allData.getContentType()).contentLength(allData.getDataLength()).inputStream(temporaryInputStream);
- } catch (IOException ioe1) {
- getResult.success(false).exception(ioe1);
- }
+ getResult.success(true).contentType(allData.getContentType()).contentLength(allData.getDataLength()).inputStream(allData.getPayloadInputStream());
completionLatch.countDown();
}
@@ -580,8 +597,8 @@ public class FcpClient implements Closeable {
/**
* Adds a peer, reading the noderef of the peer from the given file.
- * Note: the file to read the noderef from has to reside on
- * the same machine as the node!
+ * Note: the file to read the noderef from has to reside
+ * on the same machine as the node!
*
* @param file
* The name of the file containing the peerâs noderef
@@ -720,7 +737,7 @@ public class FcpClient implements Closeable {
* if an FCP error occurs
*/
public PeerNote getPeerNote(final Peer peer) throws IOException, FcpException {
- final ObjectWrapper objectWrapper = new ObjectWrapper();
+ final AtomicReference objectWrapper = new AtomicReference();
new ExtendedFcpAdapter() {
/**
@@ -806,7 +823,7 @@ public class FcpClient implements Closeable {
* if an FCP error occurs
*/
public SSKKeypair generateKeyPair() throws IOException, FcpException {
- final ObjectWrapper sskKeypairWrapper = new ObjectWrapper();
+ final AtomicReference sskKeypairWrapper = new AtomicReference();
new ExtendedFcpAdapter() {
/**
@@ -848,16 +865,12 @@ public class FcpClient implements Closeable {
* if an FCP error occurs
*/
public Collection getGetRequests(final boolean global) throws IOException, FcpException {
- return Filters.filteredCollection(getRequests(global), new Filter() {
-
- /**
- * {@inheritDoc}
- */
+ return from(getRequests(global)).filter(new Predicate() {
@Override
- public boolean filterObject(Request request) {
+ public boolean apply(Request request) {
return request instanceof GetRequest;
}
- });
+ }).toList();
}
/**
@@ -874,16 +887,12 @@ public class FcpClient implements Closeable {
* if an FCP error occurs
*/
public Collection getPutRequests(final boolean global) throws IOException, FcpException {
- return Filters.filteredCollection(getRequests(global), new Filter() {
-
- /**
- * {@inheritDoc}
- */
+ return from(getRequests(global)).filter(new Predicate() {
@Override
- public boolean filterObject(Request request) {
+ public boolean apply(Request request) {
return request instanceof PutRequest;
}
- });
+ }).toList();
}
/**
@@ -891,8 +900,8 @@ public class FcpClient implements Closeable {
*
* @param global
* true
to return requests from the global queue,
- * false
to only show requests from the client-local
- * queue
+ * false
to only show requests from the
+ * client-local queue
* @return All requests
* @throws IOException
* if an I/O error occurs
@@ -1097,7 +1106,7 @@ public class FcpClient implements Closeable {
* if an I/O error occurs
*/
public NodeData getNodeInformation(final Boolean giveOpennetRef, final Boolean withPrivate, final Boolean withVolatile) throws IOException, FcpException {
- final ObjectWrapper nodeDataWrapper = new ObjectWrapper();
+ final AtomicReference nodeDataWrapper = new AtomicReference();
new ExtendedFcpAdapter() {
@Override