♻️ Slightly improve messy expression

[Sone.git] / src / main / kotlin / net / pterodactylus / sone / web / FreenetSessionProvider.kt

diff --git a/src/main/kotlin/net/pterodactylus/sone/web/FreenetSessionProvider.kt b/src/main/kotlin/net/pterodactylus/sone/web/FreenetSessionProvider.kt
index 80bacdf..8928f12 100644 (file)
--- a/src/main/kotlin/net/pterodactylus/sone/web/FreenetSessionProvider.kt
+++ b/src/main/kotlin/net/pterodactylus/sone/web/FreenetSessionProvider.kt
@@ -41,10 +41,13 @@ class FreenetSessionProvider @Inject constructor(private val soneProvider: SoneP
                        sessionManager.useSession(toadletContext)
                                        ?.removeAttribute("Sone.CurrentSone")
                } else {
-                       (sessionManager.useSession(toadletContext)
-                                       ?: sessionManager.createSession(UUID.randomUUID().toString(), toadletContext))
+                       sessionManager.getOrCreateSession(toadletContext)
                                        ?.setAttribute("Sone.CurrentSone", sone.id)
                }
        }
 
+       private fun SessionManager.getOrCreateSession(toadletContext: ToadletContext) =
+                       useSession(toadletContext)
+                                       ?: createSession(UUID.randomUUID().toString(), toadletContext)
+
 }