From ab2a3ea23f08625477df8de194c39bdae8262c6d Mon Sep 17 00:00:00 2001
From: InsanusMokrassar <ovsyannikov.alexey95@gmail.com>
Date: Fri, 10 May 2019 11:54:57 +0800
Subject: [PATCH] add "FlowsUpdatesFilter"

---
 CHANGELOG.md                                  |  1 +
 .../updateshandlers/FlowsUpdatesFilter.kt     | 60 +++++++++++++++++++
 2 files changed, 61 insertions(+)
 create mode 100644 src/main/kotlin/com/github/insanusmokrassar/TelegramBotAPI/updateshandlers/FlowsUpdatesFilter.kt

diff --git a/CHANGELOG.md b/CHANGELOG.md
index b79c03276e..56386fae3d 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -22,6 +22,7 @@ some default library
 
 * Replace `UpdatesFilter` and `UpdatesPoller` into another package
 * Replace `WebhookPrivateKeyConfig`
+* Added `FlowsUpdatesFilter`
 
 ## 0.13.0 Telegram Polls
 
diff --git a/src/main/kotlin/com/github/insanusmokrassar/TelegramBotAPI/updateshandlers/FlowsUpdatesFilter.kt b/src/main/kotlin/com/github/insanusmokrassar/TelegramBotAPI/updateshandlers/FlowsUpdatesFilter.kt
new file mode 100644
index 0000000000..0948630dca
--- /dev/null
+++ b/src/main/kotlin/com/github/insanusmokrassar/TelegramBotAPI/updateshandlers/FlowsUpdatesFilter.kt
@@ -0,0 +1,60 @@
+package com.github.insanusmokrassar.TelegramBotAPI.updateshandlers
+
+import com.github.insanusmokrassar.TelegramBotAPI.types.update.*
+import com.github.insanusmokrassar.TelegramBotAPI.types.update.MediaGroupUpdates.*
+import com.github.insanusmokrassar.TelegramBotAPI.types.update.abstracts.Update
+import com.github.insanusmokrassar.TelegramBotAPI.utils.extensions.UpdateReceiver
+import kotlinx.coroutines.channels.BroadcastChannel
+import kotlinx.coroutines.channels.Channel
+import kotlinx.coroutines.flow.Flow
+import kotlinx.coroutines.flow.asFlow
+
+private fun <T> BroadcastChannel<T>.createUpdateReceiver(): UpdateReceiver<T> = ::send
+
+class FlowsUpdatesFilter {
+    private val messageChannel: BroadcastChannel<MessageUpdate> = BroadcastChannel(Channel.CONFLATED)
+    private val messageMediaGroupChannel: BroadcastChannel<MessageMediaGroupUpdate> = BroadcastChannel(Channel.CONFLATED)
+    private val editedMessageChannel: BroadcastChannel<EditMessageUpdate> = BroadcastChannel(Channel.CONFLATED)
+    private val editedMessageMediaGroupChannel: BroadcastChannel<EditMessageMediaGroupUpdate> = BroadcastChannel(Channel.CONFLATED)
+    private val channelPostChannel: BroadcastChannel<ChannelPostUpdate> = BroadcastChannel(Channel.CONFLATED)
+    private val channelPostMediaGroupChannel: BroadcastChannel<ChannelPostMediaGroupUpdate> = BroadcastChannel(Channel.CONFLATED)
+    private val editedChannelPostChannel: BroadcastChannel<EditChannelPostUpdate> = BroadcastChannel(Channel.CONFLATED)
+    private val editedChannelPostMediaGroupChannel: BroadcastChannel<EditChannelPostMediaGroupUpdate> = BroadcastChannel(Channel.CONFLATED)
+    private val chosenInlineResultChannel: BroadcastChannel<ChosenInlineResultUpdate> = BroadcastChannel(Channel.CONFLATED)
+    private val inlineQueryChannel: BroadcastChannel<InlineQueryUpdate> = BroadcastChannel(Channel.CONFLATED)
+    private val channelQueryChannel: BroadcastChannel<CallbackQueryUpdate> = BroadcastChannel(Channel.CONFLATED)
+    private val shippingQueryChannel: BroadcastChannel<ShippingQueryUpdate> = BroadcastChannel(Channel.CONFLATED)
+    private val preCheckoutQueryChannel: BroadcastChannel<PreCheckoutQueryUpdate> = BroadcastChannel(Channel.CONFLATED)
+
+    val filter = UpdatesFilter(
+        messageChannel.createUpdateReceiver(),
+        messageMediaGroupChannel.createUpdateReceiver(),
+        editedMessageChannel.createUpdateReceiver(),
+        editedMessageMediaGroupChannel.createUpdateReceiver(),
+        channelPostChannel.createUpdateReceiver(),
+        channelPostMediaGroupChannel.createUpdateReceiver(),
+        editedChannelPostChannel.createUpdateReceiver(),
+        editedChannelPostMediaGroupChannel.createUpdateReceiver(),
+        chosenInlineResultChannel.createUpdateReceiver(),
+        inlineQueryChannel.createUpdateReceiver(),
+        channelQueryChannel.createUpdateReceiver(),
+        shippingQueryChannel.createUpdateReceiver(),
+        preCheckoutQueryChannel.createUpdateReceiver()
+    )
+
+    val asUpdateReceiver: UpdateReceiver<Update> = filter.asUpdateReceiver
+
+    val messageChannelFlow: Flow<MessageUpdate> = messageChannel.asFlow()
+    val messageMediaGroupChannelFlow: Flow<MessageMediaGroupUpdate> = messageMediaGroupChannel.asFlow()
+    val editedMessageChannelFlow: Flow<EditMessageUpdate> = editedMessageChannel.asFlow()
+    val editedMessageMediaGroupChannelFlow: Flow<EditMessageMediaGroupUpdate> = editedMessageMediaGroupChannel.asFlow()
+    val channelPostChannelFlow: Flow<ChannelPostUpdate> = channelPostChannel.asFlow()
+    val channelPostMediaGroupChannelFlow: Flow<ChannelPostMediaGroupUpdate> = channelPostMediaGroupChannel.asFlow()
+    val editedChannelPostChannelFlow: Flow<EditChannelPostUpdate> = editedChannelPostChannel.asFlow()
+    val editedChannelPostMediaGroupChannelFlow: Flow<EditChannelPostMediaGroupUpdate> = editedChannelPostMediaGroupChannel.asFlow()
+    val chosenInlineResultChannelFlow: Flow<ChosenInlineResultUpdate> = chosenInlineResultChannel.asFlow()
+    val inlineQueryChannelFlow: Flow<InlineQueryUpdate> = inlineQueryChannel.asFlow()
+    val channelQueryChannelFlow: Flow<CallbackQueryUpdate> = channelQueryChannel.asFlow()
+    val shippingQueryChannelFlow: Flow<ShippingQueryUpdate> = shippingQueryChannel.asFlow()
+    val preCheckoutQueryChannelFlow: Flow<PreCheckoutQueryUpdate> = preCheckoutQueryChannel.asFlow()
+}
\ No newline at end of file