X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Fmain%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2Fajax%2FUnfollowSoneAjaxPage.kt;h=b1bca19b1d546e2fde420e4726145d49b4ff2f7e;hp=229765d2dbce6cdfed4a0264433f524c6decc0bb;hb=dfa9f12fd4923638b9eb5d99723af6437c307445;hpb=041a849c146bb1aa8980f97f3c2adb6653b6f8ca

diff --git a/src/main/kotlin/net/pterodactylus/sone/web/ajax/UnfollowSoneAjaxPage.kt b/src/main/kotlin/net/pterodactylus/sone/web/ajax/UnfollowSoneAjaxPage.kt
index 229765d..b1bca19 100644
--- a/src/main/kotlin/net/pterodactylus/sone/web/ajax/UnfollowSoneAjaxPage.kt
+++ b/src/main/kotlin/net/pterodactylus/sone/web/ajax/UnfollowSoneAjaxPage.kt
@@ -4,16 +4,18 @@ import net.pterodactylus.sone.data.Sone
 import net.pterodactylus.sone.utils.parameters
 import net.pterodactylus.sone.web.WebInterface
 import net.pterodactylus.sone.web.page.FreenetRequest
+import javax.inject.Inject
 
 /**
  * AJAX page that lets a Sone unfollow another Sone.
  */
-class UnfollowSoneAjaxPage(webInterface: WebInterface) : LoggedInJsonPage("unfollowSone.ajax", webInterface) {
+class UnfollowSoneAjaxPage @Inject constructor(webInterface: WebInterface) :
+		LoggedInJsonPage("unfollowSone.ajax", webInterface) {
 
 	override fun createJsonObject(currentSone: Sone, request: FreenetRequest) =
 			request.parameters["sone"]
-					?.takeIf { webInterface.core.getSone(it).isPresent }
-					?.also { webInterface.core.unfollowSone(currentSone, it) }
+					?.let(core::getSone)
+					?.also { core.unfollowSone(currentSone, it.id) }
 					?.let { createSuccessJsonObject() }
 					?: createErrorJsonObject("invalid-sone-id")