2 * Sone - NotificationHandlerModuleTest.kt - Copyright © 2019 David ‘Bombe’ Roden
4 * This program is free software: you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation, either version 3 of the License, or
7 * (at your option) any later version.
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
14 * You should have received a copy of the GNU General Public License
15 * along with this program. If not, see <http://www.gnu.org/licenses/>.
18 package net.pterodactylus.sone.web.notification
20 import com.google.inject.*
21 import com.google.inject.binder.*
22 import net.pterodactylus.sone.core.*
23 import net.pterodactylus.sone.data.*
24 import net.pterodactylus.sone.freenet.wot.*
25 import net.pterodactylus.sone.main.*
26 import net.pterodactylus.sone.notify.*
27 import net.pterodactylus.sone.text.*
28 import net.pterodactylus.util.notify.*
29 import java.util.concurrent.*
30 import java.util.concurrent.TimeUnit.*
31 import java.util.function.*
33 import javax.inject.Singleton
36 * Guice module for creating all notification handlers.
38 class NotificationHandlerModule : AbstractModule() {
40 override fun configure() {
41 bind(NotificationHandler::class.java).`in`(Singleton::class.java)
42 bind<MarkPostKnownDuringFirstStartHandler>().asSingleton()
43 bind<MarkPostReplyKnownDuringFirstStartHandler>().asSingleton()
44 bind<SoneLockedOnStartupHandler>().asSingleton()
45 bind<NewSoneHandler>().asSingleton()
46 bind<NewRemotePostHandler>().asSingleton()
47 bind<RemotePostReplyHandler>().asSingleton()
48 bind<SoneLockedHandler>().asSingleton()
49 bind<LocalPostHandler>().asSingleton()
50 bind<NewVersionHandler>().asSingleton()
51 bind<ImageInsertHandler>().asSingleton()
52 bind<FirstStartHandler>().asSingleton()
53 bind<ConfigNotReadHandler>().asSingleton()
54 bind<StartupHandler>().asSingleton()
55 bind<WebOfTrustHandler>().asSingleton()
56 bind<SoneMentionDetector>().asSingleton()
57 bind<SoneMentionedHandler>().asSingleton()
61 fun getMarkPostKnownHandler(core: Core): Consumer<Post> = Consumer { core.markPostKnown(it) }
64 fun getMarkPostReplyKnownHandler(core: Core): Consumer<PostReply> = Consumer { core.markReplyKnown(it) }
68 @Named("soneLockedOnStartup")
69 fun getSoneLockedOnStartupNotification(loaders: Loaders) =
70 ListNotification<Sone>("sone-locked-on-startup", "sones", loaders.loadTemplate("/templates/notify/soneLockedOnStartupNotification.html"))
74 fun getNewSoneNotification(loaders: Loaders) =
75 ListNotification<Sone>("new-sone-notification", "sones", loaders.loadTemplate("/templates/notify/newSoneNotification.html"), dismissable = false)
79 @Named("newRemotePost")
80 fun getNewPostNotification(loaders: Loaders) =
81 ListNotification<Post>("new-post-notification", "posts", loaders.loadTemplate("/templates/notify/newPostNotification.html"), dismissable = false)
85 @Named("newRemotePostReply")
86 fun getNewRemotePostReplyNotification(loaders: Loaders) =
87 ListNotification<PostReply>("new-reply-notification", "replies", loaders.loadTemplate("/templates/notify/newReplyNotification.html"), dismissable = false)
92 fun getSoneLockedNotification(loaders: Loaders) =
93 ListNotification<Sone>("sones-locked-notification", "sones", loaders.loadTemplate("/templates/notify/lockedSonesNotification.html"), dismissable = true)
98 fun getLocalPostNotification(loaders: Loaders) =
99 ListNotification<Post>("local-post-notification", "posts", loaders.loadTemplate("/templates/notify/newPostNotification.html"), dismissable = false)
104 fun getNewVersionNotification(loaders: Loaders) =
105 TemplateNotification("new-version-notification", loaders.loadTemplate("/templates/notify/newVersionNotification.html"))
109 @Named("imageInserting")
110 fun getImageInsertingNotification(loaders: Loaders) =
111 ListNotification<Image>("inserting-images-notification", "images", loaders.loadTemplate("/templates/notify/inserting-images-notification.html"), dismissable = true)
115 @Named("imageFailed")
116 fun getImageInsertingFailedNotification(loaders: Loaders) =
117 ListNotification<Image>("image-insert-failed-notification", "images", loaders.loadTemplate("/templates/notify/image-insert-failed-notification.html"), dismissable = true)
121 @Named("imageInserted")
122 fun getImageInsertedNotification(loaders: Loaders) =
123 ListNotification<Image>("inserted-images-notification", "images", loaders.loadTemplate("/templates/notify/inserted-images-notification.html"), dismissable = true)
128 fun getFirstStartNotification(loaders: Loaders) =
129 TemplateNotification("first-start-notification", loaders.loadTemplate("/templates/notify/firstStartNotification.html"))
133 @Named("configNotRead")
134 fun getConfigNotReadNotification(loaders: Loaders) =
135 TemplateNotification("config-not-read-notification", loaders.loadTemplate("/templates/notify/configNotReadNotification.html"))
140 fun getStartupNotification(loaders: Loaders) =
141 TemplateNotification("startup-notification", loaders.loadTemplate("/templates/notify/startupNotification.html"))
146 fun getWebOfTrustNotification(loaders: Loaders) =
147 TemplateNotification("wot-missing-notification", loaders.loadTemplate("/templates/notify/wotMissingNotification.html"))
151 @Named("webOfTrustReacher")
152 fun getWebOfTrustReacher(webOfTrustConnector: WebOfTrustConnector): Runnable =
153 Runnable { webOfTrustConnector.ping() }
157 @Named("webOfTrustReschedule")
158 fun getWebOfTrustReschedule(@Named("notification") ticker: ScheduledExecutorService) =
159 Consumer<Runnable> { ticker.schedule(it, 15, SECONDS) }
163 @Named("soneMentioned")
164 fun getSoneMentionedNotification(loaders: Loaders) =
165 ListNotification<Post>("mention-notification", "posts", loaders.loadTemplate("/templates/notify/mentionNotification.html"), dismissable = false)
167 private inline fun <reified T> bind(): AnnotatedBindingBuilder<T> = bind(T::class.java)
168 private fun ScopedBindingBuilder.asSingleton() = `in`(Singleton::class.java)