X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2Fnotification%2FNotificationHandler.kt;h=37df1ac64baca6afd85f800a758c85b93d634911;hb=b6a8b559407daefde5ce01967190cccf97e2bb00;hp=7f81f9a74fdbac99fadebf783e8c40d51d27576c;hpb=188857148a7b035bfc6033b1fd40d774bd1fac65;p=Sone.git

diff --git a/src/main/kotlin/net/pterodactylus/sone/web/notification/NotificationHandler.kt b/src/main/kotlin/net/pterodactylus/sone/web/notification/NotificationHandler.kt
index 7f81f9a..37df1ac 100644
--- a/src/main/kotlin/net/pterodactylus/sone/web/notification/NotificationHandler.kt
+++ b/src/main/kotlin/net/pterodactylus/sone/web/notification/NotificationHandler.kt
@@ -29,8 +29,11 @@ import javax.inject.*
 class NotificationHandler @Inject constructor(private val eventBus: EventBus, private val loaders: Loaders, private val notificationManager: NotificationManager) {
 
 	fun start() {
-		SoneLockedOnStartupHandler(notificationManager, loaders.loadTemplate("/templates/notify/soneLockedOnStartupNotification.html"))
-				.also(eventBus::register)
+		register { SoneLockedOnStartupHandler(it, loaders.loadTemplate("/templates/notify/soneLockedOnStartupNotification.html")) }
+		register { NewSoneHandler(it, loaders.loadTemplate("/templates/notify/newSoneNotification.html")) }
 	}
 
+	private fun register(handler: (NotificationManager) -> Any) =
+			handler(notificationManager).also(eventBus::register)
+
 }