X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Fmain%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2Fnotification%2FNotificationHandlerModule.kt;h=18aefd665732e0c38ad03319abec5190f8a3f4d2;hp=e94be5b65ad3cc9ded46785bb2364cafb35ec513;hb=438378deab1514f0f608d975ef65f5b7aea44ccb;hpb=0ec1b5b1d8bdf682f87909a1a0dea64290f6121d

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 e94be5b..18aefd6 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–2020 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
@@ -21,11 +21,13 @@ import com.google.inject.*
 import com.google.inject.binder.*
 import net.pterodactylus.sone.core.*
 import net.pterodactylus.sone.data.*
+import net.pterodactylus.sone.freenet.wot.*
 import net.pterodactylus.sone.main.*
 import net.pterodactylus.sone.notify.*
+import net.pterodactylus.sone.text.*
 import net.pterodactylus.util.notify.*
 import java.util.concurrent.*
-import java.util.concurrent.Executors.*
+import java.util.concurrent.TimeUnit.*
 import java.util.function.*
 import javax.inject.*
 import javax.inject.Singleton
@@ -38,21 +40,32 @@ class NotificationHandlerModule : AbstractModule() {
 	override fun configure() {
 		bind(NotificationHandler::class.java).`in`(Singleton::class.java)
 		bind<MarkPostKnownDuringFirstStartHandler>().asSingleton()
+		bind<MarkPostReplyKnownDuringFirstStartHandler>().asSingleton()
 		bind<SoneLockedOnStartupHandler>().asSingleton()
 		bind<NewSoneHandler>().asSingleton()
 		bind<NewRemotePostHandler>().asSingleton()
+		bind<RemotePostReplyHandler>().asSingleton()
 		bind<SoneLockedHandler>().asSingleton()
 		bind<LocalPostHandler>().asSingleton()
+		bind<LocalReplyHandler>().asSingleton()
 		bind<NewVersionHandler>().asSingleton()
 		bind<ImageInsertHandler>().asSingleton()
 		bind<FirstStartHandler>().asSingleton()
 		bind<ConfigNotReadHandler>().asSingleton()
+		bind<StartupHandler>().asSingleton()
+		bind<WebOfTrustHandler>().asSingleton()
+		bind<SoneMentionDetector>().asSingleton()
+		bind<SoneMentionedHandler>().asSingleton()
+		bind<SoneInsertHandler>().asSingleton()
 	}
 
 	@Provides
 	fun getMarkPostKnownHandler(core: Core): Consumer<Post> = Consumer { core.markPostKnown(it) }
 
 	@Provides
+	fun getMarkPostReplyKnownHandler(core: Core): Consumer<PostReply> = Consumer { core.markReplyKnown(it) }
+
+	@Provides
 	@Singleton
 	@Named("soneLockedOnStartup")
 	fun getSoneLockedOnStartupNotification(loaders: Loaders) =
@@ -71,15 +84,17 @@ class NotificationHandlerModule : AbstractModule() {
 
 	@Provides
 	@Singleton
+	@Named("newRemotePostReply")
+	fun getNewRemotePostReplyNotification(loaders: Loaders) =
+			ListNotification<PostReply>("new-reply-notification", "replies", loaders.loadTemplate("/templates/notify/newReplyNotification.html"), dismissable = false)
+
+	@Provides
+	@Singleton
 	@Named("soneLocked")
 	fun getSoneLockedNotification(loaders: Loaders) =
 			ListNotification<Sone>("sones-locked-notification", "sones", loaders.loadTemplate("/templates/notify/lockedSonesNotification.html"), dismissable = true)
 
 	@Provides
-	fun getScheduledExecutorService(): ScheduledExecutorService =
-			newScheduledThreadPool(1)
-
-	@Provides
 	@Singleton
 	@Named("localPost")
 	fun getLocalPostNotification(loaders: Loaders) =
@@ -87,6 +102,12 @@ class NotificationHandlerModule : AbstractModule() {
 
 	@Provides
 	@Singleton
+	@Named("localReply")
+	fun getLocalReplyNotification(loaders: Loaders) =
+			ListNotification<PostReply>("local-reply-notification", "replies", loaders.loadTemplate("/templates/notify/newReplyNotification.html"), dismissable = false)
+
+	@Provides
+	@Singleton
 	@Named("newVersion")
 	fun getNewVersionNotification(loaders: Loaders) =
 			TemplateNotification("new-version-notification", loaders.loadTemplate("/templates/notify/newVersionNotification.html"))
@@ -121,6 +142,50 @@ class NotificationHandlerModule : AbstractModule() {
 	fun getConfigNotReadNotification(loaders: Loaders) =
 			TemplateNotification("config-not-read-notification", loaders.loadTemplate("/templates/notify/configNotReadNotification.html"))
 
+	@Provides
+	@Singleton
+	@Named("startup")
+	fun getStartupNotification(loaders: Loaders) =
+			TemplateNotification("startup-notification", loaders.loadTemplate("/templates/notify/startupNotification.html"))
+
+	@Provides
+	@Singleton
+	@Named("webOfTrust")
+	fun getWebOfTrustNotification(loaders: Loaders) =
+			TemplateNotification("wot-missing-notification", loaders.loadTemplate("/templates/notify/wotMissingNotification.html"))
+
+	@Provides
+	@Singleton
+	@Named("webOfTrustReacher")
+	fun getWebOfTrustReacher(webOfTrustConnector: WebOfTrustConnector): Runnable =
+			Runnable { webOfTrustConnector.ping() }
+
+	@Provides
+	@Singleton
+	@Named("webOfTrustReschedule")
+	fun getWebOfTrustReschedule(@Named("notification") ticker: ScheduledExecutorService) =
+			Consumer<Runnable> { ticker.schedule(it, 15, SECONDS) }
+
+	@Provides
+	@Singleton
+	@Named("soneMentioned")
+	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)