🚧 Add handler for marking replies as known during first start

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

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 cfe5db3..2b147fc 100644 (file)
--- a/src/main/kotlin/net/pterodactylus/sone/web/notification/NotificationHandlerModule.kt
+++ b/src/main/kotlin/net/pterodactylus/sone/web/notification/NotificationHandlerModule.kt
@@ -40,6 +40,7 @@ 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()
@@ -59,6 +60,9 @@ class NotificationHandlerModule : AbstractModule() {
        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) =