🚧 Add handler for image-insert notifications
[Sone.git] / src / main / kotlin / net / pterodactylus / sone / web / notification / ImageInsertHandler.kt
1 /**
2  * Sone - ImageInsertHandler.kt - Copyright Â© 2019 David â€˜Bombe’ Roden
3  *
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.
8  *
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.
13  *
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/>.
16  */
17
18 package net.pterodactylus.sone.web.notification
19
20 import com.google.common.eventbus.*
21 import net.pterodactylus.sone.core.event.*
22 import net.pterodactylus.sone.data.*
23 import net.pterodactylus.sone.notify.*
24 import net.pterodactylus.util.notify.*
25
26 /**
27  * Notification handler for the various image-insert-related events.
28  *
29  * @see ImageInsertStartedEvent
30  * @see ImageInsertAbortedEvent
31  * @see ImageInsertFailedEvent
32  * @see ImageInsertFinishedEvent
33  */
34 class ImageInsertHandler(
35                 private val notificationManager: NotificationManager,
36                 private val imageInsertingNotification: ListNotification<Image>,
37                 private val imageFailedNotification: ListNotification<Image>,
38                 private val imageInsertedNotification: ListNotification<Image>) {
39
40         @Subscribe
41         fun imageInsertStarted(imageInsertStartedEvent: ImageInsertStartedEvent) {
42                 imageInsertingNotification.add(imageInsertStartedEvent.image)
43                 notificationManager.addNotification(imageInsertingNotification)
44         }
45
46         @Subscribe
47         fun imageInsertAborted(imageInsertAbortedEvent: ImageInsertAbortedEvent) {
48                 imageInsertingNotification.remove(imageInsertAbortedEvent.image)
49         }
50
51         @Subscribe
52         fun imageInsertFailed(imageInsertFailedEvent: ImageInsertFailedEvent) {
53                 imageInsertingNotification.remove(imageInsertFailedEvent.image)
54                 imageFailedNotification.add(imageInsertFailedEvent.image)
55                 notificationManager.addNotification(imageFailedNotification)
56         }
57
58         @Subscribe
59         fun imageInsertFinished(imageInsertFinishedEvent: ImageInsertFinishedEvent) {
60                 imageInsertingNotification.remove(imageInsertFinishedEvent.image)
61                 imageInsertedNotification.add(imageInsertFinishedEvent.image)
62                 notificationManager.addNotification(imageInsertedNotification)
63         }
64
65 }