48 lines
2.6 KiB
Kotlin
48 lines
2.6 KiB
Kotlin
package dev.inmo.tgbotapi.updateshandlers
|
|
|
|
import dev.inmo.tgbotapi.types.ALL_UPDATES_LIST
|
|
import dev.inmo.tgbotapi.types.update.*
|
|
import dev.inmo.tgbotapi.types.update.MediaGroupUpdates.*
|
|
import dev.inmo.tgbotapi.types.update.abstracts.UnknownUpdate
|
|
import dev.inmo.tgbotapi.types.update.abstracts.Update
|
|
import kotlinx.coroutines.flow.*
|
|
|
|
@Suppress("EXPERIMENTAL_API_USAGE", "unused")
|
|
class FlowsUpdatesFilter(
|
|
broadcastChannelsSize: Int = 100
|
|
): UpdatesFilter {
|
|
private val updatesSharedFlow = MutableSharedFlow<Update>(extraBufferCapacity = broadcastChannelsSize)
|
|
@Suppress("MemberVisibilityCanBePrivate")
|
|
val allUpdatesFlow: Flow<Update> = updatesSharedFlow.asSharedFlow()
|
|
@Suppress("MemberVisibilityCanBePrivate")
|
|
val allUpdatesWithoutMediaGroupsGroupingFlow: Flow<Update> = updatesSharedFlow.flatMapConcat {
|
|
(if (it is SentMediaGroupUpdate) {
|
|
it.origins
|
|
} else {
|
|
listOf(it)
|
|
}).asFlow()
|
|
}
|
|
|
|
override val allowedUpdates: List<String>
|
|
get() = ALL_UPDATES_LIST
|
|
override val asUpdateReceiver: UpdateReceiver<Update> = {
|
|
updatesSharedFlow.emit(it)
|
|
}
|
|
|
|
val messageFlow: Flow<MessageUpdate> = allUpdatesFlow.filterIsInstance()
|
|
val messageMediaGroupFlow: Flow<MessageMediaGroupUpdate> = allUpdatesFlow.filterIsInstance()
|
|
val editedMessageFlow: Flow<EditMessageUpdate> = allUpdatesFlow.filterIsInstance()
|
|
val editedMessageMediaGroupFlow: Flow<EditMessageMediaGroupUpdate> = allUpdatesFlow.filterIsInstance()
|
|
val channelPostFlow: Flow<ChannelPostUpdate> = allUpdatesFlow.filterIsInstance()
|
|
val channelPostMediaGroupFlow: Flow<ChannelPostMediaGroupUpdate> = allUpdatesFlow.filterIsInstance()
|
|
val editedChannelPostFlow: Flow<EditChannelPostUpdate> = allUpdatesFlow.filterIsInstance()
|
|
val editedChannelPostMediaGroupFlow: Flow<EditChannelPostMediaGroupUpdate> = allUpdatesFlow.filterIsInstance()
|
|
val chosenInlineResultFlow: Flow<ChosenInlineResultUpdate> = allUpdatesFlow.filterIsInstance()
|
|
val inlineQueryFlow: Flow<InlineQueryUpdate> = allUpdatesFlow.filterIsInstance()
|
|
val callbackQueryFlow: Flow<CallbackQueryUpdate> = allUpdatesFlow.filterIsInstance()
|
|
val shippingQueryFlow: Flow<ShippingQueryUpdate> = allUpdatesFlow.filterIsInstance()
|
|
val preCheckoutQueryFlow: Flow<PreCheckoutQueryUpdate> = allUpdatesFlow.filterIsInstance()
|
|
val pollFlow: Flow<PollUpdate> = allUpdatesFlow.filterIsInstance()
|
|
val pollAnswerFlow: Flow<PollAnswerUpdate> = allUpdatesFlow.filterIsInstance()
|
|
val unknownUpdateTypeFlow: Flow<UnknownUpdate> = allUpdatesFlow.filterIsInstance()
|
|
} |