X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fnet%2Fpterodactylus%2Fjsite%2Fcore%2FRequestManager.java;h=8f73b8d3a22cba89ee3cd13b808aa6f63e9b3744;hb=ffb4c30743497598db6103e976e4c35fcd880c35;hp=4e23d7986f86064aebc0c18ef6777d8cdbc46a88;hpb=9030b0e6266f4962aee282fa3ec689547775b925;p=jSite2.git

diff --git a/src/net/pterodactylus/jsite/core/RequestManager.java b/src/net/pterodactylus/jsite/core/RequestManager.java
index 4e23d79..8f73b8d 100644
--- a/src/net/pterodactylus/jsite/core/RequestManager.java
+++ b/src/net/pterodactylus/jsite/core/RequestManager.java
@@ -126,6 +126,34 @@ public class RequestManager implements NodeListener, HighLevelProgressListener {
 		}
 	}
 
+	/**
+	 * Notifies all listeners that a request has finished.
+	 * 
+	 * @see RequestListener#requestFinished(Request)
+	 * @param request
+	 *            The request that has finished
+	 */
+	private void fireRequestFinished(Request request) {
+		for (RequestListener requestListener: requestListeners) {
+			requestListener.requestFinished(request);
+		}
+	}
+
+	/**
+	 * Notifies all listeners that a request has generated a URI.
+	 * 
+	 * @see RequestListener#requestGeneratedURI(Request, String)
+	 * @param request
+	 *            The request that has generated a URI
+	 * @param uri
+	 *            The generated URI
+	 */
+	private void fireRequestGeneratedURI(Request request, String uri) {
+		for (RequestListener requestListener: requestListeners) {
+			requestListener.requestGeneratedURI(request, uri);
+		}
+	}
+
 	//
 	// ACCESSORS
 	//
@@ -308,13 +336,24 @@ public class RequestManager implements NodeListener, HighLevelProgressListener {
 			identifierRequests.put(identifier, request);
 			fireRequestAdded(request);
 		}
-		request.setTotalBlocks(highLevelProgress.getTotalBlocks());
-		request.setRequiredBlocks(highLevelProgress.getRequiredBlocks());
-		request.setSuccessfulBlocks(highLevelProgress.getSuccessfulBlocks());
-		request.setFailedBlocks(highLevelProgress.getFailedBlocks());
-		request.setFatallyFailedBlocks(highLevelProgress.getFatallyFailedBlocks());
-		request.setTotalFinalized(highLevelProgress.isTotalFinalized());
-		fireRequestProgressed(request);
+		if (highLevelProgress.isFinished()) {
+			request.setFinished(true);
+			request.setSuccessful(!highLevelProgress.isFailed());
+			fireRequestFinished(request);
+		} else if (highLevelProgress.isFetchable()) {
+			/* TODO - ignore? */
+		} else if (highLevelProgress.getURI() != null) {
+			request.setURI(highLevelProgress.getURI());
+			fireRequestGeneratedURI(request, highLevelProgress.getURI());
+		} else {
+			request.setTotalBlocks(highLevelProgress.getTotalBlocks());
+			request.setRequiredBlocks(highLevelProgress.getRequiredBlocks());
+			request.setSuccessfulBlocks(highLevelProgress.getSuccessfulBlocks());
+			request.setFailedBlocks(highLevelProgress.getFailedBlocks());
+			request.setFatallyFailedBlocks(highLevelProgress.getFatallyFailedBlocks());
+			request.setTotalFinalized(highLevelProgress.isTotalFinalized());
+			fireRequestProgressed(request);
+		}
 	}
 
 }