tgbotapi/tgbotapi.core/src/commonMain/kotlin/dev/inmo/tgbotapi/updateshandlers/FlowsUpdatesFilter.kt

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.channels.BufferOverflow
import kotlinx.coroutines.flow.*

interface FlowsUpdatesFilter : UpdatesFilter {
    override val allowedUpdates: List<String>
        get() = ALL_UPDATES_LIST
    val allUpdatesFlow: Flow<Update>
    val allUpdatesWithoutMediaGroupsGroupingFlow: Flow<Update>

    val messagesFlow: Flow<MessageUpdate>
    val messageMediaGroupsFlow: Flow<MessageMediaGroupUpdate>
    val editedMessagesFlow: Flow<EditMessageUpdate>
    val editedMessageMediaGroupsFlow: Flow<EditMessageMediaGroupUpdate>
    val channelPostsFlow: Flow<ChannelPostUpdate>
    val channelPostMediaGroupsFlow: Flow<ChannelPostMediaGroupUpdate>
    val editedChannelPostsFlow: Flow<EditChannelPostUpdate>
    val editedChannelPostMediaGroupsFlow: Flow<EditChannelPostMediaGroupUpdate>
    val chosenInlineResultsFlow: Flow<ChosenInlineResultUpdate>
    val inlineQueriesFlow: Flow<InlineQueryUpdate>
    val callbackQueriesFlow: Flow<CallbackQueryUpdate>
    val shippingQueriesFlow: Flow<ShippingQueryUpdate>
    val preCheckoutQueriesFlow: Flow<PreCheckoutQueryUpdate>
    val pollsFlow: Flow<PollUpdate>
    val pollAnswersFlow: Flow<PollAnswerUpdate>
    val chatMemberUpdatesFlow: Flow<CommonChatMemberUpdatedUpdate>
    val myChatMemberUpdatesFlow: Flow<MyChatMemberUpdatedUpdate>
    val unknownUpdatesFlow: Flow<UnknownUpdate>

    @Deprecated("Renamed", ReplaceWith("messagesFlow"))
    val messageFlow: Flow<MessageUpdate>
        get() = messagesFlow
    @Deprecated("Renamed", ReplaceWith("messageMediaGroupsFlow"))
    val messageMediaGroupFlow: Flow<MessageMediaGroupUpdate>
        get() = messageMediaGroupsFlow
    @Deprecated("Renamed", ReplaceWith("editedMessagesFlow"))
    val editedMessageFlow: Flow<EditMessageUpdate>
        get() = editedMessagesFlow
    @Deprecated("Renamed", ReplaceWith("editedMessageMediaGroupsFlow"))
    val editedMessageMediaGroupFlow: Flow<EditMessageMediaGroupUpdate>
        get() = editedMessageMediaGroupsFlow
    @Deprecated("Renamed", ReplaceWith("channelPostsFlow"))
    val channelPostFlow: Flow<ChannelPostUpdate>
        get() = channelPostsFlow
    @Deprecated("Renamed", ReplaceWith("channelPostMediaGroupsFlow"))
    val channelPostMediaGroupFlow: Flow<ChannelPostMediaGroupUpdate>
        get() = channelPostMediaGroupsFlow
    @Deprecated("Renamed", ReplaceWith("editedChannelPostsFlow"))
    val editedChannelPostFlow: Flow<EditChannelPostUpdate>
        get() = editedChannelPostsFlow
    @Deprecated("Renamed", ReplaceWith("editedChannelPostMediaGroupsFlow"))
    val editedChannelPostMediaGroupFlow: Flow<EditChannelPostMediaGroupUpdate>
        get() = editedChannelPostMediaGroupsFlow
    @Deprecated("Renamed", ReplaceWith("chosenInlineResultsFlow"))
    val chosenInlineResultFlow: Flow<ChosenInlineResultUpdate>
        get() = chosenInlineResultsFlow
    @Deprecated("Renamed", ReplaceWith("inlineQueriesFlow"))
    val inlineQueryFlow: Flow<InlineQueryUpdate>
        get() = inlineQueriesFlow
    @Deprecated("Renamed", ReplaceWith("callbackQueriesFlow"))
    val callbackQueryFlow: Flow<CallbackQueryUpdate>
        get() = callbackQueriesFlow
    @Deprecated("Renamed", ReplaceWith("shippingQueriesFlow"))
    val shippingQueryFlow: Flow<ShippingQueryUpdate>
        get() = shippingQueriesFlow
    @Deprecated("Renamed", ReplaceWith("preCheckoutQueriesFlow"))
    val preCheckoutQueryFlow: Flow<PreCheckoutQueryUpdate>
        get() = preCheckoutQueriesFlow
    @Deprecated("Renamed", ReplaceWith("pollsFlow"))
    val pollFlow: Flow<PollUpdate>
        get() = pollsFlow
    @Deprecated("Renamed", ReplaceWith("pollAnswersFlow"))
    val pollAnswerFlow: Flow<PollAnswerUpdate>
        get() = pollAnswersFlow
    @Deprecated("Renamed", ReplaceWith("chatMemberUpdatesFlow"))
    val chatMemberUpdatedFlow: Flow<CommonChatMemberUpdatedUpdate>
        get() = chatMemberUpdatesFlow
    @Deprecated("Renamed", ReplaceWith("myChatMemberUpdatesFlow"))
    val myChatMemberUpdatedFlow: Flow<MyChatMemberUpdatedUpdate>
        get() = myChatMemberUpdatesFlow
    @Deprecated("Renamed", ReplaceWith("unknownUpdatesFlow"))
    val unknownUpdateTypeFlow: Flow<UnknownUpdate>
        get() = unknownUpdatesFlow
}

/**
 * Creates [DefaultFlowsUpdatesFilter]
 */
@Suppress("FunctionName")
fun FlowsUpdatesFilter(
    broadcastChannelsSize: Int = 100
) = DefaultFlowsUpdatesFilter(broadcastChannelsSize)

@Suppress("unused")
class DefaultFlowsUpdatesFilter(
    broadcastChannelsSize: Int = 100,
    onBufferOverflow: BufferOverflow = BufferOverflow.SUSPEND
): FlowsUpdatesFilter {
    private val updatesSharedFlow = MutableSharedFlow<Update>(extraBufferCapacity = broadcastChannelsSize, onBufferOverflow = onBufferOverflow)
    @Suppress("MemberVisibilityCanBePrivate")
    override val allUpdatesFlow: Flow<Update> = updatesSharedFlow.asSharedFlow()
    @Suppress("MemberVisibilityCanBePrivate")
    override val allUpdatesWithoutMediaGroupsGroupingFlow: Flow<Update> = allUpdatesFlow.flatMapConcat {
        when (it) {
            is SentMediaGroupUpdate -> it.origins.asFlow()
            is EditMediaGroupUpdate -> flowOf(it.origin)
            else -> flowOf(it)
        }
    }

    override val asUpdateReceiver: UpdateReceiver<Update> = {
        updatesSharedFlow.emit(it)
    }

    override val messagesFlow: Flow<MessageUpdate> = allUpdatesFlow.filterIsInstance()
    override val messageMediaGroupsFlow: Flow<MessageMediaGroupUpdate> = allUpdatesFlow.filterIsInstance()
    override val editedMessagesFlow: Flow<EditMessageUpdate> = allUpdatesFlow.filterIsInstance()
    override val editedMessageMediaGroupsFlow: Flow<EditMessageMediaGroupUpdate> = allUpdatesFlow.filterIsInstance()
    override val channelPostsFlow: Flow<ChannelPostUpdate> = allUpdatesFlow.filterIsInstance()
    override val channelPostMediaGroupsFlow: Flow<ChannelPostMediaGroupUpdate> = allUpdatesFlow.filterIsInstance()
    override val editedChannelPostsFlow: Flow<EditChannelPostUpdate> = allUpdatesFlow.filterIsInstance()
    override val editedChannelPostMediaGroupsFlow: Flow<EditChannelPostMediaGroupUpdate> = allUpdatesFlow.filterIsInstance()
    override val chosenInlineResultsFlow: Flow<ChosenInlineResultUpdate> = allUpdatesFlow.filterIsInstance()
    override val inlineQueriesFlow: Flow<InlineQueryUpdate> = allUpdatesFlow.filterIsInstance()
    override val callbackQueriesFlow: Flow<CallbackQueryUpdate> = allUpdatesFlow.filterIsInstance()
    override val shippingQueriesFlow: Flow<ShippingQueryUpdate> = allUpdatesFlow.filterIsInstance()
    override val preCheckoutQueriesFlow: Flow<PreCheckoutQueryUpdate> = allUpdatesFlow.filterIsInstance()
    override val pollsFlow: Flow<PollUpdate> = allUpdatesFlow.filterIsInstance()
    override val pollAnswersFlow: Flow<PollAnswerUpdate> = allUpdatesFlow.filterIsInstance()
    override val chatMemberUpdatesFlow: Flow<CommonChatMemberUpdatedUpdate> = allUpdatesFlow.filterIsInstance()
    override val myChatMemberUpdatesFlow: Flow<MyChatMemberUpdatedUpdate> = allUpdatesFlow.filterIsInstance()
    override val unknownUpdatesFlow: Flow<UnknownUpdate> = allUpdatesFlow.filterIsInstance()
}