Add test for lock Sone ajax page
[Sone.git] / src / main / kotlin / net / pterodactylus / sone / web / ajax / LikeAjaxPage.kt
diff --git a/src/main/kotlin/net/pterodactylus/sone/web/ajax/LikeAjaxPage.kt b/src/main/kotlin/net/pterodactylus/sone/web/ajax/LikeAjaxPage.kt
new file mode 100644 (file)
index 0000000..3b206c8
--- /dev/null
@@ -0,0 +1,32 @@
+package net.pterodactylus.sone.web.ajax
+
+import net.pterodactylus.sone.data.Sone
+import net.pterodactylus.sone.utils.let
+import net.pterodactylus.sone.utils.parameters
+import net.pterodactylus.sone.web.WebInterface
+import net.pterodactylus.sone.web.page.FreenetRequest
+
+/**
+ * AJAX page that lets the user like a [net.pterodactylus.sone.data.Post].
+ */
+class LikeAjaxPage(webInterface: WebInterface) : LoggedInJsonPage("like.ajax", webInterface) {
+
+       override fun createJsonObject(currentSone: Sone, request: FreenetRequest) =
+                       when (request.parameters["type"]) {
+                               "post" -> request.parameters["post"]
+                                               .let(webInterface.core::getPost)
+                                               ?.let { currentSone.addLikedPostId(it.id) }
+                                               ?.also { webInterface.core.touchConfiguration() }
+                                               ?.let { createSuccessJsonObject() }
+                                               ?: createErrorJsonObject("invalid-post-id")
+                               "reply" -> request.parameters["reply"]
+                                               .let(webInterface.core::getPostReply)
+                                               ?.let { currentSone.addLikedReplyId(it.id) }
+                                               ?.also { webInterface.core.touchConfiguration() }
+                                               ?.let { createSuccessJsonObject() }
+                                               ?: createErrorJsonObject("invalid-reply-id")
+                               else -> createErrorJsonObject("invalid-type")
+                       }
+
+
+}