X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;ds=sidebyside;f=src%2Fmain%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2Fnotification%2FNotificationHandlerModule.kt;h=da20dafe8bdb385f483570e20350ddbf2b2c0c32;hb=2f1e414e86c61435cd8b3991c475e0832f6de2fb;hp=bb37e0956cd8a2e0b455d4085509c61735caf80e;hpb=79ef6e4644845290d9feb45a06f8588864dd0a83;p=Sone.git

diff --git a/src/main/kotlin/net/pterodactylus/sone/web/notification/NotificationHandlerModule.kt b/src/main/kotlin/net/pterodactylus/sone/web/notification/NotificationHandlerModule.kt
index bb37e09..da20daf 100644
--- a/src/main/kotlin/net/pterodactylus/sone/web/notification/NotificationHandlerModule.kt
+++ b/src/main/kotlin/net/pterodactylus/sone/web/notification/NotificationHandlerModule.kt
@@ -1,5 +1,5 @@
 /**
- * Sone - NotificationHandlerModuleTest.kt - Copyright © 2019 David ‘Bombe’ Roden
+ * Sone - NotificationHandlerModule.kt - Copyright © 2019 David ‘Bombe’ 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
@@ -56,6 +56,7 @@ class NotificationHandlerModule : AbstractModule() {
 		bind<WebOfTrustHandler>().asSingleton()
 		bind<SoneMentionDetector>().asSingleton()
 		bind<SoneMentionedHandler>().asSingleton()
+		bind<SoneInsertHandler>().asSingleton()
 	}
 
 	@Provides
@@ -171,6 +172,20 @@ class NotificationHandlerModule : AbstractModule() {
 	fun getSoneMentionedNotification(loaders: Loaders) =
 			ListNotification<Post>("mention-notification", "posts", loaders.loadTemplate("/templates/notify/mentionNotification.html"), dismissable = false)
 
+	@Provides
+	@Singleton
+	fun getSoneNotificationSupplier(loaders: Loaders): SoneInsertNotificationSupplier =
+			mutableMapOf<Sone, TemplateNotification>()
+					.let { cache ->
+						{ sone ->
+							cache.computeIfAbsent(sone) {
+								loaders.loadTemplate("/templates/notify/soneInsertNotification.html")
+										.let(::TemplateNotification)
+										.also { it["insertSone"] = sone }
+							}
+						}
+					}
+
 	private inline fun <reified T> bind(): AnnotatedBindingBuilder<T> = bind(T::class.java)
 	private fun ScopedBindingBuilder.asSingleton() = `in`(Singleton::class.java)