Replace ajax base page with Kotlin version

diff --git a/src/main/java/net/pterodactylus/sone/web/ajax/JsonPage.java b/src/main/java/net/pterodactylus/sone/web/ajax/JsonPage.java
deleted file mode 100644 (file)
index 9583437..0000000
--- a/src/main/java/net/pterodactylus/sone/web/ajax/JsonPage.java
+++ /dev/null
@@ -1,233 +0,0 @@
-/*
- * Sone - JsonPage.java - Copyright © 2010–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
- * the Free Software Foundation, either version 3 of the License, or
- * (at your option) any later version.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
- * GNU General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program.  If not, see <http://www.gnu.org/licenses/>.
- */
-
-package net.pterodactylus.sone.web.ajax;
-
-import static java.util.logging.Logger.getLogger;
-
-import java.io.ByteArrayOutputStream;
-import java.io.IOException;
-import java.io.OutputStreamWriter;
-import java.io.PrintWriter;
-import java.util.logging.Level;
-import java.util.logging.Logger;
-
-import javax.annotation.Nonnull;
-
-import net.pterodactylus.sone.data.Sone;
-import net.pterodactylus.sone.web.SessionProvider;
-import net.pterodactylus.sone.web.WebInterface;
-import net.pterodactylus.sone.web.page.FreenetRequest;
-import net.pterodactylus.util.io.Closer;
-import net.pterodactylus.util.web.Page;
-import net.pterodactylus.util.web.Response;
-
-import com.fasterxml.jackson.databind.ObjectMapper;
-import freenet.clients.http.ToadletContext;
-
-/**
- * A JSON page is a specialized {@link Page} that will always return a JSON
- * object to the browser, e.g. for use with AJAX or other scripting frameworks.
- *
- * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
- */
-public abstract class JsonPage implements Page<FreenetRequest> {
-
-       /** The logger. */
-       private static final Logger logger = getLogger(JsonPage.class.getName());
-
-       /** The JSON serializer. */
-       private static final ObjectMapper objectMapper = new ObjectMapper();
-
-       /** The path of the page. */
-       private final String path;
-
-       /** The Sone web interface. */
-       protected final WebInterface webInterface;
-       private final SessionProvider sessionProvider;
-
-       /**
-        * Creates a new JSON page at the given path.
-        *
-        * @param path
-        *            The path of the page
-        * @param webInterface
-        *            The Sone web interface
-        */
-       public JsonPage(String path, WebInterface webInterface) {
-               this.path = path;
-               this.webInterface = webInterface;
-               this.sessionProvider = webInterface;
-       }
-
-       //
-       // ACCESSORS
-       //
-
-       protected Sone getCurrentSone(ToadletContext toadletContext) {
-               return sessionProvider.getCurrentSone(toadletContext, true);
-       }
-
-       protected Sone getCurrentSone(ToadletContext toadletContext, boolean createSession) {
-               return sessionProvider.getCurrentSone(toadletContext, createSession);
-       }
-
-       //
-       // METHODS FOR SUBCLASSES TO OVERRIDE
-       //
-
-       /**
-        * This method is called to create the JSON object that is returned back to
-        * the browser.
-        *
-        * @param request
-        *            The request to handle
-        * @return The created JSON object
-        */
-       @Nonnull
-       protected abstract JsonReturnObject createJsonObject(@Nonnull FreenetRequest request);
-
-       /**
-        * Returns whether this command needs the form password for authentication
-        * and to prevent abuse.
-        *
-        * @return {@code true} if the form password (given as “formPassword”) is
-        *         required, {@code false} otherwise
-        */
-       @SuppressWarnings("static-method")
-       protected boolean needsFormPassword() {
-               return true;
-       }
-
-       /**
-        * Returns whether this page requires the user to be logged in.
-        *
-        * @return {@code true} if the user needs to be logged in to use this page,
-        *         {@code false} otherwise
-        */
-       @SuppressWarnings("static-method")
-       protected boolean requiresLogin() {
-               return true;
-       }
-
-       //
-       // PROTECTED METHODS
-       //
-
-       /**
-        * Creates a success reply.
-        *
-        * @return A reply signaling success
-        */
-       @Nonnull
-       protected static JsonReturnObject createSuccessJsonObject() {
-               return new JsonReturnObject(true);
-       }
-
-       /**
-        * Creates an error reply.
-        *
-        * @param error
-        *            The error that has occured
-        * @return The JSON object, signalling failure and the error code
-        */
-       @Nonnull
-       protected static JsonReturnObject createErrorJsonObject(String error) {
-               return new JsonErrorReturnObject(error);
-       }
-
-       //
-       // PAGE METHODS
-       //
-
-       /**
-        * {@inheritDoc}
-        */
-       @Override
-       public String getPath() {
-               return path;
-       }
-
-       /**
-        * {@inheritDoc}
-        */
-       @Override
-       public boolean isPrefixPage() {
-               return false;
-       }
-
-       /**
-        * {@inheritDoc}
-        */
-       @Override
-       public Response handleRequest(FreenetRequest request, Response response) throws IOException {
-               if (webInterface.getCore().getPreferences().isRequireFullAccess() && !request.getToadletContext().isAllowedFullAccess()) {
-                       return response.setStatusCode(403).setStatusText("Forbidden").setContentType("application/json").write(objectMapper.writeValueAsString(new JsonErrorReturnObject("auth-required")));
-               }
-               if (needsFormPassword()) {
-                       String formPassword = request.getHttpRequest().getParam("formPassword");
-                       if (!webInterface.getFormPassword().equals(formPassword)) {
-                               return response.setStatusCode(403).setStatusText("Forbidden").setContentType("application/json").write(objectMapper.writeValueAsString(new JsonErrorReturnObject("auth-required")));
-                       }
-               }
-               if (requiresLogin()) {
-                       if (getCurrentSone(request.getToadletContext(), false) == null) {
-                               return response.setStatusCode(403).setStatusText("Forbidden").setContentType("application/json").write(objectMapper.writeValueAsString(new JsonErrorReturnObject("auth-required")));
-                       }
-               }
-               try {
-                       JsonReturnObject jsonObject = createJsonObject(request);
-                       return response.setStatusCode(200).setStatusText("OK").setContentType("application/json").write(objectMapper.writeValueAsString(jsonObject));
-               } catch (Exception e1) {
-                       logger.log(Level.WARNING, "Error executing JSON page!", e1);
-                       return response.setStatusCode(500).setStatusText(e1.getMessage()).setContentType("text/plain").write(dumpStackTrace(e1));
-               }
-       }
-
-       //
-       // PRIVATE METHODS
-       //
-
-       /**
-        * Returns a byte array containing the stack trace of the given throwable.
-        *
-        * @param t
-        *            The throwable whose stack trace to dump into an array
-        * @return The array with the stack trace, or an empty array if the stack
-        *         trace could not be dumped
-        */
-       private static byte[] dumpStackTrace(Throwable t) {
-               ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
-               OutputStreamWriter writer = null;
-               PrintWriter printWriter = null;
-               try {
-                       writer = new OutputStreamWriter(byteArrayOutputStream, "UTF-8");
-                       printWriter = new PrintWriter(writer);
-                       t.printStackTrace(printWriter);
-                       printWriter.flush();
-                       return byteArrayOutputStream.toByteArray();
-               } catch (IOException ioe1) {
-                       /* quite not possible. */
-                       return new byte[0];
-               } finally {
-                       Closer.close(printWriter);
-                       Closer.close(writer);
-                       Closer.close(byteArrayOutputStream);
-               }
-       }
-
-}

diff --git a/src/main/kotlin/net/pterodactylus/sone/web/ajax/JsonPage.kt b/src/main/kotlin/net/pterodactylus/sone/web/ajax/JsonPage.kt
new file mode 100644 (file)
index 0000000..6ba842f
--- /dev/null
+++ b/src/main/kotlin/net/pterodactylus/sone/web/ajax/JsonPage.kt
@@ -0,0 +1,64 @@
+package net.pterodactylus.sone.web.ajax
+
+import com.fasterxml.jackson.databind.ObjectMapper
+import freenet.clients.http.ToadletContext
+import net.pterodactylus.sone.utils.parameters
+import net.pterodactylus.sone.web.SessionProvider
+import net.pterodactylus.sone.web.WebInterface
+import net.pterodactylus.sone.web.page.FreenetRequest
+import net.pterodactylus.util.web.Page
+import net.pterodactylus.util.web.Response
+import java.io.ByteArrayOutputStream
+import java.io.PrintStream
+
+/**
+ * A JSON page is a specialized [Page] that will always return a JSON
+ * object to the browser, e.g. for use with AJAX or other scripting frameworks.
+ */
+abstract class JsonPage(private val path: String, protected val webInterface: WebInterface) : Page<FreenetRequest> {
+
+       private val objectMapper = ObjectMapper()
+       private val sessionProvider: SessionProvider = webInterface
+
+       override fun getPath() = path
+       override fun isPrefixPage() = false
+
+       open fun needsFormPassword() = true
+       open fun requiresLogin() = true
+
+       protected fun createSuccessJsonObject() = JsonReturnObject(true)
+       protected fun createErrorJsonObject(error: String) =
+                       JsonErrorReturnObject(error)
+
+       protected fun getCurrentSone(toadletContext: ToadletContext, createSession: Boolean = true) =
+                       sessionProvider.getCurrentSone(toadletContext, createSession)
+
+       override fun handleRequest(request: FreenetRequest, response: Response): Response {
+               if (webInterface.core.preferences.isRequireFullAccess && !request.toadletContext.isAllowedFullAccess) {
+                       return response.setStatusCode(403).setStatusText("Forbidden").setContentType("application/json").write(createErrorJsonObject("auth-required").asJsonString())
+               }
+               if (needsFormPassword() && request.parameters["formPassword"] != webInterface.formPassword) {
+                       return response.setStatusCode(403).setStatusText("Forbidden").setContentType("application/json").write(createErrorJsonObject("auth-required").asJsonString())
+               }
+               if (requiresLogin() && (sessionProvider.getCurrentSone(request.toadletContext, false) == null)) {
+                       return response.setStatusCode(403).setStatusText("Forbidden").setContentType("application/json").write(createErrorJsonObject("auth-required").asJsonString())
+               }
+               return try {
+                       response.setStatusCode(200).setStatusText("OK").setContentType("application/json").write(createJsonObject(request).asJsonString())
+               } catch (e: Exception) {
+                       response.setStatusCode(500).setStatusText(e.message).setContentType("text/plain").write(e.dumpStackTrace())
+               }
+       }
+
+       abstract fun createJsonObject(request: FreenetRequest): JsonReturnObject
+
+       private fun JsonReturnObject.asJsonString(): String = objectMapper.writeValueAsString(this)
+
+       private fun Throwable.dumpStackTrace(): String = ByteArrayOutputStream().use {
+               PrintStream(it, true, "UTF-8").use {
+                       this.printStackTrace(it)
+               }
+               it.toString("UTF-8")
+       }
+
+}