Merge branch 'release-0.9.6'

[Sone.git] / src / main / java / net / pterodactylus / sone / web / GetImagePage.java

diff --git a/src/main/java/net/pterodactylus/sone/web/GetImagePage.java b/src/main/java/net/pterodactylus/sone/web/GetImagePage.java
index fdd72a8..ec38d0a 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/web/GetImagePage.java
+++ b/src/main/java/net/pterodactylus/sone/web/GetImagePage.java
@@ -1,5 +1,5 @@
 /*
- * Sone - GetImagePage.java - Copyright © 2011 David Roden
+ * Sone - GetImagePage.java - Copyright © 2011–2016 David Roden
  *
  * This program is free software: you can redistribute it and/or modify
  * it under the terms of the GNU General Public License as published by
@@ -17,15 +17,20 @@
 
 package net.pterodactylus.sone.web;
 
+import java.io.IOException;
+import java.net.URI;
+
 import net.pterodactylus.sone.data.TemporaryImage;
-import net.pterodactylus.sone.web.page.Page;
+import net.pterodactylus.sone.web.page.FreenetPage;
+import net.pterodactylus.sone.web.page.FreenetRequest;
+import net.pterodactylus.util.web.Response;
 
 /**
  * Page that delivers a {@link TemporaryImage} to the browser.
  *
  * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
  */
-public class GetImagePage implements Page {
+public class GetImagePage implements FreenetPage {
 
        /** The Sone web interface. */
        private final WebInterface webInterface;
@@ -52,13 +57,30 @@ public class GetImagePage implements Page {
         * {@inheritDoc}
         */
        @Override
-       public Response handleRequest(Request request) {
+       public boolean isPrefixPage() {
+               return false;
+       }
+
+       /**
+        * {@inheritDoc}
+        */
+       @Override
+       public Response handleRequest(FreenetRequest request, Response response) throws IOException {
                String imageId = request.getHttpRequest().getParam("image");
                TemporaryImage temporaryImage = webInterface.getCore().getTemporaryImage(imageId);
                if (temporaryImage == null) {
-                       return new Response(404, "Not found.", "text/plain; charset=utf-8", "");
+                       return response.setStatusCode(404).setStatusText("Not found.").setContentType("text/html; charset=utf-8");
                }
-               return new Response(200, "OK", temporaryImage.getMimeType(), temporaryImage.getImageData()).setHeader("Content-Disposition", "attachment; filename=" + temporaryImage.getId() + "." + temporaryImage.getMimeType().substring(temporaryImage.getMimeType().lastIndexOf("/") + 1));
+               String contentType= temporaryImage.getMimeType();
+               return response.setStatusCode(200).setStatusText("OK").setContentType(contentType).addHeader("Content-Disposition", "attachment; filename=" + temporaryImage.getId() + "." + contentType.substring(contentType.lastIndexOf('/') + 1)).write(temporaryImage.getImageData());
+       }
+
+       /**
+        * {@inheritDoc}
+        */
+       @Override
+       public boolean isLinkExcepted(URI link) {
+               return false;
        }
 
 }