🚧 Add handler for image-insert notifications
[Sone.git] / src / main / kotlin / net / pterodactylus / sone / web / notification / ImageInsertHandler.kt
diff --git a/src/main/kotlin/net/pterodactylus/sone/web/notification/ImageInsertHandler.kt b/src/main/kotlin/net/pterodactylus/sone/web/notification/ImageInsertHandler.kt
new file mode 100644 (file)
index 0000000..ea11481
--- /dev/null
@@ -0,0 +1,65 @@
+/**
+ * Sone - ImageInsertHandler.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
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+package net.pterodactylus.sone.web.notification
+
+import com.google.common.eventbus.*
+import net.pterodactylus.sone.core.event.*
+import net.pterodactylus.sone.data.*
+import net.pterodactylus.sone.notify.*
+import net.pterodactylus.util.notify.*
+
+/**
+ * Notification handler for the various image-insert-related events.
+ *
+ * @see ImageInsertStartedEvent
+ * @see ImageInsertAbortedEvent
+ * @see ImageInsertFailedEvent
+ * @see ImageInsertFinishedEvent
+ */
+class ImageInsertHandler(
+               private val notificationManager: NotificationManager,
+               private val imageInsertingNotification: ListNotification<Image>,
+               private val imageFailedNotification: ListNotification<Image>,
+               private val imageInsertedNotification: ListNotification<Image>) {
+
+       @Subscribe
+       fun imageInsertStarted(imageInsertStartedEvent: ImageInsertStartedEvent) {
+               imageInsertingNotification.add(imageInsertStartedEvent.image)
+               notificationManager.addNotification(imageInsertingNotification)
+       }
+
+       @Subscribe
+       fun imageInsertAborted(imageInsertAbortedEvent: ImageInsertAbortedEvent) {
+               imageInsertingNotification.remove(imageInsertAbortedEvent.image)
+       }
+
+       @Subscribe
+       fun imageInsertFailed(imageInsertFailedEvent: ImageInsertFailedEvent) {
+               imageInsertingNotification.remove(imageInsertFailedEvent.image)
+               imageFailedNotification.add(imageInsertFailedEvent.image)
+               notificationManager.addNotification(imageFailedNotification)
+       }
+
+       @Subscribe
+       fun imageInsertFinished(imageInsertFinishedEvent: ImageInsertFinishedEvent) {
+               imageInsertingNotification.remove(imageInsertFinishedEvent.image)
+               imageInsertedNotification.add(imageInsertFinishedEvent.image)
+               notificationManager.addNotification(imageInsertedNotification)
+       }
+
+}